#include "STDInclude.hpp"

namespace Utils
{
	namespace Compression
	{
		std::string ZLib::Compress(std::string data)
		{
			z_stream stream;
			ZeroMemory(&stream, sizeof(stream));

			char* buffer = new char[data.size() * 2];

			if (deflateInit(&stream, Z_BEST_COMPRESSION) != Z_OK) 
			{ 
				delete[] buffer;
				return "";
			}

			stream.next_out = reinterpret_cast<uint8_t*>(buffer);
			stream.next_in = reinterpret_cast<const uint8_t*>(data.data());
			stream.avail_out = data.size() * 2;
			stream.avail_in = data.size();

			if (deflate(&stream, Z_FINISH) != Z_STREAM_END)
			{ 
				delete[] buffer;
				return "";
			}

			if (deflateEnd(&stream) != Z_OK)
			{
				delete[] buffer;
				return "";
			}

			data.clear();
			data.append(buffer, stream.total_out);

			delete[] buffer;

			return data;
		}

		std::string ZLib::Decompress(std::string data)
		{
			//#error "Not implemented yet!"
			return data;
		}
	};
}