kpt-pasture/module/backend/system_service_more.go

package backend

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

	"gorm.io/gorm"

	"go.uber.org/zap"

	"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/xerr"
)

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

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

	if len(systemPastureMenuList) <= 0 {
		return nil, xerr.Customf("系统菜单未初始化，请联系管理员！")
	}

	systemMenuList := make([]*model.SystemMenu, 0)
	pref := s.DB.Model(new(model.SystemMenu)).
		Where("id IN ? ", model.SystemPastureMenuSlice(systemPastureMenuList).GetMenuIds()).
		Where("is_delete = ? ", operationPb.IsShow_OK)
	if req.Name != "" {
		pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
	}

	if err = pref.Order("parent_id ASC,id ASC").
		Find(&systemMenuList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SystemMenuTreeResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.SystemMenuSlice(systemMenuList).ToTree(),
	}, nil
}

// SearchSystemMenuList 菜单列表查询
func (s *StoreEntry) SearchSystemMenuList(ctx context.Context, req *pasturePb.SearchMenuRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchMenuResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	systemPastureMenuList := make([]*model.SystemPastureMenu, 0)
	if err = s.DB.Model(new(model.SystemPastureMenu)).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		Find(&systemPastureMenuList).
		Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	if len(systemPastureMenuList) <= 0 {
		return nil, xerr.Customf("系统菜单未初始化，请联系管理员！")
	}

	systemMenuList := make([]*model.SystemMenu, 0)
	var count int64 = 0

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

	if err = pref.Order("parent_id").
		Count(&count).
		Find(&systemMenuList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

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

func (s *StoreEntry) SearchSystemDeptList(ctx context.Context, req *pasturePb.SearchDeptRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchDeptResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	deptList := make([]*model.SystemDept, 0)
	var count int64 = 0
	pastureId := userModel.AppPasture.Id
	pref := s.DB.Model(new(model.SystemDept)).
		Where("pasture_id = ?", pastureId).
		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(pagination.PageSize)).
		Offset(int(pagination.PageOffset)).
		Find(&deptList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	systemUserList, _ := s.SystemUserList(ctx, pastureId)

	return &pasturePb.SearchDeptResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: &pasturePb.DeptData{
			List:     model.SystemDeptSlice(deptList).ToPB(systemUserList),
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) SystemDepthDelete(ctx context.Context, id int64) error {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return xerr.WithStack(err)
	}

	dept := &model.SystemDept{}
	if err = s.DB.Model(new(model.SystemDept)).
		Where("id = ?", id).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		First(dept).Error; err != nil {
		return xerr.WithStack(err)
	}

	if dept.IsInit == pasturePb.IsShow_Ok {
		return xerr.Custom("该部门不可删除")
	}

	if dept.IsShow == pasturePb.IsShow_No {
		return nil
	}
	if err = s.DB.Model(new(model.SystemDept)).
		Where("id = ?", id).
		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 {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return xerr.WithStack(err)
	}
	pastureId := userModel.AppPasture.Id
	if req.Id > 0 {
		systemDept := &model.SystemDept{}
		if err = s.DB.Model(new(model.SystemDept)).
			Where("id = ?", req.Id).
			Where("pasture_id = ?", pastureId).
			Where("is_delete = ?", pasturePb.IsShow_Ok).
			First(systemDept).Error; err != nil {
			zaplog.Error("SystemDeptCreateOrUpdate", zap.Any("req", req), zap.Any("err", err))
			return xerr.Customf("部门信息错误: %s", err.Error())
		}

		// 更新部门信息
		systemDept.DepthUpdate(req)
		if err = s.DB.Model(new(model.SystemDept)).
			Where("id = ?", systemDept.Id).
			Where("pasture_id = ?", pastureId).
			Select("name", "parent_id", "sort", "remarks", "is_show").
			Updates(systemDept).Error; err != nil {
			return xerr.WithStack(err)
		}
	} else {
		newSystemDepth := model.NewSystemDept(pastureId, req)
		if err = s.DB.Model(new(model.SystemDept)).
			Create(newSystemDepth).Error; err != nil {
			return xerr.WithStack(err)
		}
	}
	return nil
}

func (s *StoreEntry) SearchUserPastureList(ctx context.Context) (*pasturePb.SystemUserPastureListResponse, error) {
	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	farmList, err := s.FindPastureListByIds(ctx, currentUser.GetPastureIds())
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SystemUserPastureListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.AppPastureListSlice(farmList).ToPB(),
	}, err
}

func (s *StoreEntry) SystemDeptTree(ctx context.Context, req *pasturePb.SearchDeptRequest) (*pasturePb.DeptTreeResponse, error) {
	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	pastureIdsStr := currentUser.PastureIds
	if pastureIdsStr == "" {
		return nil, xerr.Customf("请先将该用户绑定牧场！谢谢！")
	}

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

	systemDeptList := make([]*model.SystemDept, 0)
	pref := s.DB.Model(new(model.SystemDept)).
		Where("is_delete = ?", pasturePb.IsShow_Ok).
		Where("pasture_id = ?", userModel.AppPasture.Id)

	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.Find(&systemDeptList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	if req.Name != "" {
		parentDepths, err := s.findAllParentDepths(ctx, systemDeptList, []int32{int32(userModel.AppPasture.Id)})
		if err != nil {
			return nil, xerr.WithStack(err)
		}
		systemDeptList = append(systemDeptList, parentDepths...)
	}

	// 没有搜到部门数据，默认返回所有牧场一级数据
	if len(systemDeptList) == 0 {
		systemDeptList = s.SearchDepthPastureList(ctx)
	}

	return &pasturePb.DeptTreeResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.SystemDeptSlice(systemDeptList).ToTree(),
	}, nil
}

func (s *StoreEntry) SearchDepthPastureList(ctx context.Context) []*model.SystemDept {
	systemDeptList := make([]*model.SystemDept, 0)
	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return systemDeptList
	}

	pastureIds := currentUser.GetPastureIds()
	if err = s.DB.Model(new(model.SystemDept)).
		Where("is_delete = ?", pasturePb.IsShow_Ok).
		Where("pasture_id IN ?", pastureIds).
		Where("parent_id = ?", 0).
		Find(&systemDeptList).Error; err != nil {
		return systemDeptList
	}
	return systemDeptList
}

