⚡ update qq/gitee/github oauth2 login

router/modules/oauth_router.go

@@ -14,8 +14,8 @@ func OauthRouter(router *gin.RouterGroup) {
 		{
 			wechatRouter.GET("/generate_client_id", oauth.GenerateClientId)
 			wechatRouter.GET("/get_temp_qrcode", oauth.GetTempQrCode)
-			wechatRouter.GET("/callback", oauth.CallbackVerify)
-			//wechatRouter.POST("/callback", oauth.CallbackNotify)
+			//wechatRouter.GET("/callback", oauth.CallbackVerify)
+			wechatRouter.POST("/callback", oauth.CallbackNotify)
 		}
 		githubRouter := group.Group("/github")
 		{

router/modules/role_router.go

@@ -0,0 +1,16 @@
+package modules
+
+import (
+	"github.com/gin-gonic/gin"
+	"schisandra-cloud-album/api"
+	"schisandra-cloud-album/middleware"
+)
+
+var roleApi = api.Api.RoleApi
+
+func RoleRouter(router *gin.RouterGroup) {
+	group := router.Group("/auth")
+	group.Use(middleware.JWTAuthMiddleware())
+	group.POST("/role/create", roleApi.CreateRole)
+	group.POST("/role/add_role_to_user", roleApi.AddRoleToUser)
+}

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
 }