sweeper_video/include/rtmp_manager.hpp

// rtmp_manager.hpp
#pragma once

#include <gst/gst.h>

#include <atomic>
#include <functional>
#include <future>
#include <memory>
#include <mutex>
#include <string>
#include <thread>
#include <unordered_map>
#include <vector>

#include "app_config.hpp"
#include "data_manager.hpp"

class RTMPManager
{
   public:
    enum class StreamResult
    {
        OK,
        PIPELINE_ERROR,
        CONNECTION_FAIL,
        EOS_RECEIVED,
        TIMEOUT,
        UNKNOWN
    };

    struct StreamStatus
    {
        bool running{false};
        StreamResult last_result{StreamResult::UNKNOWN};
        std::string last_error;
    };

    struct StreamResultInfo
    {
        int loc{-1};
        std::string url;
        int result{1};  // 0 success, 1 fail
        std::string reason;
    };

    using StreamCallback = std::function<void(const std::string &key, StreamStatus)>;

    static void init();
    static void set_status_callback(StreamCallback cb);

    // 单路接口
    static StreamResultInfo start_camera(const Camera &cam, StreamType type);
    static StreamResultInfo stop_camera(const std::string &cam_name, StreamType type);

    // 批量接口：处理 VideoPushRequest
    static std::vector<StreamResultInfo> process_push_request(const VideoPushRequest &req);

    static void stop_all();
    // 自动启动所有录像流（上电调用一次）
    static void start_all_record_streams();
    static bool is_streaming(const std::string &cam_name, StreamType type);
    static bool is_any_streaming();
    static std::string get_stream_url(const std::string &cam_name, StreamType type);

   private:
    struct StreamContext
    {
        std::atomic<bool> running{false};
        std::thread thread;

        // 启动阶段的 promise，用于首次启动结果反馈（7s 内）
        std::promise<StreamStatus> start_promise;
        std::atomic<bool> start_promise_set{false};

        StreamStatus status;

        StreamContext() = default;
        StreamContext(const StreamContext &) = delete;
        StreamContext &operator=(const StreamContext &) = delete;

        ~StreamContext()
        {
            try
            {
                running.store(false);
                if (thread.joinable())
                {
                    thread.join();  // 防止 std::terminate()
                }
            }
            catch (...)
            {
                // 安全兜底，不抛异常
            }
        }
    };

    static std::string make_stream_key(const std::string &cam_name, StreamType type);
    static GstElement *create_pipeline(const Camera &cam, StreamType type);
    static void stream_loop(Camera cam, StreamType type, StreamContext *ctx);
    static bool do_stream_once(const Camera &cam, StreamType type, StreamContext *ctx);

    static void update_status(const std::string &key, const StreamStatus &status);

    static std::unordered_map<std::string, std::unique_ptr<StreamContext>> streams;
    static std::mutex streams_mutex;
    static StreamCallback status_callback;

    // 自动重试参数
    // MAIN（record）为永久重试；SUB（live）使用限次重试
    static constexpr int LIVE_MAX_RETRIES = 5;
    static constexpr int RETRY_BASE_DELAY_MS = 2000;  // 2s -> 可指数退避
};