zeronline
116 lines
2.9 KiB
Go
116 lines
2.9 KiB
Go
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/inchstep/server/internal/middleware"
|
|
"github.com/inchstep/server/internal/services"
|
|
)
|
|
|
|
type UserHandler struct {
|
|
userSvc *services.UserService
|
|
jwtSecret string
|
|
}
|
|
|
|
func NewUserHandler(userSvc *services.UserService, jwtSecret string) *UserHandler {
|
|
return &UserHandler{userSvc: userSvc, jwtSecret: jwtSecret}
|
|
}
|
|
|
|
type registerReq struct {
|
|
Phone string `json:"phone" binding:"required,len=11"`
|
|
}
|
|
|
|
func (h *UserHandler) Register(c *gin.Context) {
|
|
var req registerReq
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
middleware.Error(c, http.StatusBadRequest, 400, "手机号格式错误")
|
|
return
|
|
}
|
|
|
|
user, err := h.userSvc.RegisterByPhone(req.Phone)
|
|
if err != nil {
|
|
middleware.Error(c, http.StatusConflict, 409, err.Error())
|
|
return
|
|
}
|
|
|
|
token, _ := middleware.GenerateToken(user.ID, h.jwtSecret)
|
|
middleware.Success(c, gin.H{
|
|
"user": user,
|
|
"token": token,
|
|
})
|
|
}
|
|
|
|
type loginReq struct {
|
|
Phone string `json:"phone" binding:"required,len=11"`
|
|
}
|
|
|
|
func (h *UserHandler) Login(c *gin.Context) {
|
|
var req loginReq
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
middleware.Error(c, http.StatusBadRequest, 400, "手机号格式错误")
|
|
return
|
|
}
|
|
|
|
user, err := h.userSvc.GetByPhone(req.Phone)
|
|
if err != nil {
|
|
// 手机号未注册时自动注册
|
|
user, err = h.userSvc.RegisterByPhone(req.Phone)
|
|
if err != nil {
|
|
middleware.Error(c, http.StatusInternalServerError, 500, "注册失败")
|
|
return
|
|
}
|
|
}
|
|
|
|
token, _ := middleware.GenerateToken(user.ID, h.jwtSecret)
|
|
middleware.Success(c, gin.H{
|
|
"user": user,
|
|
"token": token,
|
|
})
|
|
}
|
|
|
|
func (h *UserHandler) GetProfile(c *gin.Context) {
|
|
userID := middleware.GetUserID(c)
|
|
user, err := h.userSvc.GetByID(userID)
|
|
if err != nil {
|
|
middleware.Error(c, http.StatusNotFound, 404, "用户不存在")
|
|
return
|
|
}
|
|
middleware.Success(c, user)
|
|
}
|
|
|
|
func (h *UserHandler) UpdateProfile(c *gin.Context) {
|
|
userID := middleware.GetUserID(c)
|
|
var updates map[string]interface{}
|
|
if err := c.ShouldBindJSON(&updates); err != nil {
|
|
middleware.Error(c, http.StatusBadRequest, 400, "请求数据格式错误")
|
|
return
|
|
}
|
|
|
|
user, err := h.userSvc.UpdateProfile(userID, updates)
|
|
if err != nil {
|
|
middleware.Error(c, http.StatusInternalServerError, 500, err.Error())
|
|
return
|
|
}
|
|
middleware.Success(c, user)
|
|
}
|
|
|
|
func (h *UserHandler) DeleteAccount(c *gin.Context) {
|
|
userID := middleware.GetUserID(c)
|
|
if err := h.userSvc.DeleteAccount(userID); err != nil {
|
|
middleware.Error(c, http.StatusInternalServerError, 500, "注销失败")
|
|
return
|
|
}
|
|
middleware.Success(c, nil)
|
|
}
|
|
|
|
func (h *UserHandler) GetPoints(c *gin.Context) {
|
|
userID := middleware.GetUserID(c)
|
|
user, err := h.userSvc.GetByID(userID)
|
|
if err != nil {
|
|
middleware.Error(c, http.StatusNotFound, 404, "用户不存在")
|
|
return
|
|
}
|
|
middleware.Success(c, gin.H{"points": user.Points})
|
|
}
|