package com.zc.business.controller;

import java.io.IOException;
import java.util.*;
import java.util.stream.Collectors;
import javax.annotation.Resource;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;

import cn.hutool.core.date.DateUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.zc.business.domain.DcDevice;
import com.zc.business.enums.UniversalEnum;
import com.zc.common.core.httpclient.exception.HttpException;
import io.swagger.v3.oas.annotations.Parameter;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.enums.BusinessType;
import com.zc.business.domain.DcTrafficSurveyData;
import com.zc.business.service.IDcTrafficSurveyDataService;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.common.core.page.TableDataInfo;

/**
 * 一类交调数据Controller
 * 
 * @author liuwenge
 * @date 2024-10-29
 */
@RestController
@RequestMapping("/trafficSurveyData/dcTrafficSurveyData")
public class DcTrafficSurveyDataController extends BaseController
{
    @Autowired
    private IDcTrafficSurveyDataService dcTrafficSurveyDataService;

    @Resource
    private DcDeviceController dcDeviceController;

    @Scheduled(cron = "0 5 * * * ?")
    public void syncTrafficSectionData() throws HttpException, IOException {

        HashMap<String, Object> props = new HashMap<>();
        // 设置查询条件的键为“timestamp$BTW”，表示时间戳在一定范围内
        props.put("terms[0].column", "timestamp$BTW");
        ArrayList<String> dateList = new ArrayList<>();
        // 添加当前前一小时的开始和结束时间到列表，用于设定时间范围
        Date now = new Date();
        // 计算上一个小时的时间
        Date lastHourStart = DateUtil.beginOfHour(DateUtil.offsetHour(now, -1));
        Date lastHourEnd = DateUtil.endOfHour(DateUtil.offsetHour(now, -1));
        // 将上一个小时的开始和结束时间添加到列表
        dateList.add(DateUtil.format(lastHourStart, "yyyy-MM-dd HH:mm:ss"));
        dateList.add(DateUtil.format(lastHourEnd, "yyyy-MM-dd HH:mm:ss"));
        // 将日期列表以逗号分隔并设置为查询条件的值
        props.put("terms[0].value", String.join(UniversalEnum.COMMA.getValue(), dateList));
        props.put("paging", false);
        props.put("sorts[0].order", "asc");
        props.put("sorts[0].name", "timestamp");


        List<DcDevice> deviceList = dcTrafficSurveyDataService.selectDeviceList();
        String propertyId = "01"; //功能码

        List<DcTrafficSurveyData> batchData = new ArrayList<>();
        for (DcDevice dcDevice : deviceList) {

            Object data = JSON.parseObject(dcDeviceController.queryDeviceProperties(dcDevice.getIotDeviceId(), propertyId, props).get("data").toString()).get("data");
            JSONArray dataArray = JSON.parseArray(data.toString());

            Integer hezeTotal = 0;
            Integer jinanTotal = 0;
            for (Object o : dataArray) {
                JSONObject jsonObject = JSON.parseObject(o.toString());
                JSONObject formatValue = JSON.parseObject(jsonObject.get("formatValue").toString());
                hezeTotal += Integer.parseInt(formatValue.get("1").toString());
                jinanTotal += Integer.parseInt(formatValue.get("3").toString());
            }

            //菏泽方向数据
            DcTrafficSurveyData hezeData = new DcTrafficSurveyData();
            hezeData.setIotDeviceId(dcDevice.getIotDeviceId());
            hezeData.setStakeMark(dcDevice.getStakeMark());
            hezeData.setDirection("1");
            hezeData.setTimestamp(lastHourStart);
            hezeData.setTrafficVolume(Long.valueOf(hezeTotal));
            batchData.add(hezeData);
            //济南方向数据
            DcTrafficSurveyData jinanData = new DcTrafficSurveyData();
            jinanData.setIotDeviceId(dcDevice.getIotDeviceId());
            jinanData.setStakeMark(dcDevice.getStakeMark());
            jinanData.setDirection("3");
            jinanData.setTimestamp(lastHourStart);
            jinanData.setTrafficVolume(Long.valueOf(jinanTotal));
            batchData.add(jinanData);

        }

        dcTrafficSurveyDataService.batchInsert(batchData);

    }

    /**
     * 查询一类交调数据列表
     */
    @GetMapping("/list")
    public AjaxResult list(DcTrafficSurveyData dcTrafficSurveyData)
    {
        return dcTrafficSurveyDataService.selectDcTrafficSurveyDataList(dcTrafficSurveyData);
    }

