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

import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import org.jeecg.common.api.vo.CommonGenericTree;
import org.jeecg.modules.eam.entity.InspectionProjectCategory;
import org.jeecg.modules.eam.entity.TransferMethodCategory;
import org.jeecg.modules.eam.mapper.TransferMethodCategoryMapper;
import org.jeecg.modules.eam.service.ITransferMethodCategoryService;
import org.springframework.stereotype.Service;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

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

/**
 * @Description: mom_eam_transfer_method_category
 * @Author: jeecg-boot
 * @Date:   2023-03-15
 * @Version: V1.0
 */
@Service
public class TransferMethodCategoryServiceImpl extends ServiceImpl<TransferMethodCategoryMapper, TransferMethodCategory> implements ITransferMethodCategoryService {

    // 初始化工装类别
    private static final String initTransferMethodCategory = "01-增加,02-减少";

    @Override
    public List<CommonGenericTree> loadTree() {
        List<TransferMethodCategory> transferMethodCategoryList = this.lambdaQuery().eq(TransferMethodCategory::getDelFlag, "0").list();
        List<CommonGenericTree> commonGenericTrees = loadTree(transferMethodCategoryList);
        return commonGenericTrees;
    }

    @SuppressWarnings("unchecked")
    public List<CommonGenericTree> loadTree(List<TransferMethodCategory> transferMethodCategoryList) {
        @SuppressWarnings("unused")
        TransferMethodCategory transferMethodCategory = new TransferMethodCategory();
        List<CommonGenericTree> list = new ArrayList<>();
        Map<String, CommonGenericTree> map = new HashMap<>();
        CommonGenericTree<TransferMethodCategory> node = new CommonGenericTree<>();
        node.setKey("-1");
        node.setValue("-1");
        node.setTitle("增减方式分类");
        node.setRField1("");
        node.setRField2("");
        node.setDisabled(false);
        node.setEntity(new TransferMethodCategory());
        list.add(node);
        if (CollectionUtils.isNotEmpty(list)) {
            CommonGenericTree<TransferMethodCategory> pcNode;
            for (TransferMethodCategory tfc : transferMethodCategoryList) {
                if (tfc.getParentId().equals("-1")) {
                    pcNode = new CommonGenericTree<>();
                    pcNode.setKey(tfc.getId());
                    pcNode.setTitle(tfc.getNum() + "/" + tfc.getName());
                    pcNode.setParentId(node.getKey());
                    pcNode.setIcon("");
                    pcNode.setType(0);
                    pcNode.setValue(tfc.getId());
                    pcNode.setDisabled(false);
                    pcNode.setRField1(tfc.getNum());
                    pcNode.setRField2(tfc.getName());
                    pcNode.setEntity(tfc);
                    node.addChildren(pcNode);
                    map.put(tfc.getId(), pcNode);
                }
            }
            CommonGenericTree<TransferMethodCategory> childNode;
            for (TransferMethodCategory tfc : transferMethodCategoryList) {
                if (map.containsKey(tfc.getParentId())) {
                    pcNode = map.get(tfc.getParentId());
                    childNode = new CommonGenericTree<>();
                    childNode.setKey(tfc.getId());
                    childNode.setTitle(tfc.getNum() + "/" + tfc.getName());
                    childNode.setParentId(tfc.getParentId());
                    childNode.setIcon("");
                    childNode.setType(0);
                    childNode.setValue(tfc.getId());
                    childNode.setDisabled(false);
                    childNode.setRField1(tfc.getNum());
                    childNode.setRField2(tfc.getName());
                    childNode.setEntity(tfc);
                    pcNode.addChildren(childNode);
                }
            }
        }
        return list;
    }

    @Override
    public void initFristTransferMethodCategory() {
        String[] initList = initTransferMethodCategory.split(",");
        if (initList.length > 0) {
            TransferMethodCategory transferMethodCategory;
            for (String categoryNumAndName : initList) {
                if (CollectionUtils.isEmpty(
                        listByNumAndName(categoryNumAndName.split("-")[0], categoryNumAndName.split("-")[1]))) {
                    transferMethodCategory = new TransferMethodCategory();
                    transferMethodCategory.setParentId("-1");
                    transferMethodCategory.setNum(categoryNumAndName.split("-")[0]);
                    transferMethodCategory.setName(categoryNumAndName.split("-")[1]);
                    save(transferMethodCategory);
                }
            }
        }
    }

    @Override
    public List<TransferMethodCategory> listByNumAndName(String num, String name) {
        return super.lambdaQuery().eq(TransferMethodCategory::getNum, num).eq(TransferMethodCategory::getName, name)
                .orderByAsc(TransferMethodCategory::getNum).list();
    }


    @Override
    public List<TransferMethodCategory> listByType(String  type) {
        return super.lambdaQuery().eq(TransferMethodCategory::getType, type)
                .orderByAsc(TransferMethodCategory::getNum).list();
    }
}