// findAllParentDepths 递归查找所有上级部门
func (s *StoreEntry) findAllParentDepths(ctx context.Context, depths []*model.SystemDept, pastureIds []int32) ([]*model.SystemDept, error) {
	var allDepths []*model.SystemDept
	var parentIDs []int64
	foundDepths := make(map[int64]bool)

	// 收集所有需要查找的父部门ID
	for _, dept := range depths {
		if dept.ParentId > 0 && !foundDepths[dept.ParentId] {
			parentIDs = append(parentIDs, dept.ParentId)
			foundDepths[dept.ParentId] = true
		}
	}

	if len(parentIDs) == 0 {
		return allDepths, nil
	}

	// 查询这些父部门
	var parentDepths []*model.SystemDept
	if err := s.DB.Model(new(model.SystemDept)).
		Where("id IN ?", parentIDs).
		Where("pasture_id IN ?", pastureIds).
		Where("is_delete = ?", pasturePb.IsShow_Ok).
		Find(&parentDepths).Error; err != nil {
		return nil, err
	}

	// 递归查找更上级的部门
	higherParents, err := s.findAllParentDepths(ctx, parentDepths, pastureIds)
	if err != nil {
		return nil, err
	}

	allDepths = append(allDepths, parentDepths...)
	allDepths = append(allDepths, higherParents...)

	return allDepths, nil
}

