unreal-engine-mcp-server 0.4.7 → 0.5.1

package/plugins/McpAutomationBridge/Source/McpAutomationBridge/Private/McpAutomationBridge_InputHandlers.cpp

@@ -0,0 +1,209 @@
+#include "McpAutomationBridgeGlobals.h"
+#include "McpAutomationBridgeHelpers.h"
+#include "McpAutomationBridgeSubsystem.h"
+
+// Enhanced Input (Editor Only)
+#if WITH_EDITOR
+#include "AssetToolsModule.h"
+#include "EditorAssetLibrary.h"
+#include "EnhancedInputEditorSubsystem.h"
+#include "Factories/Factory.h"
+#include "InputAction.h"
+#include "InputMappingContext.h"
+
+#endif
+
+bool UMcpAutomationBridgeSubsystem::HandleInputAction(
+    const FString &RequestId, const FString &Action,
+    const TSharedPtr<FJsonObject> &Payload,
+    TSharedPtr<FMcpBridgeWebSocket> RequestingSocket) {
+  if (Action != TEXT("manage_input")) {
+    return false;
+  }
+
+#if WITH_EDITOR
+  if (!Payload.IsValid()) {
+    SendAutomationError(RequestingSocket, RequestId, TEXT("Missing payload."),
+                        TEXT("INVALID_PAYLOAD"));
+    return true;
+  }
+
+  FString SubAction;
+  if (!Payload->TryGetStringField(TEXT("action"), SubAction)) {
+    SendAutomationError(RequestingSocket, RequestId,
+                        TEXT("Missing 'action' field in payload."),
+                        TEXT("INVALID_ARGUMENT"));
+    return true;
+  }
+
+  UE_LOG(LogMcpAutomationBridgeSubsystem, Log, TEXT("HandleInputAction: %s"),
+         *SubAction);
+
+  if (SubAction == TEXT("create_input_action")) {
+    FString Name;
+    Payload->TryGetStringField(TEXT("name"), Name);
+    FString Path;
+    Payload->TryGetStringField(TEXT("path"), Path);
+
+    if (Name.IsEmpty() || Path.IsEmpty()) {
+      SendAutomationError(RequestingSocket, RequestId,
+                          TEXT("Name and path are required."),
+                          TEXT("INVALID_ARGUMENT"));
+      return true;
+    }
+
+    const FString FullPath = FString::Printf(TEXT("%s/%s"), *Path, *Name);
+    if (UEditorAssetLibrary::DoesAssetExist(FullPath)) {
+      SendAutomationError(
+          RequestingSocket, RequestId,
+          FString::Printf(TEXT("Asset already exists at %s"), *FullPath),
+          TEXT("ASSET_EXISTS"));
+      return true;
+    }
+
+    IAssetTools &AssetTools =
+        FModuleManager::Get()
+            .LoadModuleChecked<FAssetToolsModule>("AssetTools")
+            .Get();
+
+    // UInputActionFactory is not exposed directly in public headers sometimes,
+    // but we can rely on AssetTools to create it if we have the class.
+    UClass *ActionClass = UInputAction::StaticClass();
+    UObject *NewAsset =
+        AssetTools.CreateAsset(Name, Path, ActionClass, nullptr);
+
+    if (NewAsset) {
+      // Force save
+      SaveLoadedAssetThrottled(NewAsset, -1.0, true);
+      TSharedPtr<FJsonObject> Result = MakeShared<FJsonObject>();
+      Result->SetStringField(TEXT("assetPath"), NewAsset->GetPathName());
+      SendAutomationResponse(RequestingSocket, RequestId, true,
+                             TEXT("Input Action created."), Result);
+    } else {
+      SendAutomationError(RequestingSocket, RequestId,
+                          TEXT("Failed to create Input Action."),
+                          TEXT("CREATION_FAILED"));
+    }
+  } else if (SubAction == TEXT("create_input_mapping_context")) {
+    FString Name;
+    Payload->TryGetStringField(TEXT("name"), Name);
+    FString Path;
+    Payload->TryGetStringField(TEXT("path"), Path);
+
+    if (Name.IsEmpty() || Path.IsEmpty()) {
+      SendAutomationError(RequestingSocket, RequestId,
+                          TEXT("Name and path are required."),
+                          TEXT("INVALID_ARGUMENT"));
+      return true;
+    }
+
+    const FString FullPath = FString::Printf(TEXT("%s/%s"), *Path, *Name);
+    if (UEditorAssetLibrary::DoesAssetExist(FullPath)) {
+      SendAutomationError(
+          RequestingSocket, RequestId,
+          FString::Printf(TEXT("Asset already exists at %s"), *FullPath),
+          TEXT("ASSET_EXISTS"));
+      return true;
+    }
+
+    IAssetTools &AssetTools =
+        FModuleManager::Get()
+            .LoadModuleChecked<FAssetToolsModule>("AssetTools")
+            .Get();
+
+    UClass *ContextClass = UInputMappingContext::StaticClass();
+    UObject *NewAsset =
+        AssetTools.CreateAsset(Name, Path, ContextClass, nullptr);
+
+    if (NewAsset) {
+      SaveLoadedAssetThrottled(NewAsset, -1.0, true);
+      TSharedPtr<FJsonObject> Result = MakeShared<FJsonObject>();
+      Result->SetStringField(TEXT("assetPath"), NewAsset->GetPathName());
+      SendAutomationResponse(RequestingSocket, RequestId, true,
+                             TEXT("Input Mapping Context created."), Result);
+    } else {
+      SendAutomationError(RequestingSocket, RequestId,
+                          TEXT("Failed to create Input Mapping Context."),
+                          TEXT("CREATION_FAILED"));
+    }
+  } else if (SubAction == TEXT("add_mapping")) {
+    FString ContextPath;
+    Payload->TryGetStringField(TEXT("contextPath"), ContextPath);
+    FString ActionPath;
+    Payload->TryGetStringField(TEXT("actionPath"), ActionPath);
+    FString KeyName;
+    Payload->TryGetStringField(TEXT("key"), KeyName);
+
+    UInputMappingContext *Context =
+        Cast<UInputMappingContext>(UEditorAssetLibrary::LoadAsset(ContextPath));
+    UInputAction *InAction =
+        Cast<UInputAction>(UEditorAssetLibrary::LoadAsset(ActionPath));
+
+    if (!Context || !InAction || KeyName.IsEmpty()) {
+      SendAutomationError(RequestingSocket, RequestId,
+                          TEXT("Invalid context, action, or key."),
+                          TEXT("INVALID_ARGUMENT"));
+      return true;
+    }
+
+    FKey Key = FKey(FName(*KeyName));
+    if (!Key.IsValid()) {
+      SendAutomationError(RequestingSocket, RequestId,
+                          TEXT("Invalid key name."), TEXT("INVALID_ARGUMENT"));
+      return true;
+    }
+
+    FEnhancedActionKeyMapping &Mapping = Context->MapKey(InAction, Key);
+
+    // Save changes
+    SaveLoadedAssetThrottled(Context, -1.0, true);
+
+    SendAutomationResponse(RequestingSocket, RequestId, true,
+                           TEXT("Mapping added."), nullptr);
+  } else if (SubAction == TEXT("remove_mapping")) {
+    FString ContextPath;
+    Payload->TryGetStringField(TEXT("contextPath"), ContextPath);
+    FString ActionPath;
+    Payload->TryGetStringField(TEXT("actionPath"), ActionPath);
+
+    UInputMappingContext *Context =
+        Cast<UInputMappingContext>(UEditorAssetLibrary::LoadAsset(ContextPath));
+    UInputAction *InAction =
+        Cast<UInputAction>(UEditorAssetLibrary::LoadAsset(ActionPath));
+
+    if (!Context || !InAction) {
+      SendAutomationError(RequestingSocket, RequestId,
+                          TEXT("Invalid context or action."),
+                          TEXT("INVALID_ARGUMENT"));
+      return true;
+    }
+
+    // Context->UnmapAction(InAction); // Not available in 5.x
+    TArray<FKey> KeysToRemove;
+    for (const FEnhancedActionKeyMapping &Mapping : Context->GetMappings()) {
+      if (Mapping.Action == InAction) {
+        KeysToRemove.Add(Mapping.Key);
+      }
+    }
+    for (const FKey &KeyToRemove : KeysToRemove) {
+      Context->UnmapKey(InAction, KeyToRemove);
+    }
+    SaveLoadedAssetThrottled(Context, -1.0, true);
+
+    SendAutomationResponse(RequestingSocket, RequestId, true,
+                           TEXT("Mappings removed for action."), nullptr);
+  } else {
+    SendAutomationError(
+        RequestingSocket, RequestId,
+        FString::Printf(TEXT("Unknown sub-action: %s"), *SubAction),
+        TEXT("UNKNOWN_ACTION"));
+  }
+
+  return true;
+#else
+  SendAutomationError(RequestingSocket, RequestId,
+                      TEXT("Input management requires Editor build."),
+                      TEXT("NOT_AVAILABLE"));
+  return true;
+#endif
+}

