feat(client): Add ContentType() to ClientWithResponses responses

CJourneaux · CJourneaux · commit b72509c138fc · 2025-12-03T15:14:02.000-05:00
diff --git a/configuration-schema.json b/configuration-schema.json
@@ -238,6 +238,10 @@
           "type": "boolean",
           "description": "Enable the generation of a `Bytes()` method on response objects for `ClientWithResponses`"
         },
+        "client-response-content-type-function": {
+          "type": "boolean",
+          "description": "Enable the generation of a `ContentType()` method on response objects for `ClientWithResponses`"
+        },
         "prefer-skip-optional-pointer": {
           "type": "boolean",
           "description": "Allows defining at a global level whether to omit the pointer for a type to indicate that the field/type is optional. This is the same as adding `x-go-type-skip-optional-pointer` to each field (manually, or using an OpenAPI Overlay). A field can set `x-go-type-skip-optional-pointer: false` to still require the optional pointer.",
diff --git a/internal/test/issues/issue-240-2149/api.yaml b/internal/test/issues/issue-240-2149/api.yaml
diff --git a/internal/test/issues/issue-240-2149/cfg.yaml b/internal/test/issues/issue-240-2149/cfg.yaml
@@ -1,8 +1,9 @@
 # yaml-language-server: $schema=../../../../configuration-schema.json
-package: issue240
+package: clientresponsefunctions
 output: client.gen.go
 generate:
   models: true
   client: true
 output-options:
   client-response-bytes-function: true
+  client-response-content-type-function: true
diff --git a/internal/test/issues/issue-240-2149/client.gen.go b/internal/test/issues/issue-240-2149/client.gen.go
diff --git a/internal/test/issues/issue-240-2149/generate.go b/internal/test/issues/issue-240-2149/generate.go
@@ -1,3 +1,3 @@
-package issue240
+package clientresponsefunctions
 
 //go:generate go run github.com/oapi-codegen/oapi-codegen/v2/cmd/oapi-codegen -config cfg.yaml api.yaml
diff --git a/pkg/codegen/configuration.go b/pkg/codegen/configuration.go
@@ -288,6 +288,9 @@ type OutputOptions struct {
 	// ClientResponseBytesFunction decides whether to enable the generation of a `Bytes()` method on response objects for `ClientWithResponses`
 	ClientResponseBytesFunction bool `yaml:"client-response-bytes-function,omitempty"`
 
+	// ClientResponseContentTypeFunction decides whether to enable the generation of a `ContentType()` method on response objects for `ClientWithResponses`
+	ClientResponseContentTypeFunction bool `yaml:"client-response-content-type-function,omitempty"`
+
 	// PreferSkipOptionalPointer allows defining at a global level whether to omit the pointer for a type to indicate that the field/type is optional.
 	// This is the same as adding `x-go-type-skip-optional-pointer` to each field (manually, or using an OpenAPI Overlay)
 	PreferSkipOptionalPointer bool `yaml:"prefer-skip-optional-pointer,omitempty"`
diff --git a/pkg/codegen/templates/client-with-responses.tmpl b/pkg/codegen/templates/client-with-responses.tmpl
@@ -81,6 +81,17 @@ func (r {{genResponseTypeName $opid | ucFirst}}) Bytes() []byte {
     return r.Body
 }
 {{end}}
+
+{{ if opts.OutputOptions.ClientResponseContentTypeFunction }}
+// ContentType is a convenience method to retrieve the Content-Type value from the HTTP response headers
+func (r {{genResponseTypeName $opid | ucFirst}}) ContentType() string {
+    if r.HTTPResponse != nil {
+        return r.HTTPResponse.Header.Get("Content-Type")
+    }
+    return ""
+}
+{{end}}
+
 {{end}}
 
 

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,3 @@`
`1`		`-package issue240`
	`1`	`+package clientresponsefunctions`
`2`	`2`
`3`	`3`	`//go:generate go run github.com/oapi-codegen/oapi-codegen/v2/cmd/oapi-codegen -config cfg.yaml api.yaml`