kpt-pasture/module/backend/system_service.go

package backend

import (
	"context"
	"errors"
	"fmt"
	"kpt-pasture/model"
	"kpt-pasture/util"
	"net/http"
	"time"

	"gitee.com/xuyiping_admin/pkg/logger/zaplog"

	pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
	operationPb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/operation"
	"gitee.com/xuyiping_admin/pkg/jwt"
	"gitee.com/xuyiping_admin/pkg/xerr"
	"gorm.io/gorm"
)

const CurrentUserName = "userName"

// Login 用户登录
func (s *StoreEntry) Login(ctx context.Context, req *pasturePb.SearchUserRequest) (*pasturePb.SystemUserResponse, error) {
	systemUser := &model.SystemUser{}
	if err := s.DB.Where("name = ?", req.Name).Find(systemUser).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	if systemUser.Password != req.Password {
		return nil, xerr.Customf("密码错误，来自用户:%s", req.Name)
	}

	if systemUser.IsShow == pasturePb.IsShow_No {
		return nil, xerr.Customf("该账号已被禁用，请联系管理员")
	}

	jwtToken := jwt.NewJWTTokenGen(s.Cfg.AppName, s.Cfg.JwtTokenKeyConfig.PrivateKey)
	token, err := jwtToken.GenerateToken(systemUser.Name, s.Cfg.JwtExpireTime)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	if token == "" {
		return nil, xerr.Custom("获取token错误")
	}

	expires := time.Now().Add(time.Duration(s.Cfg.JwtExpireTime) * time.Second).Format(util.LayoutTime)
	return &pasturePb.SystemUserResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SystemUserData{
			AccessToken:  token,
			Expires:      expires,
			RefreshToken: expires,
			Username:     systemUser.Name,
			Roles:        []string{"admin"},
			Avatar:       systemUser.Avatar,
		},
	}, nil
}

// SearchSystemUserList 查询系统用户
func (s *StoreEntry) SearchSystemUserList(ctx context.Context, req *pasturePb.SearchUserRequest) (*pasturePb.SearchUserResponse, error) {
	systemUserList := make([]*model.SystemUser, 0)
	var count int64 = 0

	pref := s.DB.Model(new(model.SystemUser)).Where("is_delete = ?", operationPb.IsShow_OK)
	if req.Name != "" {
		pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
	}

	if req.Mobile != "" {
		pref.Where("mobile = ?", req.Mobile)
	}

	if req.DeptId > 0 {
		pref.Where("dept_id = ?", req.DeptId)
	}

	if req.IsShow > 0 {
		pref.Where("is_show = ?", req.IsShow)
	}

	if err := pref.Order("id desc").Count(&count).Limit(int(req.Pagination.PageSize)).Offset(int(req.Pagination.PageOffset)).
		Find(&systemUserList).Debug().Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	roleList, err := s.SearchSystemRoleListByIds(ctx, []int64{})
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	deptList, err := s.SearchSystemDeptListByIds(ctx, []int64{})
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SearchUserResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchUserData{
			List:     model.SystemUserSlice(systemUserList).ToPB(deptList, roleList),
			Total:    int32(count),
			PageSize: req.Pagination.PageSize,
			Page:     req.Pagination.Page,
		},
	}, nil
}

// DeleteSystemUser 删除系统用户
func (s *StoreEntry) DeleteSystemUser(ctx context.Context, userId int64) error {
	systemUser := &model.SystemUser{
		Id: userId,
	}
	if err := s.DB.First(systemUser).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该用户不存在")
		}
		return xerr.WithStack(err)
	}

	if err := s.DB.Model(systemUser).Update("is_delete", operationPb.IsShow_NO).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

// IsShowSystemUser 用户是否启用
func (s *StoreEntry) IsShowSystemUser(ctx context.Context, userId int64) error {
	systemUser := &model.SystemUser{
		Id: userId,
	}
	if err := s.DB.First(systemUser).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该用户不存在")
		}
		return xerr.WithStack(err)
	}
	isShow := pasturePb.IsShow_No
	if systemUser.IsShow == pasturePb.IsShow_No {
		isShow = pasturePb.IsShow_Ok
	}

	if err := s.DB.Model(systemUser).Update("is_show", isShow).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

