package.h

#pragma once
#pragma comment(lib, "bcrypt.lib")
#include <string>
#include <vector>
#include <array>
#include <windows.h>
#include <stdio.h>
#include <cstdio>
#include <stdlib.h>
#include <string>
#include <filesystem>
#include <bcrypt.h>
#include <set>
#include "helpers.h"
#include <unordered_map>
#include <boost/algorithm/string.hpp>
#include "E:/ww2ogg/src/wwriff.h"
#include "D:\vcpkg\packages\xxhash_x64-windows\include\xxh3.h"

//extern "C"
//{
//	#include <revorb.h>
//}

std::unordered_map<uint64_t, uint32_t> loadH64Table();
std::unordered_map<uint64_t, uint32_t> generateH64Table(std::string packagesPath);
bool saveH64Table(std::unordered_map<uint64_t, uint32_t> hash64Table);

struct PkgHeader
{
	uint16_t pkgID;
	uint16_t patchID;
	uint32_t entryTableOffset;
	uint32_t entryTableSize;
	uint32_t blockTableOffset;
	uint32_t blockTableSize;
	uint32_t hash64TableOffset;
	uint32_t hash64TableSize;
};

struct Entry
{
	std::string reference;
	std::string reference2;
	uint8_t numType;
	uint8_t numSubType;
	uint32_t startingBlock;
	uint32_t startingBlockOffset;
	uint32_t fileSize;
};

struct Block
{
	uint32_t ID;
	uint32_t offset;
	uint32_t size;
	uint16_t patchID;
	uint16_t bitFlag;
	uint8_t gcmTag[16];
};

typedef int64_t(*OodleLZ64_DecompressDef)(unsigned char* Buffer, int64_t BufferSize, unsigned char* OutputBuffer, int64_t OutputBufferSize, int32_t a, int32_t b, int64_t c, void* d, void* e, void* f, void* g, void* h, void* i, int32_t ThreadModule);

/*
* Handles the separation of a .pkg file into its constituent binary files.
* It will unpack the PatchID given, so the latest should be given if updates are being processed.
*/
class Package
{
private:
	unsigned char nonce[12] =
	{
		0x84, 0xEA, 0x11, 0xC0, 0xAC, 0xAB, 0xFA, 0x20, 0x33, 0x11, 0x26, 0x99,
	};

	unsigned char preblnonce[12] =
	{
		0x84, 0xDF, 0x11, 0xC0, 0xAC, 0xAB, 0xFA, 0x20, 0x33, 0x11, 0x26, 0x99,
	};

	const std::string CUSTOM_DIR = "output";

	std::vector<Block> blocks;

	int64_t OodleLZ_Decompress;
	HMODULE hOodleDll;
	//HMODULE hRevorbDll = LoadLibrary(L"res\\librevorb.dll");
	//typedef int (*RevorbDef)(unsigned char*& fi, unsigned char*& fo);
	//RevorbDef revorb = (RevorbDef)GetProcAddress(hRevorbDll, "revorb");

	std::vector<std::string> pkgPatchStreamPaths;
	void getBlockTable();
	void extractFiles();
	void decryptBlock(Block block, unsigned char* blockBuffer, unsigned char* &decryptBuffer);
	void decompressBlock(Block block, unsigned char* decryptBuffer, unsigned char*& decompBuffer);
	unsigned char* genericExtract(std::vector<std::string> pkgPatchStreamPaths, int i);
	//bool Extract(std::vector<Entry> entries, int i, int wemType, int wemSubType, int bnkType, int bnkSubType, int bnkSubType2, std::string wavOutput, std::vector<std::string> pkgPatchStreamPaths, std::string outputPath, std::string bnkOutputPath);
	bool getWem(int i, std::string outputPath, std::string Hambit, std::string nameID, Entry entry);
	bool getBnk(int i, std::string bnkOutputPath, Entry entry);
public:
	FILE* pkgFile;
	std::string packagesPath;
	std::string packagePath;
	std::string packageName;
	std::string version;
	bool d1prebl = false;
	PackageOptions options;

	bool ps3_x360 = false;

	PkgHeader header;
	std::vector<Entry> entries;
	std::unordered_map<std::string, std::unordered_map<std::string, XXH64_hash_t>> HashMap;

	// Constructor
	Package(std::string packageID, std::string pkgsPath, bool prebl_d1);

	bool initOodle();
	void modifyNonce();
	bool readHeader();
	bool Unpack();
	void getEntryTable();
	std::string getEntryReference(std::string hash);
	uint8_t getEntryTypes(std::string hash, uint8_t& subType);
	std::string getLatestPatchIDPath(std::string packageName);
	unsigned char* getEntryData(std::string hash, int& fileSize);
	std::vector<std::string> getAllFilesGivenRef(std::string reference);
	unsigned char* getBufferFromEntry(Entry entry);
};