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

#include "IPersonaEditorModeManager.h"

#include "ContextObjectStore.h"
#include "Tools/EdModeInteractiveToolsContext.h"
#include "SEditorViewport.h"
#include "Slate/SceneViewport.h"
#include "Framework/Application/SlateApplication.h"
#include "Framework/Application/SlateUser.h"

IPersonaEditorModeManager* UPersonaEditorModeManagerContext::GetPersonaEditorModeManager() const
{
	return ModeManager;
}

bool UPersonaEditorModeManagerContext::GetCameraTarget(FSphere& OutTarget) const
{
	check(ModeManager);
	return ModeManager->GetCameraTarget(OutTarget);
}

void UPersonaEditorModeManagerContext::GetOnScreenDebugInfo(TArray<FText>& OutDebugText) const
{
	check(ModeManager);
	return ModeManager->GetOnScreenDebugInfo(OutDebugText);
}

void UPersonaEditorModeManagerContext::SetFocusInViewport()
{
	const IPersonaEditorModeManager* PersonaEditorModeManager = GetPersonaEditorModeManager();
	if (!PersonaEditorModeManager)
	{
		return;
	}

	const FEditorViewportClient* ViewportClient = PersonaEditorModeManager->GetHoveredViewportClient();
	if (!ViewportClient)
	{
		ViewportClient = PersonaEditorModeManager->GetFocusedViewportClient();
	}
	if (!ViewportClient)
	{
		return;
	}

	const TWeakPtr<SViewport> ViewportWidget = ViewportClient->GetEditorViewportWidget()->GetSceneViewport()->GetViewportWidget();
	if (!ViewportWidget.IsValid())
	{
		return;
	}
	
	// set focus back to viewport so that hotkeys are immediately detected
	TSharedPtr<SWidget> ViewportContents = ViewportWidget.Pin()->GetContent();
	FSlateApplication::Get().ForEachUser([&ViewportContents](FSlateUser& User) 
	{
		User.SetFocus(ViewportContents.ToSharedRef());
	});
}

IPersonaEditorModeManager::IPersonaEditorModeManager() : FAssetEditorModeManager()
{
	UModeManagerInteractiveToolsContext* ToolsContext = GetInteractiveToolsContext();
	UContextObjectStore* ContextObjectStore = ToolsContext->ContextObjectStore;
	ContextObjectStore->AddContextObject(PersonaModeManagerContext.Get());
	ToolsContext->SetDragToolsEnabled(true);
}

IPersonaEditorModeManager::~IPersonaEditorModeManager()
{
	UContextObjectStore* ContextObjectStore = GetInteractiveToolsContext()->ContextObjectStore;
	ContextObjectStore->RemoveContextObject(PersonaModeManagerContext.Get());
}

void IPersonaEditorModeManager::AddReferencedObjects(FReferenceCollector& Collector)
{
	FAssetEditorModeManager::AddReferencedObjects(Collector);
	Collector.AddReferencedObject(PersonaModeManagerContext);
}