package org.jeecg.modules.system.service.impl;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import java.util.stream.Collectors;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.apache.commons.lang3.StringUtils;
import org.jeecg.common.constant.CommonConstant;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.modules.system.entity.SysDepart;
import org.jeecg.modules.system.entity.SysUser;
import org.jeecg.modules.system.entity.SysUserDepart;
import org.jeecg.modules.system.mapper.SysUserDepartMapper;
import org.jeecg.modules.system.model.DepartIdModel;
import org.jeecg.modules.system.model.DepartUserModel;
import org.jeecg.modules.system.model.SysDepartTreeModel;
import org.jeecg.modules.system.service.ISysDepartService;
import org.jeecg.modules.system.service.ISysUserDepartService;
import org.jeecg.modules.system.service.ISysUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

/**
 * <p>
 * 用户部门表实现类
 * <p/>
 *
 * @Author ZhiLin
 * @since 2019-02-22
 */
@Service
public class SysUserDepartServiceImpl extends ServiceImpl<SysUserDepartMapper, SysUserDepart> implements ISysUserDepartService {
    @Autowired
    private ISysDepartService sysDepartService;
    @Autowired
    private ISysUserService sysUserService;

    @Autowired
    private SysUserDepartMapper userDepartMapper;


    /**
     * 根据用户id查询部门信息
     */
    @Override
    public List<DepartIdModel> queryDepartIdsOfUser(String userId) {
        LambdaQueryWrapper<SysUserDepart> queryUDep = new LambdaQueryWrapper<SysUserDepart>();
        LambdaQueryWrapper<SysDepart> queryDep = new LambdaQueryWrapper<SysDepart>();
        try {
            queryUDep.eq(SysUserDepart::getUserId, userId);
            List<String> depIdList = new ArrayList<>();
            List<DepartIdModel> depIdModelList = new ArrayList<>();
            List<SysUserDepart> userDepList = this.list(queryUDep);
            if (userDepList != null && userDepList.size() > 0) {
                for (SysUserDepart userDepart : userDepList) {
                    depIdList.add(userDepart.getDepId());
                }
                queryDep.in(SysDepart::getId, depIdList);
                List<SysDepart> depList = sysDepartService.list(queryDep);
                if (depList != null || depList.size() > 0) {
                    for (SysDepart depart : depList) {
                        depIdModelList.add(new DepartIdModel().convertByUserDepart(depart));
                    }
                }
                return depIdModelList;
            }
        } catch (Exception e) {
            e.fillInStackTrace();
        }
        return null;


    }


    /**
     * 根据部门id查询用户信息
     */
    @Override
    public List<SysUser> queryUserByDepId(String depId) {
        LambdaQueryWrapper<SysUserDepart> queryUDep = new LambdaQueryWrapper<SysUserDepart>();
        queryUDep.eq(SysUserDepart::getDepId, depId);
        List<String> userIdList = new ArrayList<>();
        List<SysUserDepart> uDepList = this.list(queryUDep);
        if (uDepList != null && uDepList.size() > 0) {
            for (SysUserDepart uDep : uDepList) {
                userIdList.add(uDep.getUserId());
            }
            List<SysUser> userList = (List<SysUser>) sysUserService.listByIds(userIdList);
            //update-begin-author:taoyan date:201905047 for:接口调用查询返回结果不能返回密码相关信息
            for (SysUser sysUser : userList) {
                sysUser.setSalt("");
                sysUser.setPassword("");
            }
            //update-end-author:taoyan date:201905047 for:接口调用查询返回结果不能返回密码相关信息
            return userList;
        }
        return new ArrayList<SysUser>();
    }

    /**
     * 根据部门code，查询当前部门和下级部门的 用户信息
     */
    @Override
    public List<SysUser> queryUserByDepCode(String depCode, String realname) {
        //update-begin-author:taoyan date:20210422 for: 根据部门选择用户接口代码优化
        if (oConvertUtils.isNotEmpty(realname)) {
            realname = realname.trim();
        }
        List<SysUser> userList = this.baseMapper.queryDepartUserList(depCode, realname);
        Map<String, SysUser> map = new HashMap<String, SysUser>();
        for (SysUser sysUser : userList) {
            // 返回的用户数据去掉密码信息
            sysUser.setSalt("");
            sysUser.setPassword("");
            map.put(sysUser.getId(), sysUser);
        }
        return new ArrayList<SysUser>(map.values());
        //update-end-author:taoyan date:20210422 for: 根据部门选择用户接口代码优化

    }

