package com.ruoyi.web.controller.api;


import com.alibaba.fastjson2.JSONArray;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.dto.EquipmentDto;
import com.ruoyi.system.model.*;
import com.ruoyi.system.service.*;
import com.ruoyi.system.vo.IndexDataDetailVo;
import com.ruoyi.system.vo.IndexDataRankVo;
import com.ruoyi.system.vo.IndexDataVo;
import com.tencentcloudapi.iot.v20180123.models.AppUser;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.time.Duration;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import static java.util.stream.Collectors.groupingBy;

/**
 * <p>
 * 专业管理 前端控制器
 * </p>
 *
 * @author xiaochen
 * @since 2025-09-28
 */
@Api(tags = "数据统计")
@RestController
@RequestMapping("")
public class TDataStatisticsController {

    private final TMissionService missionService;
    private final TokenService tokenService;
    private final TMissionUserService missionUserService;
    private final TAppUserService appUserService;
    private final TDivisionTeamService divisionTeamService;
    private final TMissionAssignRuleService missionReassignService;
    private final TAppUserEquipmentService appUserEquipmentService;
    private final TEquipmentService equipmentService;
    @Autowired
    public TDataStatisticsController(TMissionService missionService, TokenService tokenService, TMissionUserService missionUserService, TAppUserService appUserService, TDivisionTeamService divisionTeamService, TMissionAssignRuleService missionReassignService1, TAppUserEquipmentService appUserEquipmentService, TEquipmentService equipmentService) {
        this.missionService = missionService;
        this.tokenService = tokenService;
        this.missionUserService = missionUserService;
        this.appUserService = appUserService;
        this.divisionTeamService = divisionTeamService;
        this.missionReassignService = missionReassignService1;
        this.appUserEquipmentService = appUserEquipmentService;
        this.equipmentService = equipmentService;
    }

    @ApiOperation(value = "获取数据-第一部分")
    @GetMapping(value = "/open/t-data-statistics/getData")
    public R<IndexDataVo> getData() {
        IndexDataVo indexDataVo = new IndexDataVo();
        List<TMission> list = missionService.list();
        if(list.isEmpty()){
            return R.ok(indexDataVo);
        }
        indexDataVo.setAllMissionNum(list.size());
        indexDataVo.setUnAssignMissionNum((int) list.stream().filter(item -> item.getStatus() == 1).count());

        indexDataVo.setFinishMissionNum((int) list.stream().filter(item -> item.getStatus() == 4 || item.getStatus() == 6).count());

        indexDataVo.setUnFinishMissionNum(indexDataVo.getAllMissionNum()-indexDataVo.getFinishMissionNum());

        indexDataVo.setSpecialMissionNum((int) list.stream().filter(item -> item.getUrgencyLevel() == 3).count());
        indexDataVo.setEmergencyMissionNum((int) list.stream().filter(item -> item.getUrgencyLevel() == 2).count());
        indexDataVo.setRegularMissionNum((int) list.stream().filter(item -> item.getUrgencyLevel() == 1).count());

        ArrayList<IndexDataDetailVo> indexDataDetailVos = new ArrayList<>();

        // 近7天
        for (int i = 6; i >=0 ; i--) {
            IndexDataDetailVo indexDataDetailVo = new IndexDataDetailVo();
            // list根据createTime 获取对应时间的数据
            String time  = LocalDate.now().minusDays(i).toString();
            List<TMission> timeList = list.stream().filter(item -> item.getCreateTime().toLocalDate().toString().equals(time)).collect(Collectors.toList());

            indexDataDetailVo.setTime(time);
            indexDataDetailVo.setSpecialMissionNum((int) timeList.stream().filter(item -> item.getUrgencyLevel() == 3).count());
            indexDataDetailVo.setEmergencyMissionNum((int) timeList.stream().filter(item -> item.getUrgencyLevel() == 2).count());
            indexDataDetailVo.setRegularMissionNum((int) timeList.stream().filter(item -> item.getUrgencyLevel() == 1).count());
            indexDataDetailVos.add(indexDataDetailVo);
        }
        indexDataVo.setMissionDetail(indexDataDetailVos);


        List<TMission> tMissions = list.stream().filter(e -> e.getReceiveTime() != null).collect(Collectors.toList());
        double v = tMissions.stream().mapToDouble(TMission::getResTime).average().orElse(0);
        indexDataVo.setAvgResponseTime(String.valueOf(v));


        List<TMission> tMissions1 = list.stream().filter(e -> e.getFinishTime() != null).collect(Collectors.toList());
        if(!tMissions1.isEmpty()){
            long count = tMissions1.stream().filter(e -> e.getStatus() == 4).count();
            String finishRate = new BigDecimal(count).divide(new BigDecimal(tMissions1.size()), 4, BigDecimal.ROUND_HALF_UP).multiply(BigDecimal.valueOf(100)).setScale(2, RoundingMode.HALF_UP).toString();
            indexDataVo.setFinishRate(finishRate);
        }


        double avgFinishTime = tMissions1.stream().mapToDouble(TMission::getSuccessTime).average().orElse(0);
        indexDataVo.setAvgFinishTime(String.valueOf(avgFinishTime));

        // 平了分的
        List<TMission> tMissions2 = list.stream().filter(e -> e.getAddressScore() != null ).collect(Collectors.toList());
        if(!tMissions2.isEmpty()){
            long count1 = tMissions2.stream().filter(e -> e.getTimeScore() + e.getAddressScore() + e.getPersonCountScore() + e.getEquipmentScore() + e.getAttemptScore() > 30).count();
            String enemyAccuracyRate = new BigDecimal(count1).divide(new BigDecimal(tMissions2.size()), 4, RoundingMode.HALF_UP).multiply(BigDecimal.valueOf(100)).setScale(2, RoundingMode.HALF_UP).toString();
            indexDataVo.setEnemyAccuracyRate(enemyAccuracyRate);
        }

        List<TMissionUser> list1 = missionUserService.list(new LambdaQueryWrapper<TMissionUser>().eq(TMissionUser::getUserType, 1).isNotNull(TMissionUser::getSuccessTime));
        if(!list1.isEmpty()){
            double setDamageRate = list1.stream().mapToDouble(TMissionUser::getDamageRate).average().orElse(0);
            indexDataVo.setDamageRate(String.valueOf(setDamageRate));
        }

        return R.ok(indexDataVo);
    }

