nepiedg
112 lines
2.5 KiB
Go
112 lines
2.5 KiB
Go
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
|
|
}
|