#ifndef MS_RTC_ICE_ICE_CANDIDATE_HPP
#define MS_RTC_ICE_ICE_CANDIDATE_HPP

#include "common.hpp"
#include "FBS/webRtcTransport.h"
#include "RTC/TcpServer.hpp"
#include "RTC/TransportTuple.hpp"
#include "RTC/UdpSocket.hpp"
#include <flatbuffers/flatbuffers.h>
#include <string>

namespace RTC
{
	namespace ICE
	{
		class IceCandidate
		{
			using Protocol = TransportTuple::Protocol;

		public:
			enum class CandidateType : uint8_t
			{
				HOST = 1
			};

		public:
			enum class TcpCandidateType : uint8_t
			{
				PASSIVE = 1
			};

		public:
			static CandidateType CandidateTypeFromFbs(FBS::WebRtcTransport::IceCandidateType type);
			static FBS::WebRtcTransport::IceCandidateType CandidateTypeToFbs(CandidateType type);
			static TcpCandidateType TcpCandidateTypeFromFbs(FBS::WebRtcTransport::IceCandidateTcpType type);
			static FBS::WebRtcTransport::IceCandidateTcpType TcpCandidateTypeToFbs(TcpCandidateType type);

		public:
			IceCandidate(RTC::UdpSocket* udpSocket, uint32_t priority)
			  : foundation("udpcandidate"),
			    priority(priority),
			    address(udpSocket->GetLocalIp()),
			    protocol(Protocol::UDP),
			    port(udpSocket->GetLocalPort())
			{
			}

			IceCandidate(RTC::UdpSocket* udpSocket, uint32_t priority, std::string& announcedAddress)
			  : foundation("udpcandidate"),
			    priority(priority),
			    address(announcedAddress),
			    protocol(Protocol::UDP),
			    port(udpSocket->GetLocalPort())
			{
			}

			IceCandidate(RTC::TcpServer* tcpServer, uint32_t priority)
			  : foundation("tcpcandidate"),
			    priority(priority),
			    address(tcpServer->GetLocalIp()),
			    protocol(Protocol::TCP),
			    port(tcpServer->GetLocalPort())

			{
			}

			IceCandidate(RTC::TcpServer* tcpServer, uint32_t priority, std::string& announcedAddress)
			  : foundation("tcpcandidate"),
			    priority(priority),
			    address(announcedAddress),
			    protocol(Protocol::TCP),
			    port(tcpServer->GetLocalPort())

			{
			}

			flatbuffers::Offset<FBS::WebRtcTransport::IceCandidate> FillBuffer(
			  flatbuffers::FlatBufferBuilder& builder) const;

		private:
			// Others.
			std::string foundation;
			uint32_t priority{ 0u };
			std::string address;
			Protocol protocol;
			uint16_t port{ 0u };
			CandidateType type{ CandidateType::HOST };
			TcpCandidateType tcpType{ TcpCandidateType::PASSIVE };
		};
	} // namespace ICE
} // namespace RTC

#endif