:toda:

http/api/user.go

@@ -0,0 +1,252 @@
+package api
+
+import (
+	"errors"
+	"github.com/gin-gonic/gin"
+	"wireguard-ui/global/constant"
+	"wireguard-ui/http/param"
+	"wireguard-ui/http/response"
+	"wireguard-ui/model"
+	"wireguard-ui/service"
+	"wireguard-ui/utils"
+)
+
+type UserApi struct{}
+
+func User() UserApi {
+	return UserApi{}
+}
+
+// GetLoginUser
+// @description: 获取登陆用户信息
+// @receiver UserApi
+// @param c
+func (UserApi) GetLoginUser(c *gin.Context) {
+	loginUser, ok := c.Get("user")
+	if !ok {
+		response.R(c).AuthorizationFailed("未登陆")
+		return
+	}
+
+	response.R(c).OkWithData(loginUser)
+}
+
+// SaveUser
+// @description: 新增/编辑用户信息
+// @receiver UserApi
+// @param c
+func (UserApi) SaveUser(c *gin.Context) {
+	var p param.SaveUser
+	if err := c.ShouldBind(&p); err != nil {
+		response.R(c).Validator(err)
+		return
+	}
+
+	// 如果是新增用户判断该用户是否已经存在
+	if p.Id == "" {
+		if len(p.Account) < 2 || len(p.Account) > 20 {
+			response.R(c).FailedWithError(errors.New("账号长度在2-20位"))
+			return
+		}
+		if len(p.Password) < 8 || len(p.Password) > 32 {
+			response.R(c).FailedWithError(errors.New("密码长度在8-32位"))
+			return
+		}
+
+		var count int64
+		if err := service.User().Model(&model.User{}).Where("account = ?", p.Account).Count(&count).Error; err != nil {
+			response.R(c).FailedWithError(err)
+			return
+		}
+
+		if count > 0 {
+			response.R(c).FailedWithError(errors.New("该账号已存在"))
+			return
+		}
+	}
+
+	userEnt := &model.User{
+		Base: model.Base{
+			Id: p.Id,
+		},
+		Account:  p.Account,
+		Password: p.Password,
+		Nickname: p.Nickname,
+		Avatar:   p.Avatar,
+		Contact:  p.Contact,
+		IsAdmin:  *p.IsAdmin,
+		Status:   *p.Status,
+	}
+
+	if err := service.User().CreateUser(userEnt); err != nil {
+		response.R(c).FailedWithError(err)
+		return
+	}
+
+	response.R(c).OK()
+}
+
+// List
+// @description: 用户列表
+// @receiver UserApi
+// @param c
+func (UserApi) List(c *gin.Context) {
+	var p param.Page
+	if err := c.ShouldBind(&p); err != nil {
+		response.R(c).Validator(err)
+		return
+	}
+
+	data, total, err := service.User().List(p)
+	if err != nil {
+		response.R(c).FailedWithError(err)
+		return
+	}
+
+	response.R(c).Paginate(data, total, p.Current, p.Size)
+}
+
+// Delete
+// @description: 删除用户
+// @receiver UserApi
+// @param c
+func (UserApi) Delete(c *gin.Context) {
+	id := c.Param("id")
+	if id == "" || id == "undefined" {
+		response.R(c).FailedWithError("id不能为空")
+		return
+	}
+
+	// 是不是自己删除自己
+	if id == GetCurrentLoginUser(c).Id && c.IsAborted() {
+		response.R(c).FailedWithError("非法操作")
+		return
+	}
+
+	// 先查询一下
+	user, err := service.User().GetUserById(id)
+	if err != nil {
+		response.R(c).FailedWithError("获取用户信息失败")
+		return
+	}
+
+	// admin用户不能被删除
+	if user.Account == "admin" {
+		response.R(c).FailedWithError("当前用户不能被删除")
+		return
+	}
+
+	if err = service.User().Delete(id); err != nil {
+		response.R(c).FailedWithError("删除用户失败")
+		return
+	}
+
+	response.R(c).OK()
+}
+
+// Status
+// @description: 设置用户状态
+// @receiver UserApi
+// @param c
+func (UserApi) Status(c *gin.Context) {
+	id := c.Param("id")
+	if id == "" || id == "undefined" {
+		response.R(c).FailedWithError("id不能为空")
+		return
+	}
+
+	// 是不是自己删除自己
+	if id == GetCurrentLoginUser(c).Id && c.IsAborted() {
+		response.R(c).FailedWithError("非法操作")
+		return
+	}
+
+	// 先查询一下
+	user, err := service.User().GetUserById(id)
+	if err != nil {
+		response.R(c).FailedWithError("获取用户信息失败")
+		return
+	}
+
+	// admin用户不能被删除
+	if user.Account == "admin" {
+		response.R(c).FailedWithError("当前用户状态不可被变更")
+		return
+	}
+
+	var state = constant.Enabled
+	if user.Status == constant.Enabled {
+		state = constant.Disabled
+	}
+
+	if err := service.User().Status(id, state); err != nil {
+		response.R(c).FailedWithError(err)
+		return
+	}
+
+	response.R(c).OK()
+}
+
+// ChangePassword
+// @description: 修改密码
+// @receiver UserApi
+// @param c
+func (UserApi) ChangePassword(c *gin.Context) {
+	var p param.ChangePassword
+	if err := c.ShouldBind(&p); err != nil {
+		response.R(c).Validator(err)
+		return
+	}
+
+	user := GetCurrentLoginUser(c)
+	if user == nil {
+		response.R(c).FailedWithError("用户信息错误")
+		return
+	}
+
+	// 判断原密码是否对
+	if !utils.Password().ComparePassword(user.Password, p.OriginalPassword) {
+		response.R(c).FailedWithError("原密码错误")
+		return
+	}
+
+	// 修改密码
+	if err := service.User().ChangePassword(user.Id, p.NewPassword); err != nil {
+		response.R(c).FailedWithError(err)
+		return
+	}
+
+	response.R(c).OK()
+}
+
+// ResetPassword
+// @description: 重置密码
+// @receiver UserApi
+// @param c
+func (UserApi) ResetPassword(c *gin.Context) {
+	var id = c.Param("id")
+	if id == "" || id == "undefined" {
+		response.R(c).FailedWithError("id不能为空")
+		return
+	}
+
+	// 先查询一下
+	user, err := service.User().GetUserById(id)
+	if err != nil {
+		response.R(c).FailedWithError("获取用户信息失败")
+		return
+	}
+
+	if user.Status != constant.Enabled {
+		response.R(c).FailedWithError("当前用户不可重置密码")
+		return
+	}
+
+	// 修改密码
+	if err := service.User().ChangePassword(user.Id, "admin123"); err != nil {
+		response.R(c).FailedWithError(err)
+		return
+	}
+
+	response.R(c).OK()
+}