    /**
     * 导出一类交调数据列表
     */
    @Log(title = "一类交调数据", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    public void export(HttpServletResponse response, DcTrafficSurveyData dcTrafficSurveyData) throws IOException {
        AjaxResult ajaxResult = dcTrafficSurveyDataService.selectDcTrafficSurveyDataList(dcTrafficSurveyData);
        if (ajaxResult.get("code").equals(UniversalEnum.TWO_HUNDRED.getNumber())) {
            Map<String,Object> data = (Map<String, Object>) ajaxResult.get("data");
            List<Map<String,String>> columnList = (List<Map<String,String>>) data.get("columnList");
            List<Map<String,Object>> rowList = (List<Map<String,Object>>) data.get("rowList");

            XSSFWorkbook workbook = new XSSFWorkbook(); // 创建工作簿
            Sheet sheet = workbook.createSheet("一类交调站"); // 创建工作表

            // 创建数据行样式
            CellStyle dataStyle = workbook.createCellStyle();
            dataStyle.setAlignment(HorizontalAlignment.CENTER);
            dataStyle.setVerticalAlignment(VerticalAlignment.CENTER);
            dataStyle.setBorderRight(BorderStyle.THIN);
            dataStyle.setRightBorderColor(IndexedColors.GREY_50_PERCENT.getIndex());
            dataStyle.setBorderLeft(BorderStyle.THIN);
            dataStyle.setLeftBorderColor(IndexedColors.GREY_50_PERCENT.getIndex());
            dataStyle.setBorderTop(BorderStyle.THIN);
            dataStyle.setTopBorderColor(IndexedColors.GREY_50_PERCENT.getIndex());
            dataStyle.setBorderBottom(BorderStyle.THIN);
            dataStyle.setBottomBorderColor(IndexedColors.GREY_50_PERCENT.getIndex());
            Font dataFont = workbook.createFont();
            dataFont.setFontName(UniversalEnum.ARIAL.getValue());
            dataFont.setFontHeightInPoints((short) UniversalEnum.TEN.getNumber());
            dataStyle.setFont(dataFont);

            // 创建表头样式
            CellStyle headerStyle = workbook.createCellStyle();
            headerStyle.cloneStyleFrom(dataStyle);
            headerStyle.setAlignment(HorizontalAlignment.CENTER);
            headerStyle.setVerticalAlignment(VerticalAlignment.CENTER);
            headerStyle.setFillForegroundColor(IndexedColors.GREY_50_PERCENT.getIndex());
            headerStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
            Font headerFont = workbook.createFont();
            headerFont.setFontName(UniversalEnum.ARIAL.getValue());
            headerFont.setFontHeightInPoints((short) UniversalEnum.TEN.getNumber());
            headerFont.setBold(true);
            headerFont.setColor(IndexedColors.WHITE.getIndex());
            headerStyle.setFont(headerFont);

            // 添加表头
            Row row = sheet.createRow(UniversalEnum.ZERO.getNumber());
            Cell cell = row.createCell(UniversalEnum.ZERO.getNumber());
            cell.setCellValue("设备名称");
            cell.setCellStyle(headerStyle);
            cell = row.createCell(UniversalEnum.ONE.getNumber());
            cell.setCellValue("方向");
            cell.setCellStyle(headerStyle);
            int i;
            for (i = 0; i < columnList.size(); i++) {
                cell = row.createCell(i + 2);
                cell.setCellValue(columnList.get(i).get("label").toString());
                cell.setCellStyle(headerStyle);;
            }
            cell = row.createCell(i + 2);
            cell.setCellValue("合计");
            cell.setCellStyle(headerStyle);


            for (int j = 0; j < rowList.size(); j++) {
                Row subHeaderRow = sheet.createRow(j+1);
                cell = subHeaderRow.createCell(UniversalEnum.ZERO.getNumber());
                cell.setCellValue("一类交调站"+rowList.get(j).get("stakeMark").toString());
                cell.setCellStyle(dataStyle);
                cell = subHeaderRow.createCell(UniversalEnum.ONE.getNumber());
                cell.setCellValue(rowList.get(j).get("direction").toString().equals("1") ? "济南方向" : "菏泽方向");
                cell.setCellStyle(dataStyle);
                int k = 0;
                for (k = 0; k < columnList.size(); k++) {
                    cell = subHeaderRow.createCell(k + 2);
                    cell.setCellValue(rowList.get(j).get(columnList.get(k).get("key")).toString());
                    cell.setCellStyle(dataStyle);;
                }
                cell = subHeaderRow.createCell(k+2);
                cell.setCellValue(rowList.get(j).get("total").toString());
                cell.setCellStyle(dataStyle);
            }


            // 写入文件
            try (ServletOutputStream outputStream = response.getOutputStream()){
                workbook.write(outputStream);
            } finally {
                workbook.close();
            }

        }
    }

    /**
     * 获取一类交调数据详细信息
     */
    @GetMapping(value = "/{id}")
    public AjaxResult getInfo(@PathVariable("id") Long id)
    {
        return AjaxResult.success(dcTrafficSurveyDataService.selectDcTrafficSurveyDataById(id));
    }

    /**
     * 新增一类交调数据
     */
    @Log(title = "一类交调数据", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody DcTrafficSurveyData dcTrafficSurveyData)
    {
        return toAjax(dcTrafficSurveyDataService.insertDcTrafficSurveyData(dcTrafficSurveyData));
    }

    /**
     * 修改一类交调数据
     */
    @Log(title = "一类交调数据", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody DcTrafficSurveyData dcTrafficSurveyData)
    {
        return toAjax(dcTrafficSurveyDataService.updateDcTrafficSurveyData(dcTrafficSurveyData));
    }

    /**
     * 删除一类交调数据
     */
    @Log(title = "一类交调数据", businessType = BusinessType.DELETE)
	@DeleteMapping("/{ids}")
    public AjaxResult remove(@PathVariable Long[] ids)
    {
        return toAjax(dcTrafficSurveyDataService.deleteDcTrafficSurveyDataByIds(ids));
    }
}