// SystemRoleCreateOrUpdate 添加角色
func (s *StoreEntry) SystemRoleCreateOrUpdate(ctx context.Context, req *pasturePb.SearchRoleRequest) error {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return xerr.WithStack(err)
	}

	if req.Id > 0 {
		systemRole := &model.SystemRole{}
		if err = s.DB.Model(systemRole).
			Where("id = ?", req.Id).
			Where("pasture_id = ?", userModel.AppPasture.Id).
			First(systemRole).Error; err != nil {
			return xerr.WithStack(err)
		}

		if systemRole.IsDelete == pasturePb.IsShow_No {
			return xerr.Custom("角色已删除")
		}

		systemRole.RoleUpdate(req)
		if err = s.DB.Model(systemRole).
			Select("name", "remarks", "is_show").
			Where("id = ?", req.Id).
			Where("pasture_id = ?", userModel.AppPasture.Id).
			Updates(systemRole).Error; err != nil {
			return xerr.WithStack(err)
		}

	} else {
		newSystemRole := model.NewSystemRole(userModel.AppPasture.Id, req)
		if err = s.DB.Model(new(model.SystemRole)).Create(newSystemRole).Error; err != nil {
			return xerr.WithStack(err)
		}
	}

	return nil
}

// DeleteSystemRole 删除系统角色
func (s *StoreEntry) DeleteSystemRole(ctx context.Context, roleId int64) error {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return xerr.WithStack(err)
	}
	systemRole := &model.SystemRole{}
	if err = s.DB.Model(new(model.SystemRole)).
		Where("id = ? ", roleId).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		Where("is_delete = ?", pasturePb.IsShow_Ok).
		First(systemRole).Error; err != nil {
		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 {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return xerr.WithStack(err)
	}
	systemRole := &model.SystemRole{}
	if err = s.DB.Model(new(model.SystemRole)).
		Where("id = ?", roleId).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		Where("is_delete = ?", pasturePb.IsShow_Ok).
		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, pagination *pasturePb.PaginationModel) (*pasturePb.SearchRoleResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	systemRoleList := make([]*model.SystemRole, 0)
	var count int64 = 0

	pref := s.DB.Model(new(model.SystemRole)).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		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("is_show asc,id desc").
		Count(&count).
		Limit(int(pagination.PageSize)).
		Offset(int(pagination.PageOffset)).
		Find(&systemRoleList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

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

// SearchSystemRoleListByIds 根据id查询角色列表
func (s *StoreEntry) SearchSystemRoleListByIds(ctx context.Context, pastureId int64, ids []int64) ([]*model.SystemRole, error) {
	systemRoleList := make([]*model.SystemRole, 0)
	if err := s.DB.Model(new(model.SystemRole)).
		Where("pasture_id = ?", pastureId).
		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("该数据已被删除")
	}

	menuList := make([]*model.SystemRoleMenu, 0)
	if err := s.DB.Where("role_id = ?", systemRole.Id).Find(&menuList).Error; err != nil {
		zaplog.Error("GetSystemRoleMenuList", zap.Any("Err", err))
	}
	return &pasturePb.SystemRoleMenuResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.SystemRoleMenuSlice(menuList).ToPB(),
	}, nil

}

// RoleMenuSave 保存系统角色对应的菜单数据
func (s *StoreEntry) RoleMenuSave(ctx context.Context, res *pasturePb.SystemRoleMenuRequest) error {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return xerr.WithStack(err)
	}

	if len(res.MenuIds) <= 0 {
		return xerr.Custom("请选择菜单")
	}

	systemRole := &model.SystemRole{}
	if err = s.DB.Model(new(model.SystemRole)).
		Where("id = ?", res.RoleId).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		First(systemRole).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return xerr.Custom("该数据不存在")
		}
		return xerr.WithStack(err)
	}

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

	if systemRole.IsShow == pasturePb.IsShow_No {
		return xerr.Custom("该数据已禁用")
	}

	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)
		}

		newSystemRoleMenuList := model.NewSystemRoleMenu(systemRole.Id, res.MenuIds)
		if err = tx.Model(new(model.SystemRoleMenu)).
			Create(newSystemRoleMenuList).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) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

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

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