fans/hx-sale/src/main/java/com/fjhx/sale/flow/EhsdPurchaseFlow.java

package com.fjhx.sale.flow;

import cn.hutool.core.util.ObjectUtil;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.dynamic.datasource.annotation.DSTransactional;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.fjhx.area.utils.CustomizeAreaUtil;
import com.fjhx.common.entity.InOutBo;
import com.fjhx.common.enums.CodingRuleEnum;
import com.fjhx.common.enums.FlowStatusEnum1;
import com.fjhx.common.enums.InOutType;
import com.fjhx.common.service.coding.CodingRuleService;
import com.fjhx.common.utils.Assert;
import com.fjhx.flow.core.FlowDelegate;
import com.fjhx.flow.enums.FlowStatusEnum;
import com.fjhx.item.enums.ProductAvailableRecordType;
import com.fjhx.item.service.product.ProductInfoService;
import com.fjhx.purchase.entity.purchase.enums.PurchaseDataResourceEnum;
import com.fjhx.purchase.entity.purchase.enums.PurchaseStatusEnum;
import com.fjhx.purchase.entity.subscribe.enums.SubscribeDetailStatusEnum;
import com.fjhx.purchase.entity.subscribe.po.SubscribeDetail;
import com.fjhx.purchase.mapper.subscribe.SubscribeDetailMapper;
import com.fjhx.purchase.service.subscribe.SubscribeDetailService;
import com.fjhx.sale.entity.contract.po.Contract;
import com.fjhx.sale.entity.purchase.dto.EhsdPurchaseDto;
import com.fjhx.sale.entity.purchase.po.EhsdPurchase;
import com.fjhx.sale.entity.purchase.po.EhsdPurchaseProduct;
import com.fjhx.sale.entity.purchase.po.EhsdPurchaseProject;
import com.fjhx.sale.entity.sample.po.Sample;
import com.fjhx.sale.entity.sample.po.SampleProduct;
import com.fjhx.sale.service.contract.ContractService;
import com.fjhx.sale.service.purchase.EhsdPurchaseProductService;
import com.fjhx.sale.service.purchase.EhsdPurchaseProjectService;
import com.fjhx.sale.service.purchase.EhsdPurchaseService;
import com.fjhx.sale.service.sample.SampleProductService;
import com.fjhx.sale.service.sample.SampleService;
import com.fjhx.wms.entity.stock.emums.StockWaitType;
import com.fjhx.wms.entity.stock.po.StockWait;
import com.fjhx.wms.entity.stock.po.StockWaitDetails;
import com.fjhx.wms.service.stock.StockWaitDetailsService;
import com.fjhx.wms.service.stock.StockWaitService;
import com.ruoyi.common.annotation.LogicIgnore;
import com.ruoyi.common.core.domain.BasePo;
import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.common.utils.SecurityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.math.BigDecimal;
import java.util.*;
import java.util.stream.Collectors;

/**
 * EHSD采购流程
 *
 * @Author:caozj
 * @DATE:2023/4/3 17:38
 */
@Component
public class EhsdPurchaseFlow extends FlowDelegate {

    @Autowired
    private EhsdPurchaseService purchaseService;

    @Autowired
    private EhsdPurchaseProductService purchaseProductService;

    @Autowired
    private EhsdPurchaseProjectService purchaseProjectService;
    @Autowired
    private SampleProductService sampleProductService;

    @Autowired
    private CodingRuleService codingRuleService;
    @Autowired
    private SubscribeDetailService subscribeDetailService;
    @Autowired
    private ContractService contractService;
    @Autowired
    private SampleService sampleService;
    @Autowired
    private ProductInfoService productInfoService;
    @Autowired
    private StockWaitService stockWaitService;
    @Autowired
    private StockWaitDetailsService stockWaitDetailsService;
    @Autowired
    private SubscribeDetailMapper subscribeDetailMapper;

    @Override
    public String getFlowKey() {
        return "purchase_flow";
    }

