package com.moral.api.service.impl;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.ObjectUtils;

import java.text.NumberFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.moral.api.entity.CityAqi;
import com.moral.api.entity.CityAqiDaily;
import com.moral.api.entity.CityAqiMonthly;
import com.moral.api.entity.Device;
import com.moral.api.entity.HistoryDaily;
import com.moral.api.entity.HistoryHourly;
import com.moral.api.entity.HistoryMonthly;
import com.moral.api.entity.Organization;
import com.moral.api.entity.Sensor;
import com.moral.api.entity.SysArea;
import com.moral.api.service.CityAqiDailyService;
import com.moral.api.service.CityAqiMonthlyService;
import com.moral.api.service.CityAqiService;
import com.moral.api.service.DeviceService;
import com.moral.api.service.HistoryDailyService;
import com.moral.api.service.HistoryHourlyService;
import com.moral.api.service.HistoryMonthlyService;
import com.moral.api.service.OrganizationService;
import com.moral.api.service.RegionService;
import com.moral.api.service.SensorService;
import com.moral.api.service.SysAreaService;
import com.moral.constant.Constants;
import com.moral.constant.RedisConstants;
import com.moral.util.AmendUtils;
import com.moral.util.DateUtils;

@Service
public class RegionServiceImpl implements RegionService {

    @Autowired
    private DeviceService deviceService;

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    private OrganizationService organizationService;

    @Autowired
    private CityAqiMonthlyService cityAqiMonthlyService;

    @Autowired
    private CityAqiDailyService cityAqiDailyService;

    @Autowired
    private CityAqiService cityAqiService;

    @Autowired
    private HistoryMonthlyService historyMonthlyService;

    @Autowired
    private HistoryDailyService historyDailyService;

    @Autowired
    private HistoryHourlyService historyHourlyService;

    @Autowired
    private SysAreaService sysAreaService;

    @Autowired
    private SensorService sensorService;

    private static Map<String, String> senorMap = new HashMap<>();

    static {
        senorMap.put(Constants.SENSOR_CODE_PM25, "PM2_5");
        senorMap.put(Constants.SENSOR_CODE_PM10, "PM10");
        senorMap.put(Constants.SENSOR_CODE_SO2, "SO2");
        senorMap.put(Constants.SENSOR_CODE_NO2, "NO2");
        senorMap.put(Constants.SENSOR_CODE_CO, "CO");
        senorMap.put(Constants.SENSOR_CODE_O3, "O3");
    }

    @Override
    public Set<Map<String, Object>> getRegionsByOrganizationId(Integer organizationId) {
        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
        queryWrapper.select("mac")
                .eq("organization_id", organizationId)
                .eq("is_delete", Constants.NOT_DELETE);
        List<Object> macs = deviceService.listObjs(queryWrapper);

        List<Map<String, Object>> devices = new ArrayList<>();
        //从redis获取设备
        Map<String, Object> entries = redisTemplate.opsForHash().entries(RedisConstants.DEVICE);
        entries.forEach((key, value) -> {
            if (macs.contains(key)) {
                devices.add((Map<String, Object>) value);
            }
        });
        return devices.stream()
                .map(device -> (Map<String, Object>) device.get("town"))
                .collect(Collectors.toSet());
    }

    @Override
    public Set<Map<String, Object>> getSensorByRegionCodesAndOrganizationId(Map<String, Object> params) {
        Set<Map<String, Object>> result = new HashSet<>();
        Integer orgId = Integer.parseInt(params.get("organizationId").toString());
        List<String> regionCodes = Arrays.asList(params.get("regionCodes").toString().split(","));
        //获取区域设备macs
        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
        queryWrapper.select("mac")
                .eq("organization_id", orgId)
                .eq("is_delete", Constants.NOT_DELETE)
                .in("town_code", regionCodes);
        List<Object> macs = deviceService.listObjs(queryWrapper);

        //根据设备mac获取设备传感器信息
        for (Object mac : macs) {
            Device device = (Device) redisTemplate.opsForHash().get(RedisConstants.DEVICE_INFO, mac.toString());
            List<Sensor> sensors = device.getVersion().getSensors();
            sensors.forEach(sensor -> {
                Map<String, Object> sensorMap = new HashMap<>();
                sensorMap.put("sensorCode", sensor.getCode());
                sensorMap.put("sensorName", sensor.getName());
                result.add(sensorMap);
            });
        }
        return result;
    }

