package cn.stylefeng.guns.utils;

import cn.hutool.core.date.DateField;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.map.MapUtil;
import cn.hutool.core.util.StrUtil;
import lombok.experimental.UtilityClass;

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

@UtilityClass
public class StatDataUtil {

    /**
     * 获取日期格式化SQL
     *
     * @param type
     * @return
     */
    public String getDateFormatSql(Integer type) {
        String dateFormatSql = "%Y-%m-%d";
        if (type != null) {
            switch (type) {
                case 2:
                    return "%Y-%u";
                case 3:
                    return "%Y-%m";
                case 4:
                    return "%Y";
            }
        }
        return dateFormatSql;
    }
    /**
     * 获取开始时间和结束时间（空使用默认）
     *
     * @param type      类型：1日，2周，3月，4年
     * @param beginTime 开始时间
     * @param endTime   结束时间
     * @return
     */
    public List<String> getStartAndEndTime(Integer type, String beginTime, String endTime) {
        Date nowDate = new Date();
        if (type == null || type == 1) {
            if (StrUtil.isEmpty(beginTime)) {
                beginTime = DateUtil.beginOfDay(DateUtil.offsetDay(nowDate, -30)).toString();
            }else{
                beginTime = beginTime + " 00:00:00";
            }
            if (StrUtil.isEmpty(endTime)) {
                endTime = DateUtil.endOfDay(nowDate).toString();
            }else{
                endTime = endTime + " 23:59:59";
            }
        } else if (type == 2) {
            if (StrUtil.isEmpty(beginTime)) {
                DateTime weekDate = DateUtil.offsetWeek(nowDate, -11);
//                int week = DateUtil.weekOfYear(weekDate);
//                beginTime = DateUtil.format(weekDate, "yyyy") + "-" + StrUtil.padPre(String.valueOf(week), 2, "0");
             beginTime = DateUtil.beginOfDay(weekDate).toString();
            }else{
                beginTime = beginTime + " 00:00:00";
            }
            if (StrUtil.isEmpty(endTime)) {
//                int week = DateUtil.weekOfYear(nowDate);
                endTime = DateUtil.endOfDay(nowDate).toString();
//                endTime = DateUtil.format(nowDate, "yyyy") + "-" + StrUtil.padPre(String.valueOf(week), 2, "0");
            }else{
                endTime = endTime + " 23:59:59";
            }
        } else if (type == 3) {
            if (StrUtil.isEmpty(beginTime)) {
                beginTime =  DateUtil.beginOfDay(DateUtil.offsetMonth(nowDate, -5)).toString();

            }else{
                beginTime = beginTime + " 00:00:00";
            }
            if (StrUtil.isEmpty(endTime)) {
                endTime = DateUtil.endOfDay(nowDate).toString();
            }else{
                endTime = endTime + " 23:59:59";
            }
        } else if (type == 4) {
            if (StrUtil.isEmpty(beginTime)) {
                beginTime = DateUtil.beginOfDay(DateUtil.offset(nowDate, DateField.YEAR, -2)).toString();
            }else{
                beginTime = beginTime + " 00:00:00";
            }
            if (StrUtil.isEmpty(endTime)) {
                endTime = DateUtil.endOfDay(nowDate).toString();
            }else{
                endTime = endTime + " 23:59:59";
            }
        }

        return Arrays.asList(beginTime, endTime);
    }

    /**
     * 获取时间范围
     *
     * @param type      类型：1日，2周，3月，4年
     * @param beginTime 开始时间
     * @param endTime   结束时间
     * @return
     */
    public List<String> getBeginEndTimeRangeList(Integer type, String beginTime, String endTime) {
        if (type == null || type == 1) {
            return DateUtil.rangeToList(DateUtil.parse(beginTime), DateUtil.parse(endTime), DateField.DAY_OF_MONTH)
                    .stream()
                    .map(date -> DateUtil.format(date, "yyyy-MM-dd"))
                    .collect(Collectors.toList());
        } else if (type == 2) {
            //return DateUtil.rangeToList(DateUtil.parse(beginTime, "yyyy-ww"), DateUtil.parse(endTime, "yyyy-ww"), DateField.WEEK_OF_YEAR)
            return DateUtil.rangeToList(DateUtil.parse(beginTime), DateUtil.parse(endTime), DateField.WEEK_OF_YEAR)
                    .stream()
                    .map(date -> DateUtil.format(date, "yyyy-ww"))
                    .collect(Collectors.toList());
        } else if (type == 3) {
            return DateUtil.rangeToList(DateUtil.parse(beginTime, "yyyy-MM"), DateUtil.parse(endTime, "yyyy-MM"), DateField.MONTH)
                    .stream()
                    .map(date -> DateUtil.format(date, "yyyy-MM"))
                    .collect(Collectors.toList());
        } else if (type == 4) {
            return DateUtil.rangeToList(DateUtil.parse(beginTime, "yyyy"), DateUtil.parse(endTime, "yyyy"), DateField.YEAR)
                    .stream()
                    .map(date -> DateUtil.format(date, "yyyy"))
                    .collect(Collectors.toList());
        }
        return null;
    }

    /**
     * 封装时间数据
     *
     * @param timeList 时间范围
     * @param dataList 数据列表
     * @param dataFieldName 数据字段名，默认：data
     * @return
     */
    public List<Map<String, Object>> wrapperTimeLIst(List<String> timeList, List<Map<String, Object>> dataList, String dataFieldName) {
        dataFieldName = StrUtil.isNotEmpty(dataFieldName) ? dataFieldName : "data";
        String finalDataFieldName = dataFieldName;
        return timeList.stream()
                .map(time -> {
                    Object data = dataList.stream()
                            .filter(map -> map.get("time").equals(time))
                            .findFirst()
                            .map(map -> map.get(finalDataFieldName))
                            .orElse(0);

                    Map<String, Object> hashMap = MapUtil.newHashMap();
                    hashMap.put("time", time);
                    hashMap.put("data", data);
                    return hashMap;
                })
                .collect(Collectors.toList());
    }

    public static void main(String[] args) {
        List<String> timeList = StatDataUtil.getStartAndEndTime(2, "", "");
        System.out.println(timeList.get(0));
        System.out.println(timeList.get(1));
        System.out.println("-------------------------------------");

//        Integer type = 1;
//        String beginTime = "2023-01-01", endTime = "2023-02-02";
        Integer type = 2;
        String beginTime = "2023-46", endTime = "2024-05";
//        Integer type = 3;
//        String beginTime = "2023-01", endTime = "2024-02";
//        Integer type = 4;
//        String beginTime = "2020", endTime = "2024";
        timeList = StatDataUtil.getBeginEndTimeRangeList(type, beginTime, endTime);
        timeList.forEach(System.out::println);
        System.out.println("-------------------------------------");

        HashMap m1 = new HashMap();
        m1.put("time", "2023-48");
        m1.put("data", "233");

        HashMap m2 = new HashMap();
        m2.put("time", "2024-02");
        m2.put("data", "123");

        HashMap m3 = new HashMap();
        m3.put("time", "2023-50");
        m3.put("data", "444");

        List<Map<String, Object>> dataList = Arrays.asList(m1, m2, m3);

        List<Map<String, Object>> maps = StatDataUtil.wrapperTimeLIst(timeList, dataList, null);
        for (Map<String, Object> map : maps) {
            System.out.println(map.get("time") + "----" + map.get("data"));
        }
    }
}