jnpf-java-boot-v6x/jnpf-permission/jnpf-permission-biz/target/classes/jnpf/permission/service/impl/PermissionGroupServiceImpl.java

package jnpf.permission.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import jnpf.base.service.SuperServiceImpl;
import jnpf.constant.PermissionConst;
import jnpf.permission.entity.*;
import jnpf.permission.mapper.PermissionGroupMapper;
import jnpf.permission.model.permissiongroup.PaginationPermissionGroup;
import jnpf.permission.service.*;
import jnpf.util.RandomUtil;
import jnpf.util.StringUtil;
import jnpf.util.type.AuthorizeType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

@Service
public class PermissionGroupServiceImpl extends SuperServiceImpl<PermissionGroupMapper, PermissionGroupEntity> implements PermissionGroupService {

    @Autowired
    private UserService userService;
    @Autowired
    private UserRelationService userRelationService;
    @Autowired
    private PositionService positionService;
    @Autowired
    private GroupService groupService;
    @Autowired
    private AuthorizeService authorizeService;
    @Autowired
    private OrganizeService organizeService;

    @Override
    public List<PermissionGroupEntity> list(PaginationPermissionGroup pagination) {
        boolean flag = false;
        QueryWrapper<PermissionGroupEntity> queryWrapper = new QueryWrapper<>();
        String keyword = pagination.getKeyword();
        if (StringUtil.isNotEmpty(keyword)) {
            flag = true;
            queryWrapper.lambda().and(
                    t -> t.like(PermissionGroupEntity::getFullName, keyword)
                            .or().like(PermissionGroupEntity::getEnCode, keyword)
                            .or().like(PermissionGroupEntity::getDescription, keyword)
            );
        }
        if (pagination.getEnabledMark() != null) {
            queryWrapper.lambda().eq(PermissionGroupEntity::getEnabledMark, pagination.getEnabledMark());
        }
        queryWrapper.lambda().orderByAsc(PermissionGroupEntity::getSortCode).orderByDesc(PermissionGroupEntity::getCreatorTime);
        if (flag) {
            queryWrapper.lambda().orderByDesc(PermissionGroupEntity::getLastModifyTime);
        }
        Page<PermissionGroupEntity> page = new Page<>(pagination.getCurrentPage(), pagination.getPageSize());
        IPage<PermissionGroupEntity> iPage = this.page(page, queryWrapper);
        return pagination.setData(iPage.getRecords(), iPage.getTotal());
    }

    @Override
    public List<PermissionGroupEntity> list(boolean filterEnabledMark, List<String> ids) {
        if (ids != null && ids.size() == 0) {
            return new ArrayList<>();
        }
        QueryWrapper<PermissionGroupEntity> queryWrapper = new QueryWrapper<>();
        if (filterEnabledMark) {
            queryWrapper.lambda().eq(PermissionGroupEntity::getEnabledMark, 1);
        }
        if (ids != null && ids.size() > 0) {
            queryWrapper.lambda().in(PermissionGroupEntity::getId, ids);
        }
        return this.list(queryWrapper);
    }

    @Override
    public PermissionGroupEntity info(String id) {
        QueryWrapper<PermissionGroupEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(PermissionGroupEntity::getId, id);
        return this.getOne(queryWrapper);
    }

    @Override
    public boolean create(PermissionGroupEntity entity) {
        entity.setId(RandomUtil.uuId());
        return this.save(entity);
    }

    @Override
    public boolean update(String id, PermissionGroupEntity entity) {
        entity.setId(id);
        return this.updateById(entity);
    }

    @Override
    public boolean delete(PermissionGroupEntity entity) {
        return this.removeById(entity);
    }

    @Override
    public boolean isExistByFullName(String id, PermissionGroupEntity entity) {
        QueryWrapper<PermissionGroupEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(PermissionGroupEntity::getFullName, entity.getFullName());
        if (StringUtil.isNotEmpty(id)) {
            queryWrapper.lambda().ne(PermissionGroupEntity::getId, id);
        }
        return this.count(queryWrapper) > 0;
    }

