package com.gkhy.hazmat.system.service.impl;

import cn.hutool.core.date.DateField;
import cn.hutool.core.date.DatePattern;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gkhy.hazmat.common.api.CommonPage;
import com.gkhy.hazmat.common.config.IdTableNameHandler;
import com.gkhy.hazmat.common.domain.entity.SysUser;
import com.gkhy.hazmat.common.enums.UserTypeEnum;
import com.gkhy.hazmat.common.exception.ApiException;
import com.gkhy.hazmat.common.utils.PageUtils;
import com.gkhy.hazmat.common.utils.SecurityUtils;
import com.gkhy.hazmat.system.domain.HzWarning;
import com.gkhy.hazmat.system.domain.vo.HzEntryRecordVO;
import com.gkhy.hazmat.system.mapper.HzWarningMapper;
import com.gkhy.hazmat.system.service.HzWarningService;
import org.springframework.stereotype.Service;

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

/**
 * <p>
 * 预警表 服务实现类
 * </p>
 *
 * @author kzy
 * @since 2024-08-05 14:41:40
 */
@Service
public class HzWarningServiceImpl extends ServiceImpl<HzWarningMapper, HzWarning> implements HzWarningService {

    @Override
    public CommonPage selectWarningList(HzWarning warning) {
        SysUser currentUser= SecurityUtils.getLoginUser().getUser();
        if(!currentUser.getUserType().equals(UserTypeEnum.SYSTEM_USER.getCode()) && !currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){
            warning.setCompanyId(currentUser.getCompanyId());
        }
        PageUtils.startPage();
        List<HzWarning> studentList=baseMapper.selectWarningList(warning);

        return CommonPage.restPage(studentList);
    }

    @Override
    public HzWarning selectWarningById(Long warningId) {
        HzWarning warning= baseMapper.selectById(warningId);
        SysUser currentUser=SecurityUtils.getLoginUser().getUser();
        if(currentUser.getUserType().equals(UserTypeEnum.SYSTEM_USER.getCode())){
            return warning;
        }else if(!warning.getCompanyId().equals(currentUser.getCompanyId())){
            throw new ApiException("无权限查看其它企业数据");
        }
        return warning;
    }

    @Override
    public int markWarning(HzWarning warning) {
        if(warning.getId()==null||warning.getCompanyId()==null||warning.getState()==null){
            throw new ApiException("标记预警参数不正确");
        }
        SysUser currentUser=SecurityUtils.getLoginUser().getUser();
        if(!currentUser.getUserType().equals(UserTypeEnum.SYSTEM_USER.getCode())) {
            if (!warning.getCompanyId().equals(currentUser.getCompanyId())) {
                throw new ApiException("无权限操作其它企业数据");
            }
        }
        int row=baseMapper.updateById(new HzWarning().setId(warning.getId()).setState(warning.getState()).setHandleTime(LocalDateTime.now()).setUpdateBy(currentUser.getUsername()));
        if(row<1){
            throw new ApiException("标记预警信息失败");
        }
        return row;
    }

    @Override
    public int deleteWarningById(Long warningId) {
        HzWarning warning=baseMapper.selectById(warningId);
        if(warning==null){
            throw new ApiException("预警信息不存在");
        }
        SysUser currentUser=SecurityUtils.getLoginUser().getUser();
        if(!currentUser.getUserType().equals(UserTypeEnum.SYSTEM_USER.getCode())) {
            if (!warning.getCompanyId().equals(currentUser.getCompanyId())) {
                throw new ApiException("无权限操作其它企业数据");
            }
        }
        int row=baseMapper.deleteById(warningId);
        return row;
    }

    @Override
    public Integer selectWarningCount(Long companyId) {
        SysUser currentUser=SecurityUtils.getLoginUser().getUser();
        if(currentUser.getUserType().equals(UserTypeEnum.SYSTEM_USER.getCode())){
            throw new ApiException("管理员不能操作");
        }
        if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){
            return baseMapper.selectWarningCount(companyId);
        }
        return baseMapper.selectWarningCount(currentUser.getCompanyId());
    }

    @Override
    public List<HzEntryRecordVO> dailywarningCount() {
        // 获取当前时间并清零时分秒
        DateTime now = DateUtil.date().setField(Calendar.HOUR_OF_DAY, 0)
                .setField(Calendar.MINUTE, 0)
                .setField(Calendar.SECOND, 0)
                .setField(Calendar.MILLISECOND, 0);

        // 设置时间范围：30天前00:00:00 到 昨天23:59:59
        DateTime startTime = DateUtil.offsetDay(now, -30);
        DateTime endTime = DateUtil.endOfDay(DateUtil.offsetDay(now, -1));

        // 格式化成数据库参数
        String startDate = DateUtil.format(startTime, DatePattern.NORM_DATETIME_FORMAT);
        String endDate = DateUtil.format(endTime, DatePattern.NORM_DATETIME_FORMAT);

        // 生成30天的日期列表（格式：dd）
        List<HzEntryRecordVO> dailyEntryList = new ArrayList<>();
        DateTime currentDay = startTime;
        while (!currentDay.isAfter(endTime)) {
            String dayStr = DateUtil.format(currentDay, "MM-dd"); // 两位天数
            HzEntryRecordVO vo = new HzEntryRecordVO();
            vo.setDay(dayStr);
            vo.setCount(0);
            dailyEntryList.add(vo);
            currentDay = DateUtil.offsetDay(currentDay, 1);
        }

        // 查询数据库按天统计（需SQL返回dd格式）
        List<HzEntryRecordVO> dbResults = baseMapper.dailyWarningStatic(startDate, endDate);
        IdTableNameHandler.removeCurrentId();

        // 合并结果到初始化列表
        if (!dbResults.isEmpty()) {
            Map<String, HzEntryRecordVO> resultMap = dbResults.stream()
                    .collect(Collectors.toMap(HzEntryRecordVO::getDay, item -> item));
            for (HzEntryRecordVO dailyVO : dailyEntryList) {
                HzEntryRecordVO matched = resultMap.get(dailyVO.getDay());
                if (matched != null) {
                    dailyVO.setCount(matched.getCount());
                }
            }
        }

        return dailyEntryList;
    }
}