// SystemUserCreateOrUpdate 创建或者更新系统用户
func (s *StoreEntry) SystemUserCreateOrUpdate(ctx context.Context, req *pasturePb.SearchUserRequest) error {
	if req.Id > 0 {
		systemUser := &model.SystemUser{}
		if err := s.DB.First(systemUser).Error; err != nil {
			return xerr.WithStack(err)
		}
		req.Password = systemUser.Password
	}

	if err := s.DB.Model(&model.SystemUser{}).Where(map[string]interface{}{
		"id": req.Id,
	}).Assign(map[string]interface{}{
		"name":      req.Name,
		"nick_name": req.NickName,
		"mobile":    req.Mobile,
		"gender":    req.Gender,
		"password":  req.Password,
		"dept_id":   req.DeptId,
		"remarks":   req.Remarks,
		"is_delete": operationPb.IsShow_OK,
		"is_show":   req.IsShow,
	}).FirstOrCreate(&model.SystemUser{}).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

// ResetPasswordSystemUser 重置系统用户密码
func (s *StoreEntry) ResetPasswordSystemUser(ctx context.Context, req *pasturePb.ResetUserPasswordRequest) error {
	systemUser := &model.SystemUser{
		Id: int64(req.Id),
	}
	if err := s.DB.First(systemUser).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该用户不存在")
		}
		return xerr.WithStack(err)
	}

	if err := s.DB.Model(systemUser).Update("password", req.Password).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

func (s *StoreEntry) SystemUserRole(ctx context.Context, userId int64) (*pasturePb.SystemUserRoleResponse, error) {
	systemUser := &model.SystemUser{Id: userId}
	if err := s.DB.First(systemUser).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, xerr.Custom("该用户不存在")
		}
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SystemUserRoleResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data:    int32(systemUser.RoleId),
	}, nil
}

func (s *StoreEntry) SystemUserRoleSave(ctx context.Context, req *pasturePb.SystemUserRoleRequest) error {
	systemUser := &model.SystemUser{Id: int64(req.UserId)}
	if err := s.DB.First(systemUser).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该用户不存在")
		}
		return xerr.WithStack(err)
	}

	systemRole := &model.SystemRole{Id: int64(req.RoleId)}
	if err := s.DB.First(systemRole).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该角色不存在")
		}
		return xerr.WithStack(err)
	}
	if err := s.DB.Model(systemUser).Update("role_id", systemRole.Id).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

// GetCurrentUserName 获取当前用户名
func (s *StoreEntry) GetCurrentUserName(ctx context.Context) (string, error) {
	userNameInter := ctx.Value(CurrentUserName)
	if userNameInter == nil {
		return "", xerr.Customf("cannot userName")
	}

	if userName, ok := userNameInter.(string); ok {
		return userName, nil
	} else {
		return "", xerr.Customf("waring userName")
	}
}

// GetUserInfo 获取用户信息
func (s *StoreEntry) GetUserInfo(ctx context.Context, token string) (*operationPb.UserAuth, error) {
	systemUser := &model.SystemUser{}
	userName, err := s.GetCurrentUserName(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	if err = s.DB.Where("name = ?", userName).First(systemUser).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	systemRole := make([]*model.SystemRole, 0)
	return systemUser.SystemUserFormat(systemRole, nil), nil
}

// DetailsSystemUser 系统用户详情
func (s *StoreEntry) DetailsSystemUser(ctx context.Context, userId int64) (*operationPb.UserDetails, error) {
	systemUser := &model.SystemUser{
		Id: userId,
	}
	if err := s.DB.First(systemUser).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, xerr.Custom("该用户不存在")
		}
		return nil, xerr.WithStack(err)
	}

	return &operationPb.UserDetails{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: systemUser.ToPb(),
	}, nil
}