    @Override
    public boolean isExistByEnCode(String id, PermissionGroupEntity entity) {
        QueryWrapper<PermissionGroupEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(PermissionGroupEntity::getEnCode, entity.getEnCode());
        if (StringUtil.isNotEmpty(id)) {
            queryWrapper.lambda().ne(PermissionGroupEntity::getId, id);
        }
        return this.count(queryWrapper) > 0;
    }

    @Override
    public PermissionGroupEntity permissionMember(String id) {
        QueryWrapper<PermissionGroupEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(PermissionGroupEntity::getId, id);
        queryWrapper.lambda().select(PermissionGroupEntity::getId, PermissionGroupEntity::getPermissionMember);
        return this.getOne(queryWrapper);
    }

    @Override
    public List<PermissionGroupEntity> getPermissionGroupByUserId(String userId, String organizeId, boolean singletonOrg, String systemId) {
        List<PermissionGroupEntity> list = new ArrayList<>();
        // 用户本身有没有权限
        UserEntity userEntity = userService.getInfo(userId);
        if (userEntity == null) {
            return list;
        }
        List<PermissionGroupEntity> allGroupList = this.list(true, null).stream().filter(t -> Objects.equals(0, t.getType())).collect(Collectors.toList());
        allGroupList.forEach(permissionGroupEntity -> {
            if (authorizeService.existAuthorize(permissionGroupEntity.getId(), systemId)) {
                list.add(permissionGroupEntity);
            }
        });
        List<PermissionGroupEntity> permissionGroupEntities = this.list(true, null).stream().filter(t -> StringUtil.isNotEmpty(t.getPermissionMember())).collect(Collectors.toList());
        String finalUserId = userId + "--" + PermissionConst.USER;
        List<PermissionGroupEntity> collect = permissionGroupEntities.stream().filter(entity -> entity.getPermissionMember().contains(finalUserId)).collect(Collectors.toList());
        collect.forEach(permissionGroupEntity -> {
            if (authorizeService.existAuthorize(permissionGroupEntity.getId(), systemId)) {
                list.add(permissionGroupEntity);
            }
        });
        // 用户关系表
        List<UserRelationEntity> listByUserId = userRelationService.getListByUserId(userEntity.getId()).stream().filter(r -> StringUtil.isNotEmpty(r.getObjectId())).collect(Collectors.toList());
        // 分组有没有权限
        List<String> groupIds = new ArrayList<>();
        List<String> groupId = listByUserId.stream().filter(t -> PermissionConst.GROUP.equals(t.getObjectType()))
                .map(UserRelationEntity::getObjectId).collect(Collectors.toList());
        List<String> groupName = groupService.getListByIds(groupId)
                .stream().map(GroupEntity::getId).collect(Collectors.toList());
        groupName.forEach(t -> groupIds.add(t + "--group"));
        for (String id : groupIds) {
            List<PermissionGroupEntity> collect1 = permissionGroupEntities.stream().filter(entity -> entity.getPermissionMember().contains(id)).collect(Collectors.toList());
            collect1.forEach(permissionGroupEntity -> {
                if (authorizeService.existAuthorize(permissionGroupEntity.getId(), systemId)) {
                    list.add(permissionGroupEntity);
                }
            });
        }

        // 组织有权限
        List<String> organizeIds = new ArrayList<>();
        List<String> orgIds = new ArrayList<>();
        if (list.size() > 0) {
            // 当前组织及组织下岗位、角色权限组
            organizeIds.add(userEntity.getOrganizeId());
        } else {
            // 此情况下不找
            if (StringUtil.isNotEmpty(organizeId)) {
                organizeIds.add(organizeId);
            } else {
                // 找到一个有权限的组织、岗位、角色
                List<UserRelationEntity> listByObjectType = userRelationService.getListByObjectType(userId, PermissionConst.ORGANIZE);
                List<String> collect2 = listByObjectType.stream().map(UserRelationEntity::getObjectId).collect(Collectors.toList());
                organizeIds.addAll(collect2);
                organizeIds.add(userEntity.getOrganizeId());
            }
        }
        // 拼上后缀
        if (organizeIds.size() > 0) {
            List<String> collect1 = new ArrayList<>();
            collect1.addAll(organizeService.getOrgEntityList(organizeIds, true)
                    .stream().map(OrganizeEntity::getId).collect(Collectors.toList()));
            orgIds.addAll(collect1);
            collect1.forEach(t -> {
                orgIds.add(t + "--" + PermissionConst.COMPANY);
                orgIds.add(t + "--" + PermissionConst.DEPARTMENT);
            });
        }
        return list;
    }