    @Override
    public Map<String, Object> regionContribution(Map<String, Object> params) {
        int orgId = Integer.parseInt(params.get("organizationId").toString());
        List<String> regionCodes = Arrays.asList(params.get("regionCodes").toString().split(","));
        String type = params.get("type").toString();
        String time = params.get("time").toString();
        String sensorCode = params.get("sensorCode").toString();
        Map<String, Object> result = new HashMap<>();
        switch (type) {
            case "year":
                result = regionContributionOfYear(orgId, regionCodes, time, sensorCode);
                break;
            case "month":
                result = regionContributionOfMonth(orgId, regionCodes, time, sensorCode);
                break;
            case "day":
                result = regionContributionOfDay(orgId, regionCodes, time, sensorCode);
                break;
            default:
                break;
        }
        return result;
    }

    //贡献率，年
    private Map<String, Object> regionContributionOfYear(Integer orgId, List<String> regionCodes, String time, String sensorCode) {
        Map<String, Object> result = new HashMap<>();

        //1.时间信息结果
        List<String> timeLag = DateUtils.getTimeLag(time);
        result.put("time", timeLag);


        //根据组织获取区域定位
        QueryWrapper<Organization> organizationQueryWrapper = new QueryWrapper<>();
        organizationQueryWrapper.select("location_level_code")
                .eq("id", orgId);
        Integer locationLevelCode = organizationService.getOne(organizationQueryWrapper).getLocationLevelCode();


        //本市本年所有月数据
        QueryWrapper<CityAqiMonthly> cityAqiMonthlyQueryWrapper = new QueryWrapper<>();
        cityAqiMonthlyQueryWrapper.select("time", "value")
                .eq("city_code", locationLevelCode)
                .likeRight("time", time);
        List<Map<String, Object>> cityAqis = cityAqiMonthlyService.listMaps(cityAqiMonthlyQueryWrapper);
        Map<String, Object> cityAqiMap = new HashMap<>();
        if (!ObjectUtils.isEmpty(cityAqis)) {
            for (Map<String, Object> cityAqi : cityAqis) {
                cityAqiMap.put(cityAqi.get("time").toString().substring(0, 7), cityAqi.get("value"));
            }
        }


        //该组织所有设备信息，本年所有月数据
        List<String> allMacs = deviceService.getMacsByOrganizationId(orgId);
        List<HistoryMonthly> allDeviceData = historyMonthlyService.getValueByMacs(allMacs, time);
        Map<String, List<HistoryMonthly>> allDeviceDataMap = allDeviceData.stream()
                .collect(Collectors.groupingBy(o -> DateUtils.dateToDateString(o.getTime()).substring(0, 7)));


        //本市数据结果集
        Map<String, Object> resultCityDataMap = new HashMap<>();
        List<Map<String, Object>> cityDataList = new ArrayList<>();
        //所有设备每个时间点，和值
        Map<String, Object> allDeviceSumMap = new HashMap<>();
        for (String yearMonth : timeLag) {
            //所有设备该因子累加值
            Double allDeviceSum = null;
            List<HistoryMonthly> historyMonthlyList = allDeviceDataMap.get(yearMonth);
            if (!ObjectUtils.isEmpty(historyMonthlyList)) {
                allDeviceSum = historyMonthlyService.calculatedValue(historyMonthlyList, sensorCode, "sum");
            }
            allDeviceSumMap.put(yearMonth, allDeviceSum);


            //本市值
            Double cityValue = null;
            if (cityAqiMap.get(yearMonth) != null) {
                Map<String, Object> dataValue = JSONObject.parseObject(cityAqiMap.get(yearMonth).toString(), Map.class);
                //判断城市aqi是否有该因子数据
                String sensorName = senorMap.get(sensorCode);
                if (sensorName != null) {
                    cityValue = Double.parseDouble(dataValue.get(sensorName).toString());
                }
            }
            Map<String, Object> cityDataMap = new HashMap<>();
            //本市数据
            cityDataMap.put("name", cityValue);
            cityDataList.add(cityDataMap);
        }

        //2.本市数据结果
        resultCityDataMap.put("data", cityDataList);
        result.put("本市", resultCityDataMap);


        for (String regionCode : regionCodes) {
            //每个乡镇区域数据结果集
            HashMap<String, Object> regionMap = new HashMap<>();
            List<Map<String, Object>> dataList = new ArrayList<>();


            //乡镇区域名称
            QueryWrapper<SysArea> queryWrapper = new QueryWrapper<>();
            queryWrapper.select("area_name").eq("area_code", Integer.parseInt(regionCode));
            String regionName = sysAreaService.getOne(queryWrapper).getAreaName();


            //获取乡镇区域下所有设备,本年所有月数据
            List<String> regionMacs = deviceService.getMacsByOrgIdAndRegionCode(orgId, Integer.parseInt(regionCode));
            List<HistoryMonthly> regionDeviceData = historyMonthlyService.getValueByMacs(regionMacs, time);
            //按time分组
            Map<String, List<HistoryMonthly>> regionDataMap = regionDeviceData.stream()
                    .collect(Collectors.groupingBy(o -> DateUtils.dateToDateString(o.getTime()).substring(0, 7)));


            for (String resultTime : timeLag) {
                //每个时间点数据
                Map<String, Object> timeDataMap = new HashMap<>();
                Object o = allDeviceSumMap.get(resultTime);
                Double allDeviceSum = null;
                if (o != null) {
                    allDeviceSum = Double.parseDouble(o.toString());
                }

                List<HistoryMonthly> regionDeviceHistoryMonthlyList = regionDataMap.get(resultTime);
                //贡献率
                String contributionRate = null;
                Double regionAvg = null;
                if (!ObjectUtils.isEmpty(regionDeviceHistoryMonthlyList)) {
                    //该乡镇区域均值计算
                    regionAvg = historyMonthlyService.calculatedValue(regionDeviceHistoryMonthlyList, sensorCode, "avg");
                    //该乡镇区域和值计算
                    Double regionSum = historyMonthlyService.calculatedValue(regionDeviceHistoryMonthlyList, sensorCode, "sum");
                    //乡镇区域贡献率计算
                    NumberFormat numberFormat = NumberFormat.getInstance();
                    numberFormat.setMaximumFractionDigits(2);
                    if (allDeviceSum != null && allDeviceSum != 0d) {
                        contributionRate = numberFormat.format(regionSum / (allDeviceSum) * 100) + "%";
                    }
                }

                //乡镇区域均值
                timeDataMap.put("name", regionAvg == null ? null : AmendUtils.sciCal(regionAvg, 0));
                //乡镇区域贡献率
                timeDataMap.put("value", contributionRate);
                dataList.add(timeDataMap);
            }
            regionMap.put("data", dataList);
            result.put(regionName, regionMap);
        }
        return result;
    }

