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

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.jeecg.common.api.vo.CommonGenericTree;
import org.jeecg.common.constant.CommonConstant;
import org.jeecg.common.util.StrUtils;
import org.jeecg.modules.tms.entity.Warehouse;
import org.jeecg.modules.tms.entity.vo.DictVo;
import org.jeecg.modules.tms.mapper.WarehouseMapper;
import org.jeecg.modules.tms.service.IWarehouseService;
import org.springframework.stereotype.Service;

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

import java.util.*;

/**
 * @Description: tms_warehouse
 * @Author: jeecg-boot
 * @Date:   2025-05-06
 * @Version: V1.0
 */
@Service
public class WarehouseServiceImpl extends ServiceImpl<WarehouseMapper, Warehouse> implements IWarehouseService {

    @Override
    public List<CommonGenericTree<Warehouse>> loadTree() {
        List<Warehouse> warehouseList = list(new LambdaQueryWrapper<Warehouse>()
                .eq(Warehouse::getStatus, CommonConstant.STATUS_1).orderByAsc(Warehouse::getSeq));
        return loadTree(warehouseList);
    }

    @Override
    public IPage<Warehouse> queryPageList(Page<Warehouse> page, Map<String, String[]> parameterMap) {
        QueryWrapper<Warehouse> queryWrapper = Wrappers.query();
        String[] warehouseNames = parameterMap.get("warehouseName");
        if (warehouseNames != null && warehouseNames.length > 0) {
            queryWrapper.like("t.warehouse_name", warehouseNames[0]);
        }
        String[] parentIds = parameterMap.get("parentId");
        if (parentIds != null && parentIds.length > 0) {
            queryWrapper.eq("t.parent_id", parentIds[0]);
        }
        String[] beginTimes = parameterMap.get("beginTime");
        String[] endTimes = parameterMap.get("endTime");
        if (beginTimes != null && beginTimes.length > 0) {
            queryWrapper.ge("t.create_time", beginTimes[0]);
        }
        if (endTimes != null && endTimes.length > 0) {
            queryWrapper.le("t.create_time", endTimes[0]);
        }
        queryWrapper.orderByAsc("t.seq");
        return this.baseMapper.queryPageList(page, queryWrapper);
    }

    @Override
    public List<DictVo> queryWarehouseDictList() {
        return this.baseMapper.queryWarehouseDictList();
    }

    private List<CommonGenericTree<Warehouse>> loadTree(List<Warehouse> warehouseList) {
        Warehouse warehouse = new Warehouse();
        List<CommonGenericTree<Warehouse>> list = new ArrayList<>();
        Map<String, CommonGenericTree<Warehouse>> map = new HashMap<>();
        CommonGenericTree<Warehouse> node = new CommonGenericTree<>();//手动创建根节点
        node.setKey("-1");
        node.setTitle("航宇救生");
        node.setRField1("");
        node.setRField2("");
        node.setEntity(new Warehouse()
                .setId("-1")
                .setWarehouseId("-1")
                .setWarehouseName("航宇救生")
                .setLeafFlag("2"));
        list.add(node);
        if (CollectionUtils.isNotEmpty(warehouseList)) {
            CommonGenericTree<Warehouse> tcNode;
            for (Warehouse wh : warehouseList) {
                if (StrUtils.isBlankOrNull(wh.getParentId()) || wh.getParentId().equals("-1")) {
                    tcNode = new CommonGenericTree<>();
                    tcNode.setKey(wh.getId());
                    tcNode.setTitle(wh.getWarehouseId() + "/" + wh.getWarehouseName());
                    tcNode.setParentId(node.getKey());
                    tcNode.setIcon("");
                    tcNode.setType(1);
                    tcNode.setValue(wh.getWarehouseName());
                    //tcNode.setDisabled(CommonConstant.STATUS_0.equals(wh.getStatus()) ? true : false);
                    tcNode.setRField1(wh.getWarehouseId());
                    tcNode.setRField2(getBaseParent(wh.getId(), 0).getWarehouseId());
                    tcNode.setEntity(wh);
                    node.addChildren(tcNode);
                    map.put(wh.getId(), tcNode);
                }
            }
            CommonGenericTree<Warehouse> childNode;
            for (Warehouse wh : warehouseList) {
                Warehouse child = wh;
                if (map.containsKey(child.getParentId())) {
                    if (StrUtils.isBlankOrNull(wh.getParentId()) || wh.getParentId().equals("-1")) {
                        warehouse = wh;
                    } else {
                        tcNode = map.get(child.getParentId());
                        childNode = new CommonGenericTree<>();
                        childNode.setKey(wh.getId());
                        childNode.setTitle(wh.getWarehouseId() + "/" + wh.getWarehouseName());
                        childNode.setParentId(wh.getParentId());
                        childNode.setIcon("");
                        childNode.setType(0);
                        childNode.setValue(wh.getWarehouseName());
                        //childNode.setDisabled(CommonConstant.STATUS_0.equals(wh.getStatus()) ? true : false);
                        childNode.setRField1(wh.getWarehouseId());
                        childNode.setRField2(getBaseParent(wh.getId(), 0).getWarehouseId());
                        childNode.setEntity(wh);
                        tcNode.addChildren(childNode);
                        map.put(child.getId(), childNode);
                    }
                }
            }
        }
        return list;
    }

    public Warehouse getBaseParent(String id, int index) {
        Warehouse warehouse = null;
        if (index < 99) {
            warehouse = getById(id);
            if (StrUtils.isNotBlankOrNull(warehouse.getParentId())) {
                if (!warehouse.getParentId().equals("-1")) {
                    warehouse = getBaseParent(warehouse.getParentId(), index++);
                }
            }
        }
        return warehouse;
    }
}