package org.jeecg.modules.eam.base.util;

import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import io.netty.util.internal.StringUtil;
import org.jeecg.common.handler.IFillRuleHandler;
import org.jeecg.common.util.SpringContextUtils;
import org.jeecg.common.util.YouBianCodeUtil;
import org.jeecg.modules.eam.base.entity.BaseFactory;
import org.jeecg.modules.eam.base.service.IBaseFactoryService;

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

/**
 * @author: LiuS
 * @create: 2023-07-04 16:15
 * @Description: 车间机构编码生成规则
 */
public class WorkShopOrgCodeProRule implements IFillRuleHandler {

    @Override
    public Object execute(JSONObject params, JSONObject formData) {
        IBaseFactoryService BaseFactoryService = (IBaseFactoryService) SpringContextUtils.getBean("BaseFactoryServiceImpl");

        LambdaQueryWrapper<BaseFactory> query = new LambdaQueryWrapper<BaseFactory>();
        LambdaQueryWrapper<BaseFactory> query1 = new LambdaQueryWrapper<BaseFactory>();
        // 创建一个List集合,存储查询返回的所有BaseFactory对象
        List<BaseFactory> BaseFactoryList = new ArrayList<>();
        String[] strArray = new String[2];
        //定义车间类型
        String orgType = "";
        // 定义新编码字符串
        String newOrgCode = "";
        // 定义旧编码字符串
        String oldOrgCode = "";

        String parentId = null;
        if (formData != null && formData.size() > 0) {
            Object obj = formData.get("parentId");
            if (obj != null) {
                parentId = obj.toString();
            }
        } else {
            if (params != null) {
                Object obj = params.get("parentId");
                if (obj != null) {
                    parentId = obj.toString();
                }
            }
        }

        //如果是最高级,则查询出同级的org_code, 调用工具类生成编码并返回
        if (StringUtil.isNullOrEmpty(parentId)) {
            // 线判断数据库中的表是否为空,空则直接返回初始编码
            query1.eq(BaseFactory::getParentId, "").or().isNull(BaseFactory::getParentId);
            query1.orderByDesc(BaseFactory::getOrgCode);
            BaseFactoryList = BaseFactoryService.list(query1);
            if (BaseFactoryList == null || BaseFactoryList.size() == 0) {
                strArray[0] = YouBianCodeUtil.getNextYouBianCode(null);
                strArray[1] = "1";
                return strArray;
            } else {
                BaseFactory BaseFactory = BaseFactoryList.get(0);
                oldOrgCode = BaseFactory.getOrgCode();
                orgType = BaseFactory.getOrgType();
                newOrgCode = YouBianCodeUtil.getNextYouBianCode(oldOrgCode);
            }
        } else {//反之则查询出所有同级的车间,获取结果后有两种情况,有同级和没有同级
            // 封装查询同级的条件
            query.eq(BaseFactory::getParentId, parentId);
            // 降序排序
            query.orderByDesc(BaseFactory::getOrgCode);
            // 查询出同级产线的集合
            List<BaseFactory> parentList = BaseFactoryService.list(query);
            // 查询出父级产线
            BaseFactory production = BaseFactoryService.getById(parentId);
            // 获取父级产线的code
            String parentCode = production.getOrgCode();
            // 根据父级产线类型算出当前产线的类型
            orgType = String.valueOf(Integer.valueOf(production.getOrgType()) + 1);
            // 处理同级产线为null的情况
            if (parentList == null || parentList.size() == 0) {
                // 直接生成当前的产线编码并返回
                newOrgCode = YouBianCodeUtil.getSubYouBianCode(parentCode, null);
            } else { //处理有同级产线的情况
                // 获取同级产线的编码,利用工具类
                String subCode = parentList.get(0).getOrgCode();
                // 返回生成的当前产线编码
                newOrgCode = YouBianCodeUtil.getSubYouBianCode(parentCode, subCode);
            }
        }
        // 返回最终封装了产线编码和产线类型的数组
        strArray[0] = newOrgCode;
        strArray[1] = orgType;
        return strArray;
    }
}