PermissionServiceImpl.java

package pwc.taxtech.atms.service.impl;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;

import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import pwc.taxtech.atms.common.CommonConstants;
import pwc.taxtech.atms.common.CommonUtils;
import pwc.taxtech.atms.dto.navtree.DevTreeDto;
import pwc.taxtech.atms.dto.navtree.RolePermissionDisplayDto;
import pwc.taxtech.atms.dto.permission.RolePermissionDto;
import pwc.taxtech.atms.entitiy.Permission;
import pwc.taxtech.atms.entitiy.PermissionExample;
import pwc.taxtech.atms.entitiy.RolePermission;
import pwc.taxtech.atms.service.PermissionService;

@Service
public class PermissionServiceImpl extends AbstractService implements PermissionService {

    /* (non-Javadoc)
     * @see pwc.taxtech.atms.service.PermissionService#getIvhTreePermissionsByRoleID(java.lang.String, java.lang.String)
     */
    @Override
    public RolePermissionDisplayDto getIvhTreePermissionsByRoleID(String roleID, String serviceType) {

        RolePermissionDisplayDto rolePermissionDisplayDto = new RolePermissionDisplayDto();
        
        List<RolePermission> rolePermissionList = rolePermissionMapper.selectByRoleAndServiceTypeWithAssociation(roleID, serviceType);
        if(rolePermissionList == null || rolePermissionList.isEmpty()   ) {
            return rolePermissionDisplayDto;
        }
        List<RolePermissionDto> rolePermissionDtoList = new ArrayList<>();
        rolePermissionList.stream().forEach(rolePermission -> {
            RolePermissionDto rolePermissionDto = new RolePermissionDto();
            rolePermissionDto.setID(rolePermission.getID());
            rolePermissionDto.setCode(rolePermission.getPermission().getCode());
            rolePermissionDto.setRoleID(rolePermission.getRoleID());
            rolePermissionDto.setPermissionID(rolePermission.getPermissionID());
            rolePermissionDto.setRoleName(rolePermission.getRole().getName());
            rolePermissionDto.setPermissionName(rolePermission.getPermission().getName());
            rolePermissionDto.setServiceType(rolePermission.getPermission().getServiceTypeID());
            rolePermissionDto.setParentID(rolePermission.getPermission().getParentID());
            rolePermissionDto.setPLevel(rolePermission.getPermission().getPLevel());
            
            rolePermissionDtoList.add(rolePermissionDto);
        });
        rolePermissionDisplayDto.setRolePermissionList(rolePermissionDtoList);
        if(rolePermissionDtoList.isEmpty()) {
            return rolePermissionDisplayDto;
        }
        
        rolePermissionDisplayDto.setPermissionDevTreeList(getRolePermissionIvhTree(rolePermissionDtoList));
        return rolePermissionDisplayDto;
    }
    
    
    
    
    /* (non-Javadoc)
     * @see pwc.taxtech.atms.service.PermissionService#getIvhTreePermissionsByRoleIDList(java.util.List, java.lang.String)
     */
    @SuppressWarnings("rawtypes")
    @Override
    public List<DevTreeDto> getIvhTreePermissionsByRoleIDList(List<String> roleList, String serviceType) {

        if(roleList == null || roleList.isEmpty()   ) {
            return new ArrayList<DevTreeDto>();
        }
        List<RolePermission> rolePermissionList = rolePermissionMapper.selectByRoleListAndServiceTypeWithAssociation(roleList, serviceType);
        if(rolePermissionList == null || rolePermissionList.isEmpty()   ) {
            return new ArrayList<DevTreeDto>();
        }
        List<RolePermissionDto> rolePermissionDtoListTemp = new ArrayList<>();
        rolePermissionList.stream().forEach(rolePermission -> {
            RolePermissionDto rolePermissionDto = new RolePermissionDto();
            rolePermissionDto.setPermissionID(rolePermission.getPermissionID());
            rolePermissionDto.setCode(rolePermission.getPermission().getCode());
            rolePermissionDto.setPermissionName(rolePermission.getPermission().getName());
            rolePermissionDto.setServiceType(rolePermission.getPermission().getServiceTypeID());
            rolePermissionDto.setParentID(rolePermission.getPermission().getParentID());
            
            rolePermissionDtoListTemp.add(rolePermissionDto);
        });
        List<RolePermissionDto>  rolePermissionDtoList = rolePermissionDtoListTemp.stream().distinct().collect(Collectors.toList());
        
        return getRolePermissionIvhTree(rolePermissionDtoList);
    }

