mediasoup-sys 0.5.4

#ifndef MS_PAYLOAD_CHANNEL_SOCKET_HPP
#define MS_PAYLOAD_CHANNEL_SOCKET_HPP

#include "common.hpp"
#include "PayloadChannel/PayloadChannelNotification.hpp"
#include "PayloadChannel/PayloadChannelRequest.hpp"
#include "handles/UnixStreamSocket.hpp"
#include <nlohmann/json.hpp>
#include <string>

using json = nlohmann::json;

namespace PayloadChannel
{
	class ConsumerSocket : public ::UnixStreamSocket
	{
	public:
		class Listener
		{
		public:
			virtual ~Listener() = default;

		public:
			virtual void OnConsumerSocketMessage(ConsumerSocket* consumerSocket, char* msg, size_t msgLen) = 0;
			virtual void OnConsumerSocketClosed(ConsumerSocket* consumerSocket) = 0;
		};

	public:
		ConsumerSocket(int fd, size_t bufferSize, Listener* listener);
		~ConsumerSocket();

		/* Pure virtual methods inherited from ::UnixStreamSocket. */
	public:
		void UserOnUnixStreamRead() override;
		void UserOnUnixStreamSocketClosed() override;

	private:
		// Passed by argument.
		Listener* listener{ nullptr };
	};

	class ProducerSocket : public ::UnixStreamSocket
	{
	public:
		ProducerSocket(int fd, size_t bufferSize);

		/* Pure virtual methods inherited from ::UnixStreamSocket. */
	public:
		void UserOnUnixStreamRead() override
		{
		}
		void UserOnUnixStreamSocketClosed() override
		{
		}
	};

	class PayloadChannelSocket : public ConsumerSocket::Listener
	{
	public:
		class RequestHandler
		{
		public:
			virtual ~RequestHandler() = default;

		public:
			virtual void HandleRequest(PayloadChannel::PayloadChannelRequest* request) = 0;
		};

		class NotificationHandler
		{
		public:
			virtual ~NotificationHandler() = default;

		public:
			virtual void HandleNotification(PayloadChannel::PayloadChannelNotification* notification) = 0;
		};

		class Listener : public RequestHandler, public NotificationHandler
		{
		public:
			virtual ~Listener() = default;

		public:
			virtual void OnPayloadChannelClosed(PayloadChannel::PayloadChannelSocket* payloadChannel) = 0;
		};

	public:
		explicit PayloadChannelSocket(int consumerFd, int producerFd);
		explicit PayloadChannelSocket(
		  PayloadChannelReadFn payloadChannelReadFn,
		  PayloadChannelReadCtx payloadChannelReadCtx,
		  PayloadChannelWriteFn payloadChannelWriteFn,
		  PayloadChannelWriteCtx payloadChannelWriteCtx);
		virtual ~PayloadChannelSocket();

	public:
		void Close();
		void SetListener(Listener* listener);
		void Send(json& jsonMessage, const uint8_t* payload, size_t payloadLen);
		void Send(const std::string& message, const uint8_t* payload, size_t payloadLen);
		void Send(json& jsonMessage);
		void Send(const std::string& message);
		bool CallbackRead();

	private:
		void SendImpl(const uint8_t* message, uint32_t messageLen);
		void SendImpl(
		  const uint8_t* message, uint32_t messageLen, const uint8_t* payload, uint32_t payloadLen);

		/* Pure virtual methods inherited from ConsumerSocket::Listener. */
	public:
		void OnConsumerSocketMessage(ConsumerSocket* consumerSocket, char* msg, size_t msgLen) override;
		void OnConsumerSocketClosed(ConsumerSocket* consumerSocket) override;

	private:
		// Passed by argument.
		Listener* listener{ nullptr };
		// Others.
		bool closed{ false };
		ConsumerSocket* consumerSocket{ nullptr };
		ProducerSocket* producerSocket{ nullptr };
		PayloadChannelReadFn payloadChannelReadFn{ nullptr };
		PayloadChannelReadCtx payloadChannelReadCtx{ nullptr };
		PayloadChannelWriteFn payloadChannelWriteFn{ nullptr };
		PayloadChannelWriteCtx payloadChannelWriteCtx{ nullptr };
		PayloadChannel::PayloadChannelNotification* ongoingNotification{ nullptr };
		PayloadChannel::PayloadChannelRequest* ongoingRequest{ nullptr };
		uv_async_t* uvReadHandle{ nullptr };
		uint8_t* writeBuffer{ nullptr };
	};
} // namespace PayloadChannel

#endif