    @Override
    public String getPermissionGroupByUserId(String userId) {
        // 用户本身有没有权限
        UserEntity userEntity = userService.getInfo(userId);
        if (userEntity == null) {
            return "";
        }
        List<PermissionGroupEntity> allGroupList = this.list(true, null).stream().filter(t -> Objects.equals(0, t.getType())).collect(Collectors.toList());
        for (PermissionGroupEntity permissionGroupEntity : allGroupList) {
            if (authorizeService.existAuthorize(permissionGroupEntity.getId(), null)) {
                return "";
            }
        }
        List<PermissionGroupEntity> permissionGroupEntities = this.list(true, null).stream().filter(t -> StringUtil.isNotEmpty(t.getPermissionMember())).collect(Collectors.toList());
        String finalUserId = userId + "--" + PermissionConst.USER;
        List<PermissionGroupEntity> collect = permissionGroupEntities.stream().filter(entity -> entity.getPermissionMember().contains(finalUserId)).collect(Collectors.toList());
        for (PermissionGroupEntity permissionGroupEntity : collect) {
            if (authorizeService.existAuthorize(permissionGroupEntity.getId(), null)) {
                return "";
            }
        }
        // 用户关系表
        List<UserRelationEntity> listByUserId = userRelationService.getListByUserId(userEntity.getId());
        // 分组有没有权限
        List<String> groupIds = new ArrayList<>();
        List<String> groupId = listByUserId.stream().filter(t -> PermissionConst.GROUP.equals(t.getObjectType()))
                .map(UserRelationEntity::getObjectId).collect(Collectors.toList());
        List<String> groupName = groupService.getListByIds(groupId)
                .stream().map(GroupEntity::getId).collect(Collectors.toList());
        groupName.forEach(t -> groupIds.add(t + "--group"));
        for (String id : groupIds) {
            List<PermissionGroupEntity> collect1 = permissionGroupEntities.stream().filter(entity -> entity.getPermissionMember().contains(id)).collect(Collectors.toList());
            for (PermissionGroupEntity permissionGroupEntity : collect1) {
                if (authorizeService.existAuthorize(permissionGroupEntity.getId(), null)) {
                    return "";
                }
            }
        }

        // 组织有权限
        List<String> orgIds = new ArrayList<>();
        List<UserRelationEntity> listByObjectType = userRelationService.getListByObjectType(userId, PermissionConst.ORGANIZE);
        List<String> collect2 = new ArrayList<>();
        collect2.add(userEntity.getOrganizeId());
        collect2.addAll(listByObjectType.stream().map(UserRelationEntity::getObjectId).collect(Collectors.toList()));
        orgIds.addAll(collect2);
        collect2.forEach(t -> {
            orgIds.add(t + "--" + PermissionConst.COMPANY);
            orgIds.add(t + "--" + PermissionConst.DEPARTMENT);
        });
        return "";
    }

