package com.ycl.service.ding.impl;

import cn.hutool.core.collection.CollUtil;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.ycl.entity.depart.UmsDepart;
import com.ycl.entity.user.UmsAdmin;
import com.ycl.entity.user.UmsAdminRoleRelation;
import com.ycl.entity.user.UmsRole;
import com.ycl.service.depart.IDepartManagerService;
import com.ycl.service.depart.UmsDepartService;
import com.ycl.service.ding.BookRemarkService;
import com.ycl.service.ding.DingService;
import com.ycl.service.user.UmsAdminRoleRelationService;
import com.ycl.service.user.UmsAdminService;
import com.ycl.service.user.UmsRoleService;
import com.ycl.vo.AddressBookVO;
import com.ycl.vo.ContactVO;
import com.ycl.vo.NewAddressBookVO;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.Stream;

@Service
@Slf4j
public class DingServiceImpl implements DingService {

    @Autowired
    private UmsDepartService departService;
    @Autowired
    private IDepartManagerService departManagerService;

    @Autowired
    private UmsAdminService umsAdminService;

    @Autowired
    private UmsRoleService umsRoleService;

    @Autowired
    private UmsAdminRoleRelationService umsAdminRoleRelationService;

    @Autowired
    private BookRemarkService bookRemarkService;

    @Override
    public List<AddressBookVO> getAddressBook(Long userId) {
        List<UmsAdmin> users = umsAdminService.list();
        Map<Long, UmsAdmin> userMap = users.stream().collect(Collectors.toMap(UmsAdmin::getId, Function.identity()));
        List<AddressBookVO> addressBookVOS = new ArrayList<>();

        List<UmsDepart> list = departService.list(new QueryWrapper<UmsDepart>().lambda().eq(UmsDepart::getStatus, 1).eq(UmsDepart::getIsDeleted, 0));
        //查询备注信息
        Map<Long, String> remarkMap = bookRemarkService.findByUserId(userId);
        list.forEach(item -> {
            //构建部门信息
            extracted(userMap, addressBookVOS, item, remarkMap);
        });
        log.info(JSON.toJSONString(addressBookVOS));

        return buildTreeUseStream(addressBookVOS, 0);
    }

    @Override
    public List<AddressBookVO> getAddressBookByParentId(Long id, Long userId) {
        List<UmsAdmin> users = umsAdminService.list();
        Map<Long, UmsAdmin> userMap = users.stream().collect(Collectors.toMap(UmsAdmin::getId, Function.identity()));
        List<AddressBookVO> addressBookVOS = new ArrayList<>();

        List<UmsDepart> list = departService.list(new QueryWrapper<UmsDepart>().lambda().eq(UmsDepart::getStatus, 1).eq(UmsDepart::getParentId, id).eq(UmsDepart::getIsDeleted, 0));
        //查询备注信息
        Map<Long, String> remarkMap = bookRemarkService.findByUserId(userId);
        list.forEach(item -> {
            //构建部门信息
            extracted(userMap, addressBookVOS, item, remarkMap);
        });
        log.info(JSON.toJSONString(addressBookVOS));

        return addressBookVOS;
    }

