hospital-admin-api/api/controller/adminUser.go

package controller

import (
	"github.com/gin-gonic/gin"
	"hospital-admin-api/api/dao"
	"hospital-admin-api/api/requests"
	"hospital-admin-api/api/responses"
	"hospital-admin-api/api/responses/adminUserResponse"
	"hospital-admin-api/api/service"
	"hospital-admin-api/global"
	"hospital-admin-api/utils"
	"strconv"
	"strings"
	"time"
)

type AdminUser struct{}

// GetUserPage 获取用户列表-分页
func (r *AdminUser) GetUserPage(c *gin.Context) {
	UserRequest := requests.UserRequest{}
	if err := c.ShouldBind(&UserRequest.GetUserPage); err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	// 参数验证
	if err := global.Validate.Struct(UserRequest.GetUserPage); err != nil {
		responses.FailWithMessage(utils.Translate(err), c)
		return
	}

	if UserRequest.GetUserPage.Page == 0 {
		UserRequest.GetUserPage.Page = 1
	}

	if UserRequest.GetUserPage.PageSize == 0 {
		UserRequest.GetUserPage.PageSize = 20
	}

	// 获取当前登陆用户数据

	AdminUserDao := dao.AdminUserDao{}
	adminUser, total, err := AdminUserDao.GetAdminUserPageSearch(UserRequest.GetUserPage, UserRequest.GetUserPage.Page, UserRequest.GetUserPage.PageSize)

	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	// 处理返回值
	getUserPageResponses := adminUserResponse.GetUserPageResponse(adminUser)

	result := make(map[string]interface{})
	result["page"] = UserRequest.GetUserPage.Page
	result["page_size"] = UserRequest.GetUserPage.PageSize
	result["total"] = total
	result["data"] = getUserPageResponses
	responses.OkWithData(result, c)
}

// AddUser 新增用户
func (r *AdminUser) AddUser(c *gin.Context) {
	UserRequest := requests.UserRequest{}
	if err := c.ShouldBind(&UserRequest.AddUser); err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	// 参数验证
	if err := global.Validate.Struct(UserRequest.AddUser); err != nil {
		responses.FailWithMessage(utils.Translate(err), c)
		return
	}

	// 获取角色身份
	roleService := service.RoleService{}
	isAdmin, err := roleService.GetRoleIden(c)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	if !isAdmin {
		responses.FailWithMessage("非管理员，无法操作", c)
		return
	}

	// 业务处理
	UserService := service.AdminUserService{}
	_, err = UserService.AddUser(c, UserRequest.AddUser)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}
	responses.Ok(c)
}

// GetUser 用户详情
func (r *AdminUser) GetUser(c *gin.Context) {
	id := c.Param("user_id")
	if id == "" {
		responses.FailWithMessage("缺少参数", c)
		return
	}

	// 将 id 转换为 int64 类型
	userId, err := strconv.ParseInt(id, 10, 64)
	if err != nil {
		responses.Fail(c)
		return
	}

	// 获取用户详情
	AdminUserDao := dao.AdminUserDao{}
	adminUser, err := AdminUserDao.GetAdminUserFirstById(userId)
	if err != nil || adminUser == nil {
		responses.FailWithMessage("用户数据错误", c)
		return
	}

	// 处理返回值
	getUserResponses := adminUserResponse.GetUserResponse(adminUser)

	responses.OkWithData(getUserResponses, c)
}

// DeleteUser 删除用户-批量
func (r *AdminUser) DeleteUser(c *gin.Context) {
	userRequest := requests.UserRequest{}
	if err := c.ShouldBindJSON(&userRequest.DeleteUser); err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	// 参数验证
	if err := global.Validate.Struct(userRequest.DeleteUser); err != nil {
		responses.FailWithMessage(utils.Translate(err), c)
		return
	}

	// 获取角色身份
	roleService := service.RoleService{}
	isAdmin, err := roleService.GetRoleIden(c)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	if !isAdmin {
		responses.FailWithMessage("非管理员，无法操作", c)
		return
	}

	// 业务处理
	userService := service.AdminUserService{}
	_, err = userService.DeleteUser(c, userRequest.DeleteUser)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}
	responses.Ok(c)
}

