package cn.iocoder.yudao.module.wms.dal.mysql.inventory;

import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.wms.api.inventory.dto.WmsInventoryQueryReqDTO;
import cn.iocoder.yudao.module.wms.controller.admin.inventory.vo.WmsInventoryReservePageReqVO;
import cn.iocoder.yudao.module.wms.dal.dataobject.inventory.WmsInventoryDO;
import cn.iocoder.yudao.module.wms.dal.dataobject.inventory.WmsInventoryReserveDO;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;

import java.math.BigDecimal;
import java.util.List;

/**
 * WMS 库存预留 Mapper
 *
 * @author 芋道源码
 */
@Mapper
public interface WmsInventoryReserveMapper extends BaseMapperX<WmsInventoryReserveDO> {

    default PageResult<WmsInventoryReserveDO> selectPage(WmsInventoryReservePageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<WmsInventoryReserveDO>()
                .likeIfPresent(WmsInventoryReserveDO::getReserveNo, reqVO.getReserveNo())
                .eqIfPresent(WmsInventoryReserveDO::getBizType, reqVO.getBizType())
                .likeIfPresent(WmsInventoryReserveDO::getBizNo, reqVO.getBizNo())
                .eqIfPresent(WmsInventoryReserveDO::getStatus, reqVO.getStatus())
                .orderByDesc(WmsInventoryReserveDO::getId));
    }

    default List<WmsInventoryReserveDO> selectListByBiz(Integer bizType, Long bizId) {
        return selectList(new LambdaQueryWrapperX<WmsInventoryReserveDO>()
                .eq(WmsInventoryReserveDO::getBizType, bizType)
                .eq(WmsInventoryReserveDO::getBizId, bizId)
                .in(WmsInventoryReserveDO::getStatus, 0, 1)); // 预留中或部分使用
    }

    default WmsInventoryReserveDO selectByReserveNo(String reserveNo) {
        return selectOne(WmsInventoryReserveDO::getReserveNo, reserveNo);
    }

    /**
     * 查询指定 SKU 和仓库的预留数量
     */
    @Select("SELECT IFNULL(SUM(reserve_quantity - used_quantity), 0) FROM wms_inventory_reserve " +
            "WHERE sku_id = #{skuId} AND warehouse_id = #{warehouseId} AND status IN (0, 1) AND deleted = 0")
    BigDecimal selectReserveQuantity(@Param("skuId") Long skuId, @Param("warehouseId") Long warehouseId);

}