    @Override
    public NewAddressBookVO getAddressBookById(Long id, Long userId) {
        List<UmsAdmin> users = umsAdminService.list();
        Map<Long, UmsAdmin> userMap = users.stream().collect(Collectors.toMap(UmsAdmin::getId, Function.identity()));
        NewAddressBookVO newAddressBookVO = new NewAddressBookVO();
        List<UmsDepart> departs = departService.list(new QueryWrapper<UmsDepart>().lambda().eq(UmsDepart::getStatus, 1).eq(UmsDepart::getParentId, id).eq(UmsDepart::getIsDeleted, 0));
        Set<Long> deptIds = departs.stream().map(UmsDepart::getParentId).collect(Collectors.toSet());
         //部门信息
        newAddressBookVO.setDeparts(departs);
        Map<Long, String> remarkMap = bookRemarkService.findByUserId(userId);
        Collection<Long> userIds = departManagerService.getByDeptId(id);
        ArrayList<ContactVO> contacts = new ArrayList<>();
        //角色信息
        List<UmsAdminRoleRelation> list = umsAdminRoleRelationService.list();
        Map<Long, List<UmsAdminRoleRelation>> rolesMap = list.stream().collect(Collectors.groupingBy(UmsAdminRoleRelation::getAdminId));
        Map<Long, String> roleName = umsRoleService.list().stream().filter(o -> o.getStatus() == 1).collect(Collectors.toMap(UmsRole::getId, UmsRole::getName));
        //构建用户信息
        if (CollUtil.isNotEmpty(userIds)) {
            userIds.forEach(u -> {
                UmsAdmin admin = userMap.get(u);
                if (Objects.nonNull(admin)) {
                    ContactVO contactVO = new ContactVO();
                    contactVO.setUserId(admin.getId());
                    contactVO.setSex(admin.getSex());
                    contactVO.setUsername(admin.getUsername());
                    contactVO.setNickName(admin.getNickName());
                    contactVO.setIcon(admin.getIcon());
                    contactVO.setMobile(admin.getMobile());
                    contactVO.setRemark(remarkMap.get(admin.getId()));
                    //添加用户角色信息
                    List<UmsAdminRoleRelation> roleRelations = rolesMap.get(admin.getId());
                    if (CollUtil.isNotEmpty(roleRelations)) {
                        ArrayList<String> roles = new ArrayList<>();
                        Set<Long> roleId = roleRelations.stream().map(UmsAdminRoleRelation::getRoleId).collect(Collectors.toSet());
                        for (Long o : roleId) {
                            String s = roleName.get(o);
                            if (StringUtils.isNotBlank(s)) {
                                roles.add(s);
                            }
                        }
                        contactVO.setRoles(roles);
                    }
                    contacts.add(contactVO);
                }
            });
            newAddressBookVO.setContacts(contacts);
        }
        return newAddressBookVO;
    }

    // 使用stream流转换
    public static List<AddressBookVO> buildTreeUseStream(List<AddressBookVO> treeList, long id) {
        List<AddressBookVO> list = treeList.stream()
                //过滤父节点与传递的id相同的TreeNode对象
                .filter(treeNode -> treeNode.getParentId() == id)
                .map(treeNode -> {
                    //递归设置孩子节点
                    treeNode.setChildren(buildTreeUseStream(treeList, treeNode.getDeptId()));
                    return treeNode;
                })
                .collect(Collectors.toList());
        return list;
    }


    private void extracted(Map<Long, UmsAdmin> userMap, List<AddressBookVO> addressBookVOS, UmsDepart depart, Map<Long, String> remarkMap) {


        AddressBookVO books = new AddressBookVO();
        books.setDeptId(depart.getId());
        books.setParentId(depart.getParentId());
        books.setDeptName(depart.getDepartName());
        books.setDeptCode(depart.getCode());
        ArrayList<ContactVO> contacts = new ArrayList<>();
        Collection<Long> userIds = departManagerService.getByDeptId(depart.getId());
        //角色信息
        List<UmsAdminRoleRelation> list = umsAdminRoleRelationService.list();
        Map<Long, List<UmsAdminRoleRelation>> rolesMap = list.stream().collect(Collectors.groupingBy(UmsAdminRoleRelation::getAdminId));
        Map<Long, String> roleName = umsRoleService.list().stream().filter(o -> o.getStatus() == 1).collect(Collectors.toMap(UmsRole::getId, UmsRole::getName));
        //构建用户信息
        if (CollUtil.isNotEmpty(userIds)) {
            userIds.forEach(u -> {
                UmsAdmin admin = userMap.get(u);
                if (Objects.nonNull(admin)) {
                    ContactVO contactVO = new ContactVO();
                    contactVO.setUserId(admin.getId());
                    contactVO.setSex(admin.getSex());
                    contactVO.setUsername(admin.getUsername());
                    contactVO.setNickName(admin.getNickName());
                    contactVO.setIcon(admin.getIcon());
                    contactVO.setMobile(admin.getMobile());
                    contactVO.setRemark(remarkMap.get(admin.getId()));
                    //添加用户角色信息
                    List<UmsAdminRoleRelation> roleRelations = rolesMap.get(admin.getId());
                    if (CollUtil.isNotEmpty(roleRelations)) {
                        ArrayList<String> roles = new ArrayList<>();
                        Set<Long> roleId = roleRelations.stream().map(UmsAdminRoleRelation::getRoleId).collect(Collectors.toSet());
                        for (Long id : roleId) {
                            String s = roleName.get(id);
                            if (StringUtils.isNotBlank(s)) {
                                roles.add(s);
                            }
                        }
                        contactVO.setRoles(roles);
                    }
                    contacts.add(contactVO);
                }
            });
            books.setContacts(contacts);
        }
        addressBookVOS.add(books);
    }
}