Sure! Pl

internal/service/auth_service.go

@@ -0,0 +1,111 @@
+package service
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"strings"
+
+	"wx_service/internal/model"
+
+	"gorm.io/gorm"
+)
+
+var ErrCodeRequired = errors.New("code is required")
+
+type AuthService struct {
+	db     *gorm.DB
+	wechat *WeChatClient
+}
+
+type LoginRequest struct {
+	Code      string
+	NickName  string
+	AvatarURL string
+	Gender    *int
+	Phone     string
+}
+
+type LoginResult struct {
+	User       *model.User
+	SessionKey string
+}
+
+func NewAuthService(db *gorm.DB, wechat *WeChatClient) *AuthService {
+	return &AuthService{
+		db:     db,
+		wechat: wechat,
+	}
+}
+
+func (s *AuthService) LoginWithCode(ctx context.Context, req LoginRequest) (*LoginResult, error) {
+	if strings.TrimSpace(req.Code) == "" {
+		return nil, ErrCodeRequired
+	}
+
+	session, err := s.wechat.Code2Session(ctx, req.Code)
+	if err != nil {
+		return nil, err
+	}
+	if session.OpenID == "" {
+		return nil, fmt.Errorf("wechat response missing openid")
+	}
+
+	tx := s.db.WithContext(ctx)
+	var user model.User
+	err = tx.Where("open_id = ?", session.OpenID).First(&user).Error
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		user = model.User{
+			OpenID:     session.OpenID,
+			UnionID:    session.UnionID,
+			NickName:   req.NickName,
+			AvatarURL:  req.AvatarURL,
+			Phone:      req.Phone,
+			SessionKey: session.SessionKey,
+		}
+		if req.Gender != nil {
+			user.Gender = *req.Gender
+		}
+		if err := tx.Create(&user).Error; err != nil {
+			return nil, fmt.Errorf("create user: %w", err)
+		}
+	} else if err != nil {
+		return nil, fmt.Errorf("query user: %w", err)
+	} else {
+		updates := map[string]interface{}{
+			"session_key": session.SessionKey,
+		}
+		if session.UnionID != "" && session.UnionID != user.UnionID {
+			updates["union_id"] = session.UnionID
+			user.UnionID = session.UnionID
+		}
+		if req.NickName != "" && req.NickName != user.NickName {
+			updates["nick_name"] = req.NickName
+			user.NickName = req.NickName
+		}
+		if req.AvatarURL != "" && req.AvatarURL != user.AvatarURL {
+			updates["avatar_url"] = req.AvatarURL
+			user.AvatarURL = req.AvatarURL
+		}
+		if req.Phone != "" && req.Phone != user.Phone {
+			updates["phone"] = req.Phone
+			user.Phone = req.Phone
+		}
+		if req.Gender != nil && user.Gender != *req.Gender {
+			updates["gender"] = *req.Gender
+			user.Gender = *req.Gender
+		}
+		if len(updates) > 0 {
+			if err := tx.Model(&user).Updates(updates).Error; err != nil {
+				return nil, fmt.Errorf("update user: %w", err)
+			}
+		}
+		user.SessionKey = session.SessionKey
+	}
+
+	result := &LoginResult{
+		User:       &user,
+		SessionKey: session.SessionKey,
+	}
+	return result, nil
+}