MountedBuild.h

#pragma once

#include "containers/cancel_flag.h"
#include "filesystem/egfs.h"
#include "web/manifest/manifest.h"
#include "storage/storage.h"

#include <filesystem>

namespace fs = std::filesystem;

typedef std::function<void(uint32_t max)> ProgressSetMaxHandler;
typedef std::function<void()> ProgressIncrHandler;
typedef std::function<void()> ProgressFinishHandler;

struct ChunkMetadata {
	std::shared_ptr<Chunk> Chunk;
	bool Downloaded;
	size_t FileSize;
	uint16_t Flags;
};

typedef std::function<void(const ChunkMetadata& meta, bool lastUpdate)> QueryChunkCallback;

class MountedBuild {
public:
	MountedBuild(Manifest manifest, fs::path mountDir, fs::path cachePath, uint32_t storageFlags, uint32_t memoryPoolCapacity);
	~MountedBuild();

	static bool SetupCacheDirectory(fs::path CacheDir);
	void SetupGameDirectory(ProgressSetMaxHandler setMax, ProgressIncrHandler onProg, ProgressFinishHandler onFinish, cancel_flag& flag, uint32_t threadCount);
	void PreloadAllChunks(ProgressSetMaxHandler setMax, ProgressIncrHandler onProg, ProgressFinishHandler onFinish, cancel_flag& flag, uint32_t threadCount);
	void VerifyAllChunks(ProgressSetMaxHandler setMax, ProgressIncrHandler onProg, ProgressFinishHandler onFinish, cancel_flag& flag, uint32_t threadCount);
	void PurgeUnusedChunks(cancel_flag& flag);
	void QueryChunks(QueryChunkCallback onQuery, cancel_flag& flag, uint32_t threadCount);
	uint32_t GetMissingChunkCount();
	void LaunchGame(const char* additionalArgs);

	const fs::path& GetCachePath() const {
		return CacheDir;
	}

	const fs::path& GetMountPath() const {
		return MountDir;
	}

private:
	void PreloadFile(File& File, uint32_t ThreadCount, cancel_flag& cancelFlag);

	void FileRead(PVOID Handle, PVOID Buffer, UINT64 offset, ULONG length, ULONG* bytesRead);

	fs::path MountDir;
	fs::path CacheDir;
	Manifest Build;
	Storage StorageData;
	std::unique_ptr<EGFS> Egfs;
};