package com.tievd.cube.modules.system.service.impl;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.tievd.cube.commons.constant.CommonConst;
import com.tievd.cube.commons.exception.CubeAppException;
import com.tievd.cube.modules.system.entity.*;
import com.tievd.cube.modules.system.mapper.*;
import com.tievd.cube.modules.system.model.TreeModel;
import com.tievd.cube.modules.system.model.api.response.LoadDataRuleResponse;
import com.tievd.cube.modules.system.model.api.response.TreeListForDeptRoleResponse;
import com.tievd.cube.modules.system.service.ISysDepartPermissionService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;
import java.util.stream.Collectors;

@Service
public class SysDepartPermissionServiceImpl extends ServiceImpl<SysDepartPermissionMapper, SysDepartPermission> implements ISysDepartPermissionService {

    @Autowired
    private SysPermissionDataRuleMapper ruleMapper;
    @Autowired
    private SysDepartRoleMapper sysDepartRoleMapper;
    @Autowired
    private SysDepartRolePermissionMapper departRolePermissionMapper;
    @Autowired
    private SysPermissionDataRuleMapper sysPermissionDataRuleMapper;
    @Autowired
    private SysPermissionMapper sysPermissionMapper;


    @Override
    @Transactional(rollbackFor = Exception.class)
    public void saveDepartPermission(String departId, String permissionIds, String lastPermissionIds) {
        List<String> add = getDiff(lastPermissionIds, permissionIds);
        if (add != null && add.size() > 0) {
            List<SysDepartPermission> list = new ArrayList<>();
            for (String p : add) {
                if (StrUtil.isNotEmpty(p)) {
                    SysDepartPermission rolepms = new SysDepartPermission(departId, p);
                    list.add(rolepms);
                }
            }
            this.saveBatch(list);
        }
        List<String> delete = getDiff(permissionIds, lastPermissionIds);
        if (delete != null && delete.size() > 0) {
            for (String permissionId : delete) {
                this.remove(new QueryWrapper<SysDepartPermission>().lambda().eq(SysDepartPermission::getDepartId, departId).eq(SysDepartPermission::getPermissionId, permissionId));
                //删除部门权限时，删除部门角色中已授权的权限
                List<SysDepartRole> sysDepartRoleList = sysDepartRoleMapper.selectList(new LambdaQueryWrapper<SysDepartRole>().eq(SysDepartRole::getDepartId, departId));
                List<String> roleIds = sysDepartRoleList.stream().map(SysDepartRole::getId).collect(Collectors.toList());
                if (roleIds.size() > 0) {
                    departRolePermissionMapper.delete(new LambdaQueryWrapper<SysDepartRolePermission>().eq(SysDepartRolePermission::getPermissionId, permissionId));
                }
            }
        }
    }

    @Override
    public List<SysPermissionDataRule> getPermRuleListByDeptIdAndPermId(String departId, String permissionId) {
        SysDepartPermission departPermission = this.getOne(new QueryWrapper<SysDepartPermission>().lambda().eq(SysDepartPermission::getDepartId, departId).eq(SysDepartPermission::getPermissionId, permissionId));
        if (departPermission != null) {
            LambdaQueryWrapper<SysPermissionDataRule> query = new LambdaQueryWrapper<>();
            query.in(SysPermissionDataRule::getId, Arrays.asList(departPermission.getDataRuleIds().split(",")));
            query.orderByDesc(SysPermissionDataRule::getCreateTime);
            return this.ruleMapper.selectList(query);
        } else {
            return null;
        }
    }

