ue-mcp 0.4.7 → 0.4.9

package/README.md

@@ -20,13 +20,15 @@ npx ue-mcp init
 
 The interactive setup will:
 
-1. Ask for your `.uproject` path
+1. Find your `.uproject` (auto-detects in current directory)
 2. Let you choose which tool categories to enable
 3. Deploy the C++ bridge plugin to your project
 4. Enable required UE plugins (Niagara, PCG, GAS, etc.)
 5. Detect and configure your MCP client (Claude Code, Claude Desktop, Cursor)
 
-Restart the editor once after setup to load the bridge plugin. Then ask your AI:
+Restart the editor once after setup to load the bridge plugin. To update later: `npx ue-mcp update`
+
+Then ask your AI:
 
 ```
 project(action="get_status")        — verify connection

package/dist/tools/blueprint.js

@@ -34,9 +34,9 @@ export const blueprintTool = categoryTool("blueprint", "Blueprint reading, autho
 - create_function: Create function. Params: assetPath, functionName
 - delete_function: Delete function. Params: assetPath, functionName
 - rename_function: Rename function. Params: assetPath, oldName, newName
-- add_node: Add graph node (K2, AnimGraph, etc — any UEdGraphNode subclass). Params: assetPath, graphName?, nodeClass, nodeParams?
+- add_node: Add graph node (any UEdGraphNode subclass). Aliases: CallFunction, Event, CustomEvent, GetVar, SetVar, Branch. nodeParams: functionName+targetClass for CallFunction, eventName+eventClass? for Event, variableName for GetVar/SetVar. Params: assetPath, graphName?, nodeClass, nodeParams?
 - delete_node: Delete node. Params: assetPath, graphName, nodeName
-- set_node_property: Set node property. Params: assetPath, graphName, nodeName, propertyName, value
+- set_node_property: Set node pin default or struct property (supports dot paths like "Node.IKBone.BoneName"). Params: assetPath, graphName, nodeName, propertyName, value
 - connect_pins: Wire nodes. Params: sourceNode, sourcePin, targetNode, targetPin, assetPath, graphName?
 - add_component: Add BP component. Params: assetPath, componentClass, componentName?
 - compile: Compile Blueprint. Params: assetPath

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ue-mcp",
-  "version": "0.4.7",
+  "version": "0.4.9",
   "description": "Unreal Engine MCP server — 19 tools, 300+ actions for AI-driven editor control",
   "type": "module",
   "main": "dist/index.js",

package/plugin/ue_mcp_bridge/Source/UE_MCP_Bridge/Private/Handlers/BlueprintHandlers.cpp

@@ -41,6 +41,22 @@
 #include "Kismet/KismetStringLibrary.h"
 #include "Kismet/KismetArrayLibrary.h"
 
+// Helper: find a UClass by short name (e.g. "AnimInstance" finds UAnimInstance)
+static UClass* FindClassByShortName(const FString& ClassName)
+{
+	UClass* PrefixedMatch = nullptr;
+	for (TObjectIterator<UClass> It; It; ++It)
+	{
+		const FString& Name = It->GetName();
+		if (Name == ClassName) return *It;
+		if (!PrefixedMatch && (Name == TEXT("U") + ClassName || Name == TEXT("A") + ClassName))
+		{
+			PrefixedMatch = *It;
+		}
+	}
+	return PrefixedMatch;
+}
+
 void FBlueprintHandlers::RegisterHandlers(FMCPHandlerRegistry& Registry)
 {
 	Registry.RegisterHandler(TEXT("create_blueprint"), &CreateBlueprint);
@@ -996,8 +1012,12 @@ TSharedPtr<FJsonValue> FBlueprintHandlers::AddNode(const TSharedPtr<FJsonObject>
 
 	// Resolve short aliases to full class names
 	FString ResolvedClass = NodeClass;
-	if (NodeClass == TEXT("CallFunction")) ResolvedClass = TEXT("K2Node_CallFunction");
-	else if (NodeClass == TEXT("Event"))   ResolvedClass = TEXT("K2Node_Event");
+	if (NodeClass == TEXT("CallFunction"))  ResolvedClass = TEXT("K2Node_CallFunction");
+	else if (NodeClass == TEXT("Event"))    ResolvedClass = TEXT("K2Node_Event");
+	else if (NodeClass == TEXT("GetVar"))   ResolvedClass = TEXT("K2Node_VariableGet");
+	else if (NodeClass == TEXT("SetVar"))   ResolvedClass = TEXT("K2Node_VariableSet");
+	else if (NodeClass == TEXT("Branch"))   ResolvedClass = TEXT("K2Node_IfThenElse");
+	else if (NodeClass == TEXT("CustomEvent")) ResolvedClass = TEXT("K2Node_CustomEvent");
 
 	// Find the UEdGraphNode subclass by name (works for K2, AnimGraph, and any other graph node types)
 	UClass* NodeUClass = nullptr;
@@ -1026,34 +1046,34 @@ TSharedPtr<FJsonValue> FBlueprintHandlers::AddNode(const TSharedPtr<FJsonObject>
 		return MakeShared<FJsonValueObject>(Result);
 	}
 
-	// Special-case initialization for known types
+	// Special-case initialization for known types (must happen BEFORE AllocateDefaultPins)
 	if (UK2Node_CallFunction* CallNode = Cast<UK2Node_CallFunction>(NewNode))
 	{
 		if (NodeParams)
 		{
 			FString FunctionName;
-			if ((*NodeParams)->TryGetStringField(TEXT("functionName"), FunctionName))
+			if (!(*NodeParams)->TryGetStringField(TEXT("functionName"), FunctionName))
+				(*NodeParams)->TryGetStringField(TEXT("memberName"), FunctionName);
+
+			if (!FunctionName.IsEmpty())
 			{
 				FString TargetClassName;
-				if ((*NodeParams)->TryGetStringField(TEXT("targetClass"), TargetClassName))
+				if (!(*NodeParams)->TryGetStringField(TEXT("targetClass"), TargetClassName))
+					(*NodeParams)->TryGetStringField(TEXT("memberParent"), TargetClassName);
+
+				UClass* TargetClass = nullptr;
+				if (!TargetClassName.IsEmpty())
 				{
-					UClass* TargetClass = FindObject<UClass>(nullptr, *TargetClassName);
-					if (!TargetClass)
-					{
-						TargetClass = FindObject<UClass>(nullptr, *(TEXT("U") + TargetClassName));
-					}
-					if (TargetClass)
-					{
-						UFunction* Func = TargetClass->FindFunctionByName(FName(*FunctionName));
-						if (Func)
-						{
-							CallNode->SetFromFunction(Func);
-						}
-					}
+					TargetClass = FindClassByShortName(TargetClassName);
+				}
+				if (!TargetClass)
+				{
+					TargetClass = Blueprint->ParentClass;
 				}
-				else if (Blueprint->ParentClass)
+
+				if (TargetClass)
 				{
-					UFunction* Func = Blueprint->ParentClass->FindFunctionByName(FName(*FunctionName));
+					UFunction* Func = TargetClass->FindFunctionByName(FName(*FunctionName));
 					if (Func)
 					{
 						CallNode->SetFromFunction(Func);
@@ -1067,9 +1087,58 @@ TSharedPtr<FJsonValue> FBlueprintHandlers::AddNode(const TSharedPtr<FJsonObject>
 		if (NodeParams)
 		{
 			FString EventName;
-			if ((*NodeParams)->TryGetStringField(TEXT("eventName"), EventName))
+			if (!(*NodeParams)->TryGetStringField(TEXT("eventName"), EventName))
+				(*NodeParams)->TryGetStringField(TEXT("memberName"), EventName);
+
+			if (!EventName.IsEmpty())
+			{
+				FString EventClassName;
+				if (!(*NodeParams)->TryGetStringField(TEXT("eventClass"), EventClassName))
+					(*NodeParams)->TryGetStringField(TEXT("memberParent"), EventClassName);
+
+				if (!EventClassName.IsEmpty())
+				{
+					// Engine event override — bind via EventReference
+					UClass* EventClass = FindClassByShortName(EventClassName);
+					if (!EventClass) EventClass = Blueprint->ParentClass;
+
+					if (EventClass)
+					{
+						UFunction* EventFunc = EventClass->FindFunctionByName(FName(*EventName));
+						if (EventFunc)
+						{
+							bool bIsSelf = Blueprint->ParentClass && Blueprint->ParentClass->IsChildOf(EventClass);
+							EventNode->EventReference.SetFromField<UFunction>(EventFunc, bIsSelf);
+						}
+					}
+				}
+				else
+				{
+					// Custom event — just set the name
+					EventNode->CustomFunctionName = FName(*EventName);
+				}
+			}
+		}
+	}
+	else if (UK2Node_VariableGet* GetNode = Cast<UK2Node_VariableGet>(NewNode))
+	{
+		if (NodeParams)
+		{
+			FString VarName;
+			if ((*NodeParams)->TryGetStringField(TEXT("variableName"), VarName))
 			{
-				EventNode->CustomFunctionName = FName(*EventName);
+				GetNode->VariableReference.SetSelfMember(FName(*VarName));
+			}
+		}
+	}
+	else if (UK2Node_VariableSet* SetNode = Cast<UK2Node_VariableSet>(NewNode))
+	{
+		if (NodeParams)
+		{
+			FString VarName;
+			if ((*NodeParams)->TryGetStringField(TEXT("variableName"), VarName))
+			{
+				SetNode->VariableReference.SetSelfMember(FName(*VarName));
 			}
 		}
 	}
@@ -1986,7 +2055,7 @@ TSharedPtr<FJsonValue> FBlueprintHandlers::SetNodeProperty(const TSharedPtr<FJso
 		return MakeShared<FJsonValueObject>(Result);
 	}
 
-	// Find the pin on the node
+	// First try to find a pin with this name
 	UEdGraphPin* TargetPin = nullptr;
 	for (UEdGraphPin* Pin : TargetNode->Pins)
 	{
@@ -1997,29 +2066,81 @@ TSharedPtr<FJsonValue> FBlueprintHandlers::SetNodeProperty(const TSharedPtr<FJso
 		}
 	}
 
-	if (!TargetPin)
+	bool bSetViaPin = false;
+	bool bSetViaProperty = false;
+
+	if (TargetPin)
 	{
-		// List available pins for better error message
-		TArray<FString> PinNames;
-		for (UEdGraphPin* Pin : TargetNode->Pins)
+		// Set pin default value using the graph's own schema
+		const UEdGraphSchema* Schema = TargetGraph->GetSchema();
+		if (Schema)
 		{
-			if (Pin)
+			Schema->TrySetDefaultValue(*TargetPin, DefaultValue);
+			TargetNode->PinDefaultValueChanged(TargetPin);
+			bSetViaPin = true;
+		}
+	}
+
+	if (!bSetViaPin)
+	{
+		// No pin found — try setting as a node property via reflection.
+		// Supports dotted paths like "Node.IKBone.BoneName" for AnimGraph inner structs.
+		TArray<FString> PathParts;
+		PinName.ParseIntoArray(PathParts, TEXT("."));
+
+		UStruct* CurrentStruct = TargetNode->GetClass();
+		void* CurrentContainer = TargetNode;
+		FProperty* FinalProp = nullptr;
+
+		for (int32 i = 0; i < PathParts.Num(); i++)
+		{
+			FProperty* Prop = CurrentStruct->FindPropertyByName(FName(*PathParts[i]));
+			if (!Prop) break;
+
+			if (i < PathParts.Num() - 1)
 			{
-				PinNames.Add(Pin->PinName.ToString());
+				// Intermediate path segment — drill into struct
+				FStructProperty* StructProp = CastField<FStructProperty>(Prop);
+				if (!StructProp) break;
+				CurrentContainer = StructProp->ContainerPtrToValuePtr<void>(CurrentContainer);
+				CurrentStruct = StructProp->Struct;
+			}
+			else
+			{
+				FinalProp = Prop;
 			}
 		}
-		FString AvailablePins = FString::Join(PinNames, TEXT(", "));
-		Result->SetStringField(TEXT("error"), FString::Printf(TEXT("Pin not found: '%s'. Available pins: [%s]"), *PinName, *AvailablePins));
-		Result->SetBoolField(TEXT("success"), false);
-		return MakeShared<FJsonValueObject>(Result);
+
+		if (FinalProp)
+		{
+			void* ValuePtr = FinalProp->ContainerPtrToValuePtr<void>(CurrentContainer);
+			FinalProp->ImportText_Direct(*DefaultValue, ValuePtr, nullptr, PPF_None);
+			TargetNode->PostEditChange();
+			bSetViaProperty = true;
+		}
 	}
 
-	// Set the default value using the schema
-	const UEdGraphSchema_K2* Schema = GetDefault<UEdGraphSchema_K2>();
-	Schema->TrySetDefaultValue(*TargetPin, DefaultValue);
+	if (!bSetViaPin && !bSetViaProperty)
+	{
+		// Neither pin nor property found — build a helpful error
+		TArray<FString> PinNames;
+		for (UEdGraphPin* Pin : TargetNode->Pins)
+		{
+			if (Pin) PinNames.Add(Pin->PinName.ToString());
+		}
+
+		TArray<FString> PropNames;
+		for (TFieldIterator<FProperty> It(TargetNode->GetClass()); It; ++It)
+		{
+			PropNames.Add(It->GetName());
+		}
 
-	// Notify the node that a pin default changed
-	TargetNode->PinDefaultValueChanged(TargetPin);
+		Result->SetStringField(TEXT("error"), FString::Printf(
+			TEXT("'%s' not found as pin or property. Pins: [%s]. Properties: [%s]"),
+			*PinName, *FString::Join(PinNames, TEXT(", ")), *FString::Join(PropNames, TEXT(", "))));
+		Result->SetBoolField(TEXT("success"), false);
+		return MakeShared<FJsonValueObject>(Result);
+	}
 
 	// Compile and save
 	FKismetEditorUtilities::CompileBlueprint(Blueprint);
@@ -2036,8 +2157,9 @@ TSharedPtr<FJsonValue> FBlueprintHandlers::SetNodeProperty(const TSharedPtr<FJso
 	Result->SetStringField(TEXT("path"), AssetPath);
 	Result->SetStringField(TEXT("graphName"), GraphName);
 	Result->SetStringField(TEXT("nodeId"), NodeId);
-	Result->SetStringField(TEXT("pinName"), PinName);
-	Result->SetStringField(TEXT("defaultValue"), DefaultValue);
+	Result->SetStringField(TEXT("propertyName"), PinName);
+	Result->SetStringField(TEXT("value"), DefaultValue);
+	Result->SetStringField(TEXT("setVia"), bSetViaPin ? TEXT("pin") : TEXT("property"));
 	Result->SetBoolField(TEXT("success"), true);
 
 	return MakeShared<FJsonValueObject>(Result);