#define MS_CLASS "RTC::RtpCodecMimeType"
// #define MS_LOG_DEV_LEVEL 3

#include "Logger.hpp"
#include "MediaSoupErrors.hpp"
#include "Utils.hpp"
#include "RTC/RtpDictionaries.hpp"

namespace RTC
{
	/* Class variables. */

	// clang-format off
	const absl::flat_hash_map<std::string, RtpCodecMimeType::Type> RtpCodecMimeType::String2Type =
	{
		{ "audio", RtpCodecMimeType::Type::AUDIO },
		{ "video", RtpCodecMimeType::Type::VIDEO }
	};
	const absl::flat_hash_map<RtpCodecMimeType::Type, std::string> RtpCodecMimeType::Type2String =
	{
		{ RtpCodecMimeType::Type::AUDIO, "audio" },
		{ RtpCodecMimeType::Type::VIDEO, "video" }
	};
	const absl::flat_hash_map<std::string, RtpCodecMimeType::Subtype> RtpCodecMimeType::String2Subtype =
	{
		// Audio codecs:
		{ "opus",            RtpCodecMimeType::Subtype::OPUS            },
		{ "multiopus",       RtpCodecMimeType::Subtype::MULTIOPUS       },
		{ "pcma",            RtpCodecMimeType::Subtype::PCMA            },
		{ "pcmu",            RtpCodecMimeType::Subtype::PCMU            },
		{ "isac",            RtpCodecMimeType::Subtype::ISAC            },
		{ "g722",            RtpCodecMimeType::Subtype::G722            },
		{ "ilbc",            RtpCodecMimeType::Subtype::ILBC            },
		{ "silk",            RtpCodecMimeType::Subtype::SILK            },
		// Video codecs:
		{ "vp8",             RtpCodecMimeType::Subtype::VP8             },
		{ "vp9",             RtpCodecMimeType::Subtype::VP9             },
		{ "h264",            RtpCodecMimeType::Subtype::H264            },
		{ "av1",             RtpCodecMimeType::Subtype::AV1             },
		// Complementary codecs:
		{ "cn",              RtpCodecMimeType::Subtype::CN              },
		{ "telephone-event", RtpCodecMimeType::Subtype::TELEPHONE_EVENT },
		// Feature codecs:
		{ "rtx",             RtpCodecMimeType::Subtype::RTX             },
		{ "ulpfec",          RtpCodecMimeType::Subtype::ULPFEC          },
		{ "flexfec",         RtpCodecMimeType::Subtype::FLEXFEC         },
		{ "x-ulpfecuc",      RtpCodecMimeType::Subtype::X_ULPFECUC      },
		{ "red",             RtpCodecMimeType::Subtype::RED             }
	};
	const absl::flat_hash_map<RtpCodecMimeType::Subtype, std::string> RtpCodecMimeType::Subtype2String =
	{
		// Audio codecs:
		{ RtpCodecMimeType::Subtype::OPUS,            "opus"            },
		{ RtpCodecMimeType::Subtype::MULTIOPUS,       "multiopus"       },
		{ RtpCodecMimeType::Subtype::PCMA,            "PCMA"            },
		{ RtpCodecMimeType::Subtype::PCMU,            "PCMU"            },
		{ RtpCodecMimeType::Subtype::ISAC,            "ISAC"            },
		{ RtpCodecMimeType::Subtype::G722,            "G722"            },
		{ RtpCodecMimeType::Subtype::ILBC,            "iLBC"            },
		{ RtpCodecMimeType::Subtype::SILK,            "SILK"            },
		// Video codecs:
		{ RtpCodecMimeType::Subtype::VP8,             "VP8"             },
		{ RtpCodecMimeType::Subtype::VP9,             "VP9"             },
		{ RtpCodecMimeType::Subtype::H264,            "H264"            },
		{ RtpCodecMimeType::Subtype::AV1,             "AV1"             },
		// Complementary codecs:
		{ RtpCodecMimeType::Subtype::CN,              "CN"              },
		{ RtpCodecMimeType::Subtype::TELEPHONE_EVENT, "telephone-event" },
		// Feature codecs:
		{ RtpCodecMimeType::Subtype::RTX,             "rtx"             },
		{ RtpCodecMimeType::Subtype::ULPFEC,          "ulpfec"          },
		{ RtpCodecMimeType::Subtype::FLEXFEC,         "flexfec"         },
		{ RtpCodecMimeType::Subtype::X_ULPFECUC,      "x-ulpfecuc"      },
		{ RtpCodecMimeType::Subtype::RED,             "red"             }
	};
	// clang-format on

	/* Instance methods. */

	void RtpCodecMimeType::SetMimeType(const std::string& mimeType)
	{
		MS_TRACE();

		auto slashPos = mimeType.find('/');

		if (slashPos == std::string::npos || slashPos == 0 || slashPos == mimeType.length() - 1)
		{
			MS_THROW_TYPE_ERROR("wrong codec MIME");
		}

		std::string type    = mimeType.substr(0, slashPos);
		std::string subtype = mimeType.substr(slashPos + 1);

		// Force lowcase names.
		Utils::String::ToLowerCase(type);
		Utils::String::ToLowerCase(subtype);

		// Set MIME type.
		{
			auto it = RtpCodecMimeType::String2Type.find(type);

			if (it == RtpCodecMimeType::String2Type.end())
			{
				MS_THROW_TYPE_ERROR("unknown codec MIME type '%s'", type.c_str());
			}

			this->type = it->second;
		}

		// Set MIME subtype.
		{
			auto it = RtpCodecMimeType::String2Subtype.find(subtype);

			if (it == RtpCodecMimeType::String2Subtype.end())
			{
				MS_THROW_TYPE_ERROR("unknown codec MIME subtype '%s'", subtype.c_str());
			}

			this->subtype = it->second;
		}

		// Set mimeType.
		this->mimeType = RtpCodecMimeType::Type2String.at(this->type) + "/" +
		                 RtpCodecMimeType::Subtype2String.at(this->subtype);
	}

	void RtpCodecMimeType::UpdateMimeType()
	{
		MS_TRACE();

		// Set mimeType.
		this->mimeType = RtpCodecMimeType::Type2String.at(this->type) + "/" +
		                 RtpCodecMimeType::Subtype2String.at(this->subtype);
	}
} // namespace RTC