    @Override
    public LoadDataRuleResponse loadDataRule(String permissionId, String departId) {
        LambdaQueryWrapper<SysPermissionDataRule> query = new LambdaQueryWrapper<>();
        query.eq(SysPermissionDataRule::getPermissionId, permissionId);
        query.orderByDesc(SysPermissionDataRule::getCreateTime);
        List<SysPermissionDataRule> dataRules = sysPermissionDataRuleMapper.selectList(query);
        if (dataRules == null || dataRules.size() == 0) {
            throw new CubeAppException("未找到菜单数据权限信息！");
        }
        LoadDataRuleResponse response = new LoadDataRuleResponse();
        LambdaQueryWrapper<SysDepartPermission> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(SysDepartPermission::getPermissionId, permissionId);
        wrapper.eq(SysDepartPermission::getDepartId, departId);
        SysDepartPermission sysDepartPermission = baseMapper.selectOne(wrapper);
        if (sysDepartPermission == null) {
            throw new CubeAppException("未找到部门菜单信息！");
        } else {
            String drChecked = sysDepartPermission.getDataRuleIds();
            if (StrUtil.isNotEmpty(drChecked)) {
                response.setDrChecked(drChecked.endsWith(",") ? drChecked.substring(0, drChecked.length() - 1) : drChecked);
            }
        }
        response.setDatarule(dataRules);
        return response;
    }

    @Override
    @Transactional
    public void saveDataRule(String permissionId, String departId, String dataRuleIds) {
        LambdaQueryWrapper<SysDepartPermission> query = new LambdaQueryWrapper<>();
        query.eq(SysDepartPermission::getPermissionId, permissionId);
        query.eq(SysDepartPermission::getDepartId, departId);
        SysDepartPermission sysDepartPermission = baseMapper.selectOne(query);
        if (sysDepartPermission == null) {
            throw new CubeAppException("请先保存部门菜单权限！");
        }
        sysDepartPermission.setDataRuleIds(dataRuleIds);
        baseMapper.updateById(sysDepartPermission);
    }

    @Override
    public TreeListForDeptRoleResponse queryTreeListForDeptRole(String departId) {
        //全部权限ids
        LambdaQueryWrapper<SysPermission> query = new LambdaQueryWrapper<>();
        query.eq(SysPermission::getDelFlag, CommonConst.NOT_DELETED);
        query.orderByAsc(SysPermission::getSortNo);
        query.inSql(SysPermission::getId, "select permission_id from sys_depart_permission where depart_id = '" + departId + "'");
        List<SysPermission> list = sysPermissionMapper.selectList(query);
        List<String> ids = new ArrayList<>();
        for (SysPermission sysPer : list) {
            ids.add(sysPer.getId());
        }
        List<TreeModel> treeList = new ArrayList<>();
        getTreeModelList(treeList, list, null);
        TreeListForDeptRoleResponse response = new TreeListForDeptRoleResponse();
        response.setTreeList(treeList);
        response.setIds(ids);
        return response;
    }

    /**
     * 从diff中找出main中没有的元素
     */
    private List<String> getDiff(String main, String diff) {
        if (StrUtil.isEmpty(diff)) {
            return null;
        }
        if (StrUtil.isEmpty(main)) {
            return Arrays.asList(diff.split(","));
        }
        String[] mainArr = main.split(",");
        String[] diffArr = diff.split(",");
        Map<String, Integer> map = new HashMap<>();
        for (String string : mainArr) {
            map.put(string, 1);
        }
        List<String> res = new ArrayList<>();
        for (String key : diffArr) {
            if (StrUtil.isNotEmpty(key) && !map.containsKey(key)) {
                res.add(key);
            }
        }
        return res;
    }

    /**
     * 递归所有部门
     *
     * @param treeList 部门数
     * @param metaList 部门信息
     * @param temp     子部门
     */
    private void getTreeModelList(List<TreeModel> treeList, List<SysPermission> metaList, TreeModel temp) {
        for (SysPermission permission : metaList) {
            String tempPid = permission.getParentId();
            TreeModel tree = new TreeModel(permission.getId(), tempPid, permission.getName(), permission.getRuleFlag(), permission.isLeaf());
            if (temp == null && StrUtil.isEmpty(tempPid)) {
                treeList.add(tree);
                if (!tree.isLeaf()) {
                    getTreeModelList(treeList, metaList, tree);
                }
            } else if (temp != null && tempPid != null && tempPid.equals(temp.getKey())) {
                temp.getChildren().add(tree);
                if (!tree.isLeaf()) {
                    getTreeModelList(treeList, metaList, tree);
                }
            }
        }
    }
}