package com.qianwen.smartman.modules.smis.mapper;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import java.util.List;
import java.util.Set;
import org.apache.ibatis.annotations.Param;
import com.qianwen.core.datascope.annotation.DataAuth;
import com.qianwen.smartman.modules.smis.dto.WorkstationRealTimeStatusDTO;
import com.qianwen.smartman.modules.smis.dto.WorkstationWcsDmpDTO;
import com.qianwen.smartman.modules.smis.entity.ProductionCalendarDay;
import com.qianwen.smartman.modules.smis.entity.Workstation;
import com.qianwen.smartman.modules.smis.vo.DmpVariablesVO;
import com.qianwen.smartman.modules.smis.vo.WorkMachineEasyVO;
import com.qianwen.smartman.modules.smis.vo.WorkstationRealTimeStatusVO;
import com.qianwen.smartman.modules.smis.vo.WorkstationVO;
import com.qianwen.smartman.modules.mdc.dto.GroupWorkDTO;
import com.qianwen.smartman.modules.mdc.dto.WorkstationDmpDTO;
import com.qianwen.smartman.modules.report.dto.WorkstationDataDTO;

public interface WorkstationMapper extends BaseMapper<Workstation> {
    List<DmpVariablesVO> queryDmpList(@Param("workstationId") String workstationId);

    /**
     * 工位分页查询
     * @param groupIdList
     * @param page 分页信息
     * @param keyWord 关键字
     * @param groupId 分组id
     * @param allWorkstationId
     * @param status 状态
     * @param type 工位类型
     * @param year
     * @return 分页数据
     */
    @DataAuth(code = "workstation")
    List<WorkstationVO> listPage(List<String> groupIdList, IPage<WorkstationVO> page, String keyWord, Long groupId, Long allWorkstationId, @Param("status")Integer status,@Param("type") Integer type, Integer year);

    @DataAuth(code = "workstation")
    <P extends IPage<Workstation>> P selectPage(P page, @Param("ew") Wrapper<Workstation> queryWrapper);

    @DataAuth(code = "workstation")
    List<Workstation> selectList(@Param("ew") Wrapper<Workstation> queryWrapper);

    List<WorkstationVO> listWorkstation(@Param("groupIds") List<String> groupIds);

    @DataAuth(code = "workstation")
    List<WorkstationRealTimeStatusVO> getWorkstationListByWorkStationGroupIdList(IPage<WorkstationRealTimeStatusVO> page, @Param("dto") WorkstationRealTimeStatusDTO dto);

    @DataAuth(code = "workstation")
    IPage<Workstation> listWorkStationOnShiftIndex(IPage<Workstation> page, @Param("year") Integer year, @Param("queryDate") String queryDate, @Param("shift") Integer shift, @Param("type") Integer type, @Param("ids") List<String> ids);

    Integer countWorkStationOnShiftIndex(@Param("year") Integer year, @Param("queryDate") String queryDate, @Param("shift") Integer shift, @Param("type") Integer type, @Param("ids") List<String> ids);

    List<Workstation> listWorkStationByGroupId(@Param("ids") List<String> ids);

    Set<Long> findByIndexAndDateAndTypeAndYear(@Param("year") Integer year, @Param("queryDate") String queryDate, @Param("shift") Integer shift, @Param("type") Integer type);

    List<Long> findByMachineShortCode(@Param("code") String code);

    List<WorkstationWcsDmpDTO> queryWcsList(@Param("workstationId") String workstationId);

    WorkstationWcsDmpDTO queryDateTypeState(@Param("code") Integer code, @Param("workstationId") String workstationId);

    List<GroupWorkDTO> queryGroupWorkStation(List<Long> ids);

    List<Workstation> listWorkStationOnShift(@Param("year") Integer year, @Param("queryDate") String queryDate, @Param("shifts") List<Integer> shifts, @Param("type") Integer type, @Param("ids") List<Long> ids);

    IPage<WorkstationVO> queryWorkstationAndGroup(IPage<WorkstationVO> page, @Param("workstationIds") List<Long> workstationIds);

    ProductionCalendarDay getDayShift(@Param("workstationId") Long workstationId, @Param("calendarDate") String calendarDate);

    IPage<WorkstationDataDTO> getWorkstationData(IPage<WorkstationDataDTO> page, @Param("keyword") String keyword, @Param("workstationIds") List<String> workstationIds);

    IPage<WorkstationDataDTO> getWorkstationDataWithIndexName(IPage<WorkstationDataDTO> page, @Param("keyword") String keyword, @Param("indexName") String indexName, @Param("startTime") String startTime, @Param("endTime") String endTime, @Param("workstationIds") List<String> workstationIds);

    List<WorkstationDmpDTO> getDmpVar(@Param("workstationId") String workstationId);

    WorkMachineEasyVO getWorkMachine(@Param("workstationId") String workstationId);

    @DataAuth(code = "workstation")
    List<Workstation> listMachine(@Param("status") Integer status, @Param("keyword") String keyword);

    List<WorkstationVO> getWorkstationListInMaintainOrRepairById(@Param("orderProcessId") Long orderProcessId, @Param("workstationIdList") Set<Long> workstationIdList);

    List<WorkstationVO> getWorkstationListInMaintainOrRepairByIdList(@Param("orderProcessIdList") List<Long> orderProcessIdList, @Param("workstationIdList") Set<Long> workstationIdList);

    @DataAuth(code = "workstation")
    List<Workstation> listWorkstationDataByGroup(@Param("groupId") String groupId);
}