// GetSystemUserPermissions 返回系统用户相关菜单权限
func (s *StoreEntry) GetSystemUserPermissions(ctx context.Context, token string) (*operationPb.SystemUserMenuPermissions, error) {
	// 解析token
	userName, err := s.GetCurrentUserName(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	// 根据用户token获取用户数据
	systemUser := &model.SystemUser{Name: userName}
	if err = s.DB.Where("name = ?", userName).First(systemUser).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, xerr.Custom("该用户数据不存在")
		}
		return nil, xerr.WithStack(err)
	}
	/*roleIds := systemUser.SystemUserRoleToSlice()*/
	roleIds := []int64{1}
	// 获取用户角色数据
	systemRoles := make([]*model.SystemRole, 0)
	if err = s.DB.Where("is_show = ?", operationPb.IsShow_OK).Find(&systemRoles, roleIds).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	systemAllPermissionsList := &SystemAllPermissionsList{
		MenuList: make([]*model.SystemRoleMenu, 0),
	}

	for _, role := range systemRoles {
		zaplog.Info(role.Name)
		/*item := s.GetSystemRoleMenuList(ctx, role.Id)
		systemAllPermissionsList.MenuList = append(systemAllPermissionsList.MenuList, item.MenuList...)*/
	}

	systemAllPermissionsList.SystemUserMenuPermissionsUnDuplicate()

	menuList, err := s.GetMenuList(ctx, systemAllPermissionsList.MenuList)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	return s.SystemPermissionsFormatPb(menuList), nil
}

func (s *StoreEntry) SystemPermissionsFormatPb(menuList []*model.SystemMenu) *operationPb.SystemUserMenuPermissions {
	systemUserMenuPermissions := &operationPb.SystemUserMenuPermissions{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: &operationPb.SystemUserMenuData{
			PastureList:     make([]*operationPb.AddPastureRequest, 0),
			MenuList:        make([]*operationPb.AddMenuRequest, 0),
			MobileList:      make([]*operationPb.AddMobileRequest, 0),
			MenuButtonsPath: make([]*operationPb.MenuButtonsPath, 0),
		},
	}

	/*level := make(map[int32][]*operationPb.AddMenuRequest, 0)
	for _, menu := range menuList {
		if _, ok := level[menu.Level]; !ok {
			level[menu.Level] = make([]*operationPb.AddMenuRequest, 0)
		}
		level[menu.Level] = append(level[menu.Level], &operationPb.AddMenuRequest{
			Id:              int32(menu.Id),
			Name:            menu.Name,
			ParentId:        int32(menu.ParentId),
			MenuType:        menu.MenuType,
			Title:           menu.Title,
			Path:            menu.Path,
			IsShow:          menu.IsShow,
			Component:       menu.Component,
			Icon:            menu.Icon,
			Sort:            menu.Sort,
			Redirect:        menu.Redirect,
			CreatedAt:       int32(menu.CreatedAt),
			CreatedAtFormat: time.Unix(menu.CreatedAt, 0).Format(model.LayoutTime),
			Level:           menu.Level,
			Affix:           true,
			KeepAlive:       true,
			Children:        make([]*operationPb.AddMenuRequest, 0),
		})
		if menu.Level == model.Level3 {
			systemUserMenuPermissions.Data.MenuButtonsPath = append(systemUserMenuPermissions.Data.MenuButtonsPath, &operationPb.MenuButtonsPath{
				Path:   menu.Path,
				MenuId: int32(menu.Id),
			})
		}
	}

	level2IsShow := make([]int32, 0)
	for _, leve2Data := range level[model.Level2] {
		for _, leve3Data := range level[model.Level3] {
			if leve3Data.ParentId == leve2Data.Id && strings.Contains(leve3Data.Path, ":page") {
				level2IsShow = append(level2IsShow, leve2Data.Id)
			}

			if leve3Data.ParentId == leve2Data.Id {
				if leve2Data.Children == nil {
					leve2Data.Children = make([]*operationPb.AddMenuRequest, 0)
				}
				leve2Data.Children = append(leve2Data.Children, leve3Data)
			}
		}
	}

	sort.SliceStable(level[model.Level2], func(i, j int) bool {
		return level[model.Level2][i].Sort > level[model.Level2][j].Sort
	})
	for _, leve2Data := range level[model.Level2] {
		var info bool
		for _, v := range level2IsShow {
			if v == leve2Data.Id {
				info = true
			}
		}
		if !info {
			continue
		}
		for _, leve1Data := range level[model.Level1] {
			if leve2Data.ParentId == leve1Data.Id {
				if leve1Data.Children == nil {
					leve1Data.Children = make([]*operationPb.AddMenuRequest, 0)
				}
				leve1Data.Children = append(leve1Data.Children, leve2Data)
			}
		}
	}

	sort.SliceStable(level[model.Level1], func(i, j int) bool {
		return level[model.Level1][i].Sort > level[model.Level1][j].Sort
	})

	systemUserMenuPermissions.Data.MenuList = level[model.Level1]*/
	return systemUserMenuPermissions
}

