// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include "CoreMinimal.h"
#include "IO/PackageId.h"
#include "IO/PackageStore.h"
#include "Misc/PackagePath.h"

struct FIoContainerHeader;
struct FFilePackageStoreEntry;

/*
 * File/container based package store.
 */
class FFilePackageStore
	: public FPackageStoreBase
{
public:
	FFilePackageStore();
	virtual ~FFilePackageStore();

	virtual void Initialize() override;
	virtual void Lock() override;
	virtual void Unlock() override;
	virtual bool DoesPackageExist(FPackageId PackageId) override;
	virtual EPackageStoreEntryStatus GetPackageStoreEntry(FPackageId PackageId, FPackageStoreEntry& OutPackageStoreEntry) override;
	virtual bool GetPackageRedirectInfo(FPackageId PackageId, FName& OutSourcePackageName, FPackageId& OutRedirectedToPackageId) override;

	void Mount(const FIoContainerHeader* ContainerHeader, uint32 Order);
	void Unmount(const FIoContainerHeader* ContainerHeader);

private:
	struct FMountedContainer
	{
		const FIoContainerHeader* ContainerHeader;
		uint32 Order;
		uint32 Sequence;
	};

#if WITH_EDITOR
	struct FUncookedPackage
	{
		FName PackageName;
		EPackageExtension HeaderExtension;
	};
#endif

	void Update();
#if WITH_EDITOR
	uint64 AddUncookedPackagesFromRoot(const FString& RootPath);
#endif

	FRWLock EntriesLock;
	FCriticalSection UpdateLock;
	TArray<FMountedContainer> MountedContainers;
	TAtomic<uint32> NextSequence{ 0 };
	TMap<FPackageId, const FFilePackageStoreEntry*> StoreEntriesMap;
	TMap<FPackageId, TTuple<FName, FPackageId>> RedirectsPackageMap;
	TMap<FPackageId, FName> LocalizedPackages;
	bool bNeedsUpdate = false;

#if WITH_EDITOR
	FDelegateHandle OnContentPathMountedDelegateHandle;
	FCriticalSection UncookedPackageRootsLock;
	TSet<FString> PendingUncookedPackageRoots;
	TMap<FPackageId, FUncookedPackage> UncookedPackagesMap;
#endif
#if WITH_EDITOR
	TMap<FPackageId, const FFilePackageStoreEntry*> OptionalSegmentStoreEntriesMap;
#endif

	static thread_local int32 LockedOnThreadCount;
};