134 lines
2.8 KiB
Go
134 lines
2.8 KiB
Go
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/niangaodev/art-code/models"
|
|
"github.com/niangaodev/art-code/repositories"
|
|
"github.com/niangaodev/art-code/utils"
|
|
)
|
|
|
|
// AdminGetUsers 获取用户列表
|
|
func AdminGetUsers(c *gin.Context) {
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("pageSize", "10"))
|
|
|
|
users, total, err := repositories.GetUsers(page, pageSize)
|
|
if err != nil {
|
|
utils.ServerError(c, err)
|
|
return
|
|
}
|
|
|
|
list := repositories.BuildUsersResponse(users)
|
|
res := gin.H{
|
|
"list": list,
|
|
"total": total,
|
|
"page": page,
|
|
"size": pageSize,
|
|
}
|
|
if list == nil {
|
|
res["list"] = []interface{}{}
|
|
}
|
|
|
|
utils.Success(c, res)
|
|
}
|
|
|
|
// AdminCreateUser 创建用户
|
|
func AdminCreateUser(c *gin.Context) {
|
|
var user models.User
|
|
if err := c.ShouldBindJSON(&user); err != nil {
|
|
utils.Error(c, 400, "Invalid request")
|
|
return
|
|
}
|
|
|
|
// 密码加密
|
|
hashedPassword, err := utils.HashPassword(user.Password)
|
|
if err != nil {
|
|
utils.ServerError(c, err)
|
|
return
|
|
}
|
|
user.Password = hashedPassword
|
|
|
|
if err := repositories.CreateUser(&user); err != nil {
|
|
utils.ServerError(c, err)
|
|
return
|
|
}
|
|
|
|
utils.SuccessWithMsg(c, "User created successfully", gin.H{"id": user.ID})
|
|
}
|
|
|
|
// AdminUpdateUser 更新用户
|
|
func AdminUpdateUser(c *gin.Context) {
|
|
idStr := c.Param("id")
|
|
var id uint
|
|
if _, err := fmt.Sscanf(idStr, "%d", &id); err != nil {
|
|
utils.Error(c, 400, "Invalid user ID")
|
|
return
|
|
}
|
|
|
|
var user models.User
|
|
if err := c.ShouldBindJSON(&user); err != nil {
|
|
utils.Error(c, 400, "Invalid request")
|
|
return
|
|
}
|
|
user.ID = id
|
|
|
|
// 如果提供了密码，则加密
|
|
if user.Password != "" {
|
|
hashedPassword, err := utils.HashPassword(user.Password)
|
|
if err != nil {
|
|
utils.ServerError(c, err)
|
|
return
|
|
}
|
|
user.Password = hashedPassword
|
|
}
|
|
|
|
if err := repositories.UpdateUser(&user); err != nil {
|
|
utils.ServerError(c, err)
|
|
return
|
|
}
|
|
|
|
utils.SuccessWithMsg(c, "User updated successfully", nil)
|
|
}
|
|
|
|
// AdminDeleteUser 删除用户
|
|
func AdminDeleteUser(c *gin.Context) {
|
|
idStr := c.Param("id")
|
|
var id uint
|
|
if _, err := fmt.Sscanf(idStr, "%d", &id); err != nil {
|
|
utils.Error(c, 400, "Invalid user ID")
|
|
return
|
|
}
|
|
|
|
if err := repositories.DeleteUser(id); err != nil {
|
|
utils.ServerError(c, err)
|
|
return
|
|
}
|
|
|
|
utils.SuccessWithMsg(c, "User deleted successfully", nil)
|
|
}
|
|
|
|
// AdminGetUser 获取单个用户
|
|
func AdminGetUser(c *gin.Context) {
|
|
idStr := c.Param("id")
|
|
var id uint
|
|
if _, err := fmt.Sscanf(idStr, "%d", &id); err != nil {
|
|
utils.Error(c, 400, "Invalid user ID")
|
|
return
|
|
}
|
|
|
|
user, err := repositories.GetUserByID(id)
|
|
if err != nil {
|
|
utils.ServerError(c, err)
|
|
return
|
|
}
|
|
if user == nil {
|
|
utils.Error(c, 404, "User not found")
|
|
return
|
|
}
|
|
|
|
utils.Success(c, repositories.BuildUserResponse(user))
|
|
}
|