package com.zhgd.xmgl.task;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.zhgd.xmgl.async.AsyncEnvironment;
import com.zhgd.xmgl.modules.policecamera.entity.PoliceCameraItem;
import com.zhgd.xmgl.modules.policecamera.entity.PoliceCameraVideoConfig;
import com.zhgd.xmgl.modules.policecamera.service.IPoliceCameraItemService;
import com.zhgd.xmgl.modules.policecamera.service.IPoliceCameraVideoConfigService;
import com.zhgd.xmgl.modules.project.entity.ProjectVideoConfig;
import com.zhgd.xmgl.modules.project.entity.bo.ProjectVideoConfigBo;
import com.zhgd.xmgl.modules.project.enums.ProjectVideoConfigVideoTypeEnum;
import com.zhgd.xmgl.modules.project.mapper.ProjectVideoConfigMapper;
import com.zhgd.xmgl.modules.video.entity.VideoItem;
import com.zhgd.xmgl.modules.video.mapper.VideoItemMapper;
import com.zhgd.xmgl.modules.video.service.impl.VideoItemServiceImpl;
import com.zhgd.xmgl.util.HikVideoUtil;
import com.zhgd.xmgl.util.YsVideoUtil;
import lombok.extern.slf4j.Slf4j;
import net.javacrumbs.shedlock.core.SchedulerLock;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Lazy;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

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

/**
 * @program: wisdomSite
 * @description: 视频定时任务
 * @author: Mr.Peng
 * @create: 2021-11-11 18:48
 **/
@Slf4j
@RestController
@RequestMapping("/xmgl/task/")
public class VideoTask {
    @Autowired
    private ProjectVideoConfigMapper projectVideoConfigMapper;
    @Autowired
    private VideoItemMapper videoItemMapper;
    @Autowired
    private AsyncEnvironment asyncEnvironment;
    @Value("${video-analysis-url}")
    private String videoAnalysisUrl;
    @Value("${serverUrl}")
    private String serverUrl;
    @Lazy
    @Autowired
    private VideoItemServiceImpl videoItemService;
    @Lazy
    @Autowired
    private IPoliceCameraVideoConfigService policeCameraVideoConfigService;
    @Lazy
    @Autowired
    private IPoliceCameraItemService policeCameraItemService;

    @SchedulerLock(name = "updateVideoState", lockAtMostFor = 1000 * 60 * 60, lockAtLeastFor = 1000 * 60 * 5)
    @Scheduled(cron = "0 */3 * * * ?")
    @RequestMapping("updateVideoState")
    public void updateVideoState() {
        try {
            QueryWrapper<ProjectVideoConfig> queryWrapper = new QueryWrapper<>();
            queryWrapper.lambda().eq(ProjectVideoConfig::getIsEnable, 1);
            List<ProjectVideoConfig> list = projectVideoConfigMapper.selectList(queryWrapper);
            if (list != null && list.size() > 0) {
                for (ProjectVideoConfig videoConfig : list) {
                    updateProjectVideo(videoConfig);
                }
            }
        } catch (Exception e) {
            log.error("error：", e);
        }
    }

