批量优化

go/controllers/auth/wechat_miniapp.go

@@ -10,13 +10,13 @@ import (
 	"server/services/wechatmini"
 
 	"github.com/beego/beego/v2/client/orm"
-	beego "github.com/beego/beego/v2/server/web"
 	"github.com/beego/beego/v2/core/logs"
+	beego "github.com/beego/beego/v2/server/web"
 )
 
 // 小程序登录失败原因（前端按此分支给出不同引导）
 const (
-	MiniReasonNotConfigured    = "not_configured"    // 服务端未配置小程序
+	MiniReasonNotConfigured    = "not_configured"     // 服务端未配置小程序
 	MiniReasonIdentityNotFound = "identity_not_found" // 微信未匹配到云泽账号
 	MiniReasonNoTenant         = "no_tenant"          // 账号未加入任何企业
 	MiniReasonWechatAPI        = "wechat_api_error"   // 微信接口报错
@@ -49,10 +49,19 @@ func (c *AuthWechatMiniController) fail(reason string, bizCode int, msg string)
 	c.serveJSON(map[string]interface{}{"code": bizCode, "msg": msg, "reason": reason})
 }
 
+// loadMiniConfig 按客户端传入的 appid 选择小程序配置（多应用支持），
+// 未传 appid 时回退第一条启用配置
+func loadMiniConfig(appID string) (*wechatmini.Config, error) {
+	if appID = strings.TrimSpace(appID); appID != "" {
+		return wechatmini.LoadEnabledAppByAppID(appID)
+	}
+	return wechatmini.LoadEnabledConfig()
+}
+
 // Config 小程序登录是否可用
-// GET /auth/wechat/miniapp/config
+// GET /auth/wechat/miniapp/config?appid=xxx（多应用时可指定，缺省取第一条启用配置）
 func (c *AuthWechatMiniController) Config() {
-	cfg, err := wechatmini.LoadEnabledConfig()
+	cfg, err := loadMiniConfig(c.GetString("appid"))
 	if err != nil {
 		c.serveJSON(map[string]interface{}{
 			"code": 200,
@@ -72,6 +81,7 @@ func (c *AuthWechatMiniController) Login() {
 		Code      string `json:"code"`
 		PhoneCode string `json:"phone_code"`
 		ClientID  string `json:"client_id"`
+		AppID     string `json:"appid"`
 	}
 	body := c.Ctx.Input.RequestBody
 	if len(body) == 0 {
@@ -92,7 +102,7 @@ func (c *AuthWechatMiniController) Login() {
 		return
 	}
 
-	cfg, err := wechatmini.LoadEnabledConfig()
+	cfg, err := loadMiniConfig(req.AppID)
 	if err != nil {
 		c.fail(MiniReasonNotConfigured, 400, "小程序登录未开通，请使用账号密码登录")
 		return
@@ -193,7 +203,8 @@ func (c *AuthWechatMiniController) Bind() {
 	}
 
 	var req struct {
-		Code string `json:"code"`
+		Code  string `json:"code"`
+		AppID string `json:"appid"`
 	}
 	body := c.Ctx.Input.RequestBody
 	if len(body) == 0 {
@@ -209,7 +220,7 @@ func (c *AuthWechatMiniController) Bind() {
 		return
 	}
 
-	cfg, err := wechatmini.LoadEnabledConfig()
+	cfg, err := loadMiniConfig(req.AppID)
 	if err != nil {
 		c.fail(MiniReasonNotConfigured, 400, "小程序登录未开通")
 		return