package/plugins/McpAutomationBridge/Source/McpAutomationBridge/Private/McpAutomationBridge_InsightsHandlers.cpp

@@ -0,0 +1,41 @@
+#include "McpAutomationBridgeSubsystem.h"
+#include "McpAutomationBridgeHelpers.h"
+#include "McpAutomationBridgeGlobals.h"
+
+bool UMcpAutomationBridgeSubsystem::HandleInsightsAction(const FString& RequestId, const FString& Action, const TSharedPtr<FJsonObject>& Payload, TSharedPtr<FMcpBridgeWebSocket> RequestingSocket)
+{
+    if (Action != TEXT("manage_insights"))
+    {
+        return false;
+    }
+
+    if (!Payload.IsValid())
+    {
+        SendAutomationError(RequestingSocket, RequestId, TEXT("Missing payload."), TEXT("INVALID_PAYLOAD"));
+        return true;
+    }
+
+    FString SubAction = Payload->GetStringField(TEXT("subAction"));
+
+    if (SubAction == TEXT("start_session"))
+    {
+        // Start trace using console command which is the standard way to control trace from editor
+        // "Trace.Start"
+        
+        FString Channels;
+        if (Payload->TryGetStringField(TEXT("channels"), Channels) && !Channels.IsEmpty())
+        {
+             GEngine->Exec(nullptr, *FString::Printf(TEXT("Trace.Start %s"), *Channels));
+        }
+        else
+        {
+             GEngine->Exec(nullptr, TEXT("Trace.Start"));
+        }
+        
+        SendAutomationResponse(RequestingSocket, RequestId, true, TEXT("Trace session started."));
+        return true;
+    }
+
+    SendAutomationError(RequestingSocket, RequestId, TEXT("Unknown subAction."), TEXT("INVALID_SUBACTION"));
+    return true;
+}