    /**
     * 发起流程
     *
     * @param flowId     流程ID
     * @param submitData 采购数据
     * @return
     */
    @Override
    public Long start(Long flowId, JSONObject submitData) {
        EhsdPurchaseDto purchase = submitData.toJavaObject(EhsdPurchaseDto.class);

        if (ObjectUtil.isEmpty(purchase.getCurrency())) {
            purchase.setCurrency("CNY");
        }

        purchase.setId(null);//清空id防止前端误传
        purchase.setFlowId(flowId);//赋值流程id
        purchase.setProcessInstanceId(getFlowKey());//赋值流程key

        //赋值产品归属公司
        Long companyId = purchase.getCompanyId();
        if (ObjectUtil.isEmpty(companyId)) {
            purchase.setCompanyId(SecurityUtils.getCompanyId());
        }

        Integer dataResource = purchase.getDataResource();
        if (ObjectUtils.isEmpty(dataResource)) {
            throw new ServiceException("数据来源类型不能为空！");
        }

        //采购单编号生成
        if (0 == dataResource) {
            //手动创建 编号规则
            purchase.setCode(codingRuleService.createCode(CodingRuleEnum.EHSD_PURCHASE.getKey(), null));
        } else if (1 == dataResource) {
            //合同 编号规则
            Contract contract = contractService.getById(purchase.getDataResourceId());
            Assert.notEmpty(contract, "查询不到合同信息，无法生成编号");
            long count = purchaseService.count(q -> q.eq(EhsdPurchase::getDataResourceId, contract.getId()));
            purchase.setCode(contract.getCode() + "-" + (count + 1));
        } else if (2 == dataResource) {
            //样品单 编号规则
            Sample sample = sampleService.getById(purchase.getDataResourceId());
            Assert.notEmpty(sample, "查询不到样品单信息，无法生成编号");
            long count = purchaseService.count(q -> q.eq(EhsdPurchase::getDataResourceId, sample.getId()));
            purchase.setCode(sample.getCode() + "-" + (count + 1));
        } else {
            throw new ServiceException("未知数据来源类型");
        }

        //公共代码
        purchase = connStart(purchase);


        //回传单号给流程引擎
        submitData.put("code", purchase.getCode());
        submitData.put("currency", purchase.getCurrency());

        return purchase.getId();
    }


    /**
     * 公共代码块--发起
     */
    public EhsdPurchaseDto connStart(EhsdPurchaseDto purchase) {
        //赋值城市省份信息
        CustomizeAreaUtil.setAreaId(purchase);
        purchase.setSellCityId(purchase.getCityId());
        purchase.setSellCountryId(purchase.getCountryId());
        purchase.setSellProvinceId(purchase.getProvinceId());


        purchase.setStatus(PurchaseStatusEnum.UNDER_REVIEW.getKey());
        String nickName = SecurityUtils.getLoginUser().getUser().getNickName();
        purchase.setUserName(nickName);
        purchaseService.saveOrUpdate(purchase);

        List<EhsdPurchaseProduct> purchaseProductList = purchase.getPurchaseProductList();
        List<EhsdPurchaseProject> purchaseProjectList = purchase.getPurchaseProjectList();

        //防止空指针
        purchaseProjectList = (purchaseProjectList == null ? new ArrayList<>() : purchaseProjectList);
        purchaseProductList = (purchaseProductList == null ? new ArrayList<>() : purchaseProductList);

        //产品（采购明细）
        for (EhsdPurchaseProduct s : purchaseProductList) {
            s.setId(ObjectUtils.isNotEmpty(s.getId()) ? s.getId() : IdWorker.getId());
            s.setPurchaseId(purchase.getId());
        }
        //修改或删除数据
        purchaseProductService.editLinked(purchaseProductList, EhsdPurchaseProduct::getPurchaseId, purchase.getId());

        //收费项目
        for (EhsdPurchaseProject s : purchaseProjectList) {
            s.setPurchaseId(purchase.getId());
        }
        purchaseProjectService.editLinked(purchaseProjectList, EhsdPurchaseProject::getPurchaseId, purchase.getId());

        return purchase;
    }

