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

#pragma once

#include "Async/ManualResetEvent.h"
#include "Compression/CompressedBuffer.h"
#include "Containers/Map.h"
#include "Containers/SharedString.h"
#include "Containers/StringView.h"
#include "DerivedDataBuildFunction.h"
#include "DerivedDataCacheKey.h"
#include "DerivedDataRequest.h"
#include "Memory/MemoryFwd.h"
#include "Serialization/CompactBinary.h"
#include "Templates/Function.h"

namespace UE::DerivedData { class FBuildOutputBuilder; }
namespace UE::DerivedData { class IBuildJob; }
namespace UE::DerivedData { class IRequestOwner; }
namespace UE::DerivedData { enum class EBuildPolicy : uint32; }
namespace UE::DerivedData { enum class ECachePolicy : uint32; }
namespace UE::DerivedData { enum class EPriority : uint8; }

namespace UE::DerivedData::Private
{

class FBuildJobContext final : public FBuildContext, public FBuildConfigContext, public FRequestBase
{
public:
	FBuildJobContext(
		IBuildJob& Job,
		const FCacheKey& CacheKey,
		const IBuildFunction& Function,
		FBuildOutputBuilder& OutputBuilder);

	void BeginBuild(IRequestOwner& Owner, TUniqueFunction<void ()>&& OnEndBuild);

	const FSharedString& GetName() const final;

	inline const FUtf8SharedString& GetTypeName() const { return TypeName; }

	inline const FCacheKey& GetCacheKey() const { return CacheKey; }

	inline ECachePolicy GetCachePolicyMask() const final { return CachePolicyMask; }
	inline EBuildPolicy GetBuildPolicyMask() const final { return BuildPolicyMask; }
	uint64 GetRequiredMemory() const override { return RequiredMemory; }
	inline bool ShouldCheckDeterministicOutput() const { return bDeterministicOutputCheck; }

	void AddConstant(FUtf8StringView Key, FCbObject&& Value);
	void AddInput(FUtf8StringView Key, const FCompressedBuffer& Value);

	void AddError(FStringView Message) final;
	void AddWarning(FStringView Message) final;
	void AddMessage(FStringView Message) final;

	void ResetInputs();

private:
	FCbObject FindConstant(FUtf8StringView Key) const final;
	FSharedBuffer FindInput(FUtf8StringView Key) const final;

	void AddMeta(FUtf8StringView Key, const FCbField& Meta) final;

	void AddValue(const FValueId& Id, const FValue& Value) final;
	void AddValue(const FValueId& Id, const FCompressedBuffer& Buffer) final;
	void AddValue(const FValueId& Id, const FCompositeBuffer& Buffer, uint64 BlockSize) final;
	void AddValue(const FValueId& Id, const FSharedBuffer& Buffer, uint64 BlockSize) final;
	void AddValue(const FValueId& Id, const FCbObject& Object) final;

	void EndBuild();
	void BeginAsyncBuild() final;
	void EndAsyncBuild() final;

	void SetTypeName(const FUtf8SharedString& TypeName) final;
	void SetCacheBucket(FCacheBucket Bucket) final;
	void SetCachePolicyMask(ECachePolicy Policy) final;
	void SetBuildPolicyMask(EBuildPolicy Policy) final;
	void SetRequiredMemory(uint64 InRequiredMemory) final { RequiredMemory = InRequiredMemory; }
	void SkipDeterministicOutputCheck() final { bDeterministicOutputCheck = false; }

	void SetPriority(EPriority Priority) final;
	void Cancel() final;
	void Wait() final;

private:
	IBuildJob& Job;
	FCacheKey CacheKey;
	FUtf8SharedString TypeName;
	const IBuildFunction& Function;
	FBuildOutputBuilder& OutputBuilder;
	TMap<FUtf8SharedString, FCbObject> Constants;
	TMap<FUtf8SharedString, FCompressedBuffer> Inputs;
	TUniqueFunction<void ()> OnEndBuild;
	IRequestOwner* Owner{};
	uint64 RequiredMemory{};
	ECachePolicy CachePolicyMask;
	EBuildPolicy BuildPolicyMask;
	FManualResetEvent BuildCompleteEvent;
	bool bIsAsyncBuild{false};
	bool bIsAsyncBuildComplete{false};
	bool bDeterministicOutputCheck{true};
};

} // UE::DerivedData::Private