update CORS middleware

lovego · Dec 21, 2021 · 79975d2 · 79975d2
1 parent b9cf445
commit 79975d2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -4,7 +4,7 @@ A golang http router with regexp and document generation support.
 [![Build Status](https://github.com/lovego/goa/actions/workflows/go.yml/badge.svg)](https://github.com/lovego/goa/actions/workflows/go.yml)
 [![Coverage Status](https://coveralls.io/repos/github/lovego/goa/badge.svg?branch=master&1)](https://coveralls.io/github/lovego/goa)
 [![Go Report Card](https://goreportcard.com/badge/github.com/lovego/goa)](https://goreportcard.com/report/github.com/lovego/goa)
-[![Documentation](https://pkg.go.dev/badge/github.com/lovego/goa)](https://pkg.go.dev/github.com/lovego/goa@v0.3.2)
+[![Documentation](https://pkg.go.dev/badge/github.com/lovego/goa)](https://pkg.go.dev/github.com/lovego/goa@v0.3.3)
 
 
 ## Usage
@@ -75,7 +75,7 @@ func main() {
 	server.ListenAndServe(router)
 }
 
-func allowOrigin(origin string) bool {
+func allowOrigin(origin string, c *goa.Context) bool {
 	u, err := url.Parse(origin)
 	if err != nil {
 		return false

diff --git a/benchmark/example/example.go b/benchmark/example/example.go
@@ -63,7 +63,7 @@ func main() {
 	server.ListenAndServe(router)
 }
 
-func allowOrigin(origin string) bool {
+func allowOrigin(origin string, c *goa.Context) bool {
 	u, err := url.Parse(origin)
 	if err != nil {
 		return false

diff --git a/middlewares/cors.go b/middlewares/cors.go
@@ -8,18 +8,18 @@ import (
 
 // crosss origin resource share
 type CORS struct {
-	allow     func(origin string) bool
+	allow     func(origin string, c *goa.Context) bool
 	SetHeader func(http.Header)
 }
 
-func NewCORS(allow func(origin string) bool) CORS {
+func NewCORS(allow func(origin string, c *goa.Context) bool) CORS {
 	return CORS{allow: allow}
 }
 
 func (cors CORS) Check(c *goa.Context) {
 	if c.Request.Header.Get(`Sec-Fetch-Site`) != "same-origin" {
 		if origin := c.Request.Header.Get(`Origin`); origin != `` && origin != c.Origin() {
-			if !cors.allow(origin) {
+			if !cors.allow(origin, c) {
 				c.WriteHeader(http.StatusForbidden)
 				c.Write([]byte(`origin not allowed.`))
 				return