package com.klp.service.impl;

import cn.hutool.core.bean.BeanUtil;
import com.klp.common.core.page.TableDataInfo;
import com.klp.common.core.domain.PageQuery;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.klp.common.utils.StringUtils;
import com.klp.domain.WmsProductBom;
import com.klp.domain.WmsPurchasePlanDetail;
import com.klp.domain.WmsRawMaterial;
import com.klp.domain.vo.WmsOrderDetailVo;
import com.klp.domain.vo.WmsPurchasePlanDetailVo;
import com.klp.mapper.WmsPurchasePlanDetailMapper;
import com.klp.mapper.WmsRawMaterialMapper;
import com.klp.service.*;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import com.klp.domain.bo.WmsPurchasePlanBo;
import com.klp.domain.vo.WmsPurchasePlanVo;
import com.klp.domain.WmsPurchasePlan;
import com.klp.mapper.WmsPurchasePlanMapper;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.math.BigDecimal;
import java.util.*;

/**
 * 采购计划主Service业务层处理
 *
 * @author Joshi
 * @date 2025-07-18
 */
@RequiredArgsConstructor
@Service
public class WmsPurchasePlanServiceImpl implements IWmsPurchasePlanService {

    private final WmsPurchasePlanMapper baseMapper;

    private final IWmsOrderDetailService wmsOrderDetailService;

    private final IWmsProductBomService wmsProductBomService;

    private final IWmsStockService wmsStockService;

    @Resource
    private  WmsPurchasePlanDetailMapper wmsPurchasePlanDetailMapper;

    @Resource
    private WmsRawMaterialMapper wmsRawMaterialMapper;

    @Transactional
    @Override
    public Boolean insertWithDetails(WmsPurchasePlanVo planVo) {
        // 1. 保存主表
        WmsPurchasePlan plan = new WmsPurchasePlan();
        BeanUtils.copyProperties(planVo, plan);
        int flag = 0;
        flag += baseMapper.insert(plan);

        // 2. 保存明细表
        for (WmsPurchasePlanDetailVo detailVo : planVo.getDetailList()) {
            WmsPurchasePlanDetail detail = new WmsPurchasePlanDetail();
            BeanUtils.copyProperties(detailVo, detail);
            detail.setPlanId(plan.getPlanId()); // 关联主表ID
            flag += wmsPurchasePlanDetailMapper.insert(detail);
        }
        return flag > 0;
    }

    @Override
    public WmsPurchasePlanVo recommendPurchasePlanByOrder(Long orderId) {
        // 1. 查询订单明细
        List<WmsOrderDetailVo> orderDetails = wmsOrderDetailService.queryListByOrderId(orderId);
        // 2. 汇总原材料需求
        Map<Long, WmsPurchasePlanDetailVo> materialMap = new HashMap<>();
        for (WmsOrderDetailVo detail : orderDetails) {
            List<WmsProductBom> bomList = wmsProductBomService.listByProductId(detail.getProductId());
            for (WmsProductBom bom : bomList) {
                BigDecimal needQty = bom.getQuantity().multiply(detail.getQuantity());
                WmsPurchasePlanDetailVo vo =
                    materialMap.getOrDefault(bom.getRawMaterialId(), new WmsPurchasePlanDetailVo());
                vo.setRawMaterialId(bom.getRawMaterialId());
                vo.setQuantity(vo.getQuantity() == null ? needQty : vo.getQuantity().add(needQty));
                vo.setUnit(bom.getUnit());
                // 挂载原材料名称编号
                WmsRawMaterial wmsRawMaterial = wmsRawMaterialMapper.selectById(bom.getRawMaterialId());
                vo.setRawMaterialName(wmsRawMaterial.getRawMaterialName());
                vo.setRawMaterialCode(wmsRawMaterial.getRawMaterialCode());
                materialMap.put(bom.getRawMaterialId(), vo);
            }
        }
        // 3. 查询库存并计算推荐采购量
        for (WmsPurchasePlanDetailVo vo : materialMap.values()) {
            // 需求量
            vo.setDemand(vo.getQuantity());
            BigDecimal stockQty = wmsStockService.getStockByItemId(vo.getRawMaterialId());
            // 库存量
            if(stockQty == null){
                stockQty = BigDecimal.ZERO;
            }
            vo.setInventory(stockQty);
            // 在途量
            BigDecimal onTheWayQty = BigDecimal.ZERO;
            BigDecimal byRawMaterialIdAndOnTheWay =
                wmsPurchasePlanDetailMapper.getByRawMaterialIdAndOnTheWay(vo.getRawMaterialId());
            if (byRawMaterialIdAndOnTheWay != null) {
                onTheWayQty = byRawMaterialIdAndOnTheWay;
            }
            vo.setOnTheWay(onTheWayQty);
            // 计算推荐采购量
            BigDecimal recommendQty = vo.getQuantity().subtract(onTheWayQty).subtract(stockQty);
            vo.setQuantity(recommendQty.compareTo(BigDecimal.ZERO) > 0 ? recommendQty : BigDecimal.ZERO);
        }
        // 4. 组装主VO
        WmsPurchasePlanVo planVo = new WmsPurchasePlanVo();
        planVo.setPlanCode("推荐计划-" + orderId);
        planVo.setOrderId(orderId);
        planVo.setOwner("系统推荐");
        planVo.setRemark("仅供参考，实际采购请确认");
        planVo.setDetailList(new ArrayList<>(materialMap.values()));
        return planVo;
    }

