workspace/code/allapp/allapp-go-v3/internal/handle/auth.go

package handle

import (
	"allapp-go/internal/errors"
	"allapp-go/internal/httpx"
	"allapp-go/internal/types"
	"allapp-go/pkg/db"
	"allapp-go/pkg/jwtx"
	"allapp-go/pkg/uniqueid"
	"allapp-go/pkg/wechat"
	"fmt"

	"github.com/gofiber/fiber/v3"
	"github.com/gofiber/fiber/v3/log"
)

// LoginQq ======================== QQ 登录 ========================
func LoginQq(c fiber.Ctx) error {
	var req types.LoginQqDTO
	vo := new(types.LoginVO)

	if err := httpx.BindAndValidate(c, &req); err != nil {
		return errors.WithStack(err)
	}

	return handleThirdLogin(c, req.Openid, req.Nickname, req.Avatar, 1, vo)
}

// LoginWechat ======================== 微信登录 ========================
func LoginWechat(c fiber.Ctx) error {
	var req types.LoginWechatDTO
	vo := new(types.LoginVO)

	if err := httpx.BindAndValidate(c, &req); err != nil {
		return errors.WithStack(err)
	}

	openid, token, err := wechat.GetWechatAccess(req.Code)
	if err != nil {
		log.Errorw("wechat access failed", "code", req.Code, "error", err)
		return httpx.Fail(c, "微信登录失败，请重试")
	}

	nickname, avatar, err := wechat.GetWechatUserInfo(token, openid)
	if err != nil {
		log.Errorw("wechat userinfo failed", "openid", openid, "error", err)
		return httpx.Fail(c, "微信登录失败，请重试")
	}

	return handleThirdLogin(c, openid, nickname, avatar, 0, vo)
}

// ======================== 第三方登录主流程 ========================
func handleThirdLogin(
	c fiber.Ctx,
	openid string,
	nickname string,
	avatar string,
	loginType int16,
	vo *types.LoginVO,
) error {

	dbClient := db.New()

	user, err := getUserByOpenID(dbClient, c, openid, loginType)
	if err != nil {
		return errors.WithStack(err)
	}

	// 不存在 -> 注册
	if user == nil {
		return registerUser(dbClient, c, openid, nickname, avatar, loginType, vo)
	}

	// ========= 校验状态 =========
	id := toInt64(user["id"])
	status := toInt64(user["status"])

	if status == 0 {
		return httpx.Fail(c, "账号已被禁用")
	}

	// ========= 更新登录时间 =========
	if err := dbClient.Update(
		c.Context(),
		"user",
		"id",
		map[string]any{
			"id":              id,
			"last_login_time": "NOW()",
		},
	); err != nil {
		return errors.WithStack(err)
	}

	// ========= token =========
	token, err := jwtx.CreateToken(c.Context(), map[string]any{
		"id": id,
	})
	if err != nil {
		return errors.WithStack(err)
	}

	// ========= 返回 =========
	vo.Token = token
	vo.UserId = id
	vo.Nickname, _ = user["nickname"].(string)
	vo.Avatar, _ = user["avatar"].(string)

	return httpx.OK(c, vo)
}

// ======================== 注册 ========================
func registerUser(
	dbClient *db.Client,
	c fiber.Ctx,
	openid, nickname, avatar string,
	loginType int16,
	vo *types.LoginVO,
) error {
	userId := uniqueid.NextId()

	if nickname == "" {
		nickname = fmt.Sprintf("用户_%d", userId)
	}
	if avatar == "" {
		avatar = "https://default-avatar-url.com/default.png"
	}

	err := dbClient.WithTx(c.Context(), func(tx *db.Client) error {

		if err := tx.Insert(c.Context(), "user", "id", map[string]any{
			"id":              userId,
			"nickname":        nickname,
			"avatar":          avatar,
			"status":          1,
			"create_time":     "NOW()",
			"update_time":     "NOW()",
			"last_login_time": "NOW()",
		}); err != nil {
			return err
		}

		if err := tx.Insert(c.Context(), "user_oauth", "id", map[string]any{
			"id":          uniqueid.NextId(),
			"user_id":     userId,
			"type":        loginType,
			"openid":      openid,
			"create_time": "NOW()",
			"update_time": "NOW()",
		}); err != nil {
			return err
		}

		return nil
	})

	if err != nil {
		return errors.WithStack(err)
	}

	token, err := jwtx.CreateToken(c.Context(), map[string]any{
		"id": userId,
	})
	if err != nil {
		return errors.WithStack(err)
	}

	vo.Token = token
	vo.UserId = userId
	vo.Nickname = nickname
	vo.Avatar = avatar

	return httpx.OK(c, vo)
}

// ======================== DB 查询封装（去重复 SQL） ========================
func getUserByOpenID(dbClient *db.Client, c fiber.Ctx, openid string, loginType int16) (map[string]any, error) {

	users, err := dbClient.LoadDataBySQL(
		c.Context(),
		"SELECT u.* FROM user_oauth o JOIN user u ON u.id = o.user_id WHERE o.openid = ? AND o.type = ?",
		[]any{openid, loginType},
	)

	if err != nil {
		return nil, err
	}

	if len(users) == 0 {
		return nil, nil
	}

	return users[0], nil
}

// ======================== 类型工具（极简版） ========================
func toInt64(v any) int64 {
	i, _ := v.(int64)
	return i
}