kpt-pasture/module/backend/system_service.go

package backend

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

	"gitee.com/xuyiping_admin/pkg/logger/zaplog"
	"go.uber.org/zap"

	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"
	CurrentFarmId   = "FarmId"
)

// 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).
		First(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("该账号已被禁用，请联系管理员")
	}

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

	if len(systemUser.PastureIds) <= 0 {
		return nil, xerr.Custom("当前用户未配置相关牧场数据,请联系管理员！")
	}

	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().Local().Add(time.Duration(s.Cfg.JwtExpireTime) * time.Second).Format(util.LayoutTime)
	farmList, err := s.FindPastureListByIds(ctx, systemUser.GetPastureIds())
	if err != nil || len(farmList) == 0 {
		return nil, xerr.Custom("牧场信息错误")
	}

	systemUserDepthRole, err := s.GetSystemUserDepthRole(ctx, farmList[0].Id, systemUser.Id)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	systemRoleList := make([]*model.SystemRole, 0)
	if err = s.DB.Model(new(model.SystemRole)).
		Where("is_show = ? and is_delete = ?", pasturePb.IsShow_Ok, pasturePb.IsShow_Ok).
		Where("id IN ?", systemUserDepthRole.GetRoleIds()).
		Find(&systemRoleList).Error; err != nil {
		zaplog.Error("Login", zap.Any("systemRole", err))
	}

	roleNameList := make([]string, 0)
	for _, role := range systemRoleList {
		roleNameList = append(roleNameList, role.Name)
	}

	return &pasturePb.SystemUserResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: &pasturePb.SystemUserData{
			AccessToken:  token,
			Expires:      expires,
			RefreshToken: token,
			Username:     systemUser.Name,
			Roles:        roleNameList,
			Avatar:       systemUser.Avatar,
			FarmId:       farmList[0].FarmId,
		},
	}, nil
}