// SystemRoleCreateOrUpdate 添加角色
func (s *StoreEntry) SystemRoleCreateOrUpdate(ctx context.Context, req *pasturePb.SearchRoleRequest) error {
	isShow := pasturePb.IsShow_Ok
	if req.Id > 0 {
		isShow = req.IsShow
	}

	if err := s.DB.Model(&model.SystemRole{}).Where(map[string]interface{}{
		"id": req.Id,
	}).Assign(map[string]interface{}{
		"name":      req.Name,
		"remarks":   req.Remarks,
		"is_delete": operationPb.IsShow_OK,
		"is_show":   isShow,
	}).FirstOrCreate(&model.SystemRole{}).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

// DeleteSystemRole 删除系统角色
func (s *StoreEntry) DeleteSystemRole(ctx context.Context, roleId int64) error {
	systemRole := &model.SystemRole{
		Id: roleId,
	}
	if err := s.DB.First(systemRole).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该数据不存在")
		}
		return xerr.WithStack(err)
	}

	if err := s.DB.Model(systemRole).Update("is_delete", pasturePb.IsShow_No).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

// IsShowSystemRole 角色是否启用
func (s *StoreEntry) IsShowSystemRole(ctx context.Context, roleId int64) error {
	systemRole := &model.SystemRole{
		Id: roleId,
	}
	if err := s.DB.First(systemRole).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该角色不存在")
		}
		return xerr.WithStack(err)
	}
	isShow := pasturePb.IsShow_No
	if systemRole.IsShow == pasturePb.IsShow_No {
		isShow = pasturePb.IsShow_Ok
	}

	if err := s.DB.Model(systemRole).Update("is_show", isShow).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

// SearchSystemRoleList 查询系统角色
func (s *StoreEntry) SearchSystemRoleList(ctx context.Context, req *pasturePb.SearchRoleRequest) (*pasturePb.SearchRoleResponse, error) {
	systemRoleList := make([]*model.SystemRole, 0)
	var count int64 = 0

	pref := s.DB.Model(new(model.SystemRole)).Where("is_delete = ?", operationPb.IsShow_OK)
	if req.Name != "" {
		pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
	}

	if req.IsShow > 0 {
		pref.Where("is_show = ?", req.IsShow)
	}

	if err := pref.Order("id desc").Count(&count).Limit(int(req.Pagination.PageSize)).
		Offset(int(req.Pagination.PageOffset)).Find(&systemRoleList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SearchRoleResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchRoleData{
			Page:     req.Pagination.Page,
			Total:    int32(count),
			PageSize: req.Pagination.PageSize,
			List:     model.SystemRoleSlice(systemRoleList).ToPB(),
		},
	}, nil
}

// SearchSystemRoleListByIds 根据id查询角色列表
func (s *StoreEntry) SearchSystemRoleListByIds(ctx context.Context, ids []int64) ([]*model.SystemRole, error) {
	systemRoleList := make([]*model.SystemRole, 0)
	if err := s.DB.Model(new(model.SystemRole)).
		Where("is_show = ?", operationPb.IsShow_OK).
		Find(&systemRoleList, ids).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return systemRoleList, nil
}

