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

import cn.hutool.core.date.DatePattern;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUnit;
import cn.hutool.core.date.DateUtil;
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.support.ExcelTypeEnum;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
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.HazmatPackageEnum;
import com.gkhy.hazmat.common.enums.HazmatStatusEnum;
import com.gkhy.hazmat.common.enums.UserTypeEnum;
import com.gkhy.hazmat.common.excel.HazmatBasicUseExcelData;
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.common.utils.StringUtils;
import com.gkhy.hazmat.system.domain.*;
import com.gkhy.hazmat.system.domain.vo.HzCompanyMessage;
import com.gkhy.hazmat.system.domain.vo.HzEntryRecordVO;
import com.gkhy.hazmat.system.domain.vo.HzHazmatUseVO;
import com.gkhy.hazmat.system.domain.vo.HzHomeDataVO;
import com.gkhy.hazmat.system.mapper.*;
import com.gkhy.hazmat.system.service.HzStatisticService;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.*;
import java.util.stream.Collectors;

@Service
public class HzStatisticServiceImpl implements HzStatisticService {
    @Autowired
    private HzEntryRecordMapper entryRecordMapper;
    @Autowired
    private HzHazmatMapper hazmatMapper;
    @Autowired
    private HzHazmatBasicMapper hazmatBasicMapper;
    @Autowired
    private HzWarehouseMapper warehouseMapper;
    @Autowired
    private HzWarningMapper warningMapper;

    @Override
    public List<HzEntryRecordVO> entryStatic(Long companyId) {
        SysUser currentUser= SecurityUtils.getLoginUser().getUser();
        Date currentDate=new Date();
        DateTime endTime = DateUtil.endOfDay(currentDate);
        DateTime startTime = DateUtil.beginOfDay(DateUtil.offsetDay(currentDate, -29));
        String startDate= DateUtil.format(startTime, DatePattern.NORM_DATETIME_FORMAT);
        String endDate=DateUtil.format(endTime,DatePattern.NORM_DATETIME_FORMAT);
        List<HzEntryRecordVO> dayEntryRecordList=new ArrayList<>();
        while (DateUtil.compare(startTime,endTime)<0){
            String day=String.valueOf(DateUtil.dayOfMonth(startTime));
            HzEntryRecordVO entryRecordVO=new HzEntryRecordVO();
            entryRecordVO.setCount(0);
            entryRecordVO.setDay(day);
            entryRecordVO.setMonth(String.valueOf(DateUtil.month(startTime)+1));
            dayEntryRecordList.add(entryRecordVO);
            startTime=DateUtil.offsetDay(startTime,1);
        }
        //todo
        List<HzEntryRecordVO> entryRecordVOList= entryRecordMapper.entryCountStatic(startDate, endDate,currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId());
        if(!entryRecordVOList.isEmpty()) {
            Map<String, HzEntryRecordVO> resMap = entryRecordVOList.stream().collect(Collectors.toMap(item -> item.getMonth()+"_"+item.getDay(), item -> item));
            for (HzEntryRecordVO er : dayEntryRecordList) {
                HzEntryRecordVO entryRecordVO= resMap.get(er.getMonth()+"_"+er.getDay());
                if (entryRecordVO == null) {
                    continue;
                }
                er.setCount(entryRecordVO.getCount());
            }
        }
        return dayEntryRecordList;
    }