// SearchSystemUserList 查询系统用户
func (s *StoreEntry) SearchSystemUserList(ctx context.Context, req *pasturePb.SearchUserRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchUserResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	systemUserList := make([]*model.SystemUser, 0)
	var count int64 = 0
	pref := s.DB.Model(new(model.SystemUser)).
		Where("is_delete = ?", operationPb.IsShow_OK).
		Where("FIND_IN_SET(?, pasture_ids) > ?", userModel.AppPasture.Id, 0)

	if req.Name != "" {
		pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
	}

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

	if len(req.DepthId) > 0 {
		depthIds := ""
		for _, v := range req.DepthId {
			depthIds += fmt.Sprintf("%d,", v)
		}
		pref.Where("dept_ids = ?", strings.TrimRight(depthIds, ","))
	}

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

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

	userIds := make([]int64, 0)
	for _, v := range systemUserList {
		userIds = append(userIds, v.Id)
	}

	systemUserDepthRoleList := make([]*model.SystemUserDepthRole, 0)
	if err = s.DB.Model(new(model.SystemUserDepthRole)).
		Where("user_id IN ?", userIds).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		Find(&systemUserDepthRoleList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	systemUserDepthRoleMap := make(map[int64]*model.SystemUserDepthRole)
	for _, v := range systemUserDepthRoleList {
		systemUserDepthRoleMap[v.UserId] = v
	}

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

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

	appPastureList, err := s.FindPastureList(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

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

// DeleteSystemUser 删除系统用户
func (s *StoreEntry) DeleteSystemUser(ctx context.Context, userId int64) error {
	systemUser := &model.SystemUser{Id: userId}
	if err := s.DB.Model(new(model.SystemUser)).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 {
	_, err := s.GetUserModel(ctx)
	if err != nil {
		return xerr.WithStack(err)
	}

	pastureIds, deptIds := "", ""
	deptIdsMap := make(map[int64]string)
	for _, pasture := range req.PastureDepthList {
		appPasture, err := s.GetAppPastureListByFarmId(ctx, pasture.FarmId)
		if err != nil {
			return xerr.WithStack(err)
		}
		pastureIds += fmt.Sprintf("%d,", appPasture.Id)
		if len(pasture.DepthList) > 0 {
			for _, dept := range pasture.DepthList {
				deptIds += fmt.Sprintf("%d,", dept.Id)
			}
		}
		if len(pastureIds) > 0 {
			pastureIds = strings.TrimSuffix(pastureIds, ",")
		}

		if len(deptIds) > 0 {
			deptIds = strings.TrimSuffix(deptIds, ",")
			deptIdsMap[appPasture.Id] = deptIds
		}
	}

	systemUser := &model.SystemUser{}
	if req.Id > 0 {
		systemUser.Id = int64(req.Id)
		if err = s.DB.Model(new(model.SystemUser)).
			First(systemUser).Error; err != nil {
			return xerr.WithStack(err)
		}

		systemUser.UserUpdate(req, pastureIds)
		if err = s.DB.Transaction(func(tx *gorm.DB) error {
			if err = tx.Model(new(model.SystemUser)).
				Select("name", "nick_name", "mobile", "gender", "pasture_ids", "remarks", "is_show").
				Where("id = ?", systemUser.Id).
				Updates(systemUser).Error; err != nil {
				return xerr.WithStack(err)
			}

			// 删除重新创建新数据
			if err = tx.Model(new(model.SystemUserDepthRole)).
				Where("user_id = ?", systemUser.Id).
				Delete(new(model.SystemUserDepthRole)).Error; err != nil {
				return xerr.WithStack(err)
			}

			newSystemUserDepthRoleList := model.NewSystemUserDepthRoleList(systemUser.Id, deptIdsMap)
			if err = tx.Model(new(model.SystemUserDepthRole)).
				Create(newSystemUserDepthRoleList).Error; err != nil {
				return xerr.WithStack(err)
			}

			return nil
		}); err != nil {
			return xerr.WithStack(err)
		}
	} else {
		var count int64
		if err = s.DB.Model(new(model.SystemUser)).
			Where("name = ?", req.Name).
			Where("mobile = ?", req.Mobile).
			Where("is_show = ?", pasturePb.IsShow_Ok).
			Count(&count).Error; err != nil {
			return xerr.WithStack(err)
		}

		if count > 0 {
			return xerr.Customf("系统中该用户名称已经存在: %s_%s", req.Name, req.Mobile)
		}

		newSystemUser := model.NewSystemUser(req, deptIds, pastureIds, req.PastureDepthList)
		if err = s.DB.Transaction(func(tx *gorm.DB) error {
			if err = tx.Model(new(model.SystemUser)).
				Create(newSystemUser).Error; err != nil {
				return xerr.WithStack(err)
			}

			if len(deptIdsMap) > 0 {
				newSystemUserDepthRoleList := model.NewSystemUserDepthRoleList(newSystemUser.Id, deptIdsMap)
				if err = tx.Model(new(model.SystemUserDepthRole)).
					Create(newSystemUserDepthRoleList).Error; err != nil {
					return xerr.WithStack(err)
				}
			}
			return nil
		}); err != nil {
			return xerr.WithStack(err)
		}
	}
	return nil
}

// GetSystemUserMenu 返回系统用户相关菜单权限
func (s *StoreEntry) GetSystemUserMenu(ctx context.Context) (*pasturePb.SystemUserMenuTreeResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	// 根据用户token获取用户数据
	systemUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return nil, xerr.Custom("该用户数据不存在")
	}

	systemUserDepthRole, err := s.GetSystemUserDepthRole(ctx, userModel.AppPasture.Id, systemUser.Id)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	// 获取用户角色数据
	systemRoleList := make([]*model.SystemRole, 0)
	if err = s.DB.Where("id IN ?", systemUserDepthRole.GetRoleIds()).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		Where("is_delete = ?", pasturePb.IsShow_Ok).
		Where("is_show = ?", pasturePb.IsShow_Ok).
		Find(&systemRoleList).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, xerr.Custom("该用户角色不存在")
		}
		return nil, xerr.WithStack(err)
	}

	if len(systemRoleList) <= 0 {
		return nil, xerr.Custom("该用户角色不存在")
	}

	roleIds := make([]int64, 0)
	for _, role := range systemRoleList {
		roleIds = append(roleIds, role.Id)
	}

	systemRoleMenuList := make([]*model.SystemRoleMenu, 0)
	if err = s.DB.Model(new(model.SystemRoleMenu)).
		Where("role_id IN ?", roleIds).
		Find(&systemRoleMenuList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	if len(systemRoleMenuList) <= 0 {
		return nil, xerr.Custom("该用户角色没有菜单权限")
	}

	// 菜单Id
	menuIds := make([]int64, 0)
	for _, roleMenu := range systemRoleMenuList {
		menuIds = append(menuIds, roleMenu.MenuId)
	}

	// 获取用户菜单
	systemMenuList, err := s.GetMenusWithParents(ctx, userModel.AppPasture.Id, menuIds)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	zaplog.Info("获取用户菜单权限",
		zap.Any("user", systemUser),
		zap.Any("menuIds", menuIds),
		zap.Any("systemMenuList", systemMenuList),
		zap.Any("data", model.SystemMenuSlice(systemMenuList).ToTree()),
	)
	// 获取菜单权限
	return &pasturePb.SystemUserMenuTreeResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.SystemMenuSlice(systemMenuList).ToTree(),
	}, nil
}

// ResetPasswordSystemUser 重置系统用户密码
func (s *StoreEntry) ResetPasswordSystemUser(ctx context.Context, req *pasturePb.ResetUserPasswordRequest) error {
	systemUser := &model.SystemUser{}
	if err := s.DB.Model(new(model.SystemUser)).
		Where("id = ?", req.Id).
		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) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	systemUser := &model.SystemUser{Id: userId}
	if err = s.DB.Model(new(model.SystemUser)).
		Where("is_delete = ?", pasturePb.IsShow_Ok).
		First(systemUser).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, xerr.Custom("该用户不存在")
		}
		return nil, xerr.WithStack(err)
	}

	if systemUser.IsShow == pasturePb.IsShow_No {
		return nil, xerr.Custom("该用户已禁用")
	}

	systemUserDepthRole := &model.SystemUserDepthRole{}
	if err = s.DB.Model(new(model.SystemUserDepthRole)).
		Where("user_id = ?", systemUser.Id).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		First(systemUserDepthRole).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, xerr.Custom("该用户没有角色")
		}
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SystemUserRoleResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: &pasturePb.UserRoleData{
			RoleIds: systemUserDepthRole.GetRoleIds(),
		},
	}, nil
}

