package org.dromara.eims.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.RequiredArgsConstructor;
import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.eims.domain.*;
import org.dromara.eims.domain.vo.EimsEquVo;
import org.dromara.eims.mapper.*;
import org.dromara.eims.service.IReportService;
import org.dromara.system.mapper.SysDeptMapper;
import org.springframework.stereotype.Service;

import java.time.*;
import java.util.*;
import java.util.stream.Collectors;

@RequiredArgsConstructor
@Service
public class ReportServiceImpl implements IReportService {

    private static final String UNCHECKED = "未点检";
    private static final String PENDING_VERIFICATION = "待验证";
    private static final String COMPLETED = "已完成";
    private static final String NOT_APPLICABLE = "/";


    private static final String MAINT_0 = "待保养";
    private static final String MAINT_1 = "保养中";
    private static final String MAINT_2 = "待验证";
    private static final String MAINT_3 = "已完成";

    private final EimsInspectRecordMapper baseMapper;
    private final EimsInspectStMapper inspectStMapper;
    private final EimsInspectPlanMapper inspectPlanMapper;
    private final SysDeptMapper sysDeptMapper;
    private final EimsEquMapper equMapper;
    private final EimsMaintOrderMapper maintOrderMapper;
    private final EimsMaintPlanMapper maintPlanMapper;

    @Override
    public TableDataInfo<Map<String, Object>> queryPageListCustom(Map<String, Object> queryParams, PageQuery pageQuery) {
        LambdaQueryWrapper<EimsEqu> equWrapper = Wrappers.lambdaQuery();

        if (queryParams.containsKey("keyword")) {
            String keyword = (String) queryParams.get("keyword");
            equWrapper.and(StringUtils.isNotBlank(keyword),
                wrapper -> wrapper
                    .like(EimsEqu::getEquName, keyword)
                    .or()
                    .like(EimsEqu::getAssetNo, keyword)
            );
        }

        Page<EimsEquVo> equList = equMapper.selectVoPage(pageQuery.build(), equWrapper);
        return TableDataInfo.build(fitInspData(equList, queryParams));
    }

    @Override
    public TableDataInfo<Map<String, Object>> queryMaintList(Map<String, Object> queryParams, PageQuery pageQuery) {
        LambdaQueryWrapper<EimsEqu> equWrapper = Wrappers.lambdaQuery();

        if (queryParams.containsKey("keyword")) {
            String keyword = (String) queryParams.get("keyword");
            equWrapper.and(StringUtils.isNotBlank(keyword),
                wrapper -> wrapper
                    .like(EimsEqu::getEquName, keyword)
                    .or()
                    .like(EimsEqu::getAssetNo, keyword)
            );
        }

        Page<EimsEquVo> equList = equMapper.selectVoPage(pageQuery.build(), equWrapper);
        return TableDataInfo.build(fitMaintData(equList, queryParams));
    }

    private  Page<Map<String, Object>> fitMaintData(Page<EimsEquVo> equPageList, Map<String, Object> queryParams) {
        Integer selectYear = getYearFromParams(queryParams);
        LocalDate firstDay = Year.of(selectYear).atDay(1); // 第一天
        LocalDate lastDay = Year.of(selectYear).atDay(Year.of(selectYear).length());
        Page<Map<String, Object>> page = new Page<>(
            equPageList.getCurrent(),
            equPageList.getSize(),
            equPageList.getTotal(),
            equPageList.searchCount()
        );
        if (equPageList.getRecords().isEmpty()) {
            return page;
        }

        List<Long> equIds = equPageList.getRecords().stream()
            .map(EimsEquVo::getEquId)
            .collect(Collectors.toList());

        Map<Long, List<EimsMaintOrder>> maintOrderMap = getMaintOrderMap(equIds, firstDay, lastDay);
        Map<Long, List<EimsMaintPlan>> maintPlanMap = getMaintPlanMap(equIds,selectYear);

        List<Map<String, Object>> resultList = equPageList.getRecords().stream()
            .map(equ -> buildMaintReport(equ, maintOrderMap, maintPlanMap))
            .collect(Collectors.toList());

        page.setRecords(resultList);

        return page;
    }

