package com.ruoyi.media.service.impl;

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

import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.device.camera.domain.ArdCameras;
import com.ruoyi.device.camera.domain.CameraCmd;
import com.ruoyi.device.camera.factory.CameraSDK;
import com.ruoyi.device.camera.factory.CameraSDKFactory;
import com.ruoyi.device.channel.domain.ArdChannel;
import com.ruoyi.media.domain.StreamInfo;
import com.ruoyi.media.service.IMediaService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import com.ruoyi.media.mapper.VtduMapper;
import com.ruoyi.media.domain.Vtdu;
import com.ruoyi.media.service.IVtduService;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

/**
 * 流媒体管理Service业务层处理
 *
 * @author ard
 * @date 2023-08-29
 */
@Service
@Slf4j(topic = "vtdu")
public class VtduServiceImpl implements IVtduService {
    @Resource
    private VtduMapper vtduMapper;
    @Resource
    private IMediaService mediaService;
    @Resource
    private CameraSDKFactory cameraSDKFactory;
    @Value("${mediamtx.host}")
    String mediamtxHost;
    /**
     * 查询流媒体管理
     *
     * @param name 流媒体管理主键
     * @return 流媒体管理
     */
    @Override
    public Vtdu selectVtduByName(String name) {
        return vtduMapper.selectVtduByName(name);
    }

    /**
     * 查询流媒体管理列表
     *
     * @param vtdu 流媒体管理
     * @return 流媒体管理
     */
    @Override
    public List<Vtdu> selectVtduList(Vtdu vtdu) {
        return vtduMapper.selectVtduList(vtdu);
    }

    /**
     * 新增流媒体管理
     *
     * @param vtdu 流媒体管理
     * @return 结果
     */
    @Override
    public int insertVtdu(Vtdu vtdu) {
        return vtduMapper.insertVtdu(vtdu);
    }

    /**
     * 修改流媒体管理
     *
     * @param vtdu 流媒体管理
     * @return 结果
     */
    @Override
    public int updateVtdu(Vtdu vtdu) {
        Map<String, String> map = mediaService.editPath(vtdu.getName(), vtdu.getRtspSource(), vtdu.getMode(), vtdu.getIsCode());
        vtdu.setName(vtdu.getName());
        vtdu.setRtspSource(vtdu.getRtspSource());
        vtdu.setIsCode(vtdu.getIsCode());
        vtdu.setMode(vtdu.getMode());
        vtdu.setRtspUrl(map.get("rtspUrl"));
        vtdu.setRtmpUrl(map.get("rtmpUrl"));
        vtdu.setWebrtcUrl(map.get("webrtcUrl"));
        vtdu.setUpdateTime(DateUtils.getNowDate());
        return vtduMapper.updateVtdu(vtdu);
    }

    /**
     * 批量删除流媒体管理
     *
     * @param names 需要删除的数据主键集合
     * @return 结果
     */
    @Override
    public int deleteVtduByNames(String[] names) {
        for (String name : names) {
            log.debug("流媒体【" + name + "】通道删除");
        }
        mediaService.removePath(names);
        return vtduMapper.deleteVtduByNames(names);
    }

    /**
     * 删除流媒体管理
     *
     * @param name 流媒体通道名称
     * @return 结果
     */
    @Override
    public int deleteVtduByName(String name) {
        mediaService.removePath(name);
        log.debug("流媒体【" + name + "】通道删除");
        return vtduMapper.deleteVtduByName(name);
    }

    /**
     * 删除流媒体管理信息
     *
     * @param cameraId 相机ID
     * @return 结果
     */
    @Override
    public int deleteVtduByCameraId(String cameraId) {
        Vtdu vtdu = new Vtdu();
        vtdu.setCameraId(cameraId);
        List<Vtdu> vtdus = vtduMapper.selectVtduList(vtdu);
        if (vtdus.size() > 0) {
            for (Vtdu v : vtdus) {
                mediaService.removePath(v.getName());
            }
        }
        return vtduMapper.deleteVtduByCameraId(cameraId);
    }

    /**
     * @return
     * @Author 刘苏义
     * @Description 清除所有流媒体数据
     * @Date 2024/7/10 13:31
     * @Param
     */
    public void clearVtdu() {
        vtduMapper.clearVtdu();
    }


    //需要更新的（流媒体和vtdu相同）
    @Override
    public List<Vtdu> sameList(List<Vtdu> vtdus, List<String> names) {
        return vtdus.stream().filter(vtdu -> names.contains(vtdu.getName())).collect(Collectors.toList());
    }

