Refactoring

Source/FunnyGirlEditor/Private/FunnyGirlEditorModule.cpp

@@ -2,6 +2,8 @@
 
 #include "FunnyGirlEditorModule.h"
 
+#include "AssetRegistry/AssetRegistryModule.h"
+#include "AssetRegistry/IAssetRegistry.h"
 #include "AssetTypeActions/AssetTypeActions_DirectorPrompt.h"
 #include "AssetTypeActions/AssetTypeActions_Identity.h"
 #include "AssetTypeActions/AssetTypeActions_LLMProvider.h"
@@ -11,12 +13,16 @@
 #include "AssetToolsModule.h"
 #include "IAssetTools.h"
 #include "IAssetTypeActions.h"
+#include "Misc/FileHelper.h"
+#include "Misc/MessageDialog.h"
+#include "Misc/Paths.h"
 #include "Modules/ModuleManager.h"
 #include "Morph/Editor/LipSyncMappingAssetTypeActions.h"
 #include "Morph/Editor/MorphPresetAssetTypeActions.h"
 #include "Morph/MorphPresetDataAsset.h"
 #include "Morph/Editor/MorphPresetDetailsCustomization.h"
 #include "PropertyEditorModule.h"
+#include "ToolMenus.h"
 
 #define LOCTEXT_NAMESPACE "FunnyGirlEditor"
 
@@ -44,10 +50,18 @@ void FFunnyGirlEditorModule::StartupModule()
 		FOnGetDetailCustomizationInstance::CreateStatic(&FMorphPresetDetailsCustomization::MakeInstance));
 	PropertyEditorModule.NotifyCustomizationModuleChanged();
 	bMorphPresetCustomizationRegistered = true;
+
+	UToolMenus::RegisterStartupCallback(FSimpleMulticastDelegate::FDelegate::CreateRaw(this, &FFunnyGirlEditorModule::RegisterMenus));
 }
 
 void FFunnyGirlEditorModule::ShutdownModule()
 {
+	if (UObjectInitialized() && FModuleManager::Get().IsModuleLoaded(TEXT("ToolMenus")))
+	{
+		UToolMenus::UnRegisterStartupCallback(this);
+		UToolMenus::UnregisterOwner(this);
+	}
+
 	if (FModuleManager::Get().IsModuleLoaded(TEXT("AssetTools")))
 	{
 		IAssetTools& AssetTools = FModuleManager::GetModuleChecked<FAssetToolsModule>("AssetTools").Get();
@@ -88,6 +102,91 @@ void FFunnyGirlEditorModule::RegisterAssetTypeAction(const TSharedRef<IAssetType
 	RegisteredAssetTypeActions.Add(AssetTypeAction);
 }
 
+void FFunnyGirlEditorModule::RegisterMenus()
+{
+	FToolMenuOwnerScoped OwnerScoped(this);
+
+	UToolMenu* ToolsMenu = UToolMenus::Get()->ExtendMenu(TEXT("LevelEditor.MainMenu.Tools"));
+	if (!ToolsMenu)
+	{
+		return;
+	}
+
+	FToolMenuSection& Section = ToolsMenu->FindOrAddSection(TEXT("FunnyGirl"));
+	Section.Label = LOCTEXT("FunnyGirlToolsSection", "Funny Girl");
+	Section.AddMenuEntry(
+		TEXT("ExportMorphPresetDescriptions"),
+		LOCTEXT("ExportMorphPresetDescriptionsLabel", "Export Morph Preset Descriptions"),
+		LOCTEXT("ExportMorphPresetDescriptionsTooltip", "Export all Morph Preset assets to Saved/Exports/MorphPresetDescriptions.tsv."),
+		FSlateIcon(),
+		FUIAction(FExecuteAction::CreateRaw(this, &FFunnyGirlEditorModule::ExportMorphPresetDescriptions)));
+}
+
+void FFunnyGirlEditorModule::ExportMorphPresetDescriptions()
+{
+	FAssetRegistryModule& AssetRegistryModule = FModuleManager::LoadModuleChecked<FAssetRegistryModule>(TEXT("AssetRegistry"));
+	IAssetRegistry& AssetRegistry = AssetRegistryModule.Get();
+	AssetRegistry.SearchAllAssets(true);
+
+	FARFilter Filter;
+	Filter.ClassPaths.Add(UMorphPresetDataAsset::StaticClass()->GetClassPathName());
+	Filter.bRecursiveClasses = true;
+
+	TArray<FAssetData> PresetAssets;
+	AssetRegistry.GetAssets(Filter, PresetAssets);
+	PresetAssets.Sort([](const FAssetData& A, const FAssetData& B)
+	{
+		return A.GetSoftObjectPath().ToString() < B.GetSoftObjectPath().ToString();
+	});
+
+	FString Output;
+	Output += TEXT("AssetName\tPresetName\tDescription\n");
+
+	for (const FAssetData& AssetData : PresetAssets)
+	{
+		const UMorphPresetDataAsset* Preset = Cast<UMorphPresetDataAsset>(AssetData.GetAsset());
+		if (!Preset)
+		{
+			continue;
+		}
+
+		FString AssetName = AssetData.AssetName.ToString();
+		FString PresetName = Preset->DisplayName.ToString().TrimStartAndEnd();
+		FString Description = Preset->Description.ToString().TrimStartAndEnd();
+
+		if (PresetName.IsEmpty())
+		{
+			PresetName = AssetName;
+		}
+
+		AssetName.ReplaceInline(TEXT("\t"), TEXT(" "));
+		PresetName.ReplaceInline(TEXT("\t"), TEXT(" "));
+		Description.ReplaceInline(TEXT("\t"), TEXT(" "));
+		Description.ReplaceInline(TEXT("\r\n"), TEXT(" "));
+		Description.ReplaceInline(TEXT("\n"), TEXT(" "));
+		Description.ReplaceInline(TEXT("\r"), TEXT(" "));
+
+		Output += FString::Printf(TEXT("%s\t%s\t%s\n"), *AssetName, *PresetName, *Description);
+	}
+
+	const FString ExportDir = FPaths::Combine(FPaths::ProjectSavedDir(), TEXT("Exports"));
+	IFileManager::Get().MakeDirectory(*ExportDir, true);
+	const FString ExportPath = FPaths::Combine(ExportDir, TEXT("MorphPresetDescriptions.tsv"));
+
+	if (!FFileHelper::SaveStringToFile(Output, *ExportPath, FFileHelper::EEncodingOptions::ForceUTF8WithoutBOM))
+	{
+		FMessageDialog::Open(EAppMsgType::Ok, FText::Format(
+			LOCTEXT("ExportMorphPresetDescriptionsFailed", "Failed to write Morph Preset descriptions to:\n{0}"),
+			FText::FromString(ExportPath)));
+		return;
+	}
+
+	FMessageDialog::Open(EAppMsgType::Ok, FText::Format(
+		LOCTEXT("ExportMorphPresetDescriptionsSuccess", "Exported {0} Morph Preset assets to:\n{1}"),
+		FText::AsNumber(PresetAssets.Num()),
+		FText::FromString(ExportPath)));
+}
+
 #undef LOCTEXT_NAMESPACE
 
 IMPLEMENT_MODULE(FFunnyGirlEditorModule, FunnyGirlEditor)