    //贡献率，月
    private Map<String, Object> regionContributionOfMonth(Integer orgId, List<String> regionCodes, String time, String sensorCode) {
        Map<String, Object> result = new HashMap<>();

        //1.时间信息结果
        List<String> timeLag = DateUtils.getTimeLag(time);
        result.put("time", timeLag);


        //根据组织获取区域定位
        QueryWrapper<Organization> organizationQueryWrapper = new QueryWrapper<>();
        organizationQueryWrapper.select("location_level_code")
                .eq("id", orgId);
        Integer locationLevelCode = organizationService.getOne(organizationQueryWrapper).getLocationLevelCode();


        //本市本月所有日数据
        QueryWrapper<CityAqiDaily> cityAqiDailyQueryWrapper = new QueryWrapper<>();
        cityAqiDailyQueryWrapper.select("time", "value")
                .eq("city_code", locationLevelCode)
                .likeRight("time", time);
        List<Map<String, Object>> cityAqis = cityAqiDailyService.listMaps(cityAqiDailyQueryWrapper);
        Map<String, Object> cityAqiMap = new HashMap<>();
        if (!ObjectUtils.isEmpty(cityAqis)) {
            for (Map<String, Object> cityAqi : cityAqis) {
                cityAqiMap.put(cityAqi.get("time").toString().substring(0, 10), cityAqi.get("value"));
            }
        }


        //该组织所有设备信息，本月所有日数据
        List<String> allMacs = deviceService.getMacsByOrganizationId(orgId);
        List<HistoryDaily> allDeviceData = historyDailyService.getValueByMacs(allMacs, time);
        Map<String, List<HistoryDaily>> allDeviceDataMap = allDeviceData.stream()
                .collect(Collectors.groupingBy(o -> DateUtils.dateToDateString(o.getTime()).substring(0, 10)));


        //本市数据结果集
        Map<String, Object> resultCityDataMap = new HashMap<>();
        List<Map<String, Object>> cityDataList = new ArrayList<>();
        //所有设备每个时间点，和值
        Map<String, Object> allDeviceSumMap = new HashMap<>();
        for (String yearMonthDay : timeLag) {
            //所有设备该因子累加值
            Double allDeviceSum = null;
            List<HistoryDaily> historyDailyList = allDeviceDataMap.get(yearMonthDay);
            if (!ObjectUtils.isEmpty(historyDailyList)) {
                allDeviceSum = historyDailyService.calculatedValue(historyDailyList, sensorCode, "sum");
            }
            allDeviceSumMap.put(yearMonthDay, allDeviceSum);


            //本市值
            Double cityValue = null;
            if (cityAqiMap.get(yearMonthDay) != null) {
                Map<String, Object> dataValue = JSONObject.parseObject(cityAqiMap.get(yearMonthDay).toString(), Map.class);
                //判断城市aqi是否有该因子数据
                String sensorName = senorMap.get(sensorCode);
                if (sensorName != null) {
                    cityValue = Double.parseDouble(dataValue.get(sensorName).toString());
                }
            }
            Map<String, Object> cityDataMap = new HashMap<>();
            //本市数据
            cityDataMap.put("name", cityValue);
            cityDataList.add(cityDataMap);
        }

        //2.本市数据结果
        resultCityDataMap.put("data", cityDataList);
        result.put("本市", resultCityDataMap);


        for (String regionCode : regionCodes) {
            //每个乡镇区域数据结果集
            HashMap<String, Object> regionMap = new HashMap<>();
            List<Map<String, Object>> dataList = new ArrayList<>();


            //乡镇区域名称
            QueryWrapper<SysArea> queryWrapper = new QueryWrapper<>();
            queryWrapper.select("area_name").eq("area_code", Integer.parseInt(regionCode));
            String regionName = sysAreaService.getOne(queryWrapper).getAreaName();


            //获取乡镇区域下所有设备,本月所有日数据
            List<String> regionMacs = deviceService.getMacsByOrgIdAndRegionCode(orgId, Integer.parseInt(regionCode));
            List<HistoryDaily> regionDeviceData = historyDailyService.getValueByMacs(regionMacs, time);
            //按time分组
            Map<String, List<HistoryDaily>> regionDataMap = regionDeviceData.stream()
                    .collect(Collectors.groupingBy(o -> DateUtils.dateToDateString(o.getTime()).substring(0, 10)));


            for (String resultTime : timeLag) {
                //每个时间点数据
                Map<String, Object> timeDataMap = new HashMap<>();
                Object o = allDeviceSumMap.get(resultTime);
                Double allDeviceSum = null;
                if (o != null) {
                    allDeviceSum = Double.parseDouble(o.toString());
                }

                List<HistoryDaily> regionDeviceHistoryDailyList = regionDataMap.get(resultTime);
                //贡献率
                String contributionRate = null;
                Double regionAvg = null;
                if (!ObjectUtils.isEmpty(regionDeviceHistoryDailyList)) {
                    //该乡镇区域均值计算
                    regionAvg = historyDailyService.calculatedValue(regionDeviceHistoryDailyList, sensorCode, "avg");
                    //该乡镇区域和值计算
                    Double regionSum = historyDailyService.calculatedValue(regionDeviceHistoryDailyList, sensorCode, "sum");
                    //乡镇区域贡献率计算
                    NumberFormat numberFormat = NumberFormat.getInstance();
                    numberFormat.setMaximumFractionDigits(2);
                    if (allDeviceSum != null && allDeviceSum != 0d) {
                        contributionRate = numberFormat.format(regionSum / (allDeviceSum) * 100) + "%";
                    }
                }

                //乡镇区域均值
                timeDataMap.put("name", regionAvg == null ? null : AmendUtils.sciCal(regionAvg, 0));
                //乡镇区域贡献率
                timeDataMap.put("value", contributionRate);
                dataList.add(timeDataMap);
            }
            regionMap.put("data", dataList);
            result.put(regionName, regionMap);
        }
        return result;
    }

