首次提交
This commit is contained in:
@@ -0,0 +1,115 @@
|
||||
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})
|
||||
}
|
||||
Reference in New Issue
Block a user