feat: add ability to set header parameters

go-fuego · Feb 21, 2024 · 6dda095 · 6dda095
1 parent 97ca254
commit 6dda095
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/examples/basic/main.go b/examples/basic/main.go
@@ -35,13 +35,18 @@ func main() {
 			return MyResponse{}, err
 		}
 
+		// read the request header test
+		if c.Request().Header.Get("test") != "test" {
+			return MyResponse{}, errors.New("test header not found")
+		}
+
 		c.Response().Header().Set("X-Hello", "World")
 
 		return MyResponse{
 			Message:       "Hello, " + data.Name,
 			BestFramework: "Fuego!",
 		}, nil
-	})
+	}).WithDescription("Say hello to the world").AddHeader("test")
 
 	// Standard net/http handler with automatic OpenAPI route declaration
 	fuego.GetStd(s, "/std", func(w http.ResponseWriter, r *http.Request) {

diff --git a/mux.go b/mux.go
@@ -117,6 +117,14 @@ func (r Route[ResponseBody, RequestBody]) SetTags(tags ...string) Route[Response
 	return r
 }
 
+func (r Route[ResponseBody, RequestBody]) AddHeader(header string) Route[ResponseBody, RequestBody] {
+	r.operation.Parameters = append(r.operation.Parameters, &openapi3.ParameterRef{
+		Value: openapi3.NewHeaderParameter(header),
+	})
+
+	return r
+}
+
 func (r Route[ResponseBody, RequestBody]) AddTags(tags ...string) Route[ResponseBody, RequestBody] {
 	r.operation.Tags = append(r.operation.Tags, tags...)
 	return r