    private void updateProjectVideo(ProjectVideoConfig videoConfig) {
        try {
            QueryWrapper<VideoItem> queryWrapper = new QueryWrapper<>();
            queryWrapper.lambda().eq(VideoItem::getVideoId, videoConfig.getId());
            List<VideoItem> list = videoItemMapper.selectList(queryWrapper);
            if (list != null && list.size() > 0) {
                if (Objects.equals(videoConfig.getVideoType(), ProjectVideoConfigVideoTypeEnum.YSY.getValue())) {
                    String accessToken = YsVideoUtil.getToken(videoConfig.getAppId(), videoConfig.getAppSecret());
                    for (VideoItem videoItem : list) {
                        Integer state = YsVideoUtil.getVideoStatus(videoConfig.getAppId(), videoConfig.getAppSecret(), videoItem.getSerialNumber(), accessToken);
                        videoItem.setDeviceState(state);
                        videoItemMapper.updateById(videoItem);
                    }
                } else if (Objects.equals(videoConfig.getVideoType(), ProjectVideoConfigVideoTypeEnum.ISC.getValue())) {
                    if (StringUtils.isNotEmpty(videoConfig.getAccount()) && StringUtils.isNotEmpty(videoConfig.getPassword())
                            && StringUtils.isNotEmpty(videoConfig.getAppId()) && StringUtils.isNotEmpty(videoConfig.getAppSecret())) {
                        for (VideoItem videoItem : list) {
                            if (StringUtils.isNotEmpty(videoItem.getSerialNumber())) {
                                Boolean online = HikVideoUtil.callPostApiOnlineStatus(videoItem.getSerialNumber(), videoConfig.getAccount(),
                                        videoConfig.getPassword(), videoConfig.getAppId(), videoConfig.getAppSecret());
                                if (online != null) {
                                    if (online) {
                                        videoItem.setDeviceState(1);
                                    } else {
                                        videoItem.setDeviceState(2);
                                    }
                                    videoItemMapper.updateById(videoItem);
                                }
                                //sendVideoData(videoItem,videoConfig);
                            }
                        }
                    }
                } else {
                    QueryWrapper<VideoItem> qu = new QueryWrapper<>();
                    qu.lambda().eq(VideoItem::getVideoId, videoConfig.getId());
                    VideoItem item = new VideoItem();
                    item.setDeviceState(2);
                    item.setVideoId(videoConfig.getId());
                    videoItemMapper.update(item, qu);
                }
            }
        } catch (Exception e) {
            log.error("error：", e);
        }
    }

    /**
     * 高频更新isc监控状态
     */
    @Scheduled(cron = "*/10 * * * * ?")
    @RequestMapping("updateIscVideoState")
    public void updateIscVideoState() {
        try {
            List<ProjectVideoConfigBo> projectVideoConfigs = projectVideoConfigMapper.getIscConfigsGroupByConfig();
            Map<Long, List<VideoItem>> videoIdMap = videoItemService.list(new LambdaQueryWrapper<VideoItem>()
                            .select(VideoItem::getItemId, VideoItem::getVideoId, VideoItem::getDeviceState, VideoItem::getSerialNumber))
                    .stream()
                    .collect(Collectors.groupingBy(VideoItem::getVideoId));
            for (ProjectVideoConfigBo videoConfig : projectVideoConfigs) {
                // 收集需要更新的视频设备
                List<VideoItem> videoItemsToUpdate = new ArrayList<>();
                JSONObject resultJo;
                JSONArray alls = new JSONArray();
                int pageNo = 0;
                if (StrUtil.isBlank(videoConfig.getIds())) {
                    continue;
                }
                List<VideoItem> itemDbs = StrUtil.split(videoConfig.getIds(), ",").stream().flatMap(vid -> {
                    List<VideoItem> videoItems = videoIdMap.get(Convert.toLong(vid));
                    if (CollUtil.isNotEmpty(videoItems)) {
                        return videoItems.stream();
                    }
                    return null;
                }).filter(Objects::nonNull).collect(Collectors.toList());
                if (CollUtil.isEmpty(itemDbs)) {
                    continue;
                }
                try {
                    do {
                        resultJo = HikVideoUtil.callPostApiOnlineStatus(videoConfig.getAccount(),
                                videoConfig.getPassword(), videoConfig.getAppId(),
                                videoConfig.getAppSecret(), ++pageNo);
                        if (resultJo != null && "0".equals(resultJo.getString("code"))) {
                            alls.addAll(resultJo.getJSONObject("data").getJSONArray("list"));
                        } else {
                            log.error("高频更新isc监控状态返回错误:项目sn:{}:code:{},msg:{}",
                                    videoConfig.getProjectSn(), resultJo != null ? resultJo.getString("code") : "",
                                    resultJo != null ? resultJo.getString("msg") : "");
                            break;
                        }
                    } while (pageNo * 1000 < resultJo.getJSONObject("data").getInteger("total"));
                } catch (Exception e) {
                    log.error("高频更新isc监控状态返回异常", e);
                }
                Map<String, JSONObject> indexCodeMap = alls.stream().collect(Collectors.toMap(j -> ((JSONObject) j).getString("indexCode"), j -> ((JSONObject) j), (o1, o2) -> o1));
                for (VideoItem videoItem : itemDbs) {
                    JSONObject jsonObject = indexCodeMap.get(videoItem.getSerialNumber());
                    Integer newOnlineStatus = jsonObject != null && Objects.equals(jsonObject.getInteger("online"), 1) ? 1 : 2;
                    // 只有当设备状态发生变化时才更新
                    if (!newOnlineStatus.equals(videoItem.getDeviceState())) {
                        videoItem.setDeviceState(newOnlineStatus);
                        videoItemsToUpdate.add(videoItem);
                    }
                }
                // 批量更新
                if (CollUtil.isNotEmpty(videoItemsToUpdate)) {
                    List<Long> onlineIds = videoItemsToUpdate.stream().filter(v -> Objects.equals(v.getDeviceState(), 1)).map(VideoItem::getItemId).collect(Collectors.toList());
                    if (CollUtil.isNotEmpty(onlineIds)) {
                        videoItemService.update(null, new LambdaUpdateWrapper<VideoItem>()
                                .set(VideoItem::getDeviceState, 1)
                                .in(VideoItem::getItemId, onlineIds)
                        );
                    }
                    List<Long> offlineIds = videoItemsToUpdate.stream().filter(v -> Objects.equals(v.getDeviceState(), 2)).map(VideoItem::getItemId).collect(Collectors.toList());
                    if (CollUtil.isNotEmpty(offlineIds)) {
                        videoItemService.update(null, new LambdaUpdateWrapper<VideoItem>()
                                .set(VideoItem::getDeviceState, 2)
                                .in(VideoItem::getItemId, offlineIds)
                        );
                    }
                }
            }
        } catch (Exception e) {
            log.error("高频更新isc监控状态错误", e);
        }
    }

