feat: def cors options

hokamsingh · Aug 31, 2024 · e37fb3a · e37fb3a
1 parent fb45b70
commit e37fb3a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/internal/core/router/router.go b/internal/core/router/router.go
@@ -29,7 +29,7 @@ type Router struct {
 type Option func(*Router)
 
 // Default CORS options
-var _ = middleware.CORSOptions{
+var defCorsOpts = middleware.CORSOptions{
 	AllowedOrigins: []string{"*"},
 	AllowedMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
 	AllowedHeaders: []string{"Content-Type", "Authorization"},

diff --git a/pkg/lessgo/less.go b/pkg/lessgo/less.go
@@ -220,7 +220,7 @@ func App(options ...router.Option) *Router {
 
 // New Cors Options.
 //
-// Example
+// Example (default to)
 //
 //	 corsOptions := LessGo.NewCorsOptions(
 //		[]string{"*"}, // Allow all origins
@@ -229,6 +229,20 @@ func App(options ...router.Option) *Router {
 //
 // )
 func NewCorsOptions(origins []string, methods []string, headers []string) *CORSOptions {
+	var defCorsOpts = middleware.CORSOptions{
+		AllowedOrigins: []string{"*"},
+		AllowedMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
+		AllowedHeaders: []string{"Content-Type", "Authorization"},
+	}
+	if len(origins) == 0 {
+		origins = append(origins, defCorsOpts.AllowedOrigins...)
+	}
+	if len(headers) == 0 {
+		headers = append(headers, defCorsOpts.AllowedHeaders...)
+	}
+	if len(methods) == 0 {
+		methods = append(methods, defCorsOpts.AllowedMethods...)
+	}
 	return middleware.NewCorsOptions(origins, methods, headers)
 }