    @Override
    public String getOrgIdByUserIdAndSystemId(String userId, String systemId) {
        // 用户本身有没有权限
        UserEntity userEntity = userService.getInfo(userId);
        if (userEntity == null) {
            return "";
        }
        // 判断有这个应用的权限组有哪些
        List<String> collect = authorizeService.getListByObjectAndItemIdAndType(systemId, PermissionConst.SYSTEM).stream().map(AuthorizeEntity::getObjectId).collect(Collectors.toList());
        List<PermissionGroupEntity> list = this.list(true, collect).stream().filter(t -> StringUtil.isNotEmpty(t.getPermissionMember())).collect(Collectors.toList());
        List<PermissionGroupEntity> allGroupList = this.list(true, collect).stream().filter(t -> Objects.equals(0, t.getType())).collect(Collectors.toList());
        if (allGroupList.size() > 0) {
            return "";
        }

        List<String> objectIds = new ArrayList<>();
        objectIds.add(userId + "--" + PermissionConst.USER);
        // 用户关系
        List<UserRelationEntity> listByUserId = userRelationService.getListByUserId(userEntity.getId())
                .stream().filter(t -> userId.equals(t.getUserId())).collect(Collectors.toList());
        // 分组
        List<String> groupId = listByUserId.stream().filter(t -> PermissionConst.GROUP.equals(t.getObjectType())).map(UserRelationEntity::getObjectId).collect(Collectors.toList());
        List<String> groupName = groupService.getListByIds(groupId)
                .stream().map(GroupEntity::getId).collect(Collectors.toList());
        groupName.forEach(t -> {
            objectIds.add(t + "--group");
        });
        for (String objectId : objectIds) {
            List<PermissionGroupEntity> collect1 = list.stream().filter(t -> t.getPermissionMember().contains(objectId)).collect(Collectors.toList());
            if (collect1.size() > 0) {
                return "";
            }
        }
        // 组织
        List<String> orgIds = listByUserId.stream().filter(t -> PermissionConst.ORGANIZE.equals(t.getObjectType())).map(UserRelationEntity::getObjectId).collect(Collectors.toList());
        List<OrganizeEntity> orgEntityList = organizeService.getOrgEntityList(orgIds, true);
        List<PositionEntity> listByOrganizeId = positionService.getListByOrganizeId(orgIds, false);
        return "";
    }

    @Override
    public List<PermissionGroupEntity> getPermissionGroupAllByUserId(String userId) {
        QueryWrapper<PermissionGroupEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().like(PermissionGroupEntity::getPermissionMember, userId);
        queryWrapper.lambda().eq(PermissionGroupEntity::getEnabledMark, 1);
        return this.list(queryWrapper);
//        UserEntity userEntity = userService.getInfo(userId);
//        // 通过用户id获取相关的组织、部门、岗位、角色、分组
//        if (userEntity == null) {
//            return new ArrayList<>();
//        }
//        if (userEntity.getIsAdministrator() == 1) {
//            return this.list(true, null);
//        }
//        Set<String> objIds = new HashSet<>();
//
//        // 用户与组织关系
//        List<String> orgIds = new ArrayList<>();
//        List<String> orgId = userRelationService.getListByObjectType(userEntity.getId(), PermissionConst.ORGANIZE).stream().map(UserRelationEntity::getObjectId).collect(Collectors.toList());
//        orgId.forEach(t -> orgIds.add(t + "--" + PermissionConst.ORGANIZE));
//        // 用户与岗位关系
//        List<String> positionIds = new ArrayList<>();
//        List<String> positionId = userRelationService.getListByObjectType(userEntity.getId(), PermissionConst.POSITION).stream().map(UserRelationEntity::getObjectId).collect(Collectors.toList());
//        positionId.forEach(t -> positionIds.add(t + "--" + PermissionConst.POSITION));
//        // 用户与角色关系
//        List<String> roleIds = new ArrayList<>();
//        List<String> roleId = userRelationService.getListByObjectType(userEntity.getId(), PermissionConst.ROLE).stream().map(UserRelationEntity::getObjectId).collect(Collectors.toList());
//        roleId.forEach(t -> roleIds.add(t + "--" + PermissionConst.ROLE));
//        // 用户与跟分组关系
//        List<String> groupIds = new ArrayList<>();
//        List<String> groupId = userRelationService.getListByObjectType(userEntity.getId(), PermissionConst.GROUP).stream().map(UserRelationEntity::getObjectId).collect(Collectors.toList());
//        groupId.forEach(t -> groupIds.add(t + "--" + PermissionConst.GROUP));
//
//        objIds.addAll(orgIds);
//        objIds.addAll(positionIds);
//        objIds.addAll(roleIds);
//        objIds.addAll(groupIds);
//
//        Set<String> permissionGroupIds = new HashSet<>();
//
//        List<PermissionGroupEntity> permissionGroupEntities = this.list(true, null);
//        objIds.forEach(objId -> {
//            List<PermissionGroupEntity> collect = permissionGroupEntities.stream().filter(entity -> entity.getPermissionMember().contains(objId)).collect(Collectors.toList());
//            if (collect.size() > 0) {
//                permissionGroupIds.addAll(collect.stream().map(PermissionGroupEntity::getId).collect(Collectors.toList()));
//            }
//        });
//        return this.list(true, new ArrayList<>(permissionGroupIds));
    }