    /**
     * 高频更新执法记录仪的isc监控状态
     */
    @Scheduled(cron = "*/10 * * * * ?")
    @RequestMapping("updatePoliceCameraItemIscVideoState")
    public void updatePoliceCameraItemIscVideoState() {
        try {
            List<PoliceCameraVideoConfig> configs = policeCameraVideoConfigService.list(new QueryWrapper<>());
            Map<String, List<PoliceCameraItem>> projectSnMap2ItemsMap = policeCameraItemService.list(new LambdaQueryWrapper<PoliceCameraItem>()).stream().collect(Collectors.groupingBy(PoliceCameraItem::getProjectSn));
            Map<String, Map<String, JSONObject>> projectSn2IndexCodeMap = new HashMap<>();
            for (PoliceCameraVideoConfig videoConfig : configs) {
                // 收集需要更新的视频设备
                List<PoliceCameraItem> videoItemsToUpdate = new ArrayList<>();
                List<PoliceCameraItem> itemDbs = projectSnMap2ItemsMap.get(videoConfig.getProjectSn());
                if (CollUtil.isEmpty(itemDbs)) {
                    continue;
                }
                Map<String, JSONObject> indexCodeMap = projectSn2IndexCodeMap.computeIfAbsent(StrUtil.format("{},{},{},{}", videoConfig.getIp(),
                        videoConfig.getPort(), videoConfig.getAppId(),
                        videoConfig.getAppSecret()), key -> {
                    if (StrUtil.isBlank(videoConfig.getIp()) || StrUtil.isBlank(videoConfig.getPort()) || StrUtil.isBlank(videoConfig.getAppId()) || StrUtil.isBlank(videoConfig.getAppSecret())) {
                        return new HashMap<>();
                    }
                    JSONArray alls = new JSONArray();
                    try {
                        JSONObject resultJo;
                        int pageNo = 0;
                        do {
                            resultJo = HikVideoUtil.callPostApiOnlineStatus(videoConfig.getIp(),
                                    videoConfig.getPort(), videoConfig.getAppId(),
                                    videoConfig.getAppSecret(), ++pageNo);
                            if (resultJo != null && "0".equals(resultJo.getString("code"))) {
                                alls.addAll(resultJo.getJSONObject("data").getJSONArray("list"));
                            } else {
                                log.error("高频更新执法记录仪的isc监控状态返回错误:项目sn:{}:code:{},msg:{}",
                                        videoConfig.getProjectSn(), resultJo != null ? resultJo.getString("code") : "",
                                        resultJo != null ? resultJo.getString("msg") : "");
                                break;
                            }
                        } while (pageNo * 1000 < resultJo.getJSONObject("data").getInteger("total"));
                    } catch (Exception e) {
                        log.error("高频更新执法记录仪的isc监控状态返回异常", e);
                    }
                    return alls.stream().collect(Collectors.toMap(j -> ((JSONObject) j).getString("indexCode"), j -> ((JSONObject) j), (o1, o2) -> o1));
                });

                for (PoliceCameraItem item : itemDbs) {
                    JSONObject jsonObject = indexCodeMap.get(item.getMonitoringNumber());
                    Integer newOnlineStatus = jsonObject != null && Objects.equals(jsonObject.getInteger("online"), 1) ? 1 : 2;
                    // 只有当设备状态发生变化时才更新
                    if (!newOnlineStatus.equals(item.getDeviceState())) {
                        item.setDeviceState(newOnlineStatus);
                        videoItemsToUpdate.add(item);
                    }
                }
                // 批量更新
                if (CollUtil.isNotEmpty(videoItemsToUpdate)) {
                    List<Long> onlineIds = videoItemsToUpdate.stream().filter(v -> Objects.equals(v.getDeviceState(), 1)).map(PoliceCameraItem::getItemId).collect(Collectors.toList());
                    if (CollUtil.isNotEmpty(onlineIds)) {
                        policeCameraItemService.update(null, new LambdaUpdateWrapper<PoliceCameraItem>()
                                .set(PoliceCameraItem::getDeviceState, 1)
                                .in(PoliceCameraItem::getItemId, onlineIds)
                        );
                    }
                    List<Long> offlineIds = videoItemsToUpdate.stream().filter(v -> Objects.equals(v.getDeviceState(), 2)).map(PoliceCameraItem::getItemId).collect(Collectors.toList());
                    if (CollUtil.isNotEmpty(offlineIds)) {
                        policeCameraItemService.update(null, new LambdaUpdateWrapper<PoliceCameraItem>()
                                .set(PoliceCameraItem::getDeviceState, 2)
                                .in(PoliceCameraItem::getItemId, offlineIds)
                        );
                    }
                }
            }
        } catch (Exception e) {
            log.error("高频更新执法记录仪的isc监控状态返回错误", e);
        }
    }

