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

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.common.api.vo.CommonGenericTree;
import org.jeecg.modules.eam.entity.ShutdownCause;
import org.jeecg.modules.eam.entity.ShutdownCauseCategory;
import org.jeecg.modules.eam.mapper.ShutdownCauseMapper;
import org.jeecg.modules.eam.service.IShutDownCauseCategoryService;
import org.jeecg.modules.eam.service.IShutdownCauseService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;

/**
 * @Description: mom_eam_shutdown_cause
 * @Author: jeecg-boot
 * @Date: 2023-03-16
 * @Version: V1.0
 */
@Service
public class ShutdownCauseServiceImpl extends ServiceImpl<ShutdownCauseMapper, ShutdownCause> implements IShutdownCauseService {

    @Autowired
    private IShutDownCauseCategoryService shutDownCauseCategoryService;


    @Override
    public List<CommonGenericTree> loadTree() {
        // 根节点
        CommonGenericTree rootNode = initRootNode();
        // 停机原因
        List<ShutdownCauseCategory> shutdownCauseCategoryList = shutDownCauseCategoryService.list();
        List<CommonGenericTree> shutdownCauseCategoryTree = new ArrayList<CommonGenericTree>();
        for (ShutdownCauseCategory shutdownCauseCategory : shutdownCauseCategoryList) {
            shutdownCauseCategoryTree.add(shutdownCauseCategory2Tree(shutdownCauseCategory));
        }
        // 停机原因
        List<ShutdownCause> shutdownCauseList = this.list();
        List<CommonGenericTree> shutdownCauseTree = new ArrayList<CommonGenericTree>();

        shutdownCauseCategoryTree = this.list2Tree(shutdownCauseCategoryTree, shutdownCauseTree);
        for (CommonGenericTree temp : shutdownCauseCategoryTree) {
            rootNode.addChildren(temp);
        }
        // 返回的结果是list
        List<CommonGenericTree> list = new ArrayList<>();
        list.add(rootNode);
        return list;
    }

    private CommonGenericTree initRootNode() {
        CommonGenericTree node = new CommonGenericTree<>();
        node.setKey("-1");
        node.setTitle("停机原因");
        node.setType(0);
        node.setDisabled(true);
        node.setEntity(null);
        node.setValue("-1");
        return node;
    }

    private List<CommonGenericTree> list2Tree(List<CommonGenericTree> parentList, List<CommonGenericTree> childList) {
        for (CommonGenericTree tree : parentList) {
            String id = tree.getKey();
            for (CommonGenericTree cTree : childList) {
                if (id.equals(cTree.getParentId())) {
                    tree.addChildren(cTree);
                }
            }
        }
        return parentList;
    }

    private CommonGenericTree shutdownCauseCategory2Tree(ShutdownCauseCategory shutdownCauseCategory) {
        CommonGenericTree<ShutdownCauseCategory> shutdownCauseCategoryNode = new CommonGenericTree<>();
        shutdownCauseCategoryNode.setKey(shutdownCauseCategory.getId());
        shutdownCauseCategoryNode.setTitle(shutdownCauseCategory.getNum() + "/" + shutdownCauseCategory.getName());
        shutdownCauseCategoryNode.setParentId("-1");
        shutdownCauseCategoryNode.setIcon("");
        shutdownCauseCategoryNode.setType(1);
        shutdownCauseCategoryNode.setDisabled(true);
        shutdownCauseCategoryNode.setEntity(shutdownCauseCategory);
        shutdownCauseCategoryNode.setValue(shutdownCauseCategory.getId());
        return shutdownCauseCategoryNode;
    }

}