/*
 * Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
 *
 * This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
 *
 * Use of this source code is governed by MIT-like 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_WEBRTC_ICE_SESSION_H
#define ZLMEDIAKIT_WEBRTC_ICE_SESSION_H

#include "Network/Session.h"
#include "IceTransport.hpp"
#include "Http/HttpRequestSplitter.h"

namespace mediakit {

class IceSession : public toolkit::Session, public RTC::IceTransport::Listener, public HttpRequestSplitter {
public:
    using Ptr = std::shared_ptr<IceSession>;
    using WeakPtr = std::weak_ptr<IceSession>;
    IceSession(const toolkit::Socket::Ptr &sock);
    ~IceSession() override;

    static toolkit::EventPoller::Ptr queryPoller(const toolkit::Buffer::Ptr &buffer);

    //// Session override////
    // void attachServer(const Server &server) override;
    void onRecv(const toolkit::Buffer::Ptr &) override;
    void onError(const toolkit::SockException &err) override;
    void onManager() override;

    // ice related callbacks ///
    void onIceTransportRecvData(const toolkit::Buffer::Ptr& buffer, const RTC::IceTransport::Pair::Ptr& pair) override;
    void onIceTransportGatheringCandidate(const RTC::IceTransport::Pair::Ptr& pair, const RTC::CandidateInfo& candidate) override;
    void onIceTransportDisconnected() override;
    void onIceTransportCompleted() override;

    //// HttpRequestSplitter override ////
    ssize_t onRecvHeader(const char *data, size_t len) override;
    const char *onSearchPacketTail(const char *data, size_t len) override;

    void onRecv_l(const char *data, size_t len);
protected:
    bool _over_tcp = false;

    RTC::IceTransport::Pair::Ptr _session_pair = nullptr;
    RTC::IceServer::Ptr _ice_transport;
};

class IceSessionManager {
public:
    static IceSessionManager &Instance();
    IceSession::Ptr getItem(const std::string& key);
    void addItem(const std::string& key, const IceSession::Ptr &ptr);
    void removeItem(const std::string& key);

private:
    IceSessionManager() = default;

private:
    std::mutex _mtx;
    std::unordered_map<std::string, std::weak_ptr<IceSession>> _map;
};
}// namespace mediakit

#endif //ZLMEDIAKIT_WEBRTC_ICE_SESSION_H
