package com.jilongda.manage.authority.service.impl;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.google.common.collect.Lists;
import com.jilongda.manage.authority.mapper.SecResourcesMapper;
import com.jilongda.manage.authority.model.SecResources;
import com.jilongda.manage.authority.service.SecResourcesService;
import com.jilongda.common.basic.ApiResult;
import com.jilongda.common.utils.SpringUtils;
import com.jilongda.manage.authority.dto.Btn;
import com.jilongda.manage.authority.dto.SysSourcesDTO;
import com.jilongda.manage.authority.dto.SysSourcesListDTO;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.jilongda.manage.authority.vo.SecResourceVO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.*;

import static java.util.Comparator.comparing;
import static java.util.stream.Collectors.toList;

/**
 * <p>
 * 资源表 服务实现类
 * </p>
 *
 * @author xiaochen
 * @since 2022-06-08
 */
@Service
public class SecResourcesServiceImpl extends ServiceImpl<SecResourcesMapper, SecResources> implements SecResourcesService {

    @Autowired
    private SecResourcesMapper secResourcesMapper;

    @Override
    public SecResourcesMapper getSecResourcesMapper() {
        return secResourcesMapper;
    }

    @Override
    public ApiResult<String> importAllSources(SysSourcesListDTO dto) {
        try {
            secResourcesMapper.delete(Wrappers.lambdaQuery(SecResources.class).eq(SecResources::getIsDelete, false));

            List<SysSourcesDTO> menus = dto.getMenus();
            for (SysSourcesDTO sources : menus) {
                importMenu(sources, 0L);
            }
            return ApiResult.success();
        } catch (Exception e) {
            return ApiResult.failed(e.getMessage());
        }

    }

    @Override
    public List<SecResourceVO> selectResources() {
        List<SecResources> allResources = secResourcesMapper.selectList(Wrappers.lambdaQuery(SecResources.class)
                .ne(SecResources::getHidden,1));
        SecResourceVO item;
        //获取根节点的集合
        List<SecResourceVO> root = new ArrayList<>();
        Map<Long, SecResourceVO> tempMap = new HashMap<>(3);
        //子节点集合
        SecResourceVO parent;
        for (SecResources resources : allResources) {
            item = SpringUtils.beanCopy(resources, SecResourceVO.class);
            //每次循环都将集合重新循环添加到父集合中
            tempMap.put(resources.getId(), item);
            parent = tempMap.get(resources.getParentId());
            if (Objects.nonNull(parent)) {
                //父集合不为空时，获取下级集合
                //子节点集合
                parent.getChildren().add(item);
                // 对子集排序,升序
                parent.getChildren().sort(comparing(SecResourceVO::getSort));
            } else {
                //根集合
                root.add(item);
                // 对根集合排序,升序
                root.sort(comparing(SecResourceVO::getSort));
            }
        }
        return root;
    }

    @Override
    public List<SecResourceVO> selectResourcesRecursion() {
        List<SecResourceVO> resourceVOS = secResourcesMapper.allList();
        // 判空
        if (CollectionUtils.isEmpty(resourceVOS))
            return Lists.newArrayList();

        // 找出所有父级
        List<SecResourceVO> parent = resourceVOS.stream().filter(resource -> resource.getParentId()==0).collect(toList());
        // 通过父级递归子集
        getChildren(resourceVOS,parent);

        return parent;
    }

    private void getChildren(List<SecResourceVO> resourceVOS, List<SecResourceVO> parentList) {
        parentList.forEach(parent -> {
            List<SecResourceVO> childrenList = resourceVOS.stream().filter(region -> region.getParentId().equals(parent.getId())).collect(toList());
            parent.setChildren(childrenList);
            if (!CollectionUtils.isEmpty(childrenList))
                getChildren(resourceVOS,childrenList);
        });
    }

    private void importMenu(SysSourcesDTO sources, Long parentId) {
        //装填本体
        SecResources resources = new SecResources();
        if ("0".equals(parentId)) {
            resources.setType(1);
        } else {
            resources.setType(2);
        }
        if (Objects.nonNull(sources.getMeta())) {
            resources.setMenu(sources.getMeta().getParent());
            resources.setTitle(sources.getMeta().getTitle());
        }
        resources.setParentId(parentId);
        resources.setIcon(sources.getIcon());
        resources.setPath(sources.getPath());
        resources.setName(sources.getName());
        resources.setComponent(sources.getComponent());
        if (Objects.nonNull(sources.getHidden())) {
            resources.setHidden(sources.getHidden());
        }
        //存储本体
        secResourcesMapper.insert(resources);
        //判断是否有btn按钮,有则存储
        if (!CollectionUtils.isEmpty(sources.getBtn())) {
            setButtons(sources.getBtn(), resources.getId());
        }
        //判断是否存在子菜单，存在则递归存储
        if (!CollectionUtils.isEmpty(sources.getChildren())) {
            for (SysSourcesDTO nextSources : sources.getChildren()) {
                importMenu(nextSources, resources.getId());
            }
        }
    }

    private void setButtons(List<Btn> buttons, Long parentId) {
        for (Btn button : buttons) {
            SecResources btn = new SecResources();
            btn.setType(3);
            btn.setTitle(button.getTitle());
            btn.setName(button.getName());
            btn.setParentId(parentId);
            secResourcesMapper.insert(btn);
        }
    }

}