package com.fzzy.work;

import com.bstek.dorado.annotation.DataProvider;
import com.bstek.dorado.data.provider.Page;
import com.fzzy.igds.utils.ContextUtil;
import com.fzzy.work.data.OrderStatus;
import com.fzzy.work.domain.WorkOrder;
import com.fzzy.work.data.WorkOrderParam;
import com.fzzy.work.service.WorkOrderService;
import com.ruoyi.common.core.domain.entity.SysDictData;

import org.springframework.stereotype.Component;

import java.util.List;

import javax.annotation.Resource;

/**
 * @Desc: 工单管理
 * @Author: Andy.jia
 * @Date: 2025/12/19
 */
@Component
public class WorkOrderPR {

    @Resource
    private WorkOrderService workOrderService;


    /**
     *
     * workOrderPR#pageQuery
     * 分页查询
     * @param page
     * @param param
     */
    @DataProvider
    public void pageQuery(Page<WorkOrder> page, WorkOrderParam param) {

        com.baomidou.mybatisplus.extension.plugins.pagination.Page<WorkOrder> corePage = new com.baomidou.mybatisplus.extension.plugins.pagination.Page<>(page.getPageNo(), page.getPageSize());

        if (null == param) param = new WorkOrderParam();
        param.setCompanyId(ContextUtil.getCompanyId());

        workOrderService.pageQuery(corePage, param);

        // 重新封装
        page.setEntities(corePage.getRecords());
        page.setEntityCount(Integer.parseInt(String.valueOf(corePage.getTotal())));
    }

    /**
     *
     * workOrderPR#pageQuery2
     * 已经完成，分页查询
     * @param page
     * @param param
     */
    @DataProvider
    public void pageQuery2(Page<WorkOrder> page, WorkOrderParam param) {

        com.baomidou.mybatisplus.extension.plugins.pagination.Page<WorkOrder> corePage = new com.baomidou.mybatisplus.extension.plugins.pagination.Page<>(page.getPageNo(), page.getPageSize());

        if (null == param) param = new WorkOrderParam();
        param.setStatus(OrderStatus.STATUS_50.getCode());
        param.setCompanyId(ContextUtil.getCompanyId());

        workOrderService.pageQuery(corePage, param);

        // 重新封装
        page.setEntities(corePage.getRecords());
        page.setEntityCount(Integer.parseInt(String.valueOf(corePage.getTotal())));
    }
    
    
    /**
     * 工单状态
     * ${dorado.getDataProvider("workOrderPR#triggerStatus").getResult()}
     *
     * @return
     */
    @DataProvider
    public List<SysDictData> triggerStatus() {
        return workOrderService.triggerStatus();
    }
}