    /**
     * 查询采购计划主
     */
    @Override
    public WmsPurchasePlanVo queryById(Long planId){
        return baseMapper.selectVoById(planId);
    }

    /**
     * 查询采购计划主列表
     */
    @Override
    public TableDataInfo<WmsPurchasePlanVo> queryPageList(WmsPurchasePlanBo bo, PageQuery pageQuery) {
        LambdaQueryWrapper<WmsPurchasePlan> lqw = buildQueryWrapper(bo);
        Page<WmsPurchasePlanVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
        return TableDataInfo.build(result);
    }

    /**
     * 查询采购计划主列表
     */
    @Override
    public List<WmsPurchasePlanVo> queryList(WmsPurchasePlanBo bo) {
        LambdaQueryWrapper<WmsPurchasePlan> lqw = buildQueryWrapper(bo);
        return baseMapper.selectVoList(lqw);
    }

    private LambdaQueryWrapper<WmsPurchasePlan> buildQueryWrapper(WmsPurchasePlanBo bo) {
        Map<String, Object> params = bo.getParams();
        LambdaQueryWrapper<WmsPurchasePlan> lqw = Wrappers.lambdaQuery();
        lqw.eq(StringUtils.isNotBlank(bo.getPlanCode()), WmsPurchasePlan::getPlanCode, bo.getPlanCode());
        lqw.eq(StringUtils.isNotBlank(bo.getOwner()), WmsPurchasePlan::getOwner, bo.getOwner());
        lqw.eq(bo.getOrderId() != null, WmsPurchasePlan::getOrderId, bo.getOrderId());
        lqw.eq(bo.getStatus() != null, WmsPurchasePlan::getStatus, bo.getStatus());
        return lqw;
    }

    /**
     * 新增采购计划主
     */
    @Override
    public Boolean insertByBo(WmsPurchasePlanBo bo) {
        WmsPurchasePlan add = BeanUtil.toBean(bo, WmsPurchasePlan.class);
        validEntityBeforeSave(add);
        boolean flag = baseMapper.insert(add) > 0;
        if (flag) {
            bo.setPlanId(add.getPlanId());
        }
        return flag;
    }

    /**
     * 修改采购计划主
     */
    @Override
    public Boolean updateByBo(WmsPurchasePlanBo bo) {
        WmsPurchasePlan update = BeanUtil.toBean(bo, WmsPurchasePlan.class);
        validEntityBeforeSave(update);
        return baseMapper.updateById(update) > 0;
    }

    /**
     * 保存前的数据校验
     */
    private void validEntityBeforeSave(WmsPurchasePlan entity){
        //TODO 做一些数据校验,如唯一约束
    }

    /**
     * 批量删除采购计划主
     */
    @Override
    public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
        // 逻辑删除采购明细
        LambdaQueryWrapper<WmsPurchasePlanDetail> lqw = Wrappers.lambdaQuery();
        lqw.in(WmsPurchasePlanDetail::getPlanId, ids);
        wmsPurchasePlanDetailMapper.delete(lqw);
        // 逻辑删除采购计划主表
        return baseMapper.deleteBatchIds(ids) > 0;
    }
}