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

#include "Tools/AssetEditorContextObject.h"

#include "EditorModeManager.h"
#include "Elements/Framework/TypedElementSelectionSet.h"
#include "Elements/Framework/TypedElementCommonActions.h"
#include "Toolkits/IToolkitHost.h"

const UTypedElementSelectionSet* UAssetEditorContextObject::GetSelectionSet() const
{
	if (GetToolkitHost())
	{
		return GetToolkitHost()->GetEditorModeManager().GetEditorSelectionSet();
	}

	return nullptr;
}

UTypedElementSelectionSet* UAssetEditorContextObject::GetMutableSelectionSet()
{
	if (GetToolkitHost())
	{
		return GetToolkitHost()->GetEditorModeManager().GetEditorSelectionSet();
	}

	return nullptr;
}

UTypedElementCommonActions* UAssetEditorContextObject::GetCommonActions()
{
	if (GetToolkitHost())
	{
		return GetToolkitHost()->GetCommonActions();
	}

	return nullptr;
}

UWorld* UAssetEditorContextObject::GetEditingWorld() const
{
	if (GetToolkitHost())
	{
		return GetToolkitHost()->GetWorld();
	}

	return nullptr;
}

const IToolkitHost* UAssetEditorContextObject::GetToolkitHost() const
{
	return ToolkitHost;
}

IToolkitHost* UAssetEditorContextObject::GetMutableToolkitHost()
{
	return ToolkitHost;
}

void UAssetEditorContextObject::SetToolkitHost(IToolkitHost* InToolkitHost)
{
	ToolkitHost = InToolkitHost;
}

UWorld* UAssetEditorContextObject::GetWorld() const
{
	return GetEditingWorld();
}