package org.jeecg.modules.dnc.service.support;

import org.jeecg.modules.dnc.dto.ComponentExt;
import org.jeecg.modules.dnc.entity.ComponentInfo;
import org.jeecg.modules.dnc.response.CommonGenericTree;
import org.springframework.beans.BeanUtils;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class ComponentTreeWrapper {
    public static List<CommonGenericTree> loadTree(List<ComponentExt> componentInfoList) {
        Map<String, CommonGenericTree> treeMap = new HashMap<>();
        for(ComponentExt ext : componentInfoList) {
            treeMap = getBaseNodeMap(ext, treeMap);
        }
        return CommonGenericTree.convertMapToList(treeMap);
    }

    private static CommonGenericTree loadChildrenNodes(ComponentExt info, CommonGenericTree node) {
        if(info.hasChild()) {
            List<ComponentExt> children = info.getChildren();
            CommonGenericTree<ComponentInfo> childNode;
            ComponentInfo componentInfo;
            for (ComponentExt ext : children) {
                childNode = new CommonGenericTree();
                childNode.setId(ext.getComponentId());
                childNode.setLabel(ext.getComponentName());
                childNode.setParentId(info.getComponentId());
                //设置菜单节点图标 Start
                String iconStr = "";
                /*判断是否设置了菜单导航图标样式*/
                if (!ext.hasChild()) {//导航页面的菜单节点
                    iconStr = "";//默认图标
                }
                childNode.setIconClass(iconStr);
                childNode.setType(2);
                componentInfo = new ComponentInfo();
                BeanUtils.copyProperties(ext, componentInfo);
                childNode.setEntity(componentInfo);
                node.setRField(ext.getProductId());
                childNode = loadChildrenNodes(ext, childNode);
                node.addChildren(childNode);
            }
        }
        return node;
    }

    private static Map<String,CommonGenericTree> getBaseNodeMap(ComponentExt ext, Map<String,CommonGenericTree> allNodeMap) {
        CommonGenericTree<ComponentInfo> node;
        ComponentInfo componentInfo;
        if(ext.getRankLevel() == 1) {
            node = new CommonGenericTree<ComponentInfo>();
            node.setId(ext.getComponentId());
            node.setLabel("[" + ext.getComponentCode()+ "]" + ext.getComponentName());
            node.setParentId(null);
            //设置菜单节点图标 Start
            String iconStr = "";
            /*判断是否设置了菜单导航图标样式*/
            if (!ext.hasChild()) {//导航页面的菜单节点
                iconStr = "";//默认图标
            }
            node.setIconClass(iconStr);
            node.setType(2);
            node.setRField(ext.getProductId());
            componentInfo = new ComponentInfo();
            BeanUtils.copyProperties(ext, componentInfo);
            node.setEntity(componentInfo);
            allNodeMap.put(node.getId(), node);
        }else {
            node = new CommonGenericTree<ComponentInfo>();
            node.setId(ext.getComponentId());
            node.setLabel("[" + ext.getComponentCode()+ "]" + ext.getComponentName());
            node.setParentId(ext.getParentId());
            //设置菜单节点图标 Start
            String iconStr = "";
            /*判断是否设置了菜单导航图标样式*/
            if (!ext.hasChild()) {//导航页面的菜单节点
                iconStr = "";//默认图标
            }
            node.setIconClass(iconStr);
            node.setType(2);
            node.setRField(ext.getProductId());
            componentInfo = new ComponentInfo();
            BeanUtils.copyProperties(ext, componentInfo);
            node.setEntity(componentInfo);
            allNodeMap.put(node.getId(), node);
        }
        return allNodeMap;
    }

}