105 lines
2.3 KiB
Go
105 lines
2.3 KiB
Go
package middleware
|
||
|
||
import (
|
||
"tyapi-server/internal/config"
|
||
|
||
"github.com/gin-contrib/cors"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// CORSMiddleware CORS中间件
|
||
type CORSMiddleware struct {
|
||
config *config.Config
|
||
}
|
||
|
||
// NewCORSMiddleware 创建CORS中间件
|
||
func NewCORSMiddleware(cfg *config.Config) *CORSMiddleware {
|
||
return &CORSMiddleware{
|
||
config: cfg,
|
||
}
|
||
}
|
||
|
||
// GetName 返回中间件名称
|
||
func (m *CORSMiddleware) GetName() string {
|
||
return "cors"
|
||
}
|
||
|
||
// GetPriority 返回中间件优先级
|
||
func (m *CORSMiddleware) GetPriority() int {
|
||
return 100 // 高优先级，最先执行
|
||
}
|
||
|
||
// Handle 返回中间件处理函数
|
||
func (m *CORSMiddleware) Handle() gin.HandlerFunc {
|
||
if !m.config.Development.EnableCors {
|
||
// 如果没有启用CORS，返回空处理函数
|
||
return func(c *gin.Context) {
|
||
c.Next()
|
||
}
|
||
}
|
||
|
||
config := cors.Config{
|
||
AllowAllOrigins: false,
|
||
AllowOrigins: m.getAllowedOrigins(),
|
||
AllowMethods: m.getAllowedMethods(),
|
||
AllowHeaders: m.getAllowedHeaders(),
|
||
ExposeHeaders: []string{
|
||
"Content-Length",
|
||
"Content-Type",
|
||
"X-Request-ID",
|
||
"X-Response-Time",
|
||
},
|
||
AllowCredentials: true,
|
||
MaxAge: 86400, // 24小时
|
||
}
|
||
|
||
return cors.New(config)
|
||
}
|
||
|
||
// IsGlobal 是否为全局中间件
|
||
func (m *CORSMiddleware) IsGlobal() bool {
|
||
return true
|
||
}
|
||
|
||
// getAllowedOrigins 获取允许的来源
|
||
func (m *CORSMiddleware) getAllowedOrigins() []string {
|
||
if m.config.Development.CorsOrigins == "" {
|
||
return []string{"http://localhost:3000", "http://localhost:8080"}
|
||
}
|
||
|
||
// TODO: 解析配置中的origins字符串
|
||
return []string{m.config.Development.CorsOrigins}
|
||
}
|
||
|
||
// getAllowedMethods 获取允许的方法
|
||
func (m *CORSMiddleware) getAllowedMethods() []string {
|
||
if m.config.Development.CorsMethods == "" {
|
||
return []string{
|
||
"GET", "POST", "PUT", "PATCH", "DELETE", "HEAD", "OPTIONS",
|
||
}
|
||
}
|
||
|
||
// TODO: 解析配置中的methods字符串
|
||
return []string{m.config.Development.CorsMethods}
|
||
}
|
||
|
||
// getAllowedHeaders 获取允许的头部
|
||
func (m *CORSMiddleware) getAllowedHeaders() []string {
|
||
if m.config.Development.CorsHeaders == "" {
|
||
return []string{
|
||
"Origin",
|
||
"Content-Length",
|
||
"Content-Type",
|
||
"Authorization",
|
||
"X-Requested-With",
|
||
"Accept",
|
||
"Accept-Encoding",
|
||
"Accept-Language",
|
||
"X-Request-ID",
|
||
}
|
||
}
|
||
|
||
// TODO: 解析配置中的headers字符串
|
||
return []string{m.config.Development.CorsHeaders}
|
||
}
|