func (s *StoreEntry) SystemUserRoleSave(ctx context.Context, req *pasturePb.SystemUserRoleRequest) error {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return xerr.WithStack(err)
	}

	systemUser := &model.SystemUser{Id: int64(req.UserId)}
	if err = s.DB.Model(new(model.SystemUser)).
		First(systemUser).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该用户不存在")
		}
		return xerr.WithStack(err)
	}

	if systemUser.IsDelete == pasturePb.IsShow_No {
		return xerr.Custom("该用户已删除")
	}

	if systemUser.IsShow == pasturePb.IsShow_No {
		return xerr.Custom("该用户已禁用")
	}

	systemRoleList := make([]*model.SystemRole, 0)
	if err = s.DB.Model(new(model.SystemRole)).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		Where("is_delete = ?", pasturePb.IsShow_Ok).
		Where("id IN ?", req.RoleIds).
		Find(&systemRoleList).Error; err != nil {
		return xerr.WithStack(err)
	}

	if len(systemRoleList) <= 0 {
		return xerr.Custom("该用户没有角色")
	}

	roleIdsStr := ""
	for _, systemRole := range systemRoleList {
		roleIdsStr += fmt.Sprintf("%d,", systemRole.Id)
	}

	if len(roleIdsStr) > 0 {
		roleIdsStr = strings.TrimRight(roleIdsStr, ",")
	}

	if err = s.DB.Transaction(func(tx *gorm.DB) error {

		var systemUserDepthRoleCount int64
		if err = tx.Model(new(model.SystemUserDepthRole)).
			Where("user_id = ?", systemUser.Id).
			Where("pasture_id = ?", userModel.AppPasture.Id).
			Count(&systemUserDepthRoleCount).Error; err != nil {
			return xerr.WithStack(err)
		}

		if systemUserDepthRoleCount > 0 {
			if err = tx.Model(new(model.SystemUserDepthRole)).
				Where("user_id = ?", systemUser.Id).
				Where("pasture_id = ?", userModel.AppPasture.Id).
				Update("role_ids", roleIdsStr).Error; err != nil {
				return xerr.WithStack(err)
			}
		} else {

			newSystemUserDepthRole := model.NewSystemUserDepthRole(userModel.AppPasture.Id, systemUser.Id, "", roleIdsStr)
			if err = tx.Model(new(model.SystemUserDepthRole)).
				Create(newSystemUserDepthRole).Error; err != nil {
				return xerr.WithStack(err)
			}
		}

		return nil
	}); err != nil {
		return xerr.WithStack(err)
	}

	return nil
}

