UI Extender

Author: nakosung

Source/JavascriptEditor/JavascriptEditorLibrary.cpp

@@ -7,6 +7,8 @@
 #include "HotReloadInterface.h"
 #include "JavascriptWindow.h"
 #include "Editor/PropertyEditor/Public/PropertyEditorModule.h"
+#include "Toolkits/AssetEditorToolkit.h"
+#include "LevelEditor.h"
 
 #if WITH_EDITOR
 ULandscapeInfo* UJavascriptEditorLibrary::GetLandscapeInfo(ALandscape* Landscape, bool bSpawnNewActor)
@@ -577,4 +579,43 @@ void UJavascriptEditorLibrary::CreatePropertyEditorToolkit(TArray<UObject*> Obje
 	FPropertyEditorModule& PropertyEditorModule = FModuleManager::LoadModuleChecked<FPropertyEditorModule>("PropertyEditor");
 	PropertyEditorModule.CreatePropertyEditorToolkit(EToolkitMode::Standalone, TSharedPtr<IToolkitHost>(), ObjectsForPropertiesMenu);
 }
+
+static FName NAME_LevelEditor("LevelEditor");
+static FName NAME_MaterialEditor("MaterialEditor");
+
+FJavascriptExtensibilityManager UJavascriptEditorLibrary::GetMenuExtensibilityManager(FName What)
+{
+	if (What == NAME_LevelEditor)
+	{
+		FLevelEditorModule& LevelEditor = FModuleManager::GetModuleChecked<FLevelEditorModule>(NAME_LevelEditor);
+		return {LevelEditor.GetMenuExtensibilityManager()};
+	}
+	return FJavascriptExtensibilityManager();
+}
+
+FJavascriptExtensibilityManager UJavascriptEditorLibrary::GetToolBarExtensibilityManager(FName What)
+{
+	if (What == NAME_LevelEditor)
+	{
+		FLevelEditorModule& LevelEditor = FModuleManager::GetModuleChecked<FLevelEditorModule>(NAME_LevelEditor);
+		return{ LevelEditor.GetToolBarExtensibilityManager() };
+	}
+	return FJavascriptExtensibilityManager();
+}
+
+void UJavascriptEditorLibrary::AddExtender(FJavascriptExtensibilityManager Manager, FJavascriptExtender Extender)
+{
+	if (Manager.Handle.IsValid() && Extender.Handle.IsValid())
+	{
+		Manager->AddExtender(Extender.Handle);
+	}
+}
+
+void UJavascriptEditorLibrary::RemoveExtender(FJavascriptExtensibilityManager Manager, FJavascriptExtender Extender)
+{
+	if (Manager.Handle.IsValid() && Extender.Handle.IsValid())
+	{
+		Manager->RemoveExtender(Extender.Handle);
+	}
+}
 #endif

Source/JavascriptEditor/JavascriptEditorLibrary.h

@@ -83,6 +83,25 @@ struct FJavascriptPDI
 	FPrimitiveDrawInterface* PDI;
 };
 
+// forward decl
+class FExtensibilityManager;
+
+USTRUCT()
+struct FJavascriptExtensibilityManager
+{
+	GENERATED_BODY()
+
+public:
+#if WITH_EDITOR
+	FExtensibilityManager* operator -> () const
+	{
+		return Handle.Get();
+	}
+
+	TSharedPtr<FExtensibilityManager> Handle;
+#endif
+};
+
 /**
  * 
  */
@@ -329,5 +348,17 @@ class JAVASCRIPTEDITOR_API UJavascriptEditorLibrary : public UBlueprintFunctionL
 
 	UFUNCTION(BlueprintCallable, Category = "Javascript | Editor")
 	static void CreatePropertyEditorToolkit(TArray<UObject*> ObjectsForPropertiesMenu);
+
+	UFUNCTION(BlueprintCallable, Category = "Javascript | Editor")
+	static FJavascriptExtensibilityManager GetMenuExtensibilityManager(FName What);
+
+	UFUNCTION(BlueprintCallable, Category = "Javascript | Editor")
+	static FJavascriptExtensibilityManager GetToolBarExtensibilityManager(FName What);
+
+	UFUNCTION(BlueprintCallable, Category = "Javascript | Editor")
+	static void AddExtender(FJavascriptExtensibilityManager Manager, FJavascriptExtender Extender);
+	
+	UFUNCTION(BlueprintCallable, Category = "Javascript | Editor")
+	static void RemoveExtender(FJavascriptExtensibilityManager Manager, FJavascriptExtender Extender);
 #endif
 };

Source/JavascriptUMG/JavascriptMenuLibrary.cpp

@@ -165,11 +165,6 @@ FJavascriptUICommandInfo UJavascriptMenuLibrary::UI_COMMAND_Function(FJavascript
 	return Out;
 }
 
-FJavascriptExtender UJavascriptMenuLibrary::CreateExtender()
-{
-	return{ MakeShareable<FExtender>(new FExtender) };
-}
-
 FJavascriptExtensionBase UJavascriptMenuLibrary::AddToolBarExtension(FJavascriptExtender Extender, FName ExtensionHook, EJavascriptExtensionHook::Type HookPosition, FJavascriptUICommandList CommandList, FJavascriptFunction Function)
 {
 	auto Copy = new FJavascriptFunction;
@@ -233,4 +228,12 @@ FJavascriptExtender UJavascriptMenuLibrary::Combine(const TArray<FJavascriptExte
 		_Extenders.Add(Extender.Handle);
 	}
 	return{ FExtender::Combine(_Extenders) };
-}
+}
+
+FJavascriptExtender::FJavascriptExtender()
+	: Handle(new FExtender)
+{}
+
+FJavascriptExtender::FJavascriptExtender(TSharedPtr<FExtender> Extender)
+	: Handle(Extender)
+{}

Source/JavascriptUMG/JavascriptMenuLibrary.h

@@ -91,11 +91,14 @@ struct FJavascriptUICommandInfo
 };
 
 USTRUCT()
-struct FJavascriptExtender
+struct JAVASCRIPTUMG_API FJavascriptExtender
 {
 	GENERATED_BODY()
 
 public:
+	FJavascriptExtender();
+	FJavascriptExtender(TSharedPtr<FExtender> Extender);
+
 	FExtender* operator -> () const
 	{
 		return Handle.Get();
@@ -144,9 +147,6 @@ class JAVASCRIPTUMG_API UJavascriptMenuLibrary : public UBlueprintFunctionLibrar
 	UFUNCTION(BlueprintCallable, Category = "Scripting | Javascript")
 	static void CreateMenuBarBuilder(FJavascriptUICommandList CommandList, FJavascriptFunction Function);
 
-	UFUNCTION(BlueprintCallable, Category = "Scripting | Javascript")
-	static FJavascriptExtender CreateExtender();
-
 	UFUNCTION(BlueprintCallable, Category = "Scripting | Javascript")
 	static FJavascriptExtensionBase AddToolBarExtension(FJavascriptExtender Extender, FName ExtensionHook, EJavascriptExtensionHook::Type HookPosition, FJavascriptUICommandList CommandList, FJavascriptFunction Function);