package com.ruoyi.procurementrecord.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
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.common.exception.ServiceException;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.procurementrecord.mapper.ProcurementDemandMapper;
import com.ruoyi.procurementrecord.pojo.ProcurementDemand;
import com.ruoyi.procurementrecord.service.ProcurementDemandService;
import org.springframework.stereotype.Service;

import java.util.List;

@Service
public class ProcurementDemandServiceImpl extends ServiceImpl<ProcurementDemandMapper, ProcurementDemand> implements ProcurementDemandService {

    @Override
    public IPage<ProcurementDemand> pageList(Page<ProcurementDemand> page, ProcurementDemand req) {
        return this.page(page, buildWrapper(req));
    }

    @Override
    public List<ProcurementDemand> listByCondition(ProcurementDemand req) {
        return this.list(buildWrapper(req));
    }

    private LambdaQueryWrapper<ProcurementDemand> buildWrapper(ProcurementDemand req) {
        ProcurementDemand condition = req == null ? new ProcurementDemand() : req;
        return Wrappers.<ProcurementDemand>lambdaQuery()
                .like(StringUtils.isNotEmpty(condition.getProductName()), ProcurementDemand::getProductName, condition.getProductName())
                .like(StringUtils.isNotEmpty(condition.getProductModel()), ProcurementDemand::getProductModel, condition.getProductModel())
                .eq(condition.getStatus() != null, ProcurementDemand::getStatus, condition.getStatus())
                .orderByDesc(ProcurementDemand::getId);
    }

    @Override
    public boolean changeStatus(List<Long> ids, Integer status) {
        if (ids == null || ids.isEmpty() || status == null || status < 0 || status > 2) {
            throw new ServiceException("请选择要操作的需求并传入合法状态(0待处理/1已转采购/2已取消)");
        }
        return this.update(Wrappers.<ProcurementDemand>lambdaUpdate()
                .set(ProcurementDemand::getStatus, status)
                .in(ProcurementDemand::getId, ids));
    }

    @Override
    public boolean removePending(List<Long> ids) {
        if (ids == null || ids.isEmpty()) {
            throw new ServiceException("请至少选择一条采购需求");
        }
        long handledCount = this.count(Wrappers.<ProcurementDemand>lambdaQuery()
                .in(ProcurementDemand::getId, ids)
                .ne(ProcurementDemand::getStatus, 0));
        if (handledCount > 0) {
            throw new ServiceException("存在非待处理状态的需求，只能删除待处理数据");
        }
        return this.removeByIds(ids);
    }
}