逐步替换代码

services/platform_auth.go

@@ -1,37 +1,72 @@
 package services
 
 import (
+	"crypto/md5"
+	"encoding/hex"
 	"errors"
+	"strings"
 
+	"server/models"
 	"server/pkg/jwtutil"
 )
 
-// PlatformLogin 平台登录业务
-// TODO: 后续接真实用户/租户表，这里先做最小可用实现。
-func PlatformLogin(username, password string) (string, error) {
-	// 临时简单校验：用户名和密码非空
-	if username == "" || password == "" {
-		return "", errors.New("用户名或密码不能为空")
-	}
-
-	// 测试账号：admin / admin123
-	if username != "admin" || password != "admin123" {
-		return "", errors.New("用户名或密码错误")
-	}
-
-	// 这里后续应：
-	// 1. 从平台用户表查询用户
-	// 2. 校验密码（含加盐加密）
-	// 3. 绑定平台/租户信息
-	// 目前先返回一个平台用户的 JWT 占位 token
-	const fakeUserID = 1
-	const fakeTenantID = 0
-	const userType = "platform"
-
-	token, err := jwtutil.GenerateToken(fakeUserID, username, fakeTenantID, userType)
-	if err != nil {
-		return "", err
-	}
-	return token, nil
+type PlatformLoginUser struct {
+	ID      uint64
+	Account string
+	Name    string
+	Rid     uint64
+	Avatar  string
+}
+
+func md5Hex(s string) string {
+	sum := md5.Sum([]byte(s))
+	return hex.EncodeToString(sum[:])
+}
+
+// PlatformLogin 平台登录业务（仅允许平台用户 yz_admin_user 登录）
+func PlatformLogin(account, password string) (string, *PlatformLoginUser, error) {
+	account = strings.TrimSpace(account)
+	password = strings.TrimSpace(password)
+	if account == "" || password == "" {
+		return "", nil, errors.New("用户名或密码不能为空")
+	}
+
+	var user models.AdminUser
+	err := models.Orm.QueryTable(new(models.AdminUser)).
+		Filter("account", account).
+		One(&user)
+	if err != nil {
+		return "", nil, errors.New("用户名或密码错误")
+	}
+	if user.Password != md5Hex(password) {
+		return "", nil, errors.New("用户名或密码错误")
+	}
+
+	if user.Status == 0 {
+		return "", nil, errors.New("账号已禁用")
+	}
+	const fakeTenantID = 0
+	const userType = "platform"
+	token, err := jwtutil.GenerateToken(int(user.ID), user.Account, fakeTenantID, userType)
+	if err != nil {
+		return "", nil, err
+	}
+
+	name := ""
+	if user.Name != nil {
+		name = *user.Name
+	}
+	avatar := ""
+	if user.Avatar != nil {
+		avatar = *user.Avatar
+	}
+	loginUser := &PlatformLoginUser{
+		ID:      user.ID,
+		Account: user.Account,
+		Name:    name,
+		Rid:     user.RoleID,
+		Avatar:  avatar,
+	}
+	return token, loginUser, nil
 }