    /* (non-Javadoc)
     * @see pwc.taxtech.atms.service.PermissionService#getAllPermissions(java.lang.String)
     */
    @SuppressWarnings("rawtypes")
    @Override
    public List<DevTreeDto> getAllPermissions(String serviceType) {

        PermissionExample permissionExample = new PermissionExample();
        permissionExample.createCriteria()
            .andIsActiveEqualTo(CommonConstants.ACTIVE_STATUS)
            .andServiceTypeIDEqualTo(serviceType);
        permissionExample.setOrderByClause("Code ASC");
        List<Permission> allPermissions = permissionMapper.selectByExample(permissionExample);
        
        List<Permission> topList = allPermissions.stream()
                .filter(permission -> StringUtils.isEmpty(permission.getParentID()))
                .collect(Collectors.toList());
        List<DevTreeDto> permissionDevTreeList = new ArrayList<>();
        for (Permission topPermission : topList) {
            DevTreeDto devTreeNode = new DevTreeDto<>();
            devTreeNode.setId(topPermission.getID());
            devTreeNode.setText(topPermission.getName());
            devTreeNode.setCode(topPermission.getCode());
            devTreeNode.setParentId(null);
            setSubPermissionListRecursive(devTreeNode, allPermissions);
            permissionDevTreeList.add(devTreeNode);
        }
        return permissionDevTreeList;
    }

    @SuppressWarnings("rawtypes")
    private List<DevTreeDto> getRolePermissionIvhTree(List<RolePermissionDto> rolePermissionDtoList) {
        
        //copy 
        List<RolePermissionDto> outputRolePermissionDtoList = new ArrayList<>();
        outputRolePermissionDtoList.addAll(rolePermissionDtoList);
        
        //get all permission list
        List<Permission> allPermissionList = permissionMapper.selectByExample(null);
        if(outputRolePermissionDtoList.size() != allPermissionList.size()) {
            Collections.copy(outputRolePermissionDtoList, rolePermissionDtoList);
            for(RolePermissionDto rolePermissionDto: rolePermissionDtoList) {
                addParentPermissionToListRecursive(rolePermissionDto, allPermissionList, outputRolePermissionDtoList);
            }
        }
        else {
            outputRolePermissionDtoList = new ArrayList<>();
            for(Permission permission: allPermissionList) {
                RolePermissionDto rolePermissionDto = new RolePermissionDto();
                CommonUtils.copyProperties(permission, rolePermissionDto);
                rolePermissionDto.setID(null);
                rolePermissionDto.setPermissionID(permission.getID());
                rolePermissionDto.setPermissionName(permission.getName());
                rolePermissionDto.setServiceType(permission.getServiceTypeID());
                outputRolePermissionDtoList.add(rolePermissionDto);
            }
        }
        
        //根据resultList中情况生成ivhtree
        List<RolePermissionDto> topList = outputRolePermissionDtoList.stream()
                .filter(rPermissionDto -> StringUtils.isEmpty(rPermissionDto.getParentID()))
                .sorted(Comparator.comparing(RolePermissionDto :: getCode))
                .collect(Collectors.toList());
        List<DevTreeDto> permissionDevTreeList = new ArrayList<>();
        for (RolePermissionDto topRolePermissionDto : topList) {
            DevTreeDto devTreeNode = new DevTreeDto<>();
            devTreeNode.setId(topRolePermissionDto.getPermissionID());
            devTreeNode.setText(topRolePermissionDto.getPermissionName());
            devTreeNode.setCode(topRolePermissionDto.getCode());
            devTreeNode.setParentId(null);
            setSubRolePermissionListRecursive(devTreeNode, outputRolePermissionDtoList);
            permissionDevTreeList.add(devTreeNode);
        }
        return permissionDevTreeList;
    }