    private Page<Map<String, Object>> fitInspData(Page<EimsEquVo> equPageList, Map<String, Object> queryParams) {
        YearMonth yearMonth = getYearMonthFromParams(queryParams);
        LocalDate firstDay = yearMonth.atDay(1);
        LocalDate lastDay = yearMonth.atEndOfMonth();

        Page<Map<String, Object>> page = new Page<>(
            equPageList.getCurrent(),
            equPageList.getSize(),
            equPageList.getTotal(),
            equPageList.searchCount()
        );

        if (equPageList.getRecords().isEmpty()) {
            return page;
        }

        List<Long> equIds = equPageList.getRecords().stream()
            .map(EimsEquVo::getEquId)
            .collect(Collectors.toList());

        Map<Long, List<EimsInspectSt>> inspectStMap = getInspectStMap(equIds, firstDay, lastDay);
        Map<Long, List<EimsInspectPlan>> inspectPlanMap = getInspectPlanMap(equIds, yearMonth);

        List<Map<String, Object>> resultList = equPageList.getRecords().stream()
            .map(equ -> buildEquipmentReport(equ, inspectStMap, inspectPlanMap, yearMonth))
            .collect(Collectors.toList());

        page.setRecords(resultList);
        return page;
    }

    private YearMonth getYearMonthFromParams(Map<String, Object> queryParams) {
        if (queryParams.containsKey("selectMonth")) {
            String[] parts = queryParams.get("selectMonth").toString().split("-");
            return YearMonth.of(Integer.parseInt(parts[0]), Integer.parseInt(parts[1]));
        }
        return YearMonth.now();
    }

    private Map<Long, List<EimsInspectSt>> getInspectStMap(List<Long> equIds, LocalDate firstDay, LocalDate lastDay) {
        LambdaQueryWrapper<EimsInspectSt> sWrapper = Wrappers.lambdaQuery();
        sWrapper.in(EimsInspectSt::getEquId, equIds)
            .between(EimsInspectSt::getPlanTime, firstDay, lastDay)
            .eq(EimsInspectSt::getType, "Day");

        return inspectStMapper.selectList(sWrapper).stream()
            .collect(Collectors.groupingBy(EimsInspectSt::getEquId));
    }

    private Map<Long, List<EimsInspectPlan>> getInspectPlanMap(List<Long> equIds, YearMonth yearMonth) {
        LocalDateTime firstDayTime = yearMonth.atDay(1).atStartOfDay();
        LocalDateTime lastDayTime = yearMonth.atEndOfMonth().atTime(23, 59, 59);

        LambdaQueryWrapper<EimsInspectPlan> pWrapper = Wrappers.lambdaQuery();
        pWrapper.in(EimsInspectPlan::getEquId, equIds)
            .between(EimsInspectPlan::getCreateTime, firstDayTime, lastDayTime);

        return inspectPlanMapper.selectList(pWrapper).stream()
            .collect(Collectors.groupingBy(EimsInspectPlan::getEquId));
    }

    private Map<String, Object> buildEquipmentReport(EimsEquVo equ,
                                                     Map<Long, List<EimsInspectSt>> inspectStMap,
                                                     Map<Long, List<EimsInspectPlan>> inspectPlanMap,
                                                     YearMonth yearMonth) {
        Map<String, Object> item = new HashMap<>();
        item.put("equName", equ.getEquName());
        item.put("assetNo", equ.getAssetNo());

        List<EimsInspectSt> equInspList = inspectStMap.getOrDefault(equ.getEquId(), Collections.emptyList());
        List<EimsInspectPlan> equPlanList = inspectPlanMap.getOrDefault(equ.getEquId(), Collections.emptyList());

        Map<String, EimsInspectSt> stMap = equInspList.stream()
            .collect(Collectors.toMap(
                inspSt -> String.valueOf(inspSt.getPlanTime().toInstant()
                    .atZone(ZoneId.systemDefault())
                    .toLocalDate().getDayOfMonth()),
                inspSt -> inspSt,
                (existing, replacement) -> existing
            ));

        Map<String, EimsInspectPlan> planMap = equPlanList.stream()
            .collect(Collectors.toMap(
                inspPlan -> String.valueOf(inspPlan.getCreateTime().toInstant()
                    .atZone(ZoneId.systemDefault())
                    .toLocalDate().getDayOfMonth()),
                inspPlan -> inspPlan,
                (existing, replacement) -> existing
            ));

        int daysInMonth = yearMonth.lengthOfMonth();
        for (int day = 1; day <= daysInMonth; day++) {
            String dayKey = "day_" + day;
            if (planMap.containsKey(String.valueOf(day))) {
                item.put(dayKey, UNCHECKED);
            } else if (stMap.containsKey(String.valueOf(day))) {
                EimsInspectSt st = stMap.get(String.valueOf(day));
                switch (st.getStatus()) {
                    case "0":
                        item.put(dayKey, UNCHECKED);
                        break;
                    case "1":
                        item.put(dayKey, PENDING_VERIFICATION);
                        break;
                    case "2":
                        item.put(dayKey, COMPLETED);
                        break;
                    default:
                        item.put(dayKey, NOT_APPLICABLE);
                }
            } else {
                item.put(dayKey, NOT_APPLICABLE);
            }
        }

        return item;
    }