    private Map<String, Object> regionContributionOfDay(Integer orgId, List<String> regionCodes, String time, String sensorCode) {
        Map<String, Object> result = new HashMap<>();

        //1.时间信息结果
        List<String> timeLag = DateUtils.getTimeLag(time);
        result.put("time", timeLag);


        //根据组织获取区域定位
        QueryWrapper<Organization> organizationQueryWrapper = new QueryWrapper<>();
        organizationQueryWrapper.select("location_level_code")
                .eq("id", orgId);
        Integer locationLevelCode = organizationService.getOne(organizationQueryWrapper).getLocationLevelCode();

        //查询因子上下限
        QueryWrapper<Sensor> sensorQueryWrapper = new QueryWrapper<>();
        sensorQueryWrapper.select("lower", "upper").eq("code", sensorCode);
        Sensor sensor = sensorService.getOne(sensorQueryWrapper);
        Double sensorLower = null;
        Double sensorUpper = null;
        if (sensor != null) {
            sensorLower = sensor.getLower();
            sensorUpper = sensor.getUpper();
        }

        //本市本日所有小时数据
        QueryWrapper<CityAqi> cityAqiQueryWrapper = new QueryWrapper<>();
        cityAqiQueryWrapper.select("time", "value")
                .eq("city_code", locationLevelCode)
                .likeRight("time", time);
        List<Map<String, Object>> cityAqis = cityAqiService.listMaps(cityAqiQueryWrapper);
        Map<String, Object> cityAqiMap = new HashMap<>();
        if (!ObjectUtils.isEmpty(cityAqis)) {
            for (Map<String, Object> cityAqi : cityAqis) {
                cityAqiMap.put(cityAqi.get("time").toString().substring(0, 13), cityAqi.get("value"));
            }
        }


        //该组织所有设备信息，本日所有小时数据
        List<String> allMacs = deviceService.getMacsByOrganizationId(orgId);
        List<HistoryHourly> allDeviceData = historyHourlyService.getValueByMacs(allMacs, time);
        Map<String, List<HistoryHourly>> allDeviceDataMap = allDeviceData.stream()
                .collect(Collectors.groupingBy(o -> DateUtils.dateToDateString(o.getTime()).substring(0, 13)));


        //本市数据结果集
        Map<String, Object> resultCityDataMap = new HashMap<>();
        List<Map<String, Object>> cityDataList = new ArrayList<>();
        //所有设备每个时间点，和值
        Map<String, Object> allDeviceSumMap = new HashMap<>();
        for (String yearMonthDay : timeLag) {
            //所有设备该因子累加值
            Double allDeviceSum = null;
            List<HistoryHourly> historyHourlyList = allDeviceDataMap.get(yearMonthDay);
            if (!ObjectUtils.isEmpty(historyHourlyList)) {
                allDeviceSum = historyHourlyService.calculatedValue(historyHourlyList, sensorCode, "sum", sensorLower, sensorUpper);
            }
            allDeviceSumMap.put(yearMonthDay, allDeviceSum);


            //本市值
            Double cityValue = null;
            if (cityAqiMap.get(yearMonthDay) != null) {
                Map<String, Object> dataValue = JSONObject.parseObject(cityAqiMap.get(yearMonthDay).toString(), Map.class);
                //判断城市aqi是否有该因子数据
                String sensorName = senorMap.get(sensorCode);
                if (sensorName != null) {
                    cityValue = Double.parseDouble(dataValue.get(sensorName).toString());
                }
            }
            Map<String, Object> cityDataMap = new HashMap<>();
            //本市数据
            cityDataMap.put("name", cityValue);
            cityDataList.add(cityDataMap);
        }

        //2.本市数据结果
        resultCityDataMap.put("data", cityDataList);
        result.put("本市", resultCityDataMap);


        for (String regionCode : regionCodes) {
            //每个乡镇区域数据结果集
            HashMap<String, Object> regionMap = new HashMap<>();
            List<Map<String, Object>> dataList = new ArrayList<>();


            //乡镇区域名称
            QueryWrapper<SysArea> queryWrapper = new QueryWrapper<>();
            queryWrapper.select("area_name").eq("area_code", Integer.parseInt(regionCode));
            String regionName = sysAreaService.getOne(queryWrapper).getAreaName();


            //获取乡镇区域下所有设备,本日所有小时数据
            List<String> regionMacs = deviceService.getMacsByOrgIdAndRegionCode(orgId, Integer.parseInt(regionCode));
            List<HistoryHourly> regionDeviceData = historyHourlyService.getValueByMacs(regionMacs, time);
            //按time分组
            Map<String, List<HistoryHourly>> regionDataMap = regionDeviceData.stream()
                    .collect(Collectors.groupingBy(o -> DateUtils.dateToDateString(o.getTime()).substring(0, 13)));


            for (String resultTime : timeLag) {
                //每个时间点数据
                Map<String, Object> timeDataMap = new HashMap<>();
                Object o = allDeviceSumMap.get(resultTime);
                Double allDeviceSum = null;
                if (o != null) {
                    allDeviceSum = Double.parseDouble(o.toString());
                }

                List<HistoryHourly> regionDeviceHistoryHourlyList = regionDataMap.get(resultTime);
                //贡献率
                String contributionRate = null;
                Double regionAvg = null;
                if (!ObjectUtils.isEmpty(regionDeviceHistoryHourlyList)) {
                    //该乡镇区域均值计算
                    regionAvg = historyHourlyService.calculatedValue(regionDeviceHistoryHourlyList, sensorCode, "avg", sensorLower, sensorUpper);
                    //该乡镇区域和值计算
                    Double regionSum = historyHourlyService.calculatedValue(regionDeviceHistoryHourlyList, sensorCode, "sum", sensorLower, sensorUpper);
                    //乡镇区域贡献率计算
                    NumberFormat numberFormat = NumberFormat.getInstance();
                    numberFormat.setMaximumFractionDigits(2);
                    if (allDeviceSum != null && allDeviceSum != 0d) {
                        contributionRate = numberFormat.format(regionSum / (allDeviceSum) * 100) + "%";
                    }
                }

                //乡镇区域均值
                timeDataMap.put("name", regionAvg == null ? null : AmendUtils.sciCal(regionAvg, 0));
                //乡镇区域贡献率
                timeDataMap.put("value", contributionRate);
                dataList.add(timeDataMap);
            }
            regionMap.put("data", dataList);
            result.put(regionName, regionMap);
        }
        return result;
    }
}