package com.sundot.airport.item.service.impl;

import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.sundot.airport.common.core.domain.entity.SysDept;
import com.sundot.airport.common.core.domain.model.LoginUser;
import com.sundot.airport.common.utils.SecurityUtils;
import com.sundot.airport.item.domain.SimpleDutySchedule;
import com.sundot.airport.item.mapper.SimpleDutyScheduleMapper;
import com.sundot.airport.item.service.ISimpleDutyScheduleService;
import com.sundot.airport.item.utils.UnifiedDutyScheduler;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Date;
import java.util.List;

/**
 * 值班安排 服务层实现
 *
 * @author wangxx
 * @date 2026-01-14
 */
@Service
public class SimpleDutyScheduleServiceImpl implements ISimpleDutyScheduleService {
    @Autowired
    private SimpleDutyScheduleMapper simpleDutyScheduleMapper;

    /**
     * 查询值班安排列表
     *
     * @param simpleDutySchedule 值班安排信息
     * @return 值班安排集合
     */
    @Override
    public List<SimpleDutySchedule> selectSimpleDutyScheduleList(SimpleDutySchedule simpleDutySchedule) {
        return simpleDutyScheduleMapper.selectSimpleDutyScheduleList(simpleDutySchedule);
    }

    /**
     * 根据日期查询值班安排
     *
     * @param dutyDate 值班日期
     * @return 值班安排
     */
    @Override
    public SimpleDutySchedule selectSimpleDutyScheduleByDate(Date dutyDate) {
        return simpleDutyScheduleMapper.selectSimpleDutyScheduleByDate(dutyDate);
    }


    /**
     * 新增值班安排
     *
     * @param simpleDutySchedule 值班安排信息
     * @return 结果
     */
    @Override
    public int insertSimpleDutySchedule(SimpleDutySchedule simpleDutySchedule) {
        return simpleDutyScheduleMapper.insertSimpleDutySchedule(simpleDutySchedule);
    }

    /**
     * 修改值班安排
     *
     * @param simpleDutySchedule 值班安排信息
     * @return 结果
     */
    @Override
    public int updateSimpleDutySchedule(SimpleDutySchedule simpleDutySchedule) {
        return simpleDutyScheduleMapper.updateSimpleDutySchedule(simpleDutySchedule);
    }

    /**
     * 批量删除值班安排对象
     *
     * @param scheduleIds 需要删除的数据ID
     * @return 结果
     */
    @Override
    public int deleteSimpleDutyScheduleByIds(Long[] scheduleIds) {
        return simpleDutyScheduleMapper.deleteSimpleDutyScheduleByScheduleIds(scheduleIds);
    }

    /**
     * 批量新增值班安排信息
     *
     * @param simpleDutyScheduleList 值班安排列表
     * @return 结果
     */
    @Override
    public int batchInsertSimpleDutySchedule(List<SimpleDutySchedule> simpleDutyScheduleList) {
        return simpleDutyScheduleMapper.batchInsertSimpleDutySchedule(simpleDutyScheduleList);
    }

    /**
     * 插入或更新单条值班安排（先尝试更新，如果不存在则插入）
     *
     * @param simpleDutySchedule 值班安排信息
     * @return 结果
     */
    @Override
    public int insertOrUpdateSimpleDutySchedule(SimpleDutySchedule simpleDutySchedule) {
        // 首先检查是否存在符合条件的记录
        SimpleDutySchedule existingRecord = simpleDutyScheduleMapper.selectSimpleDutyScheduleByStationAndDate(simpleDutySchedule);

        if (existingRecord != null) {
            // 如果存在，则更新现有记录
            simpleDutySchedule.setId(existingRecord.getId());
            return simpleDutyScheduleMapper.updateSimpleDutySchedule(simpleDutySchedule);
        } else {
            // 如果不存在，则插入新记录
            return simpleDutyScheduleMapper.insertSimpleDutySchedule(simpleDutySchedule);
        }
    }


    /**
     * 批量插入或更新值班安排（对列表中的每条记录执行插入或更新操作）
     *
     * @param simpleDutyScheduleList 值班安排列表
     * @return 成功处理的记录总数
     */
    @Override
    public int batchInsertOrUpdateSimpleDutySchedule(List<SimpleDutySchedule> simpleDutyScheduleList) {
        int successCount = 0;
        LoginUser loginUser = SecurityUtils.getLoginUser();
        for (SimpleDutySchedule schedule : simpleDutyScheduleList) {
            //当前登录人
            schedule.setCreateBy(ObjectUtils.isEmpty(loginUser) ? "admin" : loginUser.getUsername());
            schedule.setCreateTime(new Date());
            int result = insertOrUpdateSimpleDutySchedule(schedule);
            if (result > 0) {
                successCount++;
            }
        }
        return successCount;
    }

    /**
     * 生成统一排班计划（支持单班、两班倒、多班倒等模式）
     *
     * @param startDate         起始日期
     * @param startDepartment   起始部门对象，包含部门ID和名称
     * @param departments       部门信息列表，包含部门ID和名称
     * @param shifts            班次安排数组，每个元素包含[startHour, endHour]
     * @param daysToGenerate    生成天数
     * @param stationDepartment 站点部门对象，包含站点ID和名称
     * @return SimpleDutySchedule列表
     */
    @Override
    public List<SimpleDutySchedule> generateUnifiedSchedule(Date startDate, SysDept startDepartment, List<SysDept> departments, int[][] shifts, int daysToGenerate, SysDept stationDepartment) {
        return UnifiedDutyScheduler.generateUnifiedSchedule(startDate, startDepartment, departments, shifts, daysToGenerate, stationDepartment);
    }
}