package com.whdc.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.whdc.exception.MyException;
import com.whdc.model.dto.ApiDto;
import com.whdc.model.dto.GroupWarningDto;
import com.whdc.model.vo.WarningData;
import com.whdc.model.vo.WarningHistoryListVo;
import com.whdc.model.vo.WarningListVo;
import com.whdc.service.IAddressBookService;
import com.whdc.utils.HttpUtil;
import com.whdc.utils.ResultJson;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.compress.utils.Lists;
import org.apache.commons.lang3.StringUtils;
import org.apache.http.HttpStatus;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * Description:
 * Created by XuSan on 2024/5/23.
 *
 * @author XuSan
 * @version 1.0
 */
@Slf4j
@Api(tags = "预警 - Controller")
@RestController
@RequestMapping("/warning")
public class WarningController {


    @Autowired
    private IAddressBookService addressBookService;

    /**
     * 气象预警
     *
     * @param dto
     * @return
     */
    @ApiOperation(value = "气象预警")
    @PostMapping("/getGroupWarning")
    public ResultJson<List<WarningListVo>> getGroupWarning(
            @RequestBody GroupWarningDto dto) {

        ApiDto apiDto = new ApiDto();
        apiDto.setStartTime(dto.getStartTime());
        apiDto.setEndTime(dto.getEndTime());
        apiDto.setPageNumber(dto.getPageNumber());
        apiDto.setPageSize(dto.getPageSize());
        ArrayList<ApiDto.Filter> filter = Lists.newArrayList();

        String publishUnit = dto.getPublishUnit();
        if (StringUtils.isNotBlank(publishUnit)) {
            String[] split = publishUnit.split(",");
            if (split.length > 1) {
                String str = "(";
                for (String s : split) {
                    str += s + ",";
                }
                str = str.substring(0, str.length() - 1) + ")";
                filter.add(new ApiDto.Filter("publishUnit", "in " + str, ""));

            } else {
                filter.add(new ApiDto.Filter("publishUnit", "like", String.format("%s%s%s", "%", publishUnit, "%")));
            }
        }
        String warnSignalLevel = dto.getWarnSignalLevel();
        if (StringUtils.isNotBlank(warnSignalLevel)) {
            String[] split = warnSignalLevel.split(",");
            if (split.length > 1) {
                String str = "(";
                for (String s : split) {
                    str += s + ",";
                }
                str = str.substring(0, str.length() - 1) + ")";
                filter.add(new ApiDto.Filter("warnSignalLevel", "in " + str, ""));

            } else {
                filter.add(new ApiDto.Filter("warnSignalLevel", "like", String.format("%s%s%s", "%", warnSignalLevel, "%")));
            }
        }
        String warnSignalType = dto.getWarnSignalType();
        if (StringUtils.isNotBlank(warnSignalType)) {
            String[] split = warnSignalType.split(",");
            if (split.length > 1) {
                String str = "(";
                for (String s : split) {
                    str += s + ",";
                }
                str = str.substring(0, str.length() - 1) + ")";
                filter.add(new ApiDto.Filter("warnSignalType", "in " + str, ""));

            } else {
                filter.add(new ApiDto.Filter("warnSignalType", "like", String.format("%s%s%s", "%", warnSignalType, "%")));
            }
        }

        apiDto.setFilter(filter);
//        String str = HttpUtil.sendPost("http://223.75.53.141:8000/shzh/met/zyqxfw/api/warning/getGroupWarning", JSON.toJSONString(apiDto));
        String str = HttpUtil.sendPost("http://127.0.0.1:20000/shzh/met/zyqxfw/api/warning/getGroupWarning", JSON.toJSONString(apiDto));
        JSONObject json = JSON.parseObject(str);
        if (json != null && json.getInteger("code") == HttpStatus.SC_OK) {
            List<WarningData> data = json.getJSONArray("data").toJavaList(WarningData.class);
            return ResultJson.ok(getList(data));
        }

        return ResultJson.ok(json);
    }