    @Override
    public List<HzEntryRecordVO> useEverydayStatic(Long companyId) {
        SysUser currentUser= SecurityUtils.getLoginUser().getUser();
        checkUserAllowed(currentUser);
        Date currentDate=new Date();
        DateTime endTime = DateUtil.endOfDay(currentDate);
        DateTime startTime = DateUtil.beginOfDay(DateUtil.offsetDay(currentDate, -6));
        String startDate= DateUtil.format(startTime, DatePattern.NORM_DATETIME_FORMAT);
        String endDate=DateUtil.format(endTime,DatePattern.NORM_DATETIME_FORMAT);
        List<HzEntryRecordVO> dayEntryRecordList=new ArrayList<>();
        while (DateUtil.compare(startTime,endTime)<0){
            String day=String.valueOf(DateUtil.dayOfMonth(startTime));
            HzEntryRecordVO entryRecordVO=new HzEntryRecordVO();
            entryRecordVO.setCount(0);
            entryRecordVO.setDay(day);
            dayEntryRecordList.add(entryRecordVO);
            startTime=DateUtil.offsetDay(startTime,1);
        }
        if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){
            IdTableNameHandler.setCurrentId(companyId);
        }else {
            //设置分表id
            //todo
            IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
        }
        List<HzEntryRecordVO> entryRecordVOList= hazmatMapper.useCountEverydayStatic(startDate, endDate,currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId());
        IdTableNameHandler.removeCurrentId();
        if(!entryRecordVOList.isEmpty()) {
            Map<String, HzEntryRecordVO> resMap = entryRecordVOList.stream().collect(Collectors.toMap(item -> item.getDay(), item -> item));
            for (HzEntryRecordVO er : dayEntryRecordList) {
                if (resMap.get(er.getDay()) == null) {
                    continue;
                }
                er.setCount(resMap.get(er.getDay()).getCount());
            }
        }
        return dayEntryRecordList;
    }

    @Override
    public List<HzHazmatUseVO> maxUseStatic(Long companyId) {
        SysUser currentUser= SecurityUtils.getLoginUser().getUser();
        checkUserAllowed(currentUser);

        Date currentDate=new Date();
        DateTime endTime = DateUtil.endOfDay(currentDate);
        DateTime startTime = DateUtil.beginOfDay(DateUtil.offsetDay(currentDate, -6));
        String startDate= DateUtil.format(startTime, DatePattern.NORM_DATETIME_FORMAT);
        String endDate=DateUtil.format(endTime,DatePattern.NORM_DATETIME_FORMAT);
        if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){
            IdTableNameHandler.setCurrentId(companyId);
        }else {
            //设置分表id
            //todo
            IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
        }
        List<HzHazmatUseVO> hazmatUseVOList= hazmatMapper.maxUseCountStatic(startDate, endDate,currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId());
        IdTableNameHandler.removeCurrentId();
        if(!hazmatUseVOList.isEmpty()) {
            List<Long> basicIds = hazmatUseVOList.stream().map(HzHazmatUseVO::getBasicId).collect(Collectors.toList());
            List<HzHazmatBasic> hzHazmatBasicList = hazmatBasicMapper.selectHazmatBasicListByIds(basicIds);
            Map<Long, HzHazmatBasic> hazmatBasicMap = hzHazmatBasicList.stream().collect(Collectors.toMap(HzHazmatBasic::getId, item -> item));
            for (HzHazmatUseVO hu : hazmatUseVOList) {
                HzHazmatBasic hazmatBasic = hazmatBasicMap.get(hu.getBasicId());
                if (hazmatBasic != null) {
                    hu.setName(hazmatBasic.getName());
                }
            }
        }
        return hazmatUseVOList;
    }

    @Override
    public HzHomeDataVO homeDataStatistic(Long companyid) {
        SysUser currentUser=SecurityUtils.getLoginUser().getUser();
        checkUserAllowed(currentUser);
        Long companyId= null;
        if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){
            companyId = companyid;
        }else {
           companyId = currentUser.getCompanyId();
        }
        Integer warehouseCount= Math.toIntExact(warehouseMapper.selectCount(Wrappers.<HzWarehouse>lambdaQuery()
                .eq(true, HzWarehouse::getDelFlag, 0)
                .eq(companyId!=null,HzWarehouse::getCompanyId,companyId)));
        List<Integer> stateList=new ArrayList<>();
        stateList.add(HazmatStatusEnum.USED.getCode());
        stateList.add(HazmatStatusEnum.DISCARD.getCode());
        //设置分表id
        //todo
        IdTableNameHandler.setCurrentId(companyId);
        Integer hazmatCount= Math.toIntExact(hazmatMapper.selectCount(Wrappers.<HzHazmat>lambdaQuery()
                .notIn(true, HzHazmat::getState, stateList)
                .eq(companyId!=null,HzHazmat::getCompanyId,companyId)));
        Integer usedCount= Math.toIntExact(hazmatMapper.selectCount(Wrappers.<HzHazmat>lambdaQuery()
                .eq(true, HzHazmat::getState, HazmatStatusEnum.USED.getCode())
                .eq(companyId!=null,HzHazmat::getCompanyId,companyId)));
        IdTableNameHandler.removeCurrentId();

        Integer warningCount= Math.toIntExact(warningMapper.selectCount(Wrappers.<HzWarning>lambdaQuery()
                .eq(true, HzWarning::getState,0)
                .eq(companyId!=null,HzWarning::getCompanyId,companyId)));
        HzHomeDataVO homeDataVO=new HzHomeDataVO();
        homeDataVO.setWarehouseCount(warehouseCount);
        homeDataVO.setHazmatCount(hazmatCount);
        homeDataVO.setUsedCount(usedCount);
        homeDataVO.setWarningCount(warningCount);
        return homeDataVO;
    }


    @Override
    public CommonPage useStatic(String startTime, String endTime,Long companyId) {
        SysUser currentUser= SecurityUtils.getLoginUser().getUser();
        checkUserAllowed(currentUser);

        String startDate="";
        String endDate="";
        if(StringUtils.isBlank(startTime)&&StringUtils.isBlank(endTime)){
            Date currentDate=new Date();
            startDate= DateUtil.format(DateUtil.beginOfDay(DateUtil.offsetDay(DateUtil.endOfDay(currentDate), -6)), DatePattern.NORM_DATETIME_FORMAT);
            endDate=DateUtil.format(DateUtil.endOfDay(currentDate),DatePattern.NORM_DATETIME_FORMAT);
        } else if (StringUtils.isBlank(startTime)&&StringUtils.isNotBlank(endTime)) {
            startDate= DateUtil.format(DateUtil.beginOfDay(DateUtil.offsetDay(DateUtil.endOfDay(new Date(endTime)), -6)), DatePattern.NORM_DATETIME_FORMAT);
            endDate=endTime;
        }else if (StringUtils.isNotBlank(startTime)&&StringUtils.isBlank(endTime)){
            startDate=startTime;
            endDate=DateUtil.format(DateUtil.beginOfDay(DateUtil.offsetDay(DateUtil.endOfDay(new Date(endTime)), 6)), DatePattern.NORM_DATETIME_FORMAT);
        }else{
            startDate=startTime;
            endDate=endTime;
            if(DateUtil.between(DateUtil.parse(startDate),DateUtil.parse(startTime), DateUnit.DAY)>90){
                throw new ApiException("只能查询90天以内数据");
            }
        }
        if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){
            IdTableNameHandler.setCurrentId(companyId);

        }else {
            //设置分表id
            //todo
            IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
        }
        PageUtils.startPage();
        List<HzHazmatUseVO> hazmatUseVOList= hazmatMapper.useCountStatic(startDate, endDate,currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId());
        IdTableNameHandler.removeCurrentId();
        if(!hazmatUseVOList.isEmpty()) {
            List<Long> basicIds = hazmatUseVOList.stream().map(HzHazmatUseVO::getBasicId).collect(Collectors.toList());
            List<HzHazmatBasic> hzHazmatBasicList = hazmatBasicMapper.selectHazmatBasicListByIds(basicIds);
            Map<Long, HzHazmatBasic> hazmatBasicMap = hzHazmatBasicList.stream().collect(Collectors.toMap(HzHazmatBasic::getId, item -> item));
            for (HzHazmatUseVO hu : hazmatUseVOList) {
                HzHazmatBasic hazmatBasic = hazmatBasicMap.get(hu.getBasicId());
                if (hazmatBasic != null) {
                    hu.setName(hazmatBasic.getName());
                    hu.setHazmatBasic(hazmatBasic);
                }
            }
        }
        return CommonPage.restPage(hazmatUseVOList);
    }

    @Override
    public void importBaiscUse(HttpServletResponse response, String startTime, String endTime) throws IOException {
        SysUser currentUser= SecurityUtils.getLoginUser().getUser();
        checkUserAllowed(currentUser);
        OutputStream outputStream = response.getOutputStream();
        try {
            String startDate="";
            String endDate="";
            if(StringUtils.isBlank(startTime)&&StringUtils.isBlank(endTime)){
                Date currentDate=new Date();
                startDate= DateUtil.format(DateUtil.beginOfDay(DateUtil.offsetDay(DateUtil.endOfDay(currentDate), -29)), DatePattern.NORM_DATETIME_FORMAT);
                endDate=DateUtil.format(DateUtil.endOfDay(currentDate),DatePattern.NORM_DATETIME_FORMAT);
            } else if (StringUtils.isBlank(startTime)&&StringUtils.isNotBlank(endTime)) {
                startDate= DateUtil.format(DateUtil.beginOfDay(DateUtil.offsetDay(DateUtil.endOfDay(new Date(endTime)), -29)), DatePattern.NORM_DATETIME_FORMAT);
                endDate=endTime;
            }else if (StringUtils.isNotBlank(startTime)&&StringUtils.isBlank(endTime)){
                startDate=startTime;
                endDate=DateUtil.format(DateUtil.beginOfDay(DateUtil.offsetDay(DateUtil.endOfDay(new Date(endTime)), 29)), DatePattern.NORM_DATETIME_FORMAT);
            }else{
                startDate=startTime;
                endDate=endTime;
                if(DateUtil.between(DateUtil.parse(startDate),DateUtil.parse(startTime), DateUnit.DAY)>90){
                    throw new ApiException("只能查询90天以内数据");
                }
            }
            //设置分表id
            IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
//            PageUtils.startPage();
            List<HzHazmatUseVO> hazmatUseVOList= hazmatMapper.useCountStatic(startDate, endDate,currentUser.getCompanyId());
            IdTableNameHandler.removeCurrentId();
            List<HazmatBasicUseExcelData> hazmatBasicUseExcelDatas = new ArrayList<>();
            if(!hazmatUseVOList.isEmpty()) {
                List<Long> basicIds = hazmatUseVOList.stream().map(HzHazmatUseVO::getBasicId).collect(Collectors.toList());
                List<HzHazmatBasic> hzHazmatBasicList = hazmatBasicMapper.selectHazmatBasicListByIds(basicIds);
                Map<Long, HzHazmatBasic> hazmatBasicMap = hzHazmatBasicList.stream().collect(Collectors.toMap(HzHazmatBasic::getId, item -> item));
                Integer index=1;
                for (HzHazmatUseVO hu : hazmatUseVOList) {
                    HazmatBasicUseExcelData hazmatBasicUseExcelData = new HazmatBasicUseExcelData();
                    hazmatBasicUseExcelData.setCount(hu.getCount());
                    HzHazmatBasic hazmatBasic = hazmatBasicMap.get(hu.getBasicId());
                    if (hazmatBasic != null) {
                        BeanUtils.copyProperties(hazmatBasic,hazmatBasicUseExcelData);
                        hazmatBasicUseExcelData.setIndex(index);
                        hazmatBasicUseExcelData.setProductType(hazmatBasic.getHazmatType());
                        hazmatBasicUseExcelData.setProductCharacter(hazmatBasic.getHazmatCharacter());
                        hazmatBasicUseExcelData.setProductFormat(hazmatBasic.getHazmatFormat());
                        hazmatBasicUseExcelData.setMinPackage(HazmatPackageEnum.getInfoByCode(hazmatBasic.getMinPackage()));
                    }
                    index++;
                    hazmatBasicUseExcelDatas.add(hazmatBasicUseExcelData);
                }
            }
            Date date = new Date();
            long time = date.getTime();
            String fileName = "危化品用量统计_"+time;
            this.setExcelResponseProp(response,fileName);

            //使用 EasyExcel 导出数据
            EasyExcel.write(response.getOutputStream(), HazmatBasicUseExcelData.class)
                    .excelType(ExcelTypeEnum.XLSX)
                    .sheet("危化品使用记录统计表") // 设置 Sheet 名称
                    .doWrite(hazmatBasicUseExcelDatas); // 写入数据

        } catch (IOException e) {
            throw new RuntimeException(e);
        }



    }


    @Override
    public List<HzEntryRecordVO> dayUseStatistic(Long companyId) {
        SysUser currentUser = SecurityUtils.getLoginUser().getUser();
        checkUserAllowed(currentUser);
        Date currentDate = new Date();

        // 设置当天的8点作为开始时间，20点作为结束时间
        DateTime beginOfDay = DateUtil.beginOfDay(currentDate); // 当天零点

        // 从零点偏移8小时得到8点，偏移20小时得到20点
        DateTime startTime = DateUtil.offsetHour(beginOfDay, 8); // 08:00:00
        DateTime endTime = DateUtil.offsetHour(beginOfDay, 21); // 20:00:00

        String startDate = DateUtil.format(startTime, DatePattern.NORM_DATETIME_FORMAT);
        String endDate = DateUtil.format(endTime, DatePattern.NORM_DATETIME_FORMAT);

        // 设置分表ID逻辑
        if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())) {
        IdTableNameHandler.setCurrentId(companyId);
        } else {
            IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
        }

        // 生成每小时的统计列表，初始count为0
        List<HzEntryRecordVO> hourEntryList = new ArrayList<>();
        DateTime currentHour = startTime;
        Integer i = hazmatMapper.countTotal(currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId());
        while (currentHour.isBefore(endTime)) { // 从8点开始到20点前，共12个小时
            String hour = DateUtil.format(currentHour, "HH"); // 两位小时格式，如08,09,...,19
            HzEntryRecordVO entryRecordVO = new HzEntryRecordVO();
            entryRecordVO.setHour(hour);
            entryRecordVO.setCount(0);
            entryRecordVO.setTotalCount(i);
            hourEntryList.add(entryRecordVO);
            currentHour = DateUtil.offsetHour(currentHour, 1); // 增加一小时
        }
        // 查询数据库，按小时统计使用量
        List<HzEntryRecordVO> entryRecordVOList = hazmatMapper.useCountHourlyStatic(startDate, endDate, currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId()
        );
        hourEntryList.get(0).setTotalCount(i);
        IdTableNameHandler.removeCurrentId();
        // 将查询结果合并到初始化的每小时列表中
        if (!entryRecordVOList.isEmpty()) {
            Map<String, HzEntryRecordVO> resMap = entryRecordVOList.stream()
                    .collect(Collectors.toMap(HzEntryRecordVO::getHour, item -> item));
            for (HzEntryRecordVO hourEntry : hourEntryList) {
                HzEntryRecordVO matched = resMap.get(hourEntry.getHour());
                if (matched != null) {
                    hourEntry.setCount(matched.getCount());
                }
            }
        }
        return hourEntryList;
    }

    @Override
    public List<HzCompanyMessage> companyMessage(Long type) {
        return hazmatMapper.companyMessage(type);
    }


    private void setExcelResponseProp(HttpServletResponse response, String rawFileName) throws UnsupportedEncodingException {
        //设置内容类型
        response.setContentType("application/vnd.vnd.ms-excel");
        //设置编码格式
        response.setCharacterEncoding("utf-8");
        //设置导出文件名称（避免乱码）
        String fileName = URLEncoder.encode(rawFileName.concat(".xlsx"), "UTF-8");
        // 设置响应头
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName);
    }

    public void checkUserAllowed(SysUser user) {
        if (user.getUserType().equals(UserTypeEnum.SYSTEM_USER.getCode())) {
            throw new ApiException("管理员不能操作");
        }
    }
}