    @Override
    public IPage<SysUser> queryDepartUserPageList(String departId, String username, String realname, int pageSize, int pageNo) {
        IPage<SysUser> pageList = null;
        // 部门ID不存在 直接查询用户表即可
        Page<SysUser> page = new Page<SysUser>(pageNo, pageSize);
        if (oConvertUtils.isEmpty(departId)) {
            LambdaQueryWrapper<SysUser> query = new LambdaQueryWrapper<>();
            if (oConvertUtils.isNotEmpty(username)) {
                query.like(SysUser::getUsername, username);
            }
            if (oConvertUtils.isNotEmpty(realname)) {
                query.like(SysUser::getRealname, realname);
            }
            pageList = sysUserService.page(page, query);
        } else {
            // 有部门ID 需要走自定义sql
            SysDepart sysDepart = sysDepartService.getById(departId);
            pageList = this.baseMapper.queryDepartUserPageList(page, sysDepart.getOrgCode(), username, realname);
        }
        List<SysUser> userList = pageList.getRecords();
        if (userList != null && userList.size() > 0) {
            List<String> userIds = userList.stream().map(SysUser::getId).collect(Collectors.toList());
            Map<String, SysUser> map = new HashMap<String, SysUser>();
            if (userIds != null && userIds.size() > 0) {
                // 查部门名称
                Map<String, String> useDepNames = sysUserService.getDepNamesByUserIds(userIds);
                userList.forEach(item -> {
                    //TODO 临时借用这个字段用于页面展示
                    item.setOrgCodeTxt(useDepNames.get(item.getId()));
                    item.setSalt("");
                    item.setPassword("");
                    // 去重
                    map.put(item.getId(), item);
                });
            }
            pageList.setRecords(new ArrayList<SysUser>(map.values()));
        }
        return pageList;
    }

    @Override
    public List<DepartUserModel> queryDepUserList(String departId, String username, String realname) {
        List<DepartUserModel> hashMaps = userDepartMapper.queryDepUserList(username,realname);

        hashMaps.forEach(item->{
            Map<String,Object> slotMap = new HashMap<>();
            slotMap.put("icon","user");
            item.setLeaf(true);
            item.setTitle(item.getRealname());
            //key 使用id+pid 避免不同部门有同一用户 TODO
            item.setKey(item.getId()+item.getDepartId());
            item.setValue(item.getId());
            item.setParentId(item.getDepartId());
            item.setType(DepartUserModel.TYPE_USER);
            item.setScopedSlots(slotMap);
        });
        Map<String, List<DepartUserModel>> userMap = hashMaps.stream().collect(Collectors.groupingBy(t -> t.getParentId()));


        //查询所有部门
        LambdaQueryWrapper<SysDepart> query = new LambdaQueryWrapper<SysDepart>();
        query.eq(SysDepart::getDelFlag, CommonConstant.DEL_FLAG_0.toString());
        query.orderByAsc(SysDepart::getDepartOrder);
        List<SysDepart> departList = sysDepartService.list(query);

        List<DepartUserModel> modelList = new ArrayList<>();
        for (int i = 0; i < departList.size(); i++) {
            SysDepart depart = departList.get(i);
            DepartUserModel model = new DepartUserModel(depart);
            Map<String,Object> slotMap = new HashMap<>();
            slotMap.put("icon","depart");
            model.setScopedSlots(slotMap);
            //部门
            model.setType(DepartUserModel.TYPE_DEPART);
            modelList.add(model);

        }
        //如果是精确查询，只显示查询用户结果 TODO
        if(!StringUtils.isEmpty(realname) || !StringUtils.isEmpty(username)){
            return hashMaps;
        }


        // 合并查询的userList userMap
        List<DepartUserModel> res = buildDeptTreeByStream(modelList,userMap);

        return res;
    }


    /**
     *
     * @param trees     部门信息
     * @param userMap   用户信息
     * @return
     */
    public static List<DepartUserModel> buildDeptTreeByStream(List<DepartUserModel> trees, Map<String, List<DepartUserModel>> userMap){
        //获取parentId = 0的根节点
        List<DepartUserModel> list = trees.stream().filter(item -> item.getParentId().isEmpty()).collect(Collectors.toList());
        //根据parentId进行分组
        Map<String, List<DepartUserModel>> map = trees.stream().collect(Collectors.groupingBy(DepartUserModel::getParentId));


        //合并map
        for (Map.Entry<String, List<DepartUserModel>> entry : map.entrySet()) {
            if (userMap.containsKey(entry.getKey())) {
                userMap.get(entry.getKey()).addAll(entry.getValue());
                userMap.put(entry.getKey(),  userMap.get(entry.getKey()) );
            } else {
                userMap.put(entry.getKey(), entry.getValue());
            }
        }
        recursionFnTree(list, userMap);
        //recursionFnTree(list, map);
        return list;
    }

    /**
     * 递归遍历节点
     * @param list  当前父节点
     * @param map   所有子节点
     */
    public static void recursionFnTree(List<DepartUserModel> list, Map<String, List<DepartUserModel>> map){
        for (DepartUserModel treeSelect : list) {
            List<DepartUserModel> childList = map.get(treeSelect.getId());
            treeSelect.setChildren(childList);
            if (null != childList && 0 < childList.size()){
                recursionFnTree(childList,map);
            }
        }
    }





}