2023-11-08 21:53:07 +08:00

170 lines
4.2 KiB
Go

package panel
import (
"sun-panel/api/api_v1/common/apiReturn"
"sun-panel/api/api_v1/common/base"
"sun-panel/global"
"sun-panel/lib/cmn"
"sun-panel/models"
"github.com/gin-gonic/gin"
"github.com/gin-gonic/gin/binding"
)
// 此API 临时使用,后期带有管理功能,将废除!!!
type UsersApi struct {
}
func (a UsersApi) Create(c *gin.Context) {
param := models.User{}
if err := c.ShouldBindBodyWith(&param, binding.JSON); err != nil {
apiReturn.ErrorParamFomat(c, err.Error())
return
}
if errMsg, err := base.ValidateInputStruct(param); err != nil {
apiReturn.ErrorParamFomat(c, errMsg)
return
}
mUser := models.User{
Username: param.Username,
Password: cmn.PasswordEncryption(param.Password),
Name: param.Username,
HeadImage: param.HeadImage,
Status: 1,
Role: 1, // 固定管理员
Mail: param.Username,
}
// 验证账号是否存在
if _, err := mUser.CheckUsernameExist(param.Username); err != nil {
apiReturn.Error(c, global.Lang.Get("register.mail_exist"))
return
}
userInfo, err := mUser.CreateOne()
if err != nil {
apiReturn.ErrorDatabase(c, err.Error())
return
}
apiReturn.SuccessData(c, gin.H{"userId": userInfo.ID})
}
func (a UsersApi) Deletes(c *gin.Context) {
type UserIds struct {
UserIds []uint
}
param := UserIds{}
if err := c.ShouldBindBodyWith(&param, binding.JSON); err != nil {
apiReturn.Error(c, global.Lang.GetAndInsert("common.api_error_param_format", "[", err.Error(), "]"))
c.Abort()
return
}
if err := global.Db.Delete(&models.User{}, &param.UserIds).Error; err != nil {
apiReturn.ErrorDatabase(c, err.Error())
return
}
apiReturn.Success(c)
}
func (a UsersApi) Update(c *gin.Context) {
param := models.User{}
if err := c.ShouldBindBodyWith(&param, binding.JSON); err != nil {
apiReturn.Error(c, global.Lang.GetAndInsert("common.api_error_param_format", "[", err.Error(), "]"))
c.Abort()
return
}
if param.Password == "" {
param.Password = "-" // 修改不允许修改密码,为了验证通过
}
param.Mail = param.Username // 密码邮箱同时修改
if errMsg, err := base.ValidateInputStruct(param); err != nil {
apiReturn.ErrorParamFomat(c, errMsg)
return
}
allowField := []string{"Username", "Name", "Mail", "Token"}
// 密码不为默认“-”空,修改密码
if param.Password != "-" {
param.Password = cmn.PasswordEncryption(param.Password)
allowField = append(allowField, "Password")
}
mUser := models.User{}
userInfo := models.User{}
// 验证账号是否存在
if user, err := mUser.CheckUsernameExist(param.Username); err != nil {
userInfo = user
if user.ID != param.ID {
apiReturn.Error(c, global.Lang.Get("register.mail_exist"))
return
}
} else {
userInfo = user
}
param.Token = "" // 修改资料就重置token
if err := global.Db.Select(allowField).Where("id=?", param.ID).Updates(&param).Error; err != nil {
apiReturn.ErrorDatabase(c, err.Error())
return
}
// global.Logger.Debug("修改资料清空token", userInfo.Token)
global.UserToken.Delete(userInfo.Token) // 更新用户信息
// 返回token等基本信息
apiReturn.SuccessData(c, param)
}
func (a UsersApi) GetList(c *gin.Context) {
type ParamsStruct struct {
models.User
Limit int
Page int
Keyword string `json:"keyword"`
}
param := ParamsStruct{}
if err := c.ShouldBindBodyWith(&param, binding.JSON); err != nil {
apiReturn.Error(c, global.Lang.GetAndInsert("common.api_error_param_format", "[", err.Error(), "]"))
c.Abort()
return
}
var (
list []models.User
count int64
)
db := global.Db
// 查询条件
if param.Keyword != "" {
db = db.Where("name LIKE ? OR username LIKE ?", "%"+param.Keyword+"%", "%"+param.Keyword+"%")
}
if err := db.Omit("Password").Limit(param.Limit).Offset((param.Page - 1) * param.Limit).Find(&list).Limit(-1).Offset(-1).Count(&count).Error; err != nil {
apiReturn.ErrorDatabase(c, err.Error())
return
}
// resMap := []map[string]interface{}{}
// for _, v := range list {
// resMap = append(resMap, map[string]interface{}{
// "userId": v.ID,
// "name": v.Name,
// "headImage": v.HeadImage,
// "status": v.Status,
// "role": v.Role,
// "username": v.Username,
// })
// }
apiReturn.SuccessListData(c, list, count)
}