🐶 重构部分用户数据库操作代码

server/api/user.go

@@ -0,0 +1,163 @@
+package api
+
+import (
+	"strconv"
+	"strings"
+
+	"next-terminal/server/global"
+	"next-terminal/server/model"
+	"next-terminal/server/utils"
+
+	"github.com/labstack/echo/v4"
+	"github.com/sirupsen/logrus"
+)
+
+func UserCreateEndpoint(c echo.Context) error {
+	var item model.User
+	if err := c.Bind(&item); err != nil {
+		return err
+	}
+	password := item.Password
+
+	var pass []byte
+	var err error
+	if pass, err = utils.Encoder.Encode([]byte(password)); err != nil {
+		return err
+	}
+	item.Password = string(pass)
+
+	item.ID = utils.UUID()
+	item.Created = utils.NowJsonTime()
+
+	if err := userRepository.Create(&item); err != nil {
+		return err
+	}
+
+	if item.Mail != "" {
+		go model.SendMail(item.Mail, "[Next Terminal] 注册通知", "你好，"+item.Nickname+"。管理员为你注册了账号："+item.Username+" 密码："+password)
+	}
+	return Success(c, item)
+}
+
+func UserPagingEndpoint(c echo.Context) error {
+	pageIndex, _ := strconv.Atoi(c.QueryParam("pageIndex"))
+	pageSize, _ := strconv.Atoi(c.QueryParam("pageSize"))
+	username := c.QueryParam("username")
+	nickname := c.QueryParam("nickname")
+	mail := c.QueryParam("mail")
+
+	order := c.QueryParam("order")
+	field := c.QueryParam("field")
+
+	items, total, err := userRepository.Find(pageIndex, pageSize, username, nickname, mail, order, field)
+	if err != nil {
+		return err
+	}
+
+	return Success(c, H{
+		"total": total,
+		"items": items,
+	})
+}
+
+func UserUpdateEndpoint(c echo.Context) error {
+	id := c.Param("id")
+
+	var item model.User
+	if err := c.Bind(&item); err != nil {
+		return err
+	}
+	item.ID = id
+
+	if err := userRepository.Update(&item); err != nil {
+		return err
+	}
+
+	return Success(c, nil)
+}
+
+func UserDeleteEndpoint(c echo.Context) error {
+	ids := c.Param("id")
+	account, found := GetCurrentAccount(c)
+	if !found {
+		return Fail(c, -1, "获取当前登录账户失败")
+	}
+	split := strings.Split(ids, ",")
+	for i := range split {
+		userId := split[i]
+		if account.ID == userId {
+			return Fail(c, -1, "不允许删除自身账户")
+		}
+		// 将用户强制下线
+		loginLogs, err := model.FindAliveLoginLogsByUserId(userId)
+		if err != nil {
+			return err
+		}
+
+		for j := range loginLogs {
+			global.Cache.Delete(loginLogs[j].ID)
+			if err := model.Logout(loginLogs[j].ID); err != nil {
+				logrus.WithError(err).WithField("id:", loginLogs[j].ID).Error("Cache Deleted Error")
+				return Fail(c, 500, "强制下线错误")
+			}
+		}
+
+		// 删除用户
+		if err := userRepository.DeleteById(userId); err != nil {
+			return err
+		}
+	}
+
+	return Success(c, nil)
+}
+
+func UserGetEndpoint(c echo.Context) error {
+	id := c.Param("id")
+
+	item, err := userRepository.FindById(id)
+	if err != nil {
+		return err
+	}
+
+	return Success(c, item)
+}
+
+func UserChangePasswordEndpoint(c echo.Context) error {
+	id := c.Param("id")
+	password := c.QueryParam("password")
+
+	user, err := userRepository.FindById(id)
+	if err != nil {
+		return err
+	}
+
+	passwd, err := utils.Encoder.Encode([]byte(password))
+	if err != nil {
+		return err
+	}
+	u := &model.User{
+		Password: string(passwd),
+		ID:       id,
+	}
+	if err := userRepository.Update(u); err != nil {
+		return err
+	}
+
+	if user.Mail != "" {
+		go model.SendMail(user.Mail, "[Next Terminal] 密码修改通知", "你好，"+user.Nickname+"。管理员已将你的密码修改为："+password)
+	}
+
+	return Success(c, "")
+}
+
+func UserResetTotpEndpoint(c echo.Context) error {
+	id := c.Param("id")
+	u := &model.User{
+		TOTPSecret: "-",
+		ID:         id,
+	}
+	if err := userRepository.Update(u); err != nil {
+		return err
+	}
+	return Success(c, "")
+}