From dc84dbbb44c96449e72779ecaa96169cab23bb7c Mon Sep 17 00:00:00 2001
From: wucongxing8150 <815046773@qq.com>
Date: Wed, 18 Jun 2025 17:08:12 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BA=86=E7=99=BB=E5=BD=95?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/controller/Public.go | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

diff --git a/api/controller/Public.go b/api/controller/Public.go
index 64872c8..c833800 100644
--- a/api/controller/Public.go
+++ b/api/controller/Public.go
@@ -245,7 +245,7 @@ func (b *Public) LoginProgram(c *gin.Context) {
 	req := publicRequest.LoginProgram
 	if err := c.ShouldBind(&req); err != nil {
 		c.JSON(http.StatusOK, gin.H{
-			"message": "登录失败",
+			"message": "登录失败1",
 			"code":    -2,
 			"data": gin.H{
 				"source": 2,
@@ -258,7 +258,7 @@ func (b *Public) LoginProgram(c *gin.Context) {
 	// 参数验证
 	if err := global.Validate.Struct(req); err != nil {
 		c.JSON(http.StatusOK, gin.H{
-			"message": "登录失败",
+			"message": "登录失败2",
 			"code":    -2,
 			"data": gin.H{
 				"source": 2,
@@ -272,7 +272,7 @@ func (b *Public) LoginProgram(c *gin.Context) {
 	t, err := utils.ParseJwt(req.Token)
 	if err != nil {
 		c.JSON(http.StatusOK, gin.H{
-			"message": "登录失败",
+			"message": "登录失败3",
 			"code":    -2,
 			"data": gin.H{
 				"source": 2,
@@ -286,7 +286,7 @@ func (b *Public) LoginProgram(c *gin.Context) {
 	userId, err := strconv.ParseInt(t.UserId, 10, 64)
 	if err != nil {
 		c.JSON(http.StatusOK, gin.H{
-			"message": "登录失败",
+			"message": "登录失败4",
 			"code":    -2,
 			"data": gin.H{
 				"source": 2,
@@ -313,17 +313,36 @@ func (b *Public) LoginProgram(c *gin.Context) {
 		return
 	}
 
+	var platformId int
+	if user.RegisterSource == 2 {
+		platformId = 1
+	} else if user.RegisterSource == 3 {
+		platformId = 2
+	}
+
+	if platformId == 0 {
+		c.JSON(http.StatusOK, gin.H{
+			"message": "登录失败",
+			"code":    -2,
+			"data": gin.H{
+				"source": 2,
+			},
+		})
+
+		return
+	}
+
 	// 下发token
 	token := &utils.Token{
 		UserId:     fmt.Sprintf("%d", user.UserId),
-		PlatformId: fmt.Sprintf("%d", 2),
+		PlatformId: fmt.Sprintf("%d", platformId),
 	}
 
 	// 生成jwt
 	jwt, err := token.NewJWT()
 	if err != nil || jwt == "" {
 		c.JSON(http.StatusOK, gin.H{
-			"message": "登陆失败",
+			"message": "登陆失败6",
 			"code":    -1,
 			"data": gin.H{
 				"source": 2,