/*
 * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
 *
 * This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
 *
 * Use of this source code is governed by MIT license that can be found in the
 * LICENSE file in the root of the source tree. All contributing project authors
 * may be found in the AUTHORS file in the root of the source tree.
 */

#ifndef ZLMEDIAKIT_RTPSELECTOR_H
#define ZLMEDIAKIT_RTPSELECTOR_H

#if defined(ENABLE_RTPPROXY)
#include <stdint.h>
#include <mutex>
#include <unordered_map>
#include "RtpProcess.h"
#include "Common/MediaSource.h"

namespace mediakit{

class RtpSelector;
class RtpProcessHelper : public MediaSourceEvent , public std::enable_shared_from_this<RtpProcessHelper> {
public:
    using Ptr = std::shared_ptr<RtpProcessHelper>;
    RtpProcessHelper(const std::string &stream_id, const std::weak_ptr<RtpSelector > &parent);
    ~RtpProcessHelper();
    void attachEvent();
    RtpProcess::Ptr & getProcess();

protected:
    // 通知其停止推流
    bool close(MediaSource &sender) override;

private:
    std::string _stream_id;
    RtpProcess::Ptr _process;
    std::weak_ptr<RtpSelector> _parent;
};

class RtpSelector : public std::enable_shared_from_this<RtpSelector>{
public:
    RtpSelector() = default;
    ~RtpSelector() = default;

    class ProcessExisted : public std::runtime_error {
    public:
        template<typename ...T>
        ProcessExisted(T && ...args) : std::runtime_error(std::forward<T>(args)...) {}
        ~ProcessExisted() override = default;
    };

    static bool getSSRC(const char *data,size_t data_len, uint32_t &ssrc);
    static RtpSelector &Instance();

    /**
     * 清空所有对象
     */
    void clear();

    /**
     * 获取一个rtp处理器
     * @param stream_id 流id
     * @param makeNew 不存在时是否新建, 该参数为true时,必须确保之前未创建同名对象
     * @return rtp处理器
     */
    RtpProcess::Ptr getProcess(const std::string &stream_id, bool makeNew);

    /**
     * 删除rtp处理器
     * @param stream_id 流id
     * @param ptr rtp处理器指针
     */
    void delProcess(const std::string &stream_id, const RtpProcess *ptr);

private:
    void onManager();
    void createTimer();

private:
    toolkit::Timer::Ptr _timer;
    std::recursive_mutex _mtx_map;
    std::unordered_map<std::string,RtpProcessHelper::Ptr> _map_rtp_process;
};

}//namespace mediakit
#endif//defined(ENABLE_RTPPROXY)
#endif //ZLMEDIAKIT_RTPSELECTOR_H