// GetRoleMenuList 查询系统角色对应的菜单数据
func (s *StoreEntry) GetRoleMenuList(ctx context.Context, roleId int64) (*pasturePb.SystemRoleMenuResponse, error) {
	systemRole := &model.SystemRole{
		Id: roleId,
	}
	if err := s.DB.First(systemRole).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, xerr.Custom("该数据不存在")
		}
		return nil, xerr.WithStack(err)
	}

	if systemRole.IsDelete == pasturePb.IsShow_No {
		return nil, xerr.Custom("该数据已被删除")
	}

	item := s.GetSystemRoleMenuList(ctx, systemRole.Id)
	return &pasturePb.SystemRoleMenuResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data:    model.SystemRoleMenuSlice(item).ToPB(),
	}, nil

}

// RoleMenuSave 保存系统角色对应的菜单数据
func (s *StoreEntry) RoleMenuSave(ctx context.Context, res *pasturePb.SystemRoleMenuRequest) error {
	if len(res.MenuIds) <= 0 {
		return xerr.Custom("请选择菜单")
	}
	systemRole := &model.SystemRole{
		Id: int64(res.RoleId),
	}

	if err := s.DB.First(systemRole).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该数据不存在")
		}
		return xerr.WithStack(err)
	}

	if err := s.DB.Transaction(func(tx *gorm.DB) error {
		if err := tx.Model(new(model.SystemRoleMenu)).Where("role_id = ?", systemRole.Id).Delete(&model.SystemRoleMenu{}).Error; err != nil {
			return xerr.WithStack(err)
		}
		for _, menuId := range res.MenuIds {
			if err := tx.Create(&model.SystemRoleMenu{
				RoleId: systemRole.Id,
				MenuId: int64(menuId),
			}).Error; err != nil {
				return xerr.WithStack(err)
			}
		}
		return nil
	}); err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

func (s *StoreEntry) SystemRoleList(ctx context.Context) (*pasturePb.GetRoleListResponse, error) {
	systemRoleList := make([]*model.SystemRole, 0)
	if err := s.DB.Model(new(model.SystemRole)).
		Where("is_delete = ? and is_show = ?", pasturePb.IsShow_Ok, pasturePb.IsShow_Ok).
		Find(&systemRoleList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.GetRoleListResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data:    model.SystemRoleSlice(systemRoleList).ToPB(),
	}, nil
}

// CreateOrUpdateSystemMenu 添加或者更新系统菜单权限
func (s *StoreEntry) CreateOrUpdateSystemMenu(ctx context.Context, req *pasturePb.SearchMenuRequest) error {
	if req.Id > 0 {

	}

	if err := s.DB.Model(&model.SystemMenu{}).Where(map[string]interface{}{
		"id": req.Id,
	}).Assign(map[string]interface{}{
		"name":             req.Name,
		"path":             req.Path,
		"title":            req.Title,
		"menu_type":        req.MenuType,
		"parent_id":        req.ParentId,
		"frame_src":        req.FrameSrc,
		"frame_loading":    model.BooleanToIsShow(req.FrameLoading),
		"keepalive":        model.BooleanToIsShow(req.Keepalive),
		"hidden_tag":       model.BooleanToIsShow(req.HiddenTag),
		"show_link":        model.BooleanToIsShow(req.ShowLink),
		"show_parent":      model.BooleanToIsShow(req.ShowParent),
		"icon":             req.Icon,
		"component":        req.Component,
		"redirect":         req.Redirect,
		"auths":            req.Auths,
		"rank":             req.Rank,
		"extra_icon":       req.ExtraIcon,
		"enter_transition": req.EnterTransition,
		"leave_transition": req.LeaveTransition,
		"active_path":      req.ActivePath,
		"is_delete":        pasturePb.IsShow_Ok,
		"is_show":          pasturePb.IsShow_Ok,
	}).FirstOrCreate(&model.SystemMenu{}).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

func (s *StoreEntry) SystemMenuTree(ctx context.Context) (*pasturePb.SystemMenuTreeResponse, error) {
	systemMenuList := make([]*model.SystemMenu, 0)
	if err := s.DB.Model(new(model.SystemMenu)).Where("is_delete = ?", pasturePb.IsShow_Ok).
		Where("is_show = ?", pasturePb.IsShow_Ok).Find(&systemMenuList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}
	return &pasturePb.SystemMenuTreeResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data:    model.SystemMenuSlice(systemMenuList).ToPB(),
	}, nil
}

