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

#include "AssetDefinition_FontFace.h"

#include "ContentBrowserMenuContexts.h"
#include "FontEditorModule.h"
#include "ToolMenus.h"
#include "EditorReimportHandler.h"
#include "FontEditorModule.h"
#include "ToolMenu.h"
#include "Toolkits/SimpleAssetEditor.h"
#include "ToolMenuSection.h"

#define LOCTEXT_NAMESPACE "AssetDefinition_FontFace"

EAssetCommandResult UAssetDefinition_FontFace::OpenAssets(const FAssetOpenArgs& OpenArgs) const
{
	// Load the FontEditor module to ensure that FFontFaceDetailsCustomization is registered
	IFontEditorModule* FontEditorModule = &FModuleManager::LoadModuleChecked<IFontEditorModule>("FontEditor");
	
	// Open each object in turn, as the default editor would try and collapse a multi-edit together 
	// into a single editor instance which doesn't really work for font face assets
	for (UFontFace* FontFace : OpenArgs.LoadObjects<UFontFace>())
	{
		FontEditorModule->CreateFontFaceEditor(OpenArgs.GetToolkitMode(), OpenArgs.ToolkitHost, FontFace);
	}

	return EAssetCommandResult::Handled;
}

// Menu Extensions
//--------------------------------------------------------------------

namespace MenuExtension_FontFace
{
	void ExecuteReimport(const FToolMenuContext& InContext)
	{
		const UContentBrowserAssetContextMenuContext* Context = UContentBrowserAssetContextMenuContext::FindContextWithAssets(InContext);
		for (UFontFace* Font : Context->LoadSelectedObjects<UFontFace>())
		{
			// Skip fonts that don't have a source filename, as they can't be reimported
			if (!Font->SourceFilename.IsEmpty())
			{
				// Fonts fail to reimport if they ask for a new file if missing
				FReimportManager::Instance()->Reimport(Font, /*bAskForNewFileIfMissing=*/false);
			}
		}
	}
	
	static FDelayedAutoRegisterHelper DelayedAutoRegister(EDelayedRegisterRunPhase::EndOfEngineInit, []{ 
		UToolMenus::RegisterStartupCallback(FSimpleMulticastDelegate::FDelegate::CreateLambda([]()
		{
			FToolMenuOwnerScoped OwnerScoped(UE_MODULE_NAME);
			UToolMenu* Menu = UE::ContentBrowser::ExtendToolMenu_AssetContextMenu(UFontFace::StaticClass());
		
			FToolMenuSection& Section = Menu->FindOrAddSection("GetAssetActions");
			Section.AddDynamicEntry(NAME_None, FNewToolMenuSectionDelegate::CreateLambda([](FToolMenuSection& InSection)
			{
				if (const UContentBrowserAssetContextMenuContext* Context = UContentBrowserAssetContextMenuContext::FindContextWithAssets(InSection))
				{
					// TODO NDarnell Having reimporting as a right click option for something like fonts is probably a
					// bit much, I think for something more complex that isn't always re-importable maybe they shouldn't
					// be right click options.  Though...I mean shouldn't even online fonts be reimportable?  They come
					// from OTF and TTF files...those things *can* be reloaded.
					// Also why do we even bother with this when we have IsImportable Asset()?
					// TODO Investigate, see if anyone remembers.
					
					const TAttribute<FText> Label = LOCTEXT("ReimportFontLabel", "Reimport");
					const TAttribute<FText> ToolTip = LOCTEXT("ReimportFontTooltip", "Reimport the selected font(s).");
					const FSlateIcon Icon = FSlateIcon(FAppStyle::GetAppStyleSetName(), "ContentBrowser.AssetActions.ReimportAsset");
					
					FToolUIAction UIAction;
					UIAction.ExecuteAction = FToolMenuExecuteAction::CreateStatic(&ExecuteReimport);
					InSection.AddMenuEntry("ReimportFont", Label, ToolTip, Icon, UIAction);
				}
			}));
		}));
	});
}

#undef LOCTEXT_NAMESPACE