#pragma once

#include <vector>
#include <cstdint>
#include <cstring>
#include <stdexcept>
#include <algorithm>
#include <iostream>
#define ZSTD_STATIC_LINKING_ONLY
#include <zstd.h>

#if ZSTD_VERSION_NUMBER < 10400
#error "Wymagane zstd >= 1.4.0 dla ZSTD_c_format / ZSTD_f_zstd1_magicless"
#endif

#include "EncryptionManager.h"

#define BLOCK_SIZE          131072 // 128KB
#define COMPRESSION_LEVEL   3

struct BlockSize
{
	uint32_t raw;
	uint32_t zip;
};

class ChunkManager
{
public:
	ChunkManager(EncryptionManager& em);
	~ChunkManager();

	// Kompresja danych
	std::vector<char> chunked(const std::vector<char>&, const bool&, const bool&);

	// Dekompresja
	std::vector<char> dechunked(const std::vector<char>&, const bool&, const bool&);

private:
	EncryptionManager eman;
	std::vector<BlockSize> blockSizes;

	// Przekonwertuj zmienną na ciąg na vector
	template <typename T>
	void addIntToVector(std::vector<char>& vec, const T& val)
	{
		size_t tmpSize = vec.size();
		vec.resize(tmpSize + sizeof(val));
		std::memcpy(vec.data() + tmpSize, &val, sizeof(val));
	}

	// Pobierz zmienną z vectora
	template <typename T>
	T getIntFromVector(const std::vector<char>& vec, size_t& offset)
	{
		T tmp{};
		std::memcpy(&tmp, vec.data() + offset, sizeof(tmp));
		offset += sizeof(tmp);
	
		return tmp;
	}

	// Kompresja
	std::vector<char> compress_data(const std::vector<char>&);

	// Dekompresja
	std::vector<char> decompress_data(const std::vector<char>&, const size_t&);
};