Sure! Pl

2025-12-29 09:32:44 +00:00
commit a5a24562b2
13 changed files with 722 additions and 0 deletions
@@ -0,0 +1,41 @@
+package database
+
+import (
+	"fmt"
+	"log"
+	"wx_service/config"
+
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
+
+var DB *gorm.DB
+
+func InitDB() error {
+	cfg := config.AppConfig.Database
+
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		cfg.User,
+		cfg.Password,
+		cfg.Host,
+		cfg.Port,
+		cfg.DBName,
+	)
+
+	var err error
+	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
+		Logger: logger.Default.LogMode(logger.Info),
+	})
+
+	if err != nil {
+		return fmt.Errorf("连接数据库失败: %v", err)
+	}
+
+	log.Println("数据库连接成功")
+	return nil
+}
+
+func AutoMigrate(models ...interface{}) error {
+	return DB.AutoMigrate(models...)
+}
@@ -0,0 +1,76 @@
+package handler
+
+import (
+	"errors"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+
+	"wx_service/internal/model"
+	"wx_service/internal/service"
+)
+
+type AuthHandler struct {
+	authService *service.AuthService
+}
+
+func NewAuthHandler(authService *service.AuthService) *AuthHandler {
+	return &AuthHandler{
+		authService: authService,
+	}
+}
+
+type weChatLoginRequest struct {
+	Code      string `json:"code" binding:"required"`
+	NickName  string `json:"nickname"`
+	AvatarURL string `json:"avatar_url"`
+	Gender    *int   `json:"gender"`
+	Phone     string `json:"phone"`
+}
+
+func (h *AuthHandler) LoginWithWeChat(c *gin.Context) {
+	var req weChatLoginRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid request payload"))
+		return
+	}
+
+	result, err := h.authService.LoginWithCode(c.Request.Context(), service.LoginRequest{
+		Code:      req.Code,
+		NickName:  req.NickName,
+		AvatarURL: req.AvatarURL,
+		Gender:    req.Gender,
+		Phone:     req.Phone,
+	})
+	if err != nil {
+		switch {
+		case errors.Is(err, service.ErrCodeRequired):
+			c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "code is required"))
+		default:
+			var apiErr *service.WeChatError
+			if errors.As(err, &apiErr) {
+				c.JSON(http.StatusBadGateway, model.Error(http.StatusBadGateway, apiErr.Error()))
+				return
+			}
+			c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "login failed"))
+		}
+		return
+	}
+
+	userPayload := gin.H{
+		"id":         result.User.ID,
+		"open_id":    result.User.OpenID,
+		"nickname":   result.User.NickName,
+		"avatar_url": result.User.AvatarURL,
+		"gender":     result.User.Gender,
+		"phone":      result.User.Phone,
+	}
+	if result.User.UnionID != "" {
+		userPayload["union_id"] = result.User.UnionID
+	}
+
+	c.JSON(http.StatusOK, model.Success(gin.H{
+		"user":        userPayload,
+		"session_key": result.SessionKey,
+	}))
+}
@@ -0,0 +1,26 @@
+package model
+
+import (
+	"time"
+
+	"gorm.io/gorm"
+)
+
+type Product struct {
+	ID          uint           `gorm:"primarykey" json:"id"`
+	CreatedAt   time.Time      `json:"created_at"`
+	UpdatedAt   time.Time      `json:"updated_at"`
+	DeletedAt   gorm.DeletedAt `gorm:"index" json:"-"`
+
+	Name        string  `gorm:"size:200;not null" json:"name"`
+	Description string  `gorm:"type:text" json:"description"`
+	Price       float64 `gorm:"type:decimal(10,2);not null" json:"price"`
+	Stock       int     `gorm:"default:0" json:"stock"`
+	ImageURL    string  `gorm:"size:500" json:"image_url"`
+	Category    string  `gorm:"size:50" json:"category"`
+	Status      int     `gorm:"default:1" json:"status"`
+}
+
+func (Product) TableName() string {
+	return "products"
+}
@@ -0,0 +1,22 @@
+package model
+
+type Response struct {
+	Code    int         `json:"code"`
+	Message string      `json:"message"`
+	Data    interface{} `json:"data,omitempty"`
+}
+
+func Success(data interface{}) Response {
+	return Response{
+		Code:    200,
+		Message: "success",
+		Data:    data,
+	}
+}
+
+func Error(code int, message string) Response {
+	return Response{
+		Code:    code,
+		Message: message,
+	}
+}
@@ -0,0 +1,27 @@
+package model
+
+import (
+	"time"
+
+	"gorm.io/gorm"
+)
+
+type User struct {
+	ID        uint           `gorm:"primarykey" json:"id"`
+	CreatedAt time.Time      `json:"created_at"`
+	UpdatedAt time.Time      `json:"updated_at"`
+	DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
+
+	OpenID    string `gorm:"uniqueIndex;size:100" json:"open_id"`
+	UnionID   string `gorm:"size:100" json:"union_id,omitempty"`
+	NickName  string `gorm:"size:100" json:"nickname"`
+	AvatarURL string `gorm:"size:500" json:"avatar_url"`
+	Gender    int    `gorm:"default:0" json:"gender"`
+	Phone     string `gorm:"size:20" json:"phone,omitempty"`
+
+	SessionKey string `gorm:"size:100" json:"-"`
+}
+
+func (User) TableName() string {
+	return "users"
+}
@@ -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
+}
@@ -0,0 +1,89 @@
+package service
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/url"
+	"time"
+)
+
+const weChatCode2SessionURL = "https://api.weixin.qq.com/sns/jscode2session"
+
+// WeChatClient 调用微信接口获取 session/openid。
+type WeChatClient struct {
+	appID     string
+	appSecret string
+	client    *http.Client
+}
+
+type WeChatSession struct {
+	OpenID     string `json:"openid"`
+	UnionID    string `json:"unionid"`
+	SessionKey string `json:"session_key"`
+}
+
+type weChatSessionResponse struct {
+	WeChatSession
+	ErrCode int    `json:"errcode"`
+	ErrMsg  string `json:"errmsg"`
+}
+
+// WeChatError 表示微信接口级错误。
+type WeChatError struct {
+	Code int
+	Msg  string
+}
+
+func (e *WeChatError) Error() string {
+	return fmt.Sprintf("wechat error: code=%d msg=%s", e.Code, e.Msg)
+}
+
+func NewWeChatClient(appID, appSecret string, client *http.Client) *WeChatClient {
+	if client == nil {
+		client = &http.Client{
+			Timeout: 5 * time.Second,
+		}
+	}
+
+	return &WeChatClient{
+		appID:     appID,
+		appSecret: appSecret,
+		client:    client,
+	}
+}
+
+func (c *WeChatClient) Code2Session(ctx context.Context, code string) (*WeChatSession, error) {
+	query := url.Values{}
+	query.Set("appid", c.appID)
+	query.Set("secret", c.appSecret)
+	query.Set("js_code", code)
+	query.Set("grant_type", "authorization_code")
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, fmt.Sprintf("%s?%s", weChatCode2SessionURL, query.Encode()), nil)
+	if err != nil {
+		return nil, fmt.Errorf("build wechat request: %w", err)
+	}
+
+	resp, err := c.client.Do(req)
+	if err != nil {
+		return nil, fmt.Errorf("call wechat api: %w", err)
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("wechat api unexpected status: %s", resp.Status)
+	}
+
+	var raw weChatSessionResponse
+	if err := json.NewDecoder(resp.Body).Decode(&raw); err != nil {
+		return nil, fmt.Errorf("decode wechat response: %w", err)
+	}
+
+	if raw.ErrCode != 0 {
+		return nil, &WeChatError{Code: raw.ErrCode, Msg: raw.ErrMsg}
+	}
+
+	return &raw.WeChatSession, nil
+}