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

#include "LevelEditorInternalTools.h"

namespace InternalEditorLevelLibrary
{
	TSharedPtr<SLevelViewport> GetLevelViewport(const FName& ViewportConfigKey)
	{
		FLevelEditorModule& LevelEditorModule = FModuleManager::GetModuleChecked<FLevelEditorModule>("LevelEditor");

		TSharedPtr<ILevelEditor> LevelEditor = LevelEditorModule.GetFirstLevelEditor();
		if (LevelEditor.IsValid())
		{
			if (ViewportConfigKey != NAME_None)
			{
				for (TSharedPtr<SLevelViewport> LevelViewport : LevelEditor->GetViewports())
				{
					if (LevelViewport.IsValid() && LevelViewport->GetConfigKey() == ViewportConfigKey)
					{
						return LevelViewport;
					}
				}
			}

			TSharedPtr<SLevelViewport> ActiveLevelViewport = LevelEditor->GetActiveViewportInterface();
			return ActiveLevelViewport;
		}

		return nullptr;
	}

	TSharedPtr<SLevelViewport> GetActiveLevelViewport()
	{
		return GetLevelViewport(NAME_None);
	}

	bool IsEditingLevelInstanceCurrentLevel(UWorld* InWorld)
	{
		if (InWorld)
		{
			if (ULevelInstanceSubsystem* LevelInstanceSubsystem = InWorld->GetSubsystem<ULevelInstanceSubsystem>())
			{
				if (ILevelInstanceInterface* LevelInstance = LevelInstanceSubsystem->GetEditingLevelInstance())
				{
					return LevelInstanceSubsystem->GetLevelInstanceLevel(LevelInstance) == InWorld->GetCurrentLevel();
				}
			}
		}

		return false;
	}

	AActor* GetEditingLevelInstance(UWorld* InWorld)
	{
		ULevelInstanceSubsystem* LevelInstanceSubsystem = InWorld ? InWorld->GetSubsystem<ULevelInstanceSubsystem>() : nullptr;
		return LevelInstanceSubsystem ? Cast<AActor>(LevelInstanceSubsystem->GetEditingLevelInstance()) : nullptr;
	}

	bool IsActorEditorContextVisible(UWorld* InWorld)
	{
		return InWorld && (InWorld->GetCurrentLevel()->OwningWorld->GetLevels().Num() > 1 && (!InWorld->IsPartitionedWorld() || GetEditingLevelInstance(InWorld) != nullptr));
	}
}