Refactoring ue-wingman to be a command-line only tool

Plugins/UEWingman/BrokenHandlers/GraphNode_SetDefaults.h

@@ -0,0 +1,136 @@
+#pragma once
+
+#include "CoreMinimal.h"
+#include "WingBasics.h"
+#include "WingServer.h"
+#include "WingFetcher.h"
+#include "WingProperty.h"
+#include "WingUtils.h"
+#include "EdGraph/EdGraphPin.h"
+#include "EdGraphSchema_K2.h"
+#include "MaterialGraph/MaterialGraphSchema.h"
+#include "GraphNode_SetDefaults.generated.h"
+
+
+// ---------------------------------------------------------------------------
+// ---------------------------------------------------------------------------
+// ---------------------------------------------------------------------------
+
+USTRUCT()
+struct FSetNodeDefaultEntry
+{
+	GENERATED_BODY()
+
+	UPROPERTY()
+	FString Node;
+
+	UPROPERTY()
+	FString Name;
+
+	UPROPERTY()
+	FString Value;
+};
+
+
+UCLASS()
+class UWing_GraphNode_SetDefaults : public UWingHandler
+{
+	GENERATED_BODY()
+
+public:
+	UPROPERTY(EditAnywhere, meta=(Description="Target graph"))
+	FString Graph;
+
+	UPROPERTY(EditAnywhere, meta=(Description="Array of {node, name, value} objects"))
+	FWingJsonArray Pins;
+
+	virtual void Register() override
+	{
+		UWingServer::AddHandler(this,
+			TEXT("Set the default value of input pins or material expression properties on nodes."));
+	}
+	// -----------------------------------------------------------------------
+	// K2 graphs: set pin default values.
+	// -----------------------------------------------------------------------
+
+	void HandleK2Entry(const FSetNodeDefaultEntry& Entry, UEdGraph* GraphObj, const UEdGraphSchema_K2* K2Schema)
+	{
+		WingFetcher F(GraphObj, WingOut::Stdout);
+		UWingGraphPinRef* PinRef = F.Node(Entry.Node).Pin(Entry.Name).Cast<UWingGraphPinRef>();
+		if (!PinRef) return;
+		UEdGraphPin* Pin = WingUtils::CheckGetPin(PinRef->Node, PinRef->PinName, WingOut::Stdout);
+		if (!Pin) return;
+
+		UEdGraphNode* Node = Pin->GetOwningNode();
+
+		if (Pin->Direction != EGPD_Input)
+		{
+			WingOut::Stdout.Printf(TEXT("error: %s is an output pin\n"), *WingUtils::FormatName(Pin));
+			return;
+		}
+
+		FString UseDefaultValue;
+		TObjectPtr<UObject> UseDefaultObject = nullptr;
+		FText UseDefaultText;
+		K2Schema->GetPinDefaultValuesFromString(Pin->PinType, Node, Entry.Value, UseDefaultValue, UseDefaultObject, UseDefaultText, false);
+		FString Error = K2Schema->IsPinDefaultValid(Pin, UseDefaultValue, UseDefaultObject, UseDefaultText);
+		if (!Error.IsEmpty())
+		{
+			WingOut::Stdout.Printf(TEXT("error: %s: %s\n"), *WingUtils::FormatName(Pin), *Error);
+			return;
+		}
+		UWingServer::AddTouchedObject(Node);
+		K2Schema->TrySetDefaultValue(*Pin, Entry.Value);
+	}
+
+	// -----------------------------------------------------------------------
+	// Material graphs: set material expression properties.
+	// -----------------------------------------------------------------------
+
+	void HandleMaterialEntry(const FSetNodeDefaultEntry& Entry, UEdGraph* GraphObj)
+	{
+		WingFetcher F(GraphObj, WingOut::Stdout);
+		UEdGraphNode* Node = F.Node(Entry.Node).Cast<UEdGraphNode>();
+		if (!Node) return;
+
+		TArray<FWingProperty> All = FWingProperty::GetDetails(Node, true);
+		FWingProperty *P = WingUtils::FindOneWithExternalID(Entry.Name, All, TEXT("Property"), WingOut::Stdout);
+		if (!P) return;
+
+		UWingServer::AddTouchedObject(Node);
+
+		if (!P->SetText(Entry.Value, WingOut::Stdout))
+			return;
+	}
+
+	// -----------------------------------------------------------------------
+
+	virtual void Handle() override
+	{
+		// Fetch the graph once.
+		WingFetcher GraphFetcher(WingOut::Stdout);
+		UEdGraph* GraphObj = GraphFetcher.Walk(Graph).Cast<UEdGraph>();
+		if (!GraphObj) return;
+
+		const UEdGraphSchema* Schema = GraphObj->GetSchema();
+		const UEdGraphSchema_K2* K2Schema = Cast<UEdGraphSchema_K2>(Schema);
+		const UMaterialGraphSchema* MGSchema = Cast<UMaterialGraphSchema>(Schema);
+
+		if (!K2Schema && !MGSchema)
+		{
+			WingOut::Stdout.Printf(TEXT("error: unsupported graph schema %s\n"), *Schema->GetClass()->GetName());
+			return;
+		}
+
+		FSetNodeDefaultEntry Entry;
+		TArray<FWingProperty> Props = FWingProperty::GetAll(nullptr, &Entry, FSetNodeDefaultEntry::StaticStruct(), true);
+		for (const TSharedPtr<FJsonValue>& PinVal : Pins.Array)
+		{
+			if (!FWingProperty::PopulateFromJson(Props, *PinVal, false, WingOut::Stdout)) continue;
+			if (K2Schema) HandleK2Entry(Entry, GraphObj, K2Schema);
+			else if (MGSchema) HandleMaterialEntry(Entry, GraphObj);
+		}
+
+		WingOut::Stdout.Printf(TEXT("Done.\n"));
+	}
+};