iw4x-client/src/Utils/String.hpp

#pragma once

namespace Utils::String
{
	template <std::size_t Buffers, std::size_t MinBufferSize>
	class VAProvider
	{
	public:
		static_assert(Buffers != 0 && MinBufferSize != 0, "Buffers and MinBufferSize mustn't be 0");

		VAProvider() : currentBuffer(0) {}
		~VAProvider() = default;

		const char* get(const char* format, va_list ap)
		{
			++this->currentBuffer %= ARRAYSIZE(this->stringPool);
			auto entry = &this->stringPool[this->currentBuffer];

			if (!entry->size || !entry->buffer)
			{
				throw std::runtime_error("String pool not initialized");
			}

			while (true)
			{
				const auto res = vsnprintf_s(entry->buffer, entry->size, _TRUNCATE, format, ap);
				if (res > 0) break; // Success
				if (res == 0) return ""; // Error

				entry->doubleSize();
			}

			return entry->buffer;
		}

	private:
		class Entry
		{
		public:
			Entry(std::size_t _size = MinBufferSize) : size(_size), buffer(nullptr)
			{
				if (this->size < MinBufferSize) this->size = MinBufferSize;
				this->allocate();
			}

			~Entry()
			{
				if (this->buffer) Memory::GetAllocator()->free(this->buffer);
				this->size = 0;
				this->buffer = nullptr;
			}

			void allocate()
			{
				if (this->buffer) Memory::GetAllocator()->free(this->buffer);
				this->buffer = Memory::GetAllocator()->allocateArray<char>(this->size + 1);
			}

			void doubleSize()
			{
				this->size *= 2;
				this->allocate();
			}

			std::size_t size;
			char* buffer;
		};

		std::size_t currentBuffer;
		Entry stringPool[Buffers];
	};

	const char *VA(const char *fmt, ...);

	std::string ToLower(std::string text);
	std::string ToUpper(std::string text);
	bool Compare(const std::string& lhs, const std::string& rhs);
	std::vector<std::string> Split(const std::string& str, char delim);
	void Replace(std::string& str, const std::string& from, const std::string& to);
	bool StartsWith(const std::string& haystack, const std::string& needle);
	bool EndsWith(const std::string& haystack, const std::string& needle);
	bool IsNumber(const std::string& str);

	std::string& LTrim(std::string& str);
	std::string& RTrim(std::string& str);
	std::string& Trim(std::string& str);

	std::string Convert(const std::wstring& wstr);
	std::wstring Convert(const std::string& str);

	std::string FormatTimeSpan(int milliseconds);
	std::string FormatBandwidth(std::size_t bytes, int milliseconds);

	std::string DumpHex(const std::string& data, const std::string& separator = " ");

	std::string XOR(std::string str, char value);

	std::string EncodeBase64(const char* input, const unsigned long inputSize);
	std::string EncodeBase64(const std::string& input);

	std::string EncodeBase128(const std::string& input);
}