// PutUser 修改用户
func (r *AdminUser) PutUser(c *gin.Context) {
	UserRequest := requests.UserRequest{}
	if err := c.ShouldBind(&UserRequest.PutUser); err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	// 参数验证
	if err := global.Validate.Struct(UserRequest.PutUser); err != nil {
		responses.FailWithMessage(utils.Translate(err), c)
		return
	}

	id := c.Param("user_id")
	if id == "" {
		responses.FailWithMessage("缺少参数", c)
		return
	}

	// 将 id 转换为 int64 类型
	userId, err := strconv.ParseInt(id, 10, 64)
	if err != nil {
		responses.Fail(c)
		return
	}

	// 业务处理
	userService := service.AdminUserService{}
	_, err = userService.PutUser(c, userId, UserRequest.PutUser)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}
	responses.Ok(c)
}

// PutUserPassword 修改用户密码
func (r *AdminUser) PutUserPassword(c *gin.Context) {
	UserRequest := requests.UserRequest{}
	if err := c.ShouldBind(&UserRequest.PutUserPassword); err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	// 参数验证
	if err := global.Validate.Struct(UserRequest.PutUserPassword); err != nil {
		responses.FailWithMessage(utils.Translate(err), c)
		return
	}

	userId := c.GetInt64("UserId")
	if userId == 0 {
		responses.FailWithMessage("用户错误", c)
		return
	}

	// 业务处理
	userService := service.AdminUserService{}
	_, err := userService.PutUserPassword(userId, UserRequest.PutUserPassword)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}
	responses.Ok(c)
}

// LoginOut 退出登陆
func (b *AdminUser) LoginOut(c *gin.Context) {
	userId := c.GetInt64("UserId")
	if userId == 0 {
		responses.FailWithMessage("用户错误", c)
		return
	}

	// 获取用户信息
	adminUserDao := dao.AdminUserDao{}
	adminUser, err := adminUserDao.GetAdminUserFirstById(userId)
	if err != nil || adminUser == nil {
		responses.FailWithMessage("用户数据错误", c)
		return
	}

	// token加入黑名单
	authorization := c.Request.Header.Get("Authorization")
	if authorization == "" || !strings.HasPrefix(authorization, "Bearer ") {
		responses.FailWithMessage("退出登陆失败", c)
		return
	}

	// 去除Bearer
	authorization = authorization[7:] // 截取字符

	// 增加缓存
	_, err = global.Redis.Set(c, "jwt_black_"+authorization, time.Now().Unix(), 60*time.Second).Result()
	if err != nil {
		responses.FailWithMessage("退出登陆失败", c)
		return
	}

	responses.Ok(c)
}

// GetUserMenuList 获取用户菜单列表
func (r *AdminUser) GetUserMenuList(c *gin.Context) {
	roleId := c.GetInt64("RoleId")
	if roleId == 0 {
		responses.Fail(c)
		return
	}

	roleService := service.RoleService{}

	// 获取用户角色身份
	isAdmin, err := roleService.GetRoleIden(c)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	// 获取角色菜单
	roleMenuList, err := roleService.GetRoleMenuList(roleId, isAdmin)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	if len(roleMenuList) == 0 || roleMenuList == nil {
		responses.FailWithMessage("请联系管理人员设置菜单", c)
		return
	}

	responses.OkWithData(roleMenuList, c)
}

// GetUserButtonList 获取用户按钮列表
func (r *AdminUser) GetUserButtonList(c *gin.Context) {
	roleId := c.GetInt64("RoleId")
	if roleId == 0 {
		responses.Fail(c)
		return
	}

	roleService := service.RoleService{}

	// 获取用户角色身份
	isAdmin, err := roleService.GetRoleIden(c)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	var buttons []string
	if isAdmin {
		// 管理员
		buttons = append(buttons, "*")
		responses.OkWithData(buttons, c)
		return
	}

	// 获取角色菜单按钮
	buttons, err = roleService.GetRoleMenuButtonList(roleId)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	responses.OkWithData(buttons, c)
}