    /**
     * 结束流程
     *
     * @param flowId     流程ID
     * @param businessId 业务ID
     * @param submitData 数据
     */
    @Override
    public void end(Long flowId, Long businessId, JSONObject submitData) {
        //通过业务ID查询采购数据
        EhsdPurchase purchase = purchaseService.getById(businessId);
        if (ObjectUtils.isEmpty(purchase)) {
            throw new ServiceException("采购单不存在,或已被删除");
        }
        //查询采购产品
        List<EhsdPurchaseProduct> purchaseProductList = purchaseProductService.list(Wrappers.<EhsdPurchaseProduct>query().lambda().eq(EhsdPurchaseProduct::getPurchaseId, businessId));
        List<SampleProduct> upSampleProduct = new ArrayList<>();
        for (EhsdPurchaseProduct p : purchaseProductList) {
            if (ObjectUtils.isNotEmpty(p.getDataResourceId()) &&
                    p.getDataResource() == PurchaseDataResourceEnum.DATA_RESOURCE_2.getKey()) {//如果采购的是样品单
                SampleProduct sampleProduct = sampleProductService.getById(p.getDataResourceId());
                BigDecimal expendQuantity = sampleProduct.getExpendQuantity().subtract(p.getQuantity());
                sampleProduct.setExpendQuantity(expendQuantity);
                upSampleProduct.add(sampleProduct);
            }
        }
        if (CollectionUtils.isNotEmpty(upSampleProduct)) {//扣减样品单数量
            sampleProductService.updateBatchById(upSampleProduct);
        }
        //修改采购状态为审批通过
        purchase.setStatus(PurchaseStatusEnum.PASS.getKey());
        purchase.setApprovedDate(new Date());
        purchaseService.updateById(purchase);

        //修改申购明细状态
        List<Long> subscribeDetailIds = purchaseProductService.listObject(EhsdPurchaseProduct::getSubscribeDetailId, q -> q
                .eq(EhsdPurchaseProduct::getPurchaseId, businessId)
        );
        //获取已采购数量
        List<SubscribeDetail> purchaseCountByIds = subscribeDetailMapper.getPurchaseCountByIds(subscribeDetailIds);
        Map<Long, BigDecimal> purchaseCountMap = purchaseCountByIds.stream().collect(Collectors.toMap(SubscribeDetail::getId, SubscribeDetail::getCount));
        //修改状态
        List<SubscribeDetail> subscribeDetails = subscribeDetailService.listByIds(subscribeDetailIds);
        for (SubscribeDetail subscribeDetail : subscribeDetails) {
            //获取已采购数并计算状态
            BigDecimal count = purchaseCountMap.getOrDefault(subscribeDetail.getId(), BigDecimal.ZERO);
            if (count.compareTo(subscribeDetail.getCount()) >= 0) {
                //修改为已采购
                subscribeDetail.setStatus(SubscribeDetailStatusEnum.PURCHASED.getKey());
            } else {
                //修改为部分采购
                subscribeDetail.setStatus(SubscribeDetailStatusEnum.LITT_PAID_AMOUNT.getKey());
            }
        }
        subscribeDetailService.updateBatchById(subscribeDetails);


        List<InOutBo> inOutBoList = new ArrayList<>();
        for (EhsdPurchaseProduct purchaseProduct : purchaseProductList) {
            //如果数据来源是申购 操作可用库存(可用库存 = 当前可用库存 + 采购明细数量)
            if (Objects.equals(purchaseProduct.getDataResource(), PurchaseDataResourceEnum.DATA_RESOURCE_0)) {
                InOutBo inOutBo = new InOutBo();
                inOutBo.setProductId(purchaseProduct.getProductId());
                inOutBo.setQuantity(purchaseProduct.getQuantity());
                inOutBoList.add(inOutBo);
            }
        }
        productInfoService.editAvailableQuantity(inOutBoList, InOutType.IN, businessId, ProductAvailableRecordType.PURCHASE_PASS, purchase.getCompanyId());


        //采购审批通过 生成待入库数据
        StockWait stockWait = new StockWait();
        stockWait.setType(1);
        stockWait.setBusinessType(StockWaitType.PURCHASE_ARRIVAL_IN.getDetailType());
        stockWait.setBusinessId(purchase.getId());
        stockWait.setBusinessCode(purchase.getCode());
        stockWait.setStatus(0);
        stockWait.setPurchaseId(purchase.getId());
        stockWaitService.save(stockWait);
        List<StockWaitDetails> stockWaitDetailsList = new ArrayList<>();
        for (EhsdPurchaseProduct purchaseProduct : purchaseProductList) {
            StockWaitDetails stockWaitDetails = new StockWaitDetails();
            stockWaitDetails.setStockWaitId(stockWait.getId());
            stockWaitDetails.setBusinessDetailsId(purchaseProduct.getId());
            stockWaitDetails.setPurchaseDetailId(purchaseProduct.getId());

            stockWaitDetails.setProductId(purchaseProduct.getProductId());
            stockWaitDetails.setQuantity(purchaseProduct.getQuantity());
            stockWaitDetails.setStatus(0);

            stockWaitDetailsList.add(stockWaitDetails);
        }
        stockWaitDetailsService.saveBatch(stockWaitDetailsList);
    }

