Halfway through with repair of right-click menus

Plugins/UEWingman/Deprecated/StateMachine_AddState.h

@@ -0,0 +1,116 @@
+#pragma once
+
+#include "CoreMinimal.h"
+#include "WingServer.h"
+#include "WingHandler.h"
+#include "WingFetcher.h"
+#include "WingUtils.h"
+#include "EdGraph/EdGraph.h"
+#include "Kismet2/KismetEditorUtilities.h"
+#include "Animation/AnimBlueprint.h"
+#include "Animation/AnimSequence.h"
+#include "AnimGraphNode_SequencePlayer.h"
+#include "AnimStateNode.h"
+#include "AnimationStateMachineGraph.h"
+#include "StateMachine_AddState.generated.h"
+
+
+// ---------------------------------------------------------------------------
+// ---------------------------------------------------------------------------
+// ---------------------------------------------------------------------------
+
+UCLASS()
+class UWing_StateMachine_AddState : public UObject, public IWingHandler
+{
+	GENERATED_BODY()
+
+public:
+	UPROPERTY(meta=(Description="Animation Blueprint package path"))
+	FString Blueprint;
+
+	UPROPERTY(meta=(Description="State machine graph name"))
+	FString Graph;
+
+	UPROPERTY(meta=(Description="Name for the new state"))
+	FString StateName;
+
+	UPROPERTY(meta=(Optional, Description="X position of the new state node"))
+	int32 PosX = 200;
+
+	UPROPERTY(meta=(Optional, Description="Y position of the new state node"))
+	int32 PosY = 0;
+
+	UPROPERTY(meta=(Optional, Description="Animation asset package path to assign to the state"))
+	FString AnimationAsset;
+
+	virtual FString GetDescription() const override
+	{
+		return TEXT("Add a new state to an animation state machine graph. "
+			"Optionally assign an animation asset to the state.");
+	}
+
+	virtual void Handle() override
+	{
+		// Resolve the anim blueprint
+		WingFetcher F;
+		UAnimBlueprint* AnimBP = F.Walk(Blueprint).Cast<UAnimBlueprint>();
+		if (!AnimBP) return;
+
+		// Find the state machine graph
+		UAnimationStateMachineGraph* SMGraph = WingUtils::FindStateMachineGraph(AnimBP, Graph);
+		if (!SMGraph)
+		{
+			UWingServer::Printf(TEXT("ERROR: State machine graph '%s' not found in %s\n"), *Graph, *WingUtils::FormatName(AnimBP));
+			return;
+		}
+
+		// Check for duplicate state name
+		if (WingUtils::FindStateByName(SMGraph, StateName))
+		{
+			UWingServer::Printf(TEXT("ERROR: State '%s' already exists in %s\n"), *StateName, *WingUtils::FormatName(SMGraph));
+			return;
+		}
+
+		// Create the state node
+		UAnimStateNode* NewState = NewObject<UAnimStateNode>(SMGraph);
+		NewState->CreateNewGuid();
+		NewState->NodePosX = PosX;
+		NewState->NodePosY = PosY;
+
+		// Set the state name via the bound graph
+		NewState->PostPlacedNewNode();
+		NewState->AllocateDefaultPins();
+
+		// Rename the bound graph to set the state name
+		if (NewState->GetBoundGraph())
+		{
+			NewState->GetBoundGraph()->Rename(*StateName, nullptr);
+		}
+
+		SMGraph->AddNode(NewState, false, false);
+		NewState->SetFlags(RF_Transactional);
+
+		// Optionally set animation asset
+		if (!AnimationAsset.IsEmpty() && NewState->GetBoundGraph())
+		{
+			WingFetcher F2;
+			UAnimSequence* AnimSeq = F2.Asset(AnimationAsset).Cast<UAnimSequence>();
+			if (!AnimSeq) return;
+
+			UAnimGraphNode_SequencePlayer* SeqNode = NewObject<UAnimGraphNode_SequencePlayer>(NewState->GetBoundGraph());
+			SeqNode->CreateNewGuid();
+			SeqNode->PostPlacedNewNode();
+			SeqNode->AllocateDefaultPins();
+			SeqNode->SetAnimationAsset(AnimSeq);
+			SeqNode->NodePosX = 0;
+			SeqNode->NodePosY = 0;
+			NewState->GetBoundGraph()->AddNode(SeqNode, false, false);
+		}
+
+		// Compile
+		FKismetEditorUtilities::CompileBlueprint(AnimBP);
+
+		UWingServer::Printf(TEXT("Created state '%s' in %s\n"), *StateName, *WingUtils::FormatName(SMGraph));
+		UWingServer::Printf(TEXT("  node: %s\n"), *WingUtils::FormatName(NewState));
+	}
+};