feat: add cookie parameter to route method

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

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/examples/basic/main.go b/examples/basic/main.go
@@ -46,7 +46,7 @@ func main() {
 			Message:       "Hello, " + data.Name,
 			BestFramework: "Fuego!",
 		}, nil
-	}).WithDescription("Say hello to the world").AddHeader("test")
+	}).WithDescription("Say hello to the world").AddHeader("test").AddCookie("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
@@ -125,6 +125,14 @@ func (r Route[ResponseBody, RequestBody]) AddHeader(header string) Route[Respons
 	return r
 }
 
+func (r Route[ResponseBody, RequestBody]) AddCookie(cookie string) Route[ResponseBody, RequestBody] {
+	r.operation.Parameters = append(r.operation.Parameters, &openapi3.ParameterRef{
+		Value: openapi3.NewCookieParameter(cookie),
+	})
+
+	return r
+}
+
 func (r Route[ResponseBody, RequestBody]) AddTags(tags ...string) Route[ResponseBody, RequestBody] {
 	r.operation.Tags = append(r.operation.Tags, tags...)
 	return r