#include "stdafx.h"
#include <Poco/Delegate.h>
#include <Poco/Zip/Compress.h>
#include <Poco/Zip/Decompress.h>
#include <fstream>
#include <iomanip>
#include <iostream>
#include <sstream>

struct ErrorDelegateThrower;
using ErrorDelegate = Poco::Delegate<ErrorDelegateThrower, std::pair<const Poco::Zip::ZipLocalFileHeader, const std::string>>;
struct ErrorDelegateThrower: public ErrorDelegate {
	ErrorDelegateThrower()
		: ErrorDelegate(this, &ErrorDelegateThrower::dothrow)
	{
	}

	void dothrow(const void* src, std::pair<const Poco::Zip::ZipLocalFileHeader, const std::string>& s)
	{
		std::cout << "ErrorDelegateThrower::doThrow(): " << s.second << '\n';
		throw std::runtime_error("Zip: " + s.second);
	}
};

int main(int /* unreferenced argc */, char* argv[])
{
	try {
		std::string path = argv[0];
		path = path.substr(0, path.rfind('\\'));

		{
			std::cout << "Preparing a large file with random data...\n";
			srand(0);
			std::string data(5.25*1024*1024*1024 /*5.25GB*/, char(0));
			for(int d = 0; d<data.size(); ++d)
				data[d] = char(rand());
			std::ofstream dummy;
			dummy.open(path + "/dummy.bin", std::ios_base::binary | std::ios_base::out);
			if (!dummy.good())
				throw std::runtime_error("failed to open dummy file");
			dummy.write(data.data(), data.size());
			dummy.close();
		}

		{
			std::cout << "Compressing...\n";
			std::ofstream output;
			output.open(path + "/dummy.zip", std::ios_base::binary | std::ios_base::out);
			if (!output.good())
				throw std::runtime_error("failed to open output zip file");
			Poco::Zip::Compress zip(output, false);
			zip.addFile(path + "/dummy.bin", "dummy_zipped.bin", Poco::Zip::ZipCommon::CM_STORE);
			zip.close();
			output.close();
		}

		{
			std::cout << "Decompressing...\n";
			std::ifstream input;
			input.open(path + "/dummy.zip", std::ios_base::binary | std::ios_base::in);
			if (!input.good())
				throw std::runtime_error("failed to open zip file");

			Poco::Zip::Decompress zip(input, path);

			ErrorDelegateThrower t;
			zip.EError += t;

			zip.decompressAllFiles();
			input.close();
		}
	}
	catch (const Poco::Exception& e) {
		std::cout << "Poco error: " << e.displayText() << "\n";
	}
	catch (const std::exception& e) {
		std::cout << "Error: " << e.what() << "\n";
	}

	return 0;
}