mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
27 lines
586 B
Go
27 lines
586 B
Go
package middleware
|
|
|
|
import (
|
|
"openflare/common"
|
|
"strings"
|
|
|
|
"github.com/gin-contrib/cors"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func CORS() gin.HandlerFunc {
|
|
config := cors.DefaultConfig()
|
|
config.AllowCredentials = true
|
|
config.AllowHeaders = []string{"Origin", "Content-Length", "Content-Type", "Authorization", "X-Agent-Token", "Accept"}
|
|
config.AllowOriginFunc = func(origin string) bool {
|
|
serverAddr := strings.TrimRight(common.ServerAddress, "/")
|
|
if serverAddr == "" {
|
|
return true
|
|
}
|
|
if origin == serverAddr {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
return cors.New(config)
|
|
}
|