    private void sendVideoData(VideoItem videoItem, ProjectVideoConfig videoConfig) {
        List<Map<String, Object>> list = new ArrayList<>();
        Map<String, Object> video = new HashMap<>(16);
        video.put("appId", videoConfig.getAppId());
        video.put("appSecret", videoConfig.getAppSecret());
        video.put("ip", videoConfig.getAccount());
        video.put("port", videoConfig.getPassword());
        video.put("serialNumber", videoItem.getSerialNumber());
        list.add(video);
        List<Long> alarmIdList = new ArrayList<>();
        alarmIdList.add(videoItem.getItemId());
        Map<String, Object> object = new HashMap<>(16);
        object.put("projectSn", videoConfig.getProjectSn());
        object.put("videoList", list);
        object.put("dataId", alarmIdList);
        object.put("resultUrl", serverUrl + "/xmgl/api/saveVideoAnalysisResult");
        object.put("type", 2);
        log.info(JSONUtil.toJsonStr(object));
        if (StringUtils.isNotEmpty(videoAnalysisUrl)) {
            asyncEnvironment.sendVideoAnalysisData(videoAnalysisUrl, object);
        }
    }

    /*public static void main(String[] args) {
        String Ip="182.101.141.23";
        String port="18443";
        String appkey="24017757";
        String appSecret="VJz0FbzmE6drPQ7egsBi";
        String url= HikVideoUtil.callPostApiGetPreviewURL("34ad8813a79d4cbaaebdbeaddaa53598", "rtmp",null,Ip,
                port,appkey, appSecret);
        log.info(url);
        *//*boolean status= VideoUtils.getVideoOnlineSate(url);
        log.info(status);*//*

    }*/


}