package org.jeecg.modules.system.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.swagger.annotations.Api;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.system.vo.DictModel;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.modules.system.entity.SysPosition;
import org.jeecg.modules.system.entity.SysUser;
import org.jeecg.modules.system.model.DepartOrgVo;
import org.jeecg.modules.system.model.SysDepartTreeModel;
import org.jeecg.modules.system.service.ISysDepartService;
import org.jeecg.modules.system.service.ISysPositionService;
import org.jeecg.modules.system.service.ISysUserDepartService;
import org.jeecg.modules.system.service.ISysUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

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

@Api(tags = "部门组织架构")
@RestController
@RequestMapping("/sys/org")
@Slf4j
public class SysDepartOrgController {
    @Autowired
    private ISysDepartService sysDepartService;
    @Autowired
    public RedisTemplate<String, Object> redisTemplate;
    @Autowired
    private ISysUserService sysUserService;

    @Autowired
    private ISysUserDepartService sysUserDepartService;
    @Autowired
    private ISysPositionService positionService;

    @RequestMapping(value = "/depart", method = RequestMethod.GET)
    public Result<List<DepartOrgVo>> queryTreeList(@RequestParam(name = "ids", required = false) String ids) {
        Result<List<DepartOrgVo>> result = new Result<>();
        try {
            //1.查询所有部门
            List<SysDepartTreeModel> departList = new ArrayList<>();
            if (oConvertUtils.isNotEmpty(ids)) {
                departList = sysDepartService.queryTreeList(ids);
            } else {
                departList = sysDepartService.queryTreeList();
                //List<SysUser> sysUsers = sysUserDepartService.queryUserByDepId(id);
            }
            List<DepartOrgVo> departOrgVoList = new ArrayList<>();
            coverDepartData(departList,null,departOrgVoList);
            result.setResult(departOrgVoList);

            //2.查询部门中的所有人
//            departList.forEach(depart->{
//
//            });

        } catch (Exception e) {
            log.error(e.getMessage(), e);
        }
        return result;
    }

    /**
     * 转换数据
     * @param departList
     * @param parent
     * @param departOrgVoList
     */
    private void coverDepartData(List<SysDepartTreeModel>  departList, DepartOrgVo parent, List<DepartOrgVo> departOrgVoList){
        for (int i = 0; i <departList.size() ; i++) {
            SysDepartTreeModel model = departList.get(i);
            DepartOrgVo vo = new DepartOrgVo();
            vo.setId(model.getId());
            vo.setPid(model.getParentId());
            vo.setLabel(model.getDepartName());
            vo.setLeaf(model.getIsLeaf());
            vo.setSort(model.getDepartOrder());
            //转换数据时查询用户数据
            fitUserData(vo);

            if(parent!=null){
                if(parent.getChildren()!=null){
                    parent.getChildren().add(vo);
                }else {
                    List<DepartOrgVo> children = new ArrayList<>();
                    children.add(vo);
                    parent.setChildren(children);
                }
            }else{
                departOrgVoList.add(vo);
            }

            List<SysDepartTreeModel> children = model.getChildren();
            if(children!=null){
                coverDepartData(children,vo,departOrgVoList);
            }
        }

    }

    /**
     * 根据部门统计用户数据
     */
    private void  fitUserData( DepartOrgVo model){
        List<DepartOrgVo.PostVo> postVoList  = new ArrayList<>();
        Map<String, List<SysUser>> map = new HashMap<>();
        //1.查询部门用户
        List<SysUser> sysUsers = sysUserDepartService.queryUserByDepId(model.getId());
        //2.部门设置用户
        model.setUserList(sysUsers);
        //3.查询部门及下级部门所有用户
        Page<SysUser> page = new Page<SysUser>(1, 1000);
        List<String> subDepids = sysDepartService.getMySubDepIdsByDepId(model.getId());
        IPage<SysUser> pageList = sysUserService.getUserByDepIds(page, subDepids, null);
        model.setSubUserList(pageList.getRecords());


        //3.根据用户职务统计
        sysUsers.forEach(item->{
            if(StringUtils.isEmpty(item.getPost())){
                if(map.containsKey("未分配")){
                    map.get("未分配").add(item);
                }else {
                    List<SysUser> userList = new ArrayList<>();
                    userList.add(item);
                    map.put("未分配",userList);
                }
            }else {
                if(map.containsKey(item.getPost())){
                    map.get(item.getPost()).add(item);
                }else {
                    List<SysUser> userList = new ArrayList<>();
                    userList.add(item);
                    map.put(item.getPost(),userList);
                }
            }
        });
        //查询所有职务
        List<SysPosition> list = positionService.list();
        Map<String, Double> positionMap = list.stream().filter(item->StringUtils.isNotEmpty(item.getName())&&item.getSort()!=null).collect(Collectors.toMap(SysPosition::getName, SysPosition::getSort));

        //4.设置部门下职务
        map.forEach((key,value)->{
            DepartOrgVo.PostVo postVo =  new DepartOrgVo.PostVo();
            postVo.setPostName(key);
            postVo.setUserList(value);
            if(positionMap.containsKey(key)){
                postVo.setSort(positionMap.get(key));
            }else {
                postVo.setSort(9999.0);
                if(key.equals("未分配")){
                    postVo.setSort(99999.0);
                }
            }

            postVoList.add(postVo);

        });
        Collections.sort(postVoList,(o1,o2)->{
            if(o1.getSort()==null || o2.getSort() ==null){
                return  0;
            }
             return  o1.getSort().compareTo(o2.getSort());
        });
        model.setPostList(postVoList);

    }

}