#pragma once

#include "CKDefines.hpp"
#include "CKEnums.hpp"

namespace LibCmo::CK2 {

	class CKBufferParser {
	private:
		char* m_MemBegin;
		size_t m_MemPos;
		bool m_NeedManualFree;
		size_t m_MemSize;

	public:
		CKBufferParser(void* ptr, size_t rsize, bool need_manual_free);
		CKBufferParser(const CKBufferParser&) = delete;
		CKBufferParser& operator=(const CKBufferParser&) = delete;
		~CKBufferParser();

		inline const void* GetPtr(void) { return (this->m_MemBegin + m_MemPos); }
		inline void Read(void* data, size_t data_size) {
			std::memcpy(data, (this->m_MemBegin + m_MemPos), data_size);
			this->m_MemPos += data_size;
		}
		inline void Write(const void* data, size_t data_size) {
			std::memcpy((this->m_MemBegin + m_MemPos), data, data_size);
			this->m_MemPos += data_size;
		}
		inline void* GetBase(void) { return this->m_MemBegin; }
		inline size_t GetSize(void) { return this->m_MemSize; }
		inline size_t GetCursor(void) { return this->m_MemPos; }
		inline void MoveCursor(size_t off) { this->m_MemPos += off; }
		inline void SetCursor(size_t off) { this->m_MemPos = off; }
	};

#pragma pack(push)
#pragma pack(1)
	struct CKRawFileInfo {
		CKBYTE NeMo[8];
		CKDWORD Crc;
		CKDWORD CKVersion;
		CKDWORD FileVersion;
		CKDWORD Zero;
		CKDWORD FileWriteMode;
		CKDWORD Hdr1PackSize;

		CKDWORD DataPackSize;
		CKDWORD DataUnPackSize;
		CKDWORD ManagerCount;
		CKDWORD ObjectCount;
		CKDWORD MaxIDSaved;
		CKDWORD ProductVersion;
		CKDWORD ProductBuild;
		CKDWORD Hdr1UnPackSize;
	};
#pragma pack(pop)

	class CKFileInfo {
	public:
		CKFileInfo();
		CKFileInfo(const CKFileInfo&);
		CKFileInfo& operator=(const CKFileInfo&);
		~CKFileInfo();

		CKDWORD ProductVersion;		// Virtools Version (Dev/Creation). (CK_VIRTOOLS_VERSION)
		CKDWORD ProductBuild;		// Virtools Build Number.
		CK_FILE_WRITEMODE FileWriteMode;		// Options used to save this file. (CK_FILE_WRITEMODE)
		CKDWORD FileVersion;		// Version of file format when file was saved.
		CKDWORD CKVersion;			// Version of CK when file was saved.
		size_t FileSize;			// Size of file in bytes.
		CKDWORD ObjectCount;		// Number of objects stored in the file. 	
		CKDWORD ManagerCount;		// Number of managers which saved data in the file.
		CKDWORD MaxIDSaved;			// Maximum Object identifier saved
		CKDWORD Crc;				// Crc of data
		CKDWORD Hdr1PackSize;		// Reserved
		CKDWORD Hdr1UnPackSize;		// Reserved
		CKDWORD DataPackSize;		// Reserved
		CKDWORD DataUnPackSize;		// Reserved
	private:

	};

	class CKFileObject {
	public:
		CKFileObject();
		CKFileObject(const CKFileObject&);
		CKFileObject& operator=(const CKFileObject&);
		~CKFileObject();

		CK_ID ObjectId;							// ID of the object being load/saved (as it will be/was saved in the file)
		CK_ID CreatedObject;	// ID of the object being created
		CK_CLASSID ObjectCid;					// Class Identifier of the object
		CKObjectImplements::CKObject* ObjPtr;	// A pointer to the object itself (as CreatedObject when loading)
		std::string Name;						// Name of the Object
		CKStateChunk* Data;						// A CKStateChunk that contains object information
		CKDWORD FileIndex;						// Position of the object data inside uncompressed file buffer
	private:

	};

	class CKFileManagerData {
	public:
		CKFileManagerData();
		CKFileManagerData(const CKFileManagerData&);
		CKFileManagerData& operator=(const CKFileManagerData&);
		~CKFileManagerData();

		CKManagerImplements::CKBaseManager* MgrPtr;
		CKStateChunk* Data;
		CKGUID Manager;
	private:

	};

	class CKFilePluginDependencies {
	public:
		CKFilePluginDependencies();
		CKFilePluginDependencies(const CKFilePluginDependencies&);
		CKFilePluginDependencies& operator=(const CKFilePluginDependencies&);
		~CKFilePluginDependencies();

		CK_PLUGIN_TYPE m_PluginCategory;
		XArray<CKGUID> m_Guids;
	private:

	};

	class CKFileDocument {
	public:
		CKFileDocument();
		~CKFileDocument();

		int32_t m_SaveIDMax;
		CKFileInfo m_FileInfo;

		XArray<CKFileObject> m_FileObjects;
		XArray<CKFileManagerData> m_FileManagersData;
		XClassArray<CKFilePluginDependencies> m_PluginDep;
		/*XClassArray<XIntArray> m_IndexByClassId;*/
		XClassArray<XString> m_IncludedFiles;
	private:

	};

	class CKFile {
	public:
		CKFile(CKMinContext* ctx);
		CKFile(const CKFile&) = delete;
		CKFile& operator=(const CKFile&) = delete;
		~CKFile();

		void ClearData(void);

		CKERROR ShallowLoad(CKSTRING u8_filename, CKFileDocument** out_doc);
		CKERROR DeepLoad(CKSTRING u8_filename, CKFileDocument** out_doc);

		CKERROR Save(CKSTRING u8_filename, CKFileDocument* in_doc);

		//CKERROR Load(CKSTRING u8_filename, /*CKObjectArray list, */ CK_LOAD_FLAGS flags);
		//CKERROR OpenFile(CKSTRING u8_filename, CK_LOAD_FLAGS flags);
		//CKERROR OpenMemory(void* MemoryBuffer, size_t BufferSize, CK_LOAD_FLAGS Flags);
		//CKERROR ReadFileHeaders(CKBufferParser** ParserPtr);
		//CKERROR ReadFileData(CKBufferParser** ParserPtr);
		//CKERROR LoadFileData(void/*CKObjectArray list*/);
		//CKERROR FinishLoading(/*CKObjectArray list, */CK_LOAD_FLAGS flags);

		//int32_t m_SaveIDMax;
		//XArray<CKFileObject> m_FileObject;
		//XArray<CKFileManagerData> m_ManagersData;
		//XClassArray<CKFilePluginDependencies> m_PluginDep;
		//XClassArray<XIntArray> m_IndexByClassId;
		//XClassArray<XString> m_IncludedFiles;

		//CKFileInfo m_FileInfo;

		//CK_LOAD_FLAGS m_Flags;


		//bool m_ReadFileDataDone;

	private:
		// reader function and variables
		CKERROR ReadFileHeader(CKBufferParser* ParserPtr, CKFileDocument* doc);
		CKERROR ReadFileData(CKBufferParser* ParserPtr, CKFileDocument* doc);

		// writer function and varibales

		// shared function and variables
		CKMinContext* m_MinCtx;
	};

}