package com.ruoyi.production.service.impl;

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 com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.production.mapper.ProductionLineMapper;
import com.ruoyi.production.pojo.ProductionLine;
import com.ruoyi.production.service.ProductionLineService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

/**
 * @author :yys
 * @date : 2025/12/4 11:19
 */
@Service
@Slf4j
public class ProductionLineServiceImpl extends ServiceImpl<ProductionLineMapper, ProductionLine> implements ProductionLineService {

    @Autowired
    private ProductionLineMapper productionLineMapper;

    /**
     * 递归查询产线树
     * @return
     */
    @Override
    public List<ProductionLine> listTree() {
        List<ProductionLine> productionLines = productionLineMapper.selectList(Wrappers.lambdaQuery(ProductionLine.class)
                .eq(ProductionLine::getParentId, 0)
                .eq(ProductionLine::getType, 1));
        // 2. 递归为每个根节点加载子节点（工序/子产线）
        for (ProductionLine rootLine : productionLines) {
            loadChildren(rootLine,1);
        }
        return productionLines;
    }

    /**
     * 递归加载子节点（核心方法）
     * @param parentNode 父节点对象
     */
    private void loadChildren(ProductionLine parentNode,Integer type) {
        // 1. 查询当前父节点的所有子节点
        List<ProductionLine> children = productionLineMapper.selectList(Wrappers.lambdaQuery(ProductionLine.class)
                .eq(ProductionLine::getParentId, parentNode.getId())
                .eq(ProductionLine::getType, type));

        // 2. 若有子节点，继续递归加载子节点的子节点
        if (!children.isEmpty()) {
            parentNode.setChildren(children); // 设置子节点
            for (ProductionLine child : children) {
                loadChildren(child,type); // 递归调用，加载孙子节点
            }
        }
    }

    /**
     * 按需提供：根据父节点ID查询单个子树（如查询某个产线下的所有工序）
     */
    @Override
    public IPage<ProductionLine> getSubTreeByParentId(Page page, ProductionLine productionLine) {
        // 1. 查询父节点本身
        ProductionLine parentNode = productionLineMapper.selectById(productionLine.getId());
        if (parentNode == null) {
            return null;
        }
        // 2. 递归加载子节点
        loadChildren(productionLine,1);
        List<Long> ids = new ArrayList<>();
        ids.add(productionLine.getId());
        if(CollectionUtils.isNotEmpty(productionLine.getChildren())){
            // 递归获取所有子节点的id
            getAllChildIds(productionLine,ids);
        }
        return productionLineMapper.selectPage(page, Wrappers.lambdaQuery(ProductionLine.class)
                .in(ProductionLine::getParentId, ids)
                .eq(ProductionLine::getType, 2));
    }

    /**
     * 递归获取所有子节点的id
     */
    public List<Long> getAllChildIds(ProductionLine node,List<Long> ids) {
        if (node.getChildren() != null) {
            for (ProductionLine child : node.getChildren()) {
                ids.add(child.getId());
                ids.addAll(getAllChildIds(child,ids));
            }
        }
        return ids;
    }
}