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

#pragma once

#include "CoreMinimal.h"
#include "AnimNextRigVMAssetEntry.h"
#include "IAnimNextRigVMExportInterface.h"
#include "Variables/IAnimNextRigVMVariableInterface.h"
#include "Param/ParamType.h"
#include "Kismet/BlueprintFunctionLibrary.h"
#include "Logging/StructuredLog.h"
#include "Variables/AnimNextVariableBinding.h"
#include "AnimNextVariableEntry.generated.h"

class UAnimNextRigVMAssetEditorData;
class UAnimNextModule_EditorData;
class UAnimNextDataInterface_EditorData;
class UAnimNextVariableBinding;
class UAnimNextDataInterfaceEntry;
class FAnimationAnimNextRuntimeTest_Variables;

namespace UE::AnimNext::Editor
{
	class FVariableCustomization;
	class FVariableBindingPropertyCustomization;
	class FUniversalObjectLocatorBindingCustomization;
	class FVariablesOutlinerMode;
	class SAddVariablesDialog;
	class FVariablesOutlinerHierarchy;
	class SVariablesOutlinerValue;
	class FVariableProxyCustomization;
	class FAnimNextEditorModule;
}

namespace UE::AnimNext::Tests
{
	class FEditor_Variables;
	class FVariables_UOLBindings;
	class FEditor_AnimGraph_Variables;
	class FDataInterfaceCompile;
}

namespace UE::AnimNext::UncookedOnly
{
	struct FPublicVariablesImpl;
	struct FUtils;
}

UCLASS(MinimalAPI, Category = "Variables", DisplayName = "Variable")
class UAnimNextVariableEntry : public UAnimNextRigVMAssetEntry, public IAnimNextRigVMVariableInterface, public IAnimNextRigVMExportInterface
{
	GENERATED_BODY()

	friend class UAnimNextRigVMAssetEditorData;
	friend class UAnimNextModule_EditorData;
	friend class UAnimNextDataInterface_EditorData;
	friend class UAnimNextDataInterfaceEntry;
	friend class UAnimNextStateTreeTreeEditorData;
	friend class UE::AnimNext::Tests::FEditor_Variables;
	friend class UE::AnimNext::Tests::FEditor_AnimGraph_Variables;
	friend class UE::AnimNext::Tests::FVariables_UOLBindings;
	friend class UE::AnimNext::Tests::FDataInterfaceCompile;
	friend class FAnimationAnimNextRuntimeTest_Variables;
	friend class UE::AnimNext::Editor::FVariableCustomization;
	friend class UE::AnimNext::Editor::FVariableBindingPropertyCustomization;
	friend class UE::AnimNext::Editor::FUniversalObjectLocatorBindingCustomization;
	friend class UE::AnimNext::Editor::FVariablesOutlinerMode;
	friend class UE::AnimNext::Editor::SAddVariablesDialog;
	friend class UE::AnimNext::Editor::SVariablesOutlinerValue;
	friend class UE::AnimNext::Editor::FVariablesOutlinerHierarchy;
	friend struct UE::AnimNext::UncookedOnly::FUtils;
	friend class UE::AnimNext::Editor::FVariableProxyCustomization;
	friend struct UE::AnimNext::UncookedOnly::FPublicVariablesImpl;
	friend class UE::AnimNext::Editor::FAnimNextEditorModule;

	UAnimNextVariableEntry();

	// UObject interface
	virtual void Serialize(FArchive& Ar) override;
	virtual void PostLoad() override;
#if WITH_EDITOR
	virtual void PostEditChangeProperty(FPropertyChangedEvent& PropertyChangedEvent) override;
#endif

	// IAnimNextRigVMExportInterface interface
	virtual FAnimNextParamType GetExportType() const override;
	virtual FName GetExportName() const override;
	virtual EAnimNextExportAccessSpecifier GetExportAccessSpecifier() const override;
	virtual void SetExportAccessSpecifier(EAnimNextExportAccessSpecifier InAccessSpecifier, bool bSetupUndoRedo = true) override;

	// UAnimNextRigVMAssetEntry interface
	virtual FName GetEntryName() const override;
	virtual void SetEntryName(FName InName, bool bSetupUndoRedo = true) override;
	virtual FText GetDisplayName() const override;
	virtual FText GetDisplayNameTooltip() const override;

	// IAnimNextRigVMVariableInterface interface
	virtual FAnimNextParamType GetType() const override;
	virtual bool SetType(const FAnimNextParamType& InType, bool bSetupUndoRedo = true) override;
	virtual FName GetVariableName() const override;
	virtual void SetVariableName(FName InName, bool bSetupUndoRedo = true) override;
	virtual bool SetDefaultValue(TConstArrayView<uint8> InValue, bool bSetupUndoRedo = true) override;
	virtual bool SetDefaultValueFromString(const FString& InDefaultValue, bool bSetupUndoRedo = true) override;
	virtual const FInstancedPropertyBag& GetPropertyBag() const override;
	virtual bool GetDefaultValue(const FProperty*& OutProperty, TConstArrayView<uint8>& OutValue) const override;
	virtual void SetBindingType(UScriptStruct* InBindingTypeStruct, bool bSetupUndoRedo = true) override;
	virtual void SetBinding(TInstancedStruct<FAnimNextVariableBindingData>&& InBinding, bool bSetupUndoRedo = true) override;
	virtual TConstStructView<FAnimNextVariableBindingData> GetBinding() const override;

	// Set the default value
	template<typename ValueType>
	bool SetDefaultValue(const ValueType& InValue, bool bSetupUndoRedo = true)
	{
		const FPropertyBagPropertyDesc* Desc = DefaultValue.FindPropertyDescByName(IAnimNextRigVMVariableInterface::ValueName);
		if(Desc == nullptr)
		{
			UE_LOGFMT(LogAnimation, Error, "UAnimNextVariableEntry::SetDefaultValue: Could not find default value in property bag");
			return false;
		}

		FAnimNextParamType SuppliedType = FAnimNextParamType::GetType<ValueType>();
		FAnimNextParamType InternalType = FAnimNextParamType(Desc->ValueType, Desc->ContainerTypes.GetFirstContainerType(), Desc->ValueTypeObject);
		if(SuppliedType != InternalType)
		{
			UE_LOGFMT(LogAnimation, Error, "UAnimNextVariableEntry::SetDefaultValue: Mismatched types");
			return false;
		}

		return SetDefaultValue(TConstArrayView<uint8>((uint8*)&InValue, sizeof(ValueType)), bSetupUndoRedo);
	}

	/** Access specifier - whether the variable is visible external to this asset */
	UPROPERTY(EditAnywhere, Category = "Variable")
	EAnimNextExportAccessSpecifier Access = EAnimNextExportAccessSpecifier::Private;

	/** Parameter name we reference */
	UPROPERTY(VisibleAnywhere, Category = "Variable")
	FName ParameterName;

	/** The variable's type */
	UPROPERTY(EditAnywhere, Category = "Variable")
	FAnimNextParamType Type = FAnimNextParamType::GetType<bool>();

	/** Binding data */
	UPROPERTY(EditAnywhere, Category = "Variable")
	FAnimNextVariableBinding Binding;

	/** Comment to display in editor */
	UPROPERTY(EditAnywhere, Category = "Variable", meta=(MultiLine))
	FString Comment;

	/** Property bag holding the default value of the variable */
	UPROPERTY()
	FInstancedPropertyBag DefaultValue;
};