package com.ard.work.sdk.zlxd.record;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.SmartLifecycle;
import org.springframework.stereotype.Component;
import java.io.File;
import java.util.HashMap;
import java.util.Map;

/**
 * 录制管理器（Spring 单例 Bean）
 */
@Slf4j
@Component // 声明为 Spring 组件，默认单例
public class RecorderManager implements SmartLifecycle {

    private String saveDir;
    // 存储摄像头ID -> 录制器的映射
    private final Map<String, RtspRecorder> recorderMap1 = new HashMap<>();
    private final Map<String, RtspToMP4> recorderMap = new HashMap<>();
    // Spring 生命周期标记
    private boolean isRunning = false;

    // 启动单个摄像头录制
    public boolean startRecorder(String uniqueCamId, String rtspUrl,String saveDir) {
        this.saveDir=saveDir;
        try {
            // 检查是否已在录制
            if (recorderMap.containsKey(uniqueCamId)) {
                log.warn("摄像头 {} 已在录制中，无需重复启动", uniqueCamId);
                return true;
            }
            // 构建录制目录
            // 创建录制器并启动
//            RtspRecorder recorder = new RtspRecorder(uniqueCamId, rtspUrl, tempDir);
//            recorder.start();
            RtspToMP4 recorder = new RtspToMP4();
            File dir = new File(saveDir);
            if (!dir.exists()) {
                boolean mkdirs = dir.mkdirs();
                if (!mkdirs) {
                    log.error("[{}] 创建保存目录失败：{}", uniqueCamId, saveDir);
                }
            }
            // ⭐ 可配置参数
            Long maxSize = 1073741824L; // 1GB
            Integer maxDuration = null; // 不限制时间
            String filePath = saveDir + File.separator + uniqueCamId + ".mp4";
            recorder.start(rtspUrl, filePath, maxSize, maxDuration);

            recorderMap.put(uniqueCamId, recorder);
            log.info("摄像头 {} 录制启动成功，存储路径：{}", uniqueCamId, saveDir);
            return true;
        } catch (Exception e) {
            log.error("摄像头 {} 录制启动失败", uniqueCamId, e);
            return false;
        }
    }

    // 停止单个摄像头录制
    public String stopRecorder(String uniqueCamId) {
        try {
            //RtspRecorder recorder = recorderMap.get(uniqueCamId);
            RtspToMP4  recorder = recorderMap.get(uniqueCamId);
            if (recorder == null) {
                log.warn("摄像头 {} 未在录制中，无需停止", uniqueCamId);
                return "摄像头未在录制";
            }
            // 停止录制
            recorder.stop();
            recorderMap.remove(uniqueCamId);
            // 返回录制文件路径（根据实际生成的文件名调整）
            String recordPath = saveDir + File.separator + uniqueCamId + ".mp4";
            log.info("摄像头 {} 录制停止成功，文件路径：{}", uniqueCamId, recordPath);
            return recordPath;
        } catch (Exception e) {
            log.error("摄像头 {} 录制停止失败", uniqueCamId, e);
            return "停止失败：" + e.getMessage();
        }
    }

    // 停止所有录制
    public void stopAllRecorders() {
        log.info("开始停止所有摄像头录制...");
        for (String camId : recorderMap.keySet()) {
            stopRecorder(camId);
        }
        recorderMap.clear();
        log.info("所有摄像头录制已停止");
    }

    // 检查摄像头是否在录制
    public boolean isRecording(String uniqueCamId) {
        //RtspRecorder recorder = recorderMap.get(uniqueCamId);
        RtspToMP4  recorder = recorderMap.get(uniqueCamId);
        return recorder != null && recorder.isRunning();
    }

    // ====================== Spring 生命周期方法 ======================
    @Override
    public void start() {
        isRunning = true;
    }

    @Override
    public void stop() {
        // Spring 关闭时，自动停止所有录制
        stopAllRecorders();
        isRunning = false;
        log.info("RecorderManager 已销毁");
    }

    @Override
    public boolean isRunning() {
        return isRunning;
    }

    // 优先级：数字越小，越先启动、越后销毁
    @Override
    public int getPhase() {
        return 0;
    }

    public static void main(String[] args) throws InterruptedException {
        RecorderManager recorder = new RecorderManager();
        recorder.startRecorder("cam1", "rtsp://admin:admin@112.98.104.134:8237/000100","D:\\home\\temp\\record");

        // 录制30秒
        log.info("开始录制，持续30秒...");
        Thread.sleep(30000);

        // 停止录制
        recorder.stop();
        log.info("录制已停止，文件可正常操作");

        // 验证文件是否可用
        File videoFile = new File("D:\\home\\temp\\cam1.mp4");
        if (videoFile.exists()) {
            log.info("录制文件已生成，大小：{} KB", videoFile.length() / 1024);
        } else {
            log.error("录制文件未生成");
        }
    }
}