    private void addParentPermissionToListRecursive(RolePermissionDto rolePermissionDto, 
            List<Permission> allPermissionList, List<RolePermissionDto> outputRolePermissionDtoList) {
        
        Optional<Permission> optional = allPermissionList.stream().
                filter(p -> p.getID().equals(rolePermissionDto.getParentID())).findFirst();
        if(optional.isPresent()) {
            Permission parentPermission = optional.get();
            RolePermissionDto parentRolePermissionDto = new RolePermissionDto();
            CommonUtils.copyProperties(parentPermission, parentRolePermissionDto);
            parentRolePermissionDto.setID(null);
            parentRolePermissionDto.setPermissionID(parentPermission.getID());
            parentRolePermissionDto.setPermissionName(parentPermission.getName());
            parentRolePermissionDto.setServiceType(parentPermission.getServiceTypeID());
            
            if(outputRolePermissionDtoList.stream()
                    .filter(rpDto -> rpDto.getPermissionID().equals(parentPermission.getID())).count()==0) {
                outputRolePermissionDtoList.add(parentRolePermissionDto);
            }
            addParentPermissionToListRecursive(parentRolePermissionDto, allPermissionList, outputRolePermissionDtoList);
        }
    }
    
    @SuppressWarnings({ "rawtypes", "unchecked" })
    private void setSubRolePermissionListRecursive(DevTreeDto currentNode, List<RolePermissionDto> allRolePermissionDtoList) {
        
        List<RolePermissionDto> subRolePermissionDtoList = allRolePermissionDtoList.stream()
                .filter(rPermissionDto -> currentNode.getId().equals(rPermissionDto.getParentID())).collect(Collectors.toList());
        if(subRolePermissionDtoList!=null && !subRolePermissionDtoList.isEmpty()) {
            currentNode.setExpanded(true);
            currentNode.setItems(new ArrayList());
            for(RolePermissionDto rolePermissionDto : subRolePermissionDtoList) {
                DevTreeDto subNode = new DevTreeDto();
                subNode.setId(rolePermissionDto.getPermissionID());
                subNode.setText(rolePermissionDto.getPermissionName());
                subNode.setCode(rolePermissionDto.getCode());
                subNode.setParentId(rolePermissionDto.getParentID());
                currentNode.getItems().add(subNode);
                setSubRolePermissionListRecursive(subNode, allRolePermissionDtoList);
            }
        }
        else {
            currentNode.setExpanded(false);
            currentNode.setItems(new ArrayList());
        }
    }
    
    @SuppressWarnings({ "rawtypes", "unchecked" })
    private void setSubPermissionListRecursive(DevTreeDto currentNode, List<Permission> allPermissionList) {
        
        List<Permission> subPermissionList = allPermissionList.stream()
                .filter(permission -> currentNode.getId().equals(permission.getParentID())).collect(Collectors.toList());
        if(subPermissionList!=null && !subPermissionList.isEmpty()) {
            currentNode.setExpanded(true);
            currentNode.setItems(new ArrayList());
            for(Permission permission : subPermissionList) {
                DevTreeDto subNode = new DevTreeDto();
                subNode.setId(permission.getID());
                subNode.setText(permission.getName());
                subNode.setCode(permission.getCode());
                subNode.setParentId(permission.getParentID());
                subNode.setRelyOnCodes(permission.getRelyOnCodes());
                currentNode.getItems().add(subNode);
                setSubPermissionListRecursive(subNode, allPermissionList);
            }
        }
        else {
            currentNode.setExpanded(false);
            currentNode.setItems(new ArrayList());
        }
    }
}