feat: Add a general redirect method

---
添加通用跳转方式

Author: dengbinbox

cmd/main.go

@@ -143,10 +143,11 @@ var serveCmd = &cobra.Command{
 		go func() {
 			log.Info(nil, "Starting server...")
 			server := handler.Server{
-				ServerPort: globalConfig.Server.ServerPort,
-				BaseURL:    globalConfig.Server.BaseURL,
-				HTTPClient: initHTTPClient(nil),
-				IsPrivate:  globalConfig.Server.IsPrivate,
+				ServerPort:  globalConfig.Server.ServerPort,
+				BaseURL:     globalConfig.Server.BaseURL,
+				HTTPClient:  initHTTPClient(nil),
+				IsPrivate:   globalConfig.Server.IsPrivate,
+				RedirectURL: globalConfig.Redirect.Uris,
 			}
 			if err := server.StartServer(); err != nil {
 				log.Error(nil, "Server error: %v", err)

config/config.yaml

@@ -152,3 +152,7 @@ log:
 
   # Compress old log files to save disk space
   compress: true  # Compress old logs
+
+redirect:
+  uris:
+    "act_2026": "/credit/manager/annual-summary"

internal/config/config.go

@@ -21,6 +21,7 @@ type AppConfig struct {
 	SMS          SMSConfig                 `json:"sms" mapstructure:"sms" validate:"required"`
 	Providers    map[string]ProviderConfig `json:"providers" mapstructure:"providers"`
 	QuotaManager QuotaConfig               `json:"quotaManager" mapstructure:"quotaManager"`
+	Redirect     RedirectConfig            `json:"redirect" mapstructure:"redirect"`
 }
 
 type Server struct {
@@ -30,6 +31,10 @@ type Server struct {
 	IsPrivate  bool              `json:"isPrivate" mapstructure:"isPrivate"`
 }
 
+type RedirectConfig struct {
+	Uris map[string]string `json:"uris" mapstructure:"uris"`
+}
+
 type HTTPClientConfig struct {
 	Timeout               time.Duration `json:"timeout" mapstructure:"timeout" validate:"gte=0"`
 	DialTimeout           time.Duration `json:"dialTimeout" mapstructure:"dialTimeout" validate:"gte=0"`

internal/handler/router.go

@@ -1,18 +1,20 @@
 package handler
 
 import (
-	"github.com/gin-gonic/gin"
 	"net/http"
 
+	"github.com/gin-gonic/gin"
+
 	"github.com/zgsm-ai/oidc-auth/internal/middleware"
 	"github.com/zgsm-ai/oidc-auth/pkg/log"
 )
 
 type Server struct {
-	ServerPort string
-	BaseURL    string
-	HTTPClient *http.Client
-	IsPrivate  bool
+	ServerPort  string
+	BaseURL     string
+	HTTPClient  *http.Client
+	IsPrivate   bool
+	RedirectURL map[string]string
 }
 
 type ParameterCarrier struct {
@@ -50,6 +52,7 @@ func (s *Server) SetupRouter(r *gin.Engine) {
 		webOauthServer.GET("userinfo", s.userInfoHandler)
 		webOauthServer.GET("login", s.webLoginHandler)
 		webOauthServer.GET("login/callback", s.webLoginCallbackHandler)
+		webOauthServer.GET("login/callback/:service", s.webLoginCallbackHandler)
 		webOauthServer.GET("invite-code", s.getUserInviteCodeHandler)
 	}
 	r.POST("/oidc-auth/api/v1/send/sms", s.SMSHandler)

internal/handler/web_handler.go

@@ -26,6 +26,7 @@ type WebParameterCarrier struct {
 func (s *Server) webLoginHandler(c *gin.Context) {
 	provider := c.DefaultQuery("provider", "casdoor")
 	inviterCode := c.DefaultQuery("inviter_code", "")
+	redirectService := c.DefaultQuery("redirect_service", "")
 
 	oauthManager := providers.GetManager()
 	providerInstance, err := oauthManager.GetProvider(provider)
@@ -37,7 +38,9 @@ func (s *Server) webLoginHandler(c *gin.Context) {
 	// Use inviterCode as state parameter
 	state := inviterCode
 	authURL := providerInstance.GetAuthURL(state, s.BaseURL+constants.WebLoginCallbackURI)
-
+	if redirectService != "" {
+		authURL = fmt.Sprintf("%s/%s", authURL, redirectService)
+	}
 	response.JSONSuccess(c, "", map[string]interface{}{
 		"state":        state,
 		"inviter_code": inviterCode,
@@ -49,7 +52,8 @@ func (s *Server) webLoginHandler(c *gin.Context) {
 func (s *Server) webLoginCallbackHandler(c *gin.Context) {
 	code := c.DefaultQuery("code", "")
 	state := c.DefaultQuery("state", "")
-	inviterCode := state // inviter code is in the state parameter
+	inviterCode := state          // inviter code is in the state parameter
+	service := c.Param("service") // service parameter for custom redirect
 
 	if code == "" {
 		response.JSONError(c, http.StatusBadRequest, errs.ErrBadRequestParam,
@@ -130,8 +134,22 @@ func (s *Server) webLoginCallbackHandler(c *gin.Context) {
 		tokenHash = user.Devices[0].AccessTokenHash
 	}
 
-	// Redirect to bind account page with tokenHash as state parameter
-	redirectURL := providerInstance.GetEndpoint(false) + constants.BindAccountBindURI + "?state=" + tokenHash
+	// Determine redirect URL based on service parameter
+	var redirectURL string
+	if service != "" {
+		// Check RedirectConfig for custom redirect URI
+		if s.RedirectURL != nil {
+			if uri, ok := s.RedirectURL[service]; ok && uri != "" {
+				redirectURL = fmt.Sprintf("%s%s?state=%s", providerInstance.GetEndpoint(false), uri, tokenHash)
+			}
+		}
+	}
+
+	// If no custom redirect configured, use default
+	if redirectURL == "" {
+		redirectURL = providerInstance.GetEndpoint(false) + constants.BindAccountBindURI + "?state=" + tokenHash
+	}
+
 	c.Redirect(http.StatusFound, redirectURL)
 }