130 lines
3.0 KiB
Go
130 lines
3.0 KiB
Go
package logic
|
|
|
|
import (
|
|
"admin/internal/dao"
|
|
"admin/internal/model"
|
|
"admin/validator"
|
|
"context"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"admin/admin"
|
|
"admin/internal/svc"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"pkg.local/modelbase"
|
|
"pkg.local/utils"
|
|
)
|
|
|
|
type AdminEditLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
BaseLogic
|
|
}
|
|
|
|
func NewAdminEditLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminEditLogic {
|
|
return &AdminEditLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
func (l *AdminEditLogic) AdminEdit(in *admin.AdminEditRequest) (*admin.Response, error) {
|
|
var v validator.AdminEditValidator
|
|
if fail := l.checkParams(in, &v); fail != nil {
|
|
return fail, nil
|
|
}
|
|
adminInfo := utils.GetUserFromCtx(l.ctx)
|
|
if adminInfo.ID < utils.NumberOne {
|
|
return l.fail(utils.ErrorNoLoginInfo), nil
|
|
}
|
|
encryPhone, cErr := utils.EncryptPhone(in.Phone)
|
|
if cErr != nil {
|
|
l.Logger.Error(cErr)
|
|
return l.fail(utils.ErrorEncryptAesError), nil
|
|
}
|
|
|
|
info, check := l.checkPhone(encryPhone)
|
|
if !check {
|
|
return l.fail(utils.ErrorEncryptAesError), nil
|
|
}
|
|
modelObj := model.AdminModel{}.Init()
|
|
if in.Id > utils.NumberZero && info.Id < utils.NumberOne {
|
|
w := modelbase.Params{Eq: map[string]string{"id": strconv.Itoa(int(in.Id))}}
|
|
info = dao.Admin{}
|
|
err := modelObj.GetOne(w, &info)
|
|
if err != nil {
|
|
l.Logger.Error(cErr)
|
|
return l.fail(utils.Fail), nil
|
|
|
|
}
|
|
if info.Id < utils.NumberZero {
|
|
return l.fail(utils.ErrorNotFund), nil
|
|
}
|
|
}
|
|
//编辑
|
|
if in.Id > utils.NumberZero {
|
|
if info.Id < utils.NumberOne {
|
|
return l.fail(utils.ErrorNotFund), nil
|
|
}
|
|
if info.Id != in.Id {
|
|
return l.fail(utils.ErrorExistUser), nil
|
|
}
|
|
} else {
|
|
if len(in.Password) < utils.NumberOne {
|
|
return l.out(utils.ErrorParams, "密码不能为空"), nil
|
|
|
|
}
|
|
if info.Id > utils.NumberZero {
|
|
return l.fail(utils.ErrorExistUser), nil
|
|
}
|
|
}
|
|
|
|
info.AdminId = int(adminInfo.ID)
|
|
info.AdminName = adminInfo.Name
|
|
var roles []string
|
|
if len(in.Roles) > utils.NumberZero {
|
|
for _, role := range in.Roles {
|
|
roles = append(roles, strconv.Itoa(int(role)))
|
|
}
|
|
}
|
|
|
|
info.Name = in.Name
|
|
info.Phone = encryPhone
|
|
info.Roles = strings.Join(roles, utils.DecollatorComma)
|
|
info.Email = in.Email
|
|
info.Avatar = in.Avatar
|
|
var err error
|
|
if info.Id > utils.NumberZero {
|
|
w := modelbase.Params{Eq: map[string]string{"id": strconv.Itoa(int(info.Id))}}
|
|
_, err = modelObj.Edit(w, info)
|
|
} else {
|
|
salt := utils.GetRandstring(utils.NumberFive)
|
|
pwd := utils.GetSaltPassword(salt, in.Password)
|
|
pwd, _ = utils.EncryptPassword(pwd)
|
|
info.Password = pwd
|
|
info.Salt = salt
|
|
err = modelObj.Init().Create(&info)
|
|
}
|
|
|
|
if err != nil {
|
|
l.Logger.Error(err)
|
|
return l.fail(utils.Fail), nil
|
|
}
|
|
return l.ok(info.Id), nil
|
|
}
|
|
|
|
func (l *AdminEditLogic) checkPhone(phone string) (dao.Admin, bool) {
|
|
w := modelbase.Params{Eq: map[string]string{"phone": phone}}
|
|
info := dao.Admin{}
|
|
err := model.AdminModel{}.Init().GetOne(w, &info)
|
|
if err != nil {
|
|
l.Logger.Error(err)
|
|
return info, false
|
|
}
|
|
|
|
return info, true
|
|
}
|