⚡ update qq/gitee/github oauth2 login

router/router.go

@@ -12,14 +12,21 @@ import (
 func InitRouter() *gin.Engine {
 	gin.SetMode(global.CONFIG.System.Env)
 	router := gin.Default()
-	err := router.SetTrustedProxies([]string{"127.0.0.1"})
+	err := router.SetTrustedProxies([]string{global.CONFIG.System.Ip})
 	if err != nil {
 		global.LOG.Error(err)
 		return nil
 	}
+	router.Use(middleware.RateLimitMiddleware(time.Millisecond*100, 20)) // 限流中间件
 	publicGroup := router.Group("api")
 	// 跨域设置
-	publicGroup.Use(cors.Default()).Use(middleware.RateLimitMiddleware(time.Millisecond*100, 20))
+	publicGroup.Use(cors.New(cors.Config{
+		AllowOrigins:     []string{global.CONFIG.System.Web},
+		AllowMethods:     []string{"GET", "POST", "PUT", "PATCH", "DELETE", "HEAD"},
+		AllowHeaders:     []string{"Origin", "Content-Length", "Content-Type", "Authorization", "X-CSRF-Token", "Accept-Language"},
+		AllowCredentials: true,
+		MaxAge:           12 * time.Hour,
+	}))
 	// 国际化设置
 	publicGroup.Use(middleware.I18n())
 
@@ -29,5 +36,6 @@ func InitRouter() *gin.Engine {
 	modules.SmsRouter(publicGroup)       // 注册短信验证码路由
 	modules.OauthRouter(publicGroup)     // 注册oauth路由
 	modules.WebsocketRouter(publicGroup) // 注册websocket路由
+	modules.RoleRouter(publicGroup)      // 注册角色路由
 	return router
 }