From 5881d18c5166b4d0762d8b3bbd0189422150e25b Mon Sep 17 00:00:00 2001 From: Fenny Date: Mon, 3 Feb 2020 13:55:37 +0100 Subject: [PATCH] Update middlewares --- middleware/cors.go | 8 ++++++-- middleware/helmet.go | 2 +- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/middleware/cors.go b/middleware/cors.go index 170219bb..e7a0b79f 100644 --- a/middleware/cors.go +++ b/middleware/cors.go @@ -3,8 +3,12 @@ package middleware import "github.com/gofiber/fiber" // Cors : Enable cross-origin resource sharing (CORS) with various options. -func Cors(c *fiber.Ctx, d string) { - c.Set("Access-Control-Allow-Origin", d) // Set d to "*" for allow all domains +func Cors(c *fiber.Ctx, origin ...string) { + o := "*" + if len(origin) > 0 { + o = origin[0] + } + c.Set("Access-Control-Allow-Origin", o) c.Set("Access-Control-Allow-Headers", "X-Requested-With") c.Next() } diff --git a/middleware/helmet.go b/middleware/helmet.go index 4d39683d..620396f4 100644 --- a/middleware/helmet.go +++ b/middleware/helmet.go @@ -8,6 +8,6 @@ import ( // Helmet : Helps secure your apps by setting various HTTP headers. func Helmet(c *fiber.Ctx) { - fmt.Println("Helmet is still under development, disable until v1.0.0") + fmt.Println("Helmet is still under development, this middleware does nothing yet.") c.Next() }