allow to hide Flow Asset and Flow Node from "Create Asset" menu

MothCocoon · Aug 16, 2021 · 53ed6ca · 53ed6ca
1 parent edf0760
commit 53ed6ca
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 6 deletions.
diff --git a/Source/FlowEditor/Private/Asset/AssetTypeActions_FlowAsset.cpp b/Source/FlowEditor/Private/Asset/AssetTypeActions_FlowAsset.cpp
@@ -1,5 +1,6 @@
 #include "Asset/AssetTypeActions_FlowAsset.h"
 #include "FlowEditorModule.h"
+#include "Graph/FlowGraphSettings.h"
 
 #include "FlowAsset.h"
 
@@ -14,7 +15,7 @@ FText FAssetTypeActions_FlowAsset::GetName() const
 
 uint32 FAssetTypeActions_FlowAsset::GetCategories()
 {
-	return FFlowEditorModule::FlowAssetCategory;
+	return UFlowGraphSettings::Get()->bExposeFlowAssetCreation ? FFlowEditorModule::FlowAssetCategory : 0;
 }
 
 UClass* FAssetTypeActions_FlowAsset::GetSupportedClass() const

diff --git a/Source/FlowEditor/Private/Graph/FlowGraphSettings.cpp b/Source/FlowEditor/Private/Graph/FlowGraphSettings.cpp
@@ -2,12 +2,14 @@
 
 UFlowGraphSettings::UFlowGraphSettings(const FObjectInitializer& ObjectInitializer)
 	: Super(ObjectInitializer)
+	, bExposeFlowAssetCreation(true)
+	, bExposeFlowNodeCreation(true)
+	, bShowAssetToolbarAboveLevelEditor(true)
 	, bShowDefaultPinNames(false)
 	, ExecPinColorModifier(0.75f, 0.75f, 0.75f, 1.0f)
 	, NodeDescriptionBackground(FLinearColor(0.0625f, 0.0625f, 0.0625f, 1.0f))
 	, NodeStatusBackground(FLinearColor(0.12f, 0.12f, 0.12f, 1.0f))
 	, NodePreloadedBackground(FLinearColor(0.12f, 0.12f, 0.12f, 1.0f))
-	, bShowAssetToolbarAboveLevelEditor(true)
 	, InactiveWireColor(FLinearColor(0.364f, 0.364f, 0.364f, 1.0f))
 	, InactiveWireThickness(1.5f)
 	, RecentWireDuration(3.0f)

diff --git a/Source/FlowEditor/Private/Nodes/AssetTypeActions_FlowNodeBlueprint.cpp b/Source/FlowEditor/Private/Nodes/AssetTypeActions_FlowNodeBlueprint.cpp
@@ -1,6 +1,7 @@
 #include "Nodes/AssetTypeActions_FlowNodeBlueprint.h"
 #include "Nodes/FlowNodeBlueprintFactory.h"
 #include "FlowEditorModule.h"
+#include "Graph/FlowGraphSettings.h"
 
 #include "Nodes/FlowNodeBlueprint.h"
 
@@ -13,7 +14,7 @@ FText FAssetTypeActions_FlowNodeBlueprint::GetName() const
 
 uint32 FAssetTypeActions_FlowNodeBlueprint::GetCategories()
 {
-	return FFlowEditorModule::FlowAssetCategory;
+	return UFlowGraphSettings::Get()->bExposeFlowNodeCreation ? FFlowEditorModule::FlowAssetCategory : 0;
 }
 
 UClass* FAssetTypeActions_FlowNodeBlueprint::GetSupportedClass() const

diff --git a/Source/FlowEditor/Public/Graph/FlowGraphSettings.h b/Source/FlowEditor/Public/Graph/FlowGraphSettings.h
@@ -14,6 +14,21 @@ class UFlowGraphSettings final : public UDeveloperSettings
 	GENERATED_UCLASS_BODY()
 	static UFlowGraphSettings* Get() { return CastChecked<UFlowGraphSettings>(UFlowGraphSettings::StaticClass()->GetDefaultObject()); }
 
+	/** Show Flow Asset in Flow category of "Create Asset" menu?
+	* Requires restart after making a change. */
+	UPROPERTY(EditAnywhere, config, Category = "Default UI")
+	bool bExposeFlowAssetCreation;
+
+	/** Show Flow Node blueprint in Flow category of "Create Asset" menu?
+	* Requires restart after making a change. */
+	UPROPERTY(EditAnywhere, config, Category = "Default UI")
+	bool bExposeFlowNodeCreation;
+
+	/** Show Flow Asset toolbar?
+	* Requires restart after making a change. */
+	UPROPERTY(EditAnywhere, config, Category = "Default UI")
+	bool bShowAssetToolbarAboveLevelEditor;
+
 	/** Hide specific nodes from the Flow Palette without changing the source code.
 	* Requires restart after making a change. */
 	UPROPERTY(EditAnywhere, config, Category = "Nodes")
@@ -37,9 +52,6 @@ class UFlowGraphSettings final : public UDeveloperSettings
 
 	UPROPERTY(EditAnywhere, config, Category = "NodePopups")
 	FLinearColor NodePreloadedBackground;
-
-	UPROPERTY(EditAnywhere, config, Category = "World")
-	bool bShowAssetToolbarAboveLevelEditor;
 
 	UPROPERTY(EditAnywhere, config, Category = "Wires")
 	FLinearColor InactiveWireColor;