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

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.modules.base.entity.Unit;
import org.jeecg.modules.base.entity.UnitCategory;
import org.jeecg.modules.base.mapper.UnitMapper;
import org.jeecg.modules.base.mapper.UnitCategoryMapper;
import org.jeecg.modules.base.service.IUnitCategoryService;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

@Service
public class UnitCategoryServiceImpl extends ServiceImpl<UnitCategoryMapper, UnitCategory> implements IUnitCategoryService {

    @Resource
    private UnitCategoryMapper unitCategoryMapper;
    @Resource
    private UnitMapper unitMapper;

    // 初始化工装类别
//    private static final String initUnitCategory = "计量单位分类";
    /**
     * 查询计量单位分类树
     * @return
     */
    @Override
    public List<UnitCategory> getTreeList() {
        List<UnitCategory> treeList = new ArrayList<>();
        LambdaQueryWrapper<UnitCategory> query = new LambdaQueryWrapper<UnitCategory>();
        List<UnitCategory> list = this.list(query);
        for(UnitCategory unitCategory1 :list){
            LambdaQueryWrapper<UnitCategory> momBaseUnitCategoryLambdaQueryWrapper = new LambdaQueryWrapper<UnitCategory>();
            momBaseUnitCategoryLambdaQueryWrapper.eq(UnitCategory::getId, unitCategory1.getPid());
            UnitCategory unitCategory = unitCategoryMapper.selectOne(momBaseUnitCategoryLambdaQueryWrapper);
            if(unitCategory ==null||"".equals(unitCategory)){
                unitCategory1.setPName("");
            }else{
                unitCategory1.setPName(unitCategory.getCode()+ unitCategory.getName());
            }
            if(Objects.equals(unitCategory1.getPid(), "-1")){
                treeList.add(findChild(list, unitCategory1));
            }
        }
        return treeList;
    }

    @Override
    public void initFristUnitCategory() {

        UnitCategory unitCategory;
        List<UnitCategory> unitCategoryList = super.lambdaQuery()
                .eq(UnitCategory::getName, "计量单位分类")
                .eq(UnitCategory::getDelFlag, "0").list();
        if (unitCategoryList.size() == 0) {
            unitCategory = new UnitCategory();
            unitCategory.setCode("0");
//            unitCategory.setId("-1");
            unitCategory.setCodeName("计量单位分类");
            unitCategory.setPid("-1");
            super.save(unitCategory);
        }
    }

    /**
     * 查询计量单位树
     * @return
     */
    @Override
    public List<UnitCategory> getUnitTreeList() {
        List<UnitCategory> treeList = new ArrayList<>();
        LambdaQueryWrapper<UnitCategory> query = new LambdaQueryWrapper<UnitCategory>();
        List<UnitCategory> list = this.list(query);
        for(UnitCategory unitCategory1 :list){
            LambdaQueryWrapper<UnitCategory> momBaseUnitCategoryLambdaQueryWrapper = new LambdaQueryWrapper<UnitCategory>();
            momBaseUnitCategoryLambdaQueryWrapper.eq(UnitCategory::getId, unitCategory1.getPid());
            UnitCategory unitCategory = unitCategoryMapper.selectOne(momBaseUnitCategoryLambdaQueryWrapper);
            if(unitCategory ==null||"".equals(unitCategory)){
                unitCategory1.setPName("");
            }else{
                unitCategory1.setPName(unitCategory.getCode()+ unitCategory.getName());
            }
            if(Objects.equals(unitCategory1.getPid(), "-1")){
                treeList.add(findUnitChild(list, unitCategory1));
            }
        }
        return treeList;
    }

    /**
     * 查询计量单位分类的下级计量单位分类
     * @param id
     * @return
     */
    @Override
    public List<UnitCategory> getList(String id) {
        List<UnitCategory> list = unitCategoryMapper.getList(id);
        list.forEach( l ->{
            LambdaQueryWrapper<UnitCategory>  query = new LambdaQueryWrapper<>();
            query.eq(UnitCategory::getId,id);
            UnitCategory unitCategory = unitCategoryMapper.selectOne(query);
            l.setPName(l.getCode()+l.getName());
            l.setCodeName(unitCategory.getCode()+ unitCategory.getName());
        });
        return list;
    }
    /**
     * 查询计量单位分类的所有下级计量单位分类
     * @return
     */
    private static UnitCategory findChild(List<UnitCategory> list, UnitCategory unitCategory){
        for(UnitCategory unitCategory1 : list){
            unitCategory1.setCodeName(unitCategory1.getCode()+ unitCategory1.getName());
            if(Objects.equals(unitCategory1.getPid(), unitCategory.getId())){
                if(unitCategory.getChildren() == null){
                    unitCategory.setChildren(new ArrayList<UnitCategory>());
                }
                unitCategory.getChildren().add(findChild(list, unitCategory1));
            }
        }
        return unitCategory;
    }

    /**
     * 查询计量单位树
     * @return
     */
    private UnitCategory findUnitChild(List<UnitCategory> list, UnitCategory unitCategory){
        for(UnitCategory unitCategory1 : list){
            unitCategory1.setCodeName(unitCategory1.getCode()+ unitCategory1.getName());
            if(Objects.equals(unitCategory1.getPid(), unitCategory.getId())){
                if(unitCategory.getChildren() == null){
                    unitCategory.setChildren(new ArrayList<UnitCategory>());
                }
                unitCategory.getChildren().add(findUnitChild(list, unitCategory1));
            }else{
                if(unitCategory.getChildren() == null){
                    unitCategory.setChildren(new ArrayList<UnitCategory>());
                    List<Unit> units = unitMapper.getListByUnitCategoryId(unitCategory.getId());
                    for(Unit l : units){

                        UnitCategory mom = new UnitCategory();
                        mom.setCodeName(l.getNum()+l.getName());
                        mom.setId(l.getId());
                        mom.setCode(l.getNum());
                        mom.setName(l.getName());
                        unitCategory.getChildren().add(mom);

                    }

                }

            }
        }
        return unitCategory;
    }

}