Merge pull request #11 from go-cinch/piupuer/dev

[fix]get method/uri from header if exist
go-cinch · Oct 5, 2024 · a95b59f · a95b59f
2 parents 77004ab + 5881f73
commit a95b59f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/server/middleware/whitelist.go b/internal/server/middleware/whitelist.go
@@ -30,6 +30,15 @@ func Whitelist(whitelist *biz.WhitelistUseCase) middleware.Middleware {
 					err = biz.ErrNoPermission(ctx)
 					return
 				}
+				// get from header if exist
+				method := tr.RequestHeader().Get("x-original-method")
+				uri := tr.RequestHeader().Get("x-permission-uri")
+				if v.Method == nil && method != "" {
+					v.Method = &method
+				}
+				if v.Uri == nil && uri != "" {
+					v.Uri = &uri
+				}
 				// check whitelist
 				pass, err = hasPermissionWhitelist(ctx, whitelist, v)
 				if err != nil {