    @Override
    @Transactional
    public boolean updateByUser(String fromId, String toId, List<String> permissionList) {
        if (StringUtil.isEmpty(fromId)) {
            return false;
        }
        String fromIds = fromId + "--" + PermissionConst.USER;
        List<PermissionGroupEntity> permissionGroupAllByUserId = this.getPermissionGroupAllByUserId(fromId);
        permissionGroupAllByUserId.forEach(t -> {
            if (permissionList.contains(t.getId())) {
                t.setPermissionMember(StringUtil.isNotEmpty(t.getPermissionMember()) ? t.getPermissionMember().replaceAll(fromIds, toId + "--" + PermissionConst.USER) : "");
                this.updateById(t);
            }
        });
        return true;
    }

    @Override
    public List<PermissionGroupEntity> getPermissionGroupByModuleId(String moduleId) {
        // 获取到菜单和权限组的关系
        List<AuthorizeEntity> authorizeEntityList = authorizeService.getListByObjectAndItem(moduleId, AuthorizeType.MODULE);
        // 获取权限组信息
        List<PermissionGroupEntity> list = this.list(true,
                authorizeEntityList.stream().map(AuthorizeEntity::getObjectId).collect(Collectors.toList()));
        return list;
    }

    @Override
    public List<PermissionGroupEntity> list(List<String> ids) {
        if (ids.size() == 0) {
            return new ArrayList<>();
        }
        QueryWrapper<PermissionGroupEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().in(PermissionGroupEntity::getId, ids);
        return this.list(queryWrapper);
    }

    @Override
    public List<PermissionGroupEntity> getPermissionGroupByObjectId(String objectId, String objectType) {
        List<PermissionGroupEntity> permissionGroupEntities = this.list(true, null).stream().filter(t -> StringUtil.isNotEmpty(t.getPermissionMember())).collect(Collectors.toList());
        String id = objectId + "--" + objectType;
        List<String> idList = new ArrayList<>();
        idList.add(id);
        if (PermissionConst.USER.equalsIgnoreCase(objectType)) {
            List<UserRelationEntity> allOrgRelationByUserId = userRelationService.getListByObjectType(objectId, null);
            for (UserRelationEntity item : allOrgRelationByUserId) {
                String oType = item.getObjectType();
                if (PermissionConst.ORGANIZE.equals(oType)) {
                    oType = PermissionConst.COMPANY;
                    idList.add(item.getObjectId() + "--" + PermissionConst.DEPARTMENT);
                }
                idList.add(item.getObjectId() + "--" + oType.toLowerCase());
            }
        }
        List<PermissionGroupEntity> collect = permissionGroupEntities.stream().filter(entity -> {
            boolean hashOperations = false;
            for (String s : idList) {
                if (entity.getPermissionMember().contains(s)) {
                    hashOperations = true;
                    break;
                }
            }
            return hashOperations;
        }).collect(Collectors.toList());
        List<PermissionGroupEntity> allGroupList = this.list(true, null).stream().filter(t -> Objects.equals(0, t.getType())).collect(Collectors.toList());
        for (PermissionGroupEntity entity : allGroupList) {
            if (!collect.contains(entity)) {
                collect.add(entity);
            }
        }
        return collect;
    }

}