quyun-v2/backend/app/services/user.go

package services

import (
	"context"
	"errors"
	"time"

	"quyun/v2/app/errorx"
	auth_dto "quyun/v2/app/http/v1/dto"
	user_dto "quyun/v2/app/http/v1/dto"
	"quyun/v2/database/models"
	"quyun/v2/pkg/consts"
	"quyun/v2/providers/jwt"

	"github.com/spf13/cast"
	"go.ipao.vip/gen/types"
	"gorm.io/gorm"
)

// @provider
type user struct {
	jwt *jwt.JWT
}

// SendOTP 发送验证码
// 当前仅模拟发送，实际应对接短信服务
func (s *user) SendOTP(ctx context.Context, phone string) error {
	// TODO: 对接短信服务
	// 模拟发送成功
	return nil
}

// LoginWithOTP 手机号验证码登录/注册
func (s *user) LoginWithOTP(ctx context.Context, phone, otp string) (*auth_dto.LoginResponse, error) {
	// 1. 校验验证码 (模拟：固定 123456)
	if otp != "123456" {
		return nil, errorx.ErrInvalidCredentials.WithMsg("验证码错误")
	}

	// 2. 查询或创建用户
	tbl, query := models.UserQuery.QueryContext(ctx)
	u, err := query.Where(tbl.Phone.Eq(phone)).First()
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			// 创建新用户
			u = &models.User{
				Phone:    phone,
				Username: phone, // 默认用户名 = 手机号
				Password: "",    // 免密登录
				Nickname: "User_" + phone[len(phone)-4:],
				Status:   string(consts.UserStatusVerified), // 默认已审核？需确认业务逻辑
				Roles:    types.Array[consts.Role]{consts.RoleUser},
			}
			if err := query.Create(u); err != nil {
				return nil, errorx.ErrDatabaseError.WithMsg("创建用户失败")
			}
		} else {
			return nil, errorx.ErrDatabaseError.WithMsg("查询用户失败")
		}
	}

	// 3. 检查状态
	if u.Status == string(consts.UserStatusBanned) {
		return nil, errorx.ErrAccountDisabled
	}

	// 4. 生成 Token
	token, err := s.jwt.CreateToken(s.jwt.CreateClaims(jwt.BaseClaims{
		UserID: u.ID,
		// TenantID: 0, // 初始登录无租户上下文
	}))
	if err != nil {
		return nil, errorx.ErrInternalError.WithMsg("生成令牌失败")
	}

	return &auth_dto.LoginResponse{
		Token: token,
		User:  s.toAuthUserDTO(u),
	}, nil
}

// Me 获取当前用户信息
func (s *user) Me(ctx context.Context) (*auth_dto.User, error) {
	userID := ctx.Value(consts.CtxKeyUser)
	if userID == nil {
		return nil, errorx.ErrUnauthorized
	}

	uid := cast.ToInt64(userID)
	tbl, query := models.UserQuery.QueryContext(ctx)
	u, err := query.Where(tbl.ID.Eq(uid)).First()
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, errorx.ErrRecordNotFound
		}
		return nil, errorx.ErrDatabaseError
	}

	return s.toAuthUserDTO(u), nil
}

// Update 更新用户信息
func (s *user) Update(ctx context.Context, form *user_dto.UserUpdate) error {
	userID := ctx.Value(consts.CtxKeyUser)
	if userID == nil {
		return errorx.ErrUnauthorized
	}
	uid := cast.ToInt64(userID)

	tbl, query := models.UserQuery.QueryContext(ctx)
	_, err := query.Where(tbl.ID.Eq(uid)).Updates(&models.User{
		Nickname: form.Nickname,
		Avatar:   form.Avatar,
		Gender:   form.Gender,
		Bio:      form.Bio,
		// Birthday: form.Birthday, // 类型转换需处理
	})
	if err != nil {
		return errorx.ErrDatabaseError
	}

	return nil
}

// RealName 实名认证
func (s *user) RealName(ctx context.Context, form *user_dto.RealNameForm) error {
	userID := ctx.Value(consts.CtxKeyUser)
	if userID == nil {
		return errorx.ErrUnauthorized
	}
	uid := cast.ToInt64(userID)

	// TODO: 调用实名认证接口校验

	tbl, query := models.UserQuery.QueryContext(ctx)
	_, err := query.Where(tbl.ID.Eq(uid)).Updates(&models.User{
		IsRealNameVerified: true,
		// RealName: form.Realname, // 需在 user 表添加字段? payout_accounts 有 realname
	})
	if err != nil {
		return errorx.ErrDatabaseError
	}
	return nil
}

// GetNotifications 获取通知
func (s *user) GetNotifications(ctx context.Context, typeArg string) ([]user_dto.Notification, error) {
	userID := ctx.Value(consts.CtxKeyUser)
	if userID == nil {
		return nil, errorx.ErrUnauthorized
	}
	uid := cast.ToInt64(userID)

	tbl, query := models.NotificationQuery.QueryContext(ctx)
	query = query.Where(tbl.UserID.Eq(uid))
	if typeArg != "" && typeArg != "all" {
		query = query.Where(tbl.Type.Eq(typeArg))
	}

	list, err := query.Order(tbl.CreatedAt.Desc()).Find()
	if err != nil {
		return nil, errorx.ErrDatabaseError
	}

	result := make([]user_dto.Notification, len(list))
	for i, v := range list {
		result[i] = user_dto.Notification{
			ID:      cast.ToString(v.ID),
			Type:    v.Type,
			Title:   v.Title,
			Content: v.Content,
			Read:    v.IsRead,
			Time:    v.CreatedAt.Format(time.RFC3339),
		}
	}
	return result, nil
}

func (s *user) toAuthUserDTO(u *models.User) *auth_dto.User {
	return &auth_dto.User{
		ID:                 cast.ToString(u.ID),
		Phone:              u.Phone,
		Nickname:           u.Nickname,
		Avatar:             u.Avatar,
		Gender:             consts.Gender(u.Gender),
		Bio:                u.Bio,
		Balance:            float64(u.Balance) / 100.0,
		Points:             u.Points,
		IsRealNameVerified: u.IsRealNameVerified,
	}
}