func (s *StoreEntry) GetMenusWithParents(ctx context.Context, pastureId int64, menuIds []int64) ([]*model.SystemMenu, error) {
	if len(menuIds) <= 0 {
		return nil, xerr.Customf("菜单id不能为空")
	}

	systemPastureMenuList := make([]*model.SystemPastureMenu, 0)
	if err := s.DB.Model(new(model.SystemPastureMenu)).
		Where("pasture_id = ?", pastureId).
		Find(&systemPastureMenuList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	pastureMenuIds := make([]int64, 0)
	for _, v := range systemPastureMenuList {
		pastureMenuIds = append(pastureMenuIds, v.MenuId)
	}

	systemMenuList := make([]*model.SystemMenu, 0)
	if err := s.DB.Model(new(model.SystemMenu)).
		Where("id IN ?", pastureMenuIds).
		Find(&systemMenuList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	systemMenuMap := make(map[int64]*model.SystemMenu)
	for _, v := range systemMenuList {
		systemMenuMap[v.Id] = v
	}

	userMenuList := make([]*model.SystemMenu, 0)
	if err := s.DB.Model(new(model.SystemMenu)).
		Where("id IN ?", menuIds).
		Find(&userMenuList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	// 收集所有需要的菜单ID（包括各级父菜单ID）
	processed := make(map[int64]struct{}) // 防止重复处理
	for _, menu := range userMenuList {
		currentID := menu.Id

		// 向上追溯所有父级菜单
		for currentID != 0 {
			// 如果已经处理过，跳过
			if _, exists := processed[currentID]; exists {
				break
			}

			// 添加到结果集
			menuIds = append(menuIds, currentID)
			processed[currentID] = struct{}{}

			// 获取父菜单
			parentMenu, ok := systemMenuMap[currentID]
			if !ok || parentMenu.ParentId == 0 {
				break
			}
			currentID = parentMenu.ParentId
		}
	}

	// 去重
	menuIds = util.DeduplicateInt64(menuIds)
	allSystemMenuList := make([]*model.SystemMenu, 0)
	if err := s.DB.Model(new(model.SystemMenu)).
		Where("id IN (?) AND is_delete = ?", menuIds, pasturePb.IsShow_Ok).
		Order("parent_id ASC,id ASC").
		Group("id").
		Find(&allSystemMenuList).Error; err != nil {
		return nil, fmt.Errorf("failed to query menus: %v", err)
	}

	return allSystemMenuList, nil
}