    /**
     * 历史气象预警统计
     *
     * @param dto
     * @return
     */
    @ApiOperation(value = "历史气象预警统计")
    @PostMapping("/getHistoryGroupWarning")
    public ResultJson<List<WarningListVo>> getHistoryGroupWarning(@RequestBody GroupWarningDto dto) {

        ApiDto apiDto = new ApiDto();
        apiDto.setStartTime(dto.getStartTime());
        apiDto.setEndTime(dto.getEndTime());
        apiDto.setPageNumber(dto.getPageNumber());
        apiDto.setPageSize(dto.getPageSize());
        ArrayList<ApiDto.Filter> filter = Lists.newArrayList();

        String publishUnit = dto.getPublishUnit();
        if (StringUtils.isNotBlank(publishUnit)) {
            String[] split = publishUnit.split(",");
            if (split.length > 1) {
                String str = "(";
                for (String s : split) {
                    str += s + ",";
                }
                str = str.substring(0, str.length() - 1) + ")";
                filter.add(new ApiDto.Filter("publishUnit", "in " + str, ""));

            } else {
                filter.add(new ApiDto.Filter("publishUnit", "like", String.format("%s%s%s", "%", publishUnit, "%")));
            }
        }
        String warnSignalLevel = dto.getWarnSignalLevel();
        if (StringUtils.isNotBlank(warnSignalLevel)) {
            String[] split = warnSignalLevel.split(",");
            if (split.length > 1) {
                String str = "(";
                for (String s : split) {
                    str += s + ",";
                }
                str = str.substring(0, str.length() - 1) + ")";
                filter.add(new ApiDto.Filter("warnSignalLevel", "in " + str, ""));

            } else {
                filter.add(new ApiDto.Filter("warnSignalLevel", "like", String.format("%s%s%s", "%", warnSignalLevel, "%")));
            }
        }
        String warnSignalType = dto.getWarnSignalType();
        if (StringUtils.isNotBlank(warnSignalType)) {
            String[] split = warnSignalType.split(",");
            if (split.length > 1) {
                String str = "(";
                for (String s : split) {
                    str += s + ",";
                }
                str = str.substring(0, str.length() - 1) + ")";
                filter.add(new ApiDto.Filter("warnSignalType", "in " + str, ""));

            } else {
                filter.add(new ApiDto.Filter("warnSignalType", "like", String.format("%s%s%s", "%", warnSignalType, "%")));
            }
        }


        apiDto.setFilter(filter);
//        String str = HttpUtil.sendPost("http://223.75.53.141:8000/shzh/met/zyqxfw/api/warning/getGroupWarning", JSON.toJSONString(apiDto));
        String str = HttpUtil.sendPost("http://127.0.0.1:20000/shzh/met/zyqxfw/api/warning/getGroupWarning", JSON.toJSONString(apiDto));
        JSONObject json = JSON.parseObject(str);
        if (json != null && json.getInteger("code") == HttpStatus.SC_OK) {
            List<WarningData> data = json.getJSONArray("data").toJavaList(WarningData.class);
            return ResultJson.ok(getCount(data));
        }

        return ResultJson.ok(json);
    }

    /**
     * 正在生效预警
     *
     * @param data
     * @return
     */
    private List<WarningListVo> getList(List<WarningData> data) {

        List<WarningListVo> respList = Lists.newArrayList();
        for (WarningData datum : data) {
            for (WarningData.TypeList typeList : datum.getTypeList()) {
                // 预警类型
                String type = typeList.getType();
                for (WarningData.Warning warning : typeList.getWarnList()) {
                    WarningListVo vo = new WarningListVo();
                    String publishUnit = warning.getPublishUnit();
                    vo.setTitle(publishUnit + "发布" + type + "预警");
                    vo.setPublishTime(warning.getPublishTime());
                    vo.setContent(warning.getContent());
                    vo.setWarnSignalType(warning.getWarnSignalType());
                    vo.setWarnSignalLevel(warning.getWarnSignalLevel());
                    vo.setAddressBooks(addressBookService.getListByAdnm(publishUnit.replace("气象台", "")));
                    respList.add(vo);
                }
            }
        }

        return respList;
    }

    /**
     * 正在生效预警
     *
     * @param data
     * @return
     */
    private List<WarningHistoryListVo> getCount(List<WarningData> data) {

        List<WarningData.Warning> warnList = Lists.newArrayList();
        for (WarningData datum : data) {
            WarningHistoryListVo vo = new WarningHistoryListVo();
            vo.setWarnSignalType("全部");

            for (WarningData.TypeList typeList : datum.getTypeList()) {
                warnList.addAll(typeList.getWarnList());
            }
        }

        Map<String, List<WarningData.Warning>> warnMap = warnList.stream()
                .collect(Collectors
                        .groupingBy(WarningData.Warning::getPublishUnit, Collectors.toList()));

        List<WarningHistoryListVo> respList = Lists.newArrayList();

        warnMap.forEach((k,v) ->{
            WarningHistoryListVo vo = new WarningHistoryListVo();
            vo.setEffectArea(k.split("气象")[0]);

            // 类型
            Map<String, List<WarningData.Warning>> types = v.stream()
                    .collect(Collectors
                            .groupingBy(WarningData.Warning::getWarnSignalType, Collectors.toList()));
            types.forEach((kt,vt) ->{
                WarningHistoryListVo voType = new WarningHistoryListVo();
                voType.setWarnSignalType(kt);

                // 等级分组
                Map<String, Long> levels = vt.stream()
                        .collect(Collectors.groupingBy(WarningData.Warning::getWarnSignalLevel, Collectors.counting()));

                levels.forEach((k1, v1) -> {
                    switch (k1) {
                        case "红色":
                            voType.setRedCount(v1.intValue());
                            vo.setRedCount(v1.intValue() + vo.getRedCount());
                            break;
                        case "橙色":
                            voType.setOraCount(v1.intValue());
                            vo.setOraCount(v1.intValue() + vo.getOraCount());
                            break;
                        case "黄色":
                            voType.setYelCount(v1.intValue());
                            vo.setYelCount(v1.intValue() + vo.getYelCount());
                            break;
                        case "蓝色":
                            voType.setBluCount(v1.intValue());
                            vo.setBluCount(v1.intValue() + vo.getBluCount());
                            break;
                        default:
                            throw new MyException("当前预警等级不存在: " + k1);
                    }
                });


            });
            vo.setSumCount(v.size());
            respList.add(vo);
        });

        return respList.stream().sorted(Comparator.comparing(WarningHistoryListVo::getSumCount).reversed()).collect(Collectors.toList());
    }

}