    //需要删除的（流媒体中多的）
    @Override
    public List<String> diffListToDel(List<Vtdu> vtdus, List<String> names) {
        if (vtdus.size() >= names.size()) {
            List<String> nameList = vtdus.stream().map(Vtdu::getName)
                    .filter(item -> !names.contains(item))
                    .collect(Collectors.toList());
            return nameList;
        } else {
            List<String> nameList = names.stream().filter(item -> !vtdus.stream().map(Vtdu::getName).collect(Collectors.toList()).contains(item))
                    .collect(Collectors.toList());
            return nameList;
        }
    }

    //需要新增的（流媒体中少的）
    @Override
    public List<Vtdu> diffListToAdd(List<Vtdu> vtdus, List<String> names) {
        return vtdus.stream().filter(vtdu -> !names.contains(vtdu.getName())).collect(Collectors.toList());
    }

    /**
     * @return
     * @Author 刘苏义
     * @Description 同步本地vtdu库和流媒体中的数据
     * @Date 2024/7/10 15:26
     * @Param vtdus vtdu库的集合
     * @Param names 流媒体的name集合
     */
    @Override
    public void asyncVtdu(List<Vtdu> vtdus, List<String> names) {
        //需要更新的数据,参数顺序注意
        sameList(vtdus, names).stream().forEach(vtdu -> {
            StreamInfo streamInfo = mediaService.getPathInfo(vtdu.getName());
            if (!streamInfo.getRtspSource().equals(vtdu.getRtspSource())) {
                mediaService.editPath(vtdu.getName(), vtdu.getRtspSource(), vtdu.getMode(), vtdu.getIsCode());
            }
        });
        //需要删除的数据
        diffListToDel(vtdus, names).stream().forEach(name -> {
            mediaService.removePath(name);
        });
        //需要新增的数据
        diffListToAdd(vtdus, names).stream().forEach(vtdu -> {
            mediaService.addPath(vtdu.getName(), vtdu.getRtspSource(), vtdu.getMode(), vtdu.getIsCode());
        });
    }
    @Override
    public void addChanToVtdu(ArdCameras camera, ArdChannel channel) {
        String name = camera.getId() + "_" + channel.getChanNo();
        String rtspSource="";
        switch (camera.getFactory())
        {
            case "1": rtspSource = "rtsp://" + camera.getUsername() + ":" + camera.getPassword() + "@" + camera.getIp() + ":" + camera.getRtspPort() + "/h264/ch" + channel.getChanNo() + "/main/av_stream";break;
            case "2": rtspSource = "rtsp://" + camera.getUsername() + ":" + camera.getPassword() + "@" + camera.getIp() + ":" + camera.getRtspPort() + "/cam/realmonitor?channel=" + channel.getChanNo() + "&subtype=0";break;
        }

        //删除流媒体
        if (selectVtduByName(name) != null) {
            deleteVtduByName(name);
        }
        //添加到流媒体
        CameraCmd cmd = new CameraCmd(camera.getId(), channel.getChanNo());
        CameraSDK cameraSDK = cameraSDKFactory.createCameraSDK(camera.getFactory());
        Map<String, Object> videoCompressionCfg = cameraSDK.getVideoCompressionCfg(cmd);
        Vtdu vtdu = new Vtdu();
        if (videoCompressionCfg.get("videoEncType") != null) {
            if (videoCompressionCfg.get("videoEncType").equals("标准h264")) {
                vtdu.setIsCode("0");//默认不转码
            } else {
                vtdu.setIsCode("1");//默认转码
            }
        } else {
            vtdu.setIsCode("0");//默认不转码
        }
        vtdu.setRtspSource(rtspSource);
        vtdu.setName(name);
        vtdu.setMode("1");//默认CPU软解码
        vtdu.setCameraId(camera.getId());
        String rtspUrl = "rtsp://" + mediamtxHost + ":8554/" + name;
        String rtmpUrl = "rtmp://" + mediamtxHost + ":1935/" + name;
        String webrtcUrl = "http://" + mediamtxHost + ":8889/" + name;
        vtdu.setRtmpUrl(rtmpUrl);
        vtdu.setWebrtcUrl(webrtcUrl);
        vtdu.setRtspUrl(rtspUrl);
        insertVtdu(vtdu);
    }

}