    @ApiOperation(value = "获取数据-第二部分")
    @GetMapping(value = "/open/t-data-statistics/getDataTwo")
    public R<IndexDataVo> getDataTwo() {
        IndexDataVo indexDataVo = new IndexDataVo();
        List<TMission> listAll = missionService.list();

        List<TMission> list = listAll.stream().filter(item -> item.getStatus() == 4).collect(Collectors.toList());
        if(listAll.isEmpty()){
            return R.ok(indexDataVo);
        }

        if (!list.isEmpty()) {
            List<String> missionIds = list.stream().map(TMission::getId).collect(Collectors.toList());
            List<TMissionUser> missionUsers = missionUserService.list(new LambdaQueryWrapper<TMissionUser>().in(TMissionUser::getMissionId, missionIds));
            List<String> userIds = missionUsers.stream().map(TMissionUser::getAppUserId).collect(Collectors.toList());
            List<TAppUser> appUsers = appUserService.list(new LambdaQueryWrapper<TAppUser>().in(TAppUser::getId, userIds));
            Map<String, List<TAppUser>> collect = appUsers.stream().collect(groupingBy(TAppUser::getTeamId));

            ArrayList<IndexDataRankVo> indexDataRankVos = new ArrayList<>();
            for (Map.Entry<String, List<TAppUser>> entry : collect.entrySet()) {
                IndexDataRankVo indexDataRankVo = new IndexDataRankVo();
                indexDataRankVo.setId(entry.getKey());
                List<String> userIds1 = entry.getValue().stream().map(TAppUser::getId).collect(Collectors.toList());
                long count = missionUsers.stream().filter(e -> userIds1.contains(e.getAppUserId())).count();
                indexDataRankVo.setCount((int) count);
                indexDataRankVos.add(indexDataRankVo);
            }
            List<String> teamIds = indexDataRankVos.stream().map(IndexDataRankVo::getId).collect(Collectors.toList());
            List<TDivisionTeam> list2 = divisionTeamService.list(new LambdaQueryWrapper<TDivisionTeam>().in(TDivisionTeam::getId, teamIds));
            indexDataRankVos.forEach(e -> {
                TDivisionTeam tDivisionTeam = list2.stream().filter(s -> s.getId().equals(e.getId())).findFirst().get();
                e.setName(tDivisionTeam.getTeamName());
            });

            // indexDataRankVos取出count排名前5的 若没有5个全取
            if(indexDataRankVos.size()>5){
                List<IndexDataRankVo> collect1 = indexDataRankVos.stream().sorted(Comparator.comparingInt(IndexDataRankVo::getCount).reversed()).limit(5).collect(Collectors.toList());
                indexDataVo.setRank(collect1);
            }else {
                List<IndexDataRankVo> collect1 = indexDataRankVos.stream().sorted(Comparator.comparingInt(IndexDataRankVo::getCount).reversed()).collect(Collectors.toList());
                indexDataVo.setRank(collect1);
            }

        }


        Map<String, List<TMission>> collect = listAll.stream().collect(groupingBy(TMission::getMissionTypeId));

        List<IndexDataRankVo> indexDataDetailVos = new ArrayList<>();
        for (Map.Entry<String, List<TMission>> entry : collect.entrySet()) {
            IndexDataRankVo indexDataDetailVo = new IndexDataRankVo();
            indexDataDetailVo.setId(entry.getKey());
            indexDataDetailVo.setCount(entry.getValue().size());
            indexDataDetailVos.add(indexDataDetailVo);
        }
        if(!indexDataDetailVos.isEmpty()){
            List<String> typeIds = indexDataDetailVos.stream().map(IndexDataRankVo::getId).collect(Collectors.toList());
            List<TMissionAssignRule> tMissionAssignRules = missionReassignService.list(new LambdaQueryWrapper<TMissionAssignRule>().in(TMissionAssignRule::getId, typeIds));
            indexDataDetailVos.forEach(e -> {
                TMissionAssignRule tMissionAssignRule = tMissionAssignRules.stream().filter(s -> s.getId().equals(e.getId())).findFirst().get();
                e.setName(tMissionAssignRule.getTypeName());
            });

        }
        indexDataVo.setMissionType(indexDataDetailVos);


        // 装备库统计
        List<TAppUserEquipment> list1 = appUserEquipmentService.list();
        ArrayList<EquipmentDto> equipmentDtos = new ArrayList<>();

        if (!list1.isEmpty()) {
            List<String> equipments = list1.stream().map(TAppUserEquipment::getEquipmentCount).collect(Collectors.toList());
            for (String equipment : equipments) {
                List<EquipmentDto> dtos = JSONArray.parseArray(equipment, EquipmentDto.class);
                dtos =  dtos.stream().filter(e->!e.getId().equals("0")).collect(Collectors.toList());
                equipmentDtos.addAll(dtos);
            }
        }
        Map<String, List<EquipmentDto>> listMap = equipmentDtos.stream().collect(groupingBy(EquipmentDto::getId));
        List<IndexDataRankVo> indexDataRankVos = new ArrayList<>();
        for (Map.Entry<String, List<EquipmentDto>> entry : listMap.entrySet()) {
            IndexDataRankVo indexDataRankVo = new IndexDataRankVo();
            indexDataRankVo.setId(entry.getKey());
            indexDataRankVo.setCount(entry.getValue().stream().mapToInt(EquipmentDto::getCount).sum());
            indexDataRankVos.add(indexDataRankVo);
        }
        if(!indexDataRankVos.isEmpty()){
            List<String> equipIds = indexDataRankVos.stream().map(IndexDataRankVo::getId).collect(Collectors.toList());
            List<TEquipment> equipmentList = equipmentService.list(new LambdaQueryWrapper<TEquipment>().in(TEquipment::getId, equipIds));
            indexDataRankVos.forEach(e -> {
                TEquipment tEquipment = equipmentList.stream().filter(s -> s.getId().equals(e.getId())).findFirst().orElse(null);
                if(tEquipment!=null){
                    e.setName(tEquipment.getEquipmentName());
                }
            });
        }
        indexDataVo.setEquip(indexDataRankVos);


        return R.ok(indexDataVo);
    }


    public static void main(String[] args) {
        String format = String.format("%05d", 2);
        System.out.println(format);
    }
}