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

#pragma once

#include "CoreMinimal.h"
#include "Templates/SharedPointer.h"
#include "UObject/Object.h"
#include "UObject/ObjectMacros.h"
#include "UObject/UObjectGlobals.h"
#include "PersonaModule.h"

#include "PersonaToolMenuContext.generated.h"

class IPersonaToolkit;
class UAnimBlueprint;
class UAnimationAsset;
class UDebugSkelMeshComponent;
class USkeletalMesh;
class USkeleton;
struct FFrame;

UCLASS(BlueprintType)
class PERSONA_API UPersonaToolMenuContext : public UObject
{
	GENERATED_BODY()
public:
	
	/** Get the skeleton that we are editing */
	UFUNCTION(BlueprintCallable, Category = PersonaEditorExtensions)
	USkeleton* GetSkeleton() const;

	/** Get the preview component that we are using */
	UFUNCTION(BlueprintCallable, Category = PersonaEditorExtensions)
	UDebugSkelMeshComponent* GetPreviewMeshComponent() const;

	/** Get the skeletal mesh that we are editing */
	UFUNCTION(BlueprintCallable, Category = PersonaEditorExtensions)
	USkeletalMesh* GetMesh() const;

	/** Get the anim blueprint that we are editing */
	UFUNCTION(BlueprintCallable, Category = PersonaEditorExtensions)
	UAnimBlueprint* GetAnimBlueprint() const;

	/** Get the animation asset that we are editing */
	UFUNCTION(BlueprintCallable, Category = PersonaEditorExtensions)
	UAnimationAsset* GetAnimationAsset() const;

	/** Get the persona toolkit */
	void SetToolkit(TSharedRef<IPersonaToolkit> InToolkit);

	/** Get a weak ptr to the persona toolkit */
	TWeakPtr<IPersonaToolkit> GetToolkit() const { return WeakToolkit; }

protected:
	bool HasValidToolkit() const;
private:
	TWeakPtr<IPersonaToolkit> WeakToolkit;
};