package com.fzzy.igds.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fzzy.igds.data.GrainParam;
import com.fzzy.igds.data.InoutParam;
import com.fzzy.igds.domain.Grain;
import com.fzzy.igds.domain.InoutRecord;
import com.fzzy.igds.mapper.GrainMapper;
import com.fzzy.igds.utils.ContextUtil;
import com.fzzy.igds.utils.DateUtil;
import com.ruoyi.common.utils.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @Description
 * @Author CZT
 * @Date 2025/12/9 10:00
 */
@Slf4j
@Service
public class GrainService {

    @Resource
    private GrainMapper grainMapper;

    /**
     * 查询数据
     * @param param
     */
    public List<Grain> listData(GrainParam param) {
        if(StringUtils.isBlank(param.getCompanyId())){
            param.setCompanyId(ContextUtil.getCompanyId());
        }
        if(StringUtils.isBlank(param.getDeptId())){
            param.setDeptId(ContextUtil.subDeptId(null));
        }

        QueryWrapper<Grain> queryWrapper = new QueryWrapper<>();
        if (StringUtils.isNotBlank(param.getCompanyId())) {
            queryWrapper.eq("company_id", param.getCompanyId());
        }
        if (StringUtils.isNotBlank(param.getDeptId())) {
            queryWrapper.eq("dept_id", param.getDeptId());
        }
        if (StringUtils.isNotBlank(param.getDepotId())) {
            queryWrapper.eq("depot_id", param.getDepotId());
        }
        if (StringUtils.isNotBlank(param.getBatchId())) {
            queryWrapper.eq("batch_id", param.getBatchId());
        }
        if (null != param.getStart()) {
            queryWrapper.ge("receive_date", DateUtil.getCurZero(param.getStart()));
        }
        if (null != param.getEnd()) {
            queryWrapper.le("receive_date", DateUtil.getNextZero(param.getEnd()));
        }
        queryWrapper.orderByDesc("batch_id");
        return grainMapper.selectList(queryWrapper);
    }

    /**
     * 分页查询数据
     * @param page
     * @param param
     */
    public void pageData(Page<Grain> page, Map<String, Object> param) {

        QueryWrapper<Grain> queryWrapper = new QueryWrapper<>();

        queryWrapper.eq("company_id", ContextUtil.getCompanyId());
        queryWrapper.eq("dept_id", ContextUtil.subDeptId(null));

        String ids = (String) param.get("ids");
        if (StringUtils.isNotBlank(ids)) {
            queryWrapper.in("depot_id", ids);
        }
        Date date = (Date) param.get(("start"));
        if (null != date) {
            queryWrapper.ge("receive_date", DateUtil.getCurZero(date));
        }
        date = (Date) param.get(("end"));
        if (null != date) {
            queryWrapper.le("receive_date", DateUtil.getNextZero(date));
        }
        queryWrapper.orderByDesc("batch_id");
        grainMapper.selectPage(page, queryWrapper);
    }

}