    private Integer getYearFromParams(Map<String, Object> queryParams) {
        if (queryParams.containsKey("selectYear")) {
            String year = queryParams.get("selectYear").toString();
            return  Integer.parseInt(year);
        }
        return LocalDate.now().getYear();
    }
    private Map<Long, List<EimsMaintOrder>> getMaintOrderMap(List<Long> equIds, LocalDate firstDay, LocalDate lastDay) {
        LambdaQueryWrapper<EimsMaintOrder> sWrapper = Wrappers.lambdaQuery();
        sWrapper.in(EimsMaintOrder::getEquId, equIds)
            .between(EimsMaintOrder::getPlanTime, firstDay, lastDay);

        return maintOrderMapper.selectList(sWrapper).stream()
            .collect(Collectors.groupingBy(EimsMaintOrder::getEquId));
    }

    private Map<Long, List<EimsMaintPlan>> getMaintPlanMap(List<Long> equIds, Integer year) {
        LocalDateTime firstDay = Year.of(year).atDay(1).atStartOfDay();
        LocalDateTime lastDay = Year.of(year).atDay(Year.of(year).length())
            .atTime(23, 59, 59);

        LambdaQueryWrapper<EimsMaintPlan> pWrapper = Wrappers.lambdaQuery();
        pWrapper.in(EimsMaintPlan::getEquId, equIds)
            .between(EimsMaintPlan::getCreateTime, firstDay, lastDay);

        return maintPlanMapper.selectList(pWrapper).stream()
            .collect(Collectors.groupingBy(EimsMaintPlan::getEquId));
    }

    private Map<String, Object> buildMaintReport(EimsEquVo equ,
                                                     Map<Long, List<EimsMaintOrder>> maintOrderMap,
                                                     Map<Long, List<EimsMaintPlan>> maintPlanMap) {
        Map<String, Object> item = new HashMap<>();
        item.put("equName", equ.getEquName());
        item.put("assetNo", equ.getAssetNo());

        List<EimsMaintOrder> equMaintOrderList = maintOrderMap.getOrDefault(equ.getEquId(), Collections.emptyList());
        List<EimsMaintPlan> equMaintPlanList = maintPlanMap.getOrDefault(equ.getEquId(), Collections.emptyList());


        Map<String, String> orderMap = equMaintOrderList.stream()
            .collect(Collectors.groupingBy(
                order -> String.valueOf(order.getPlanTime().toInstant()
                    .atZone(ZoneId.systemDefault())
                    .toLocalDate().getMonthValue()),
                // 按月份分组后，判断该月是否全部status=3
                Collectors.collectingAndThen(
                    Collectors.toList(),
                    orders -> orders.stream().allMatch(o -> o.getStatus().equals("3"))
                        ? "已完成"
                        : "未完成"
                )
            ));
        Map<String, EimsMaintPlan> planMap = equMaintPlanList.stream()
            .collect(Collectors.toMap(
                maintPlan -> String.valueOf(maintPlan.getCreateTime().toInstant()
                    .atZone(ZoneId.systemDefault())
                    .toLocalDate().getMonthValue()),
                maintPlan -> maintPlan,
                (existing, replacement) -> existing
            ));

        int monthInYear = 12;
        for (int month = 1; month <= monthInYear; month++) {
            String monthKey = "month_" + month;
            if (planMap.containsKey(String.valueOf(month))) {
                item.put(monthKey, UNCHECKED);
            } else if (orderMap.containsKey(String.valueOf(month))) {
                String result  = orderMap.get(String.valueOf(month));
                item.put(monthKey, result);

            } else {
                item.put(monthKey, NOT_APPLICABLE);
            }
        }

        return item;
    }
}