// SearchSystemMenuList 菜单列表查询
func (s *StoreEntry) SearchSystemMenuList(ctx context.Context, req *pasturePb.SearchMenuRequest) (*pasturePb.SearchMenuResponse, error) {
	systemMenuList := make([]*model.SystemMenu, 0)
	var count int64 = 0

	if err := s.DB.Model(new(model.SystemMenu)).Where("is_delete = ? ", operationPb.IsShow_OK).Order("parent_id").Count(&count).Find(&systemMenuList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SearchMenuResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchMenuData{
			Page:     req.Pagination.Page,
			Total:    int32(count),
			PageSize: req.Pagination.PageSize,
			List:     model.SystemMenuSlice(systemMenuList).ToPB(),
		},
	}, nil
}

// DeleteSystemMenu 删除系统菜单
func (s *StoreEntry) DeleteSystemMenu(ctx context.Context, menuId int64) error {
	systemMenu := &model.SystemMenu{Id: menuId}
	if err := s.DB.First(systemMenu).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该数据不存在")
		}
		return xerr.WithStack(err)
	}

	if err := s.DB.Model(systemMenu).Update("is_delete", pasturePb.IsShow_No).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

func (s *StoreEntry) SearchSystemDeptList(ctx context.Context, req *pasturePb.SearchDeptRequest) (*pasturePb.SearchDeptResponse, error) {
	deptList := make([]*model.SystemDept, 0)
	var count int64 = 0
	pref := s.DB.Model(new(model.SystemDept)).Where("is_delete = ?", operationPb.IsShow_OK)
	if req.Name != "" {
		pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
	}

	if req.IsShow > 0 {
		pref.Where("is_show = ?", req.IsShow)
	}

	if err := pref.Order("sort desc").Count(&count).Limit(int(req.Pagination.PageSize)).Offset(int(req.Pagination.PageOffset)).
		Find(&deptList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SearchDeptResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data:    model.SystemDeptSlice(deptList).ToPB(),
	}, nil
}

func (s *StoreEntry) SystemDepDelete(ctx context.Context, id int64) error {
	dept := &model.SystemDept{Id: id}
	if err := s.DB.First(dept).Error; err != nil {
		return xerr.WithStack(err)
	}

	if dept.IsShow == pasturePb.IsShow_No {
		return nil
	}
	if err := s.DB.Model(dept).Update("is_delete", operationPb.IsShow_NO).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

func (s *StoreEntry) SystemDeptCreateOrUpdate(ctx context.Context, req *pasturePb.SearchDeptRequest) error {
	if err := s.DB.Model(&model.SystemDept{}).Where(map[string]interface{}{
		"id": req.Id,
	}).Assign(map[string]interface{}{
		"name":      req.Name,
		"sort":      req.Sort,
		"parent_id": req.ParentId,
		"remarks":   req.Remarks,
		"is_delete": operationPb.IsShow_OK,
		"is_show":   req.IsShow,
		"dept_type": pasturePb.Depth_Department,
		"leader_id": req.LeaderId,
	}).FirstOrCreate(&model.SystemDept{}).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

// SearchSystemDeptListByIds 根据id查询部门列表
func (s *StoreEntry) SearchSystemDeptListByIds(ctx context.Context, ids []int64) ([]*model.SystemDept, error) {
	systemDeptList := make([]*model.SystemDept, 0)
	if err := s.DB.Model(new(model.SystemDept)).
		Where("is_show = ?", operationPb.IsShow_OK).
		Find(&systemDeptList, ids).Error; err != nil {
		return nil, xerr.WithStack(err)
	}
	return systemDeptList, nil
}