    /**
     * 重新发起
     */
    @Override
    @LogicIgnore
    @DSTransactional
    public void relaunch(Long flowId, Long businessId, FlowStatusEnum flowStatus, JSONObject submitData) {
        //删除采购合同
        EhsdPurchaseDto purchase = submitData.toJavaObject(EhsdPurchaseDto.class);
        if (ObjectUtils.isEmpty(purchase)) {
            throw new ServiceException("采购数据不能为空");
        }
        purchase.setFlowId(flowId);
        connStart(purchase);
    }

    /**
     * 驳回
     */
    @Override
    public void reject(Long flowId, Long businessId, FlowStatusEnum flowStatus) {
        purchaseService.update(q -> q
                .eq(EhsdPurchase::getId, businessId)
                .set(EhsdPurchase::getStatus, FlowStatusEnum1.REJECT.getKey())
                .set(BasePo::getUpdateTime, new Date())
                .set(BasePo::getUpdateUser, SecurityUtils.getUserId())
        );
    }

    /**
     * 作废
     */
    @Override
    public void cancellation(Long flowId, Long businessId, FlowStatusEnum flowStatus) {
        EhsdPurchase purchase = purchaseService.getById(businessId);

        super.cancellation(flowId, businessId, flowStatus);
        purchaseService.update(q -> q
                .eq(EhsdPurchase::getId, businessId)
                .set(EhsdPurchase::getStatus, FlowStatusEnum1.CANCELLATION.getKey())
                .set(BasePo::getUpdateTime, new Date())
                .set(BasePo::getUpdateUser, SecurityUtils.getUserId())
        );

        //修改可用库存
        List<InOutBo> inOutBoList = new ArrayList<>();
        List<EhsdPurchaseProduct> purchaseProductList = purchaseProductService.list(q -> q.eq(EhsdPurchaseProduct::getPurchaseId, businessId));
        for (EhsdPurchaseProduct purchaseProduct : purchaseProductList) {
            //如果数据来源是申购 操作可用库存(可用库存 = 当前可用库存 + 采购明细数量)
            if (Objects.equals(purchaseProduct.getDataResource(), PurchaseDataResourceEnum.DATA_RESOURCE_0)) {
                InOutBo inOutBo = new InOutBo();
                inOutBo.setProductId(purchaseProduct.getProductId());
                inOutBo.setQuantity(purchaseProduct.getQuantity());
                inOutBoList.add(inOutBo);
            }
        }
        productInfoService.editAvailableQuantity(inOutBoList, InOutType.OUT, businessId, ProductAvailableRecordType.PURCHASE_CANCEL, purchase.getCompanyId());
    }
}