unreal-engine-mcp-server 0.5.4 → 0.5.6

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

@@ -1,762 +0,0 @@
-#include "McpAutomationBridgeGlobals.h"
-#include "McpAutomationBridgeHelpers.h"
-#include "McpAutomationBridgeSubsystem.h"
-
-#if WITH_EDITOR
-#include "Editor.h"
-#include "EditorLevelUtils.h"
-#include "Engine/LevelStreaming.h"
-#include "Engine/LevelStreamingAlwaysLoaded.h"
-#include "Engine/LevelStreamingDynamic.h"
-#include "Engine/World.h"
-#include "FileHelpers.h"
-#include "LevelEditor.h"
-#include "RenderingThread.h"
-
-// Check for LevelEditorSubsystem
-#if defined(__has_include)
-#if __has_include("Subsystems/LevelEditorSubsystem.h")
-#include "Subsystems/LevelEditorSubsystem.h"
-#define MCP_HAS_LEVELEDITOR_SUBSYSTEM 1
-#elif __has_include("LevelEditorSubsystem.h")
-#include "LevelEditorSubsystem.h"
-#define MCP_HAS_LEVELEDITOR_SUBSYSTEM 1
-#else
-#define MCP_HAS_LEVELEDITOR_SUBSYSTEM 0
-#endif
-#else
-#define MCP_HAS_LEVELEDITOR_SUBSYSTEM 0
-#endif
-#endif
-
-bool UMcpAutomationBridgeSubsystem::HandleLevelAction(
-    const FString &RequestId, const FString &Action,
-    const TSharedPtr<FJsonObject> &Payload,
-    TSharedPtr<FMcpBridgeWebSocket> RequestingSocket) {
-  const FString Lower = Action.ToLower();
-  const bool bIsLevelAction =
-      (Lower == TEXT("manage_level") || Lower == TEXT("save_current_level") ||
-       Lower == TEXT("create_new_level") || Lower == TEXT("stream_level") ||
-       Lower == TEXT("spawn_light") || Lower == TEXT("build_lighting") ||
-       Lower == TEXT("spawn_light") || Lower == TEXT("build_lighting") ||
-       Lower == TEXT("bake_lightmap") || Lower == TEXT("list_levels") ||
-       Lower == TEXT("export_level") || Lower == TEXT("import_level") ||
-       Lower == TEXT("add_sublevel"));
-  if (!bIsLevelAction)
-    return false;
-
-  FString EffectiveAction = Lower;
-
-  // Unpack manage_level
-  if (Lower == TEXT("manage_level")) {
-    if (!Payload.IsValid()) {
-      SendAutomationError(RequestingSocket, RequestId,
-                          TEXT("manage_level payload missing"),
-                          TEXT("INVALID_PAYLOAD"));
-      return true;
-    }
-    FString SubAction;
-    Payload->TryGetStringField(TEXT("action"), SubAction);
-    const FString LowerSub = SubAction.ToLower();
-
-    if (LowerSub == TEXT("load") || LowerSub == TEXT("load_level")) {
-      // Map to Open command
-      FString LevelPath;
-      Payload->TryGetStringField(TEXT("levelPath"), LevelPath);
-
-      // Determine invalid characters for checks
-      if (LevelPath.IsEmpty()) {
-        SendAutomationError(RequestingSocket, RequestId,
-                            TEXT("levelPath required"),
-                            TEXT("INVALID_ARGUMENT"));
-      }
-
-      // Auto-resolve short names
-      if (!LevelPath.StartsWith(TEXT("/")) && !FPaths::FileExists(LevelPath)) {
-        FString TryPath = FString::Printf(TEXT("/Game/Maps/%s"), *LevelPath);
-        if (FPackageName::DoesPackageExist(TryPath)) {
-          LevelPath = TryPath;
-        }
-      }
-
-#if WITH_EDITOR
-      if (!GEditor) {
-        SendAutomationResponse(RequestingSocket, RequestId, false,
-                               TEXT("Editor not available"), nullptr,
-                               TEXT("EDITOR_NOT_AVAILABLE"));
-        return true;
-      }
-
-      // Try to resolve package path to filename
-      FString Filename;
-      bool bGotFilename = false;
-      if (FPackageName::IsPackageFilename(LevelPath)) {
-        Filename = LevelPath;
-        bGotFilename = true;
-      } else {
-        // Assume package path
-        if (FPackageName::TryConvertLongPackageNameToFilename(
-                LevelPath, Filename, FPackageName::GetMapPackageExtension())) {
-          bGotFilename = true;
-        }
-      }
-
-      // If conversion failed, it might be a short name? But LoadMap usually
-      // needs full path. Let's try to load what we have if conversion returned
-      // something, else fallback to input.
-      const FString FileToLoad = bGotFilename ? Filename : LevelPath;
-
-      // Force any pending work to complete
-      FlushRenderingCommands();
-
-      // LoadMap prompts for save if dirty. To avoid blocking automation, we
-      // should carefuly consider. But for now, we assume user wants standard
-      // behavior or has saved. There isn't a simple "Force Load" via FileUtils
-      // without clearing dirty flags manually. We will proceed with LoadMap.
-      const bool bLoaded = FEditorFileUtils::LoadMap(FileToLoad);
-
-      if (bLoaded) {
-        TSharedPtr<FJsonObject> Resp = MakeShared<FJsonObject>();
-        Resp->SetStringField(TEXT("levelPath"), LevelPath);
-        SendAutomationResponse(RequestingSocket, RequestId, true,
-                               TEXT("Level loaded"), Resp, FString());
-        return true;
-      } else {
-        // Fallback to ExecuteConsoleCommand "Open" if LoadMap failed (e.g.
-        // maybe it was a raw asset path or something) But actually if LoadMap
-        // fails, Open likely fails too.
-        SendAutomationResponse(
-            RequestingSocket, RequestId, false,
-            FString::Printf(TEXT("Failed to load map: %s"), *LevelPath),
-            nullptr, TEXT("LOAD_FAILED"));
-        return true;
-      }
-#else
-      return false;
-#endif
-    } else if (LowerSub == TEXT("save")) {
-      EffectiveAction = TEXT("save_current_level");
-    } else if (LowerSub == TEXT("save_as") ||
-               LowerSub == TEXT("save_level_as")) {
-      EffectiveAction = TEXT("save_level_as");
-    } else if (LowerSub == TEXT("create_level")) {
-      EffectiveAction = TEXT("create_new_level");
-    } else if (LowerSub == TEXT("stream")) {
-      EffectiveAction = TEXT("stream_level");
-    } else if (LowerSub == TEXT("create_light")) {
-      EffectiveAction = TEXT("spawn_light");
-    } else if (LowerSub == TEXT("list") || LowerSub == TEXT("list_levels")) {
-      EffectiveAction = TEXT("list_levels");
-    } else if (LowerSub == TEXT("export_level")) {
-      EffectiveAction = TEXT("export_level");
-    } else if (LowerSub == TEXT("import_level")) {
-    } else if (LowerSub == TEXT("import_level")) {
-      EffectiveAction = TEXT("import_level");
-    } else if (LowerSub == TEXT("add_sublevel")) {
-      EffectiveAction = TEXT("add_sublevel");
-    } else {
-      SendAutomationError(
-          RequestingSocket, RequestId,
-          FString::Printf(TEXT("Unknown manage_level action: %s"), *SubAction),
-          TEXT("UNKNOWN_ACTION"));
-      return true;
-    }
-  }
-
-#if WITH_EDITOR
-  if (EffectiveAction == TEXT("save_current_level")) {
-    if (!GEditor) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("Editor not available"), nullptr,
-                             TEXT("EDITOR_NOT_AVAILABLE"));
-      return true;
-    }
-
-    UWorld *World = GEditor->GetEditorWorldContext().World();
-    if (!World) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("No world loaded"), nullptr,
-                             TEXT("NO_WORLD"));
-      return true;
-    }
-
-    bool bSaved = FEditorFileUtils::SaveCurrentLevel();
-    if (bSaved) {
-      TSharedPtr<FJsonObject> Resp = MakeShared<FJsonObject>();
-      Resp->SetStringField(TEXT("levelPath"), World->GetOutermost()->GetName());
-      SendAutomationResponse(RequestingSocket, RequestId, true,
-                             TEXT("Level saved"), Resp, FString());
-    } else {
-      // Provide detailed error information
-      TSharedPtr<FJsonObject> ErrorDetail = MakeShared<FJsonObject>();
-      FString PackageName = World->GetOutermost()->GetName();
-      ErrorDetail->SetStringField(TEXT("attemptedPath"), PackageName);
-
-      FString Filename;
-      FString ErrorReason = TEXT("Unknown save failure");
-
-      if (PackageName.Contains(TEXT("Untitled")) ||
-          PackageName.StartsWith(TEXT("/Temp/"))) {
-        ErrorReason = TEXT(
-            "Level is unsaved/temporary. Use save_level_as with a path first.");
-        ErrorDetail->SetStringField(
-            TEXT("hint"),
-            TEXT(
-                "Use manage_level with action='save_as' and provide savePath"));
-      } else if (FPackageName::TryConvertLongPackageNameToFilename(
-                     PackageName, Filename,
-                     FPackageName::GetMapPackageExtension())) {
-        if (IFileManager::Get().IsReadOnly(*Filename)) {
-          ErrorReason = TEXT("File is read-only or locked by another process");
-          ErrorDetail->SetStringField(TEXT("filename"), Filename);
-        } else if (!IFileManager::Get().DirectoryExists(
-                       *FPaths::GetPath(Filename))) {
-          ErrorReason = TEXT("Target directory does not exist");
-          ErrorDetail->SetStringField(TEXT("directory"),
-                                      FPaths::GetPath(Filename));
-        } else {
-          ErrorReason =
-              TEXT("Save operation failed - check Output Log for details");
-          ErrorDetail->SetStringField(TEXT("filename"), Filename);
-        }
-      }
-
-      ErrorDetail->SetStringField(TEXT("reason"), ErrorReason);
-      SendAutomationResponse(
-          RequestingSocket, RequestId, false,
-          FString::Printf(TEXT("Failed to save level: %s"), *ErrorReason),
-          ErrorDetail, TEXT("SAVE_FAILED"));
-    }
-    return true;
-  }
-  if (EffectiveAction == TEXT("save_level_as")) {
-    // Force cleanup to prevent potential deadlocks with HLODs/WorldPartition
-    // during save
-    if (GEditor) {
-      FlushRenderingCommands();
-      GEditor->ForceGarbageCollection(true);
-      FlushRenderingCommands();
-    }
-
-    FString SavePath;
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("savePath"), SavePath);
-    if (SavePath.IsEmpty()) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("savePath required for save_level_as"),
-                             nullptr, TEXT("INVALID_ARGUMENT"));
-      return true;
-    }
-
-#if defined(MCP_HAS_LEVELEDITOR_SUBSYSTEM)
-    if (ULevelEditorSubsystem *LevelEditorSS =
-            GEditor->GetEditorSubsystem<ULevelEditorSubsystem>()) {
-      bool bSaved = false;
-#if __has_include("FileHelpers.h")
-      if (UWorld *World = GEditor->GetEditorWorldContext().World()) {
-        bSaved = FEditorFileUtils::SaveMap(World, SavePath);
-      }
-#endif
-      if (bSaved) {
-        TSharedPtr<FJsonObject> Resp = MakeShared<FJsonObject>();
-        Resp->SetStringField(TEXT("levelPath"), SavePath);
-        SendAutomationResponse(
-            RequestingSocket, RequestId, true,
-            FString::Printf(TEXT("Level saved as %s"), *SavePath), Resp,
-            FString());
-      } else {
-        SendAutomationResponse(RequestingSocket, RequestId, false,
-                               TEXT("Failed to save level as"), nullptr,
-                               TEXT("SAVE_FAILED"));
-      }
-      return true;
-    }
-#endif
-    SendAutomationResponse(RequestingSocket, RequestId, false,
-                           TEXT("LevelEditorSubsystem not available"), nullptr,
-                           TEXT("SUBSYSTEM_MISSING"));
-    return true;
-  }
-  if (EffectiveAction == TEXT("build_lighting") ||
-      EffectiveAction == TEXT("bake_lightmap")) {
-    TSharedPtr<FJsonObject> P = MakeShared<FJsonObject>();
-    P->SetStringField(TEXT("functionName"), TEXT("BUILD_LIGHTING"));
-    if (Payload.IsValid()) {
-      FString Q;
-      if (Payload->TryGetStringField(TEXT("quality"), Q) && !Q.IsEmpty())
-        P->SetStringField(TEXT("quality"), Q);
-    }
-    return HandleExecuteEditorFunction(
-        RequestId, TEXT("execute_editor_function"), P, RequestingSocket);
-  }
-  if (EffectiveAction == TEXT("create_new_level")) {
-    FString LevelName;
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("levelName"), LevelName);
-
-    FString LevelPath;
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("levelPath"), LevelPath);
-
-    // Construct valid package path
-    FString SavePath = LevelPath;
-    if (SavePath.IsEmpty() && !LevelName.IsEmpty()) {
-      if (LevelName.StartsWith(TEXT("/")))
-        SavePath = LevelName;
-      else
-        SavePath = FString::Printf(TEXT("/Game/Maps/%s"), *LevelName);
-    }
-
-    if (SavePath.IsEmpty()) {
-      SendAutomationResponse(
-          RequestingSocket, RequestId, false,
-          TEXT("levelName or levelPath required for create_level"), nullptr,
-          TEXT("INVALID_ARGUMENT"));
-      return true;
-    }
-
-    // Check if map already exists
-    if (FPackageName::DoesPackageExist(SavePath)) {
-      // If exists, just open it
-      const FString Cmd = FString::Printf(TEXT("Open %s"), *SavePath);
-      TSharedPtr<FJsonObject> P = MakeShared<FJsonObject>();
-      P->SetStringField(TEXT("command"), Cmd);
-      return HandleExecuteEditorFunction(
-          RequestId, TEXT("execute_console_command"), P, RequestingSocket);
-    }
-
-    // Create new map
-#if defined(MCP_HAS_LEVELEDITOR_SUBSYSTEM) && __has_include("FileHelpers.h")
-    if (GEditor->IsPlaySessionInProgress()) {
-      GEditor->RequestEndPlayMap();
-      SendAutomationResponse(
-          RequestingSocket, RequestId, false,
-          TEXT("Cannot create level while Play In Editor is active."), nullptr,
-          TEXT("PIE_ACTIVE"));
-      return true;
-    }
-
-    // Force cleanup of previous world/resources to prevent RenderCore/Driver
-    // crashes (monza/D3D12) especially when tests run back-to-back triggering
-    // thumbnail generation or world partition shutdown.
-    if (GEditor) {
-      FlushRenderingCommands();
-      GEditor->ForceGarbageCollection(true);
-      FlushRenderingCommands();
-    }
-
-    if (UWorld *NewWorld =
-            GEditor->NewMap(true)) // true = force new map (creates untitled)
-    {
-      GEditor->GetEditorWorldContext().SetCurrentWorld(NewWorld);
-
-      // Save it to valid path
-      // ISSUE #1 FIX: Ensure directory exists
-      FString Filename;
-      if (FPackageName::TryConvertLongPackageNameToFilename(
-              SavePath, Filename, FPackageName::GetMapPackageExtension())) {
-        IFileManager::Get().MakeDirectory(*FPaths::GetPath(Filename), true);
-      }
-
-      if (FEditorFileUtils::SaveMap(NewWorld, SavePath)) {
-        TSharedPtr<FJsonObject> Resp = MakeShared<FJsonObject>();
-        Resp->SetStringField(TEXT("levelPath"), SavePath);
-        Resp->SetStringField(TEXT("packagePath"), SavePath);
-        Resp->SetStringField(TEXT("objectPath"),
-                             SavePath + TEXT(".") +
-                                 FPaths::GetBaseFilename(SavePath));
-        SendAutomationResponse(
-            RequestingSocket, RequestId, true,
-            FString::Printf(TEXT("Level created: %s"), *SavePath), Resp,
-            FString());
-      } else {
-        SendAutomationResponse(RequestingSocket, RequestId, false,
-                               TEXT("Failed to save new level"), nullptr,
-                               TEXT("SAVE_FAILED"));
-      }
-    } else {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("Failed to create new map"), nullptr,
-                             TEXT("CREATION_FAILED"));
-    }
-    return true;
-#else
-    // Fallback for missing headers (shouldn't happen given build.cs)
-    const FString Cmd = FString::Printf(TEXT("Open %s"), *SavePath);
-    TSharedPtr<FJsonObject> P = MakeShared<FJsonObject>();
-    P->SetStringField(TEXT("command"), Cmd);
-    return HandleExecuteEditorFunction(
-        RequestId, TEXT("execute_console_command"), P, RequestingSocket);
-#endif
-  }
-  if (EffectiveAction == TEXT("stream_level")) {
-    FString LevelName;
-    bool bLoad = true;
-    bool bVis = true;
-    if (Payload.IsValid()) {
-      Payload->TryGetStringField(TEXT("levelName"), LevelName);
-      Payload->TryGetBoolField(TEXT("shouldBeLoaded"), bLoad);
-      Payload->TryGetBoolField(TEXT("shouldBeVisible"), bVis);
-      if (LevelName.IsEmpty())
-        Payload->TryGetStringField(TEXT("levelPath"), LevelName);
-    }
-    if (LevelName.TrimStartAndEnd().IsEmpty()) {
-      SendAutomationResponse(
-          RequestingSocket, RequestId, false,
-          TEXT("stream_level requires levelName or levelPath"), nullptr,
-          TEXT("INVALID_ARGUMENT"));
-      return true;
-    }
-    const FString Cmd =
-        FString::Printf(TEXT("StreamLevel %s %s %s"), *LevelName,
-                        bLoad ? TEXT("Load") : TEXT("Unload"),
-                        bVis ? TEXT("Show") : TEXT("Hide"));
-    TSharedPtr<FJsonObject> P = MakeShared<FJsonObject>();
-    P->SetStringField(TEXT("command"), Cmd);
-    return HandleExecuteEditorFunction(
-        RequestId, TEXT("execute_console_command"), P, RequestingSocket);
-  }
-  if (EffectiveAction == TEXT("spawn_light")) {
-    FString LightType = TEXT("Point");
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("lightType"), LightType);
-    const FString LT = LightType.ToLower();
-    FString ClassName;
-    if (LT == TEXT("directional"))
-      ClassName = TEXT("DirectionalLight");
-    else if (LT == TEXT("spot"))
-      ClassName = TEXT("SpotLight");
-    else if (LT == TEXT("rect"))
-      ClassName = TEXT("RectLight");
-    else
-      ClassName = TEXT("PointLight");
-    TSharedPtr<FJsonObject> Params = MakeShared<FJsonObject>();
-    if (Payload.IsValid()) {
-      const TSharedPtr<FJsonObject> *L = nullptr;
-      if (Payload->TryGetObjectField(TEXT("location"), L) && L &&
-          (*L).IsValid())
-        Params->SetObjectField(TEXT("location"), *L);
-      const TSharedPtr<FJsonObject> *R = nullptr;
-      if (Payload->TryGetObjectField(TEXT("rotation"), R) && R &&
-          (*R).IsValid())
-        Params->SetObjectField(TEXT("rotation"), *R);
-    }
-    TSharedPtr<FJsonObject> P = MakeShared<FJsonObject>();
-    P->SetStringField(TEXT("functionName"), TEXT("SPAWN_ACTOR_AT_LOCATION"));
-    P->SetStringField(TEXT("class_path"), ClassName);
-    P->SetObjectField(TEXT("params"), Params.ToSharedRef());
-    return HandleExecuteEditorFunction(
-        RequestId, TEXT("execute_editor_function"), P, RequestingSocket);
-  }
-  if (EffectiveAction == TEXT("list_levels")) {
-    TSharedPtr<FJsonObject> Resp = MakeShared<FJsonObject>();
-    TArray<TSharedPtr<FJsonValue>> LevelsArray;
-
-    UWorld *World =
-        GEditor ? GEditor->GetEditorWorldContext().World() : nullptr;
-
-    // Add current persistent level
-    if (World) {
-      TSharedPtr<FJsonObject> CurrentLevel = MakeShared<FJsonObject>();
-      CurrentLevel->SetStringField(TEXT("name"), World->GetMapName());
-      CurrentLevel->SetStringField(TEXT("path"),
-                                   World->GetOutermost()->GetName());
-      CurrentLevel->SetBoolField(TEXT("isPersistent"), true);
-      CurrentLevel->SetBoolField(TEXT("isLoaded"), true);
-      CurrentLevel->SetBoolField(TEXT("isVisible"), true);
-      LevelsArray.Add(MakeShared<FJsonValueObject>(CurrentLevel));
-
-      // Add streaming levels
-      for (const ULevelStreaming *StreamingLevel :
-           World->GetStreamingLevels()) {
-        if (!StreamingLevel)
-          continue;
-
-        TSharedPtr<FJsonObject> LevelEntry = MakeShared<FJsonObject>();
-        LevelEntry->SetStringField(TEXT("name"),
-                                   StreamingLevel->GetWorldAssetPackageName());
-        LevelEntry->SetStringField(
-            TEXT("path"),
-            StreamingLevel->GetWorldAssetPackageFName().ToString());
-        LevelEntry->SetBoolField(TEXT("isPersistent"), false);
-        LevelEntry->SetBoolField(TEXT("isLoaded"),
-                                 StreamingLevel->IsLevelLoaded());
-        LevelEntry->SetBoolField(TEXT("isVisible"),
-                                 StreamingLevel->IsLevelVisible());
-        LevelEntry->SetStringField(
-            TEXT("streamingState"),
-            StreamingLevel->IsStreamingStatePending() ? TEXT("Pending")
-            : StreamingLevel->IsLevelLoaded()         ? TEXT("Loaded")
-                                                      : TEXT("Unloaded"));
-        LevelsArray.Add(MakeShared<FJsonValueObject>(LevelEntry));
-      }
-    }
-
-    // Also query Asset Registry for all map assets
-    IAssetRegistry &AssetRegistry =
-        FModuleManager::LoadModuleChecked<FAssetRegistryModule>("AssetRegistry")
-            .Get();
-    TArray<FAssetData> MapAssets;
-    AssetRegistry.GetAssetsByClass(
-        FTopLevelAssetPath(TEXT("/Script/Engine"), TEXT("World")), MapAssets,
-        false);
-
-    TArray<TSharedPtr<FJsonValue>> AllMapsArray;
-    for (const FAssetData &MapAsset : MapAssets) {
-      TSharedPtr<FJsonObject> MapEntry = MakeShared<FJsonObject>();
-      MapEntry->SetStringField(TEXT("name"), MapAsset.AssetName.ToString());
-      MapEntry->SetStringField(TEXT("path"), MapAsset.PackageName.ToString());
-      MapEntry->SetStringField(TEXT("objectPath"),
-                               MapAsset.GetObjectPathString());
-      AllMapsArray.Add(MakeShared<FJsonValueObject>(MapEntry));
-    }
-
-    Resp->SetArrayField(TEXT("currentWorldLevels"), LevelsArray);
-    Resp->SetNumberField(TEXT("currentWorldLevelCount"), LevelsArray.Num());
-    Resp->SetArrayField(TEXT("allMaps"), AllMapsArray);
-    Resp->SetNumberField(TEXT("allMapsCount"), AllMapsArray.Num());
-
-    if (World) {
-      Resp->SetStringField(TEXT("currentMap"), World->GetMapName());
-      Resp->SetStringField(TEXT("currentMapPath"),
-                           World->GetOutermost()->GetName());
-    }
-
-    SendAutomationResponse(RequestingSocket, RequestId, true,
-                           TEXT("Levels listed"), Resp, FString());
-    return true;
-  }
-  if (EffectiveAction == TEXT("export_level")) {
-    FString LevelPath;
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("levelPath"), LevelPath);
-    FString ExportPath;
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("exportPath"), ExportPath);
-    if (ExportPath.IsEmpty())
-      if (Payload.IsValid())
-        Payload->TryGetStringField(TEXT("destinationPath"), ExportPath);
-
-    if (ExportPath.IsEmpty()) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("exportPath required"), nullptr,
-                             TEXT("INVALID_ARGUMENT"));
-      return true;
-    }
-
-    if (!GEditor) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("Editor not available"), nullptr,
-                             TEXT("EDITOR_NOT_AVAILABLE"));
-      return true;
-    }
-
-    UWorld *WorldToExport = nullptr;
-    if (!LevelPath.IsEmpty()) {
-      // If levelPath provided, we should probably load it first? Or export from
-      // asset. Exporting unloaded level asset usually involves loading it. For
-      // now, if levelPath is current, use current. If not, error (or attempt
-      // load).
-      UWorld *Current = GEditor->GetEditorWorldContext().World();
-      if (Current && (Current->GetOutermost()->GetName() == LevelPath ||
-                      Current->GetPathName() == LevelPath)) {
-        WorldToExport = Current;
-      } else {
-        // Should we load?
-        // SendAutomationError(RequestingSocket, RequestId, TEXT("Level must be
-        // loaded to export"), TEXT("LEVEL_NOT_LOADED")); return true; For
-        // robustness, let's assume export current if path matches or empty.
-      }
-    }
-    if (!WorldToExport)
-      WorldToExport = GEditor->GetEditorWorldContext().World();
-
-    if (!WorldToExport) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("No world loaded"), nullptr,
-                             TEXT("NO_WORLD"));
-      return true;
-    }
-
-    // Ensure directory
-    IFileManager::Get().MakeDirectory(*FPaths::GetPath(ExportPath), true);
-
-    // FEditorFileUtils::ExportMap(WorldToExport, ExportPath); // Legacy/Removed
-    // Use SaveMap for .umap or FEditorFileUtils::SaveLevel
-    FEditorFileUtils::SaveMap(WorldToExport, ExportPath);
-
-    SendAutomationResponse(RequestingSocket, RequestId, true,
-                           TEXT("Level exported"), nullptr);
-    return true;
-  }
-  if (EffectiveAction == TEXT("import_level")) {
-    FString DestinationPath;
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("destinationPath"), DestinationPath);
-    FString SourcePath;
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("sourcePath"), SourcePath);
-    if (SourcePath.IsEmpty())
-      if (Payload.IsValid())
-        Payload->TryGetStringField(TEXT("packagePath"), SourcePath); // Mapping
-
-    if (SourcePath.IsEmpty()) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("sourcePath/packagePath required"), nullptr,
-                             TEXT("INVALID_ARGUMENT"));
-      return true;
-    }
-
-    // If SourcePath is a package (starts with /Game), handle as Duplicate/Copy
-    if (SourcePath.StartsWith(TEXT("/"))) {
-      if (DestinationPath.IsEmpty()) {
-        SendAutomationResponse(RequestingSocket, RequestId, false,
-                               TEXT("destinationPath required for asset copy"),
-                               nullptr, TEXT("INVALID_ARGUMENT"));
-        return true;
-      }
-      if (UEditorAssetLibrary::DuplicateAsset(SourcePath, DestinationPath)) {
-        SendAutomationResponse(RequestingSocket, RequestId, true,
-                               TEXT("Level imported (duplicated)"), nullptr);
-      } else {
-        SendAutomationResponse(RequestingSocket, RequestId, false,
-                               TEXT("Failed to duplicate level asset"), nullptr,
-                               TEXT("IMPORT_FAILED"));
-      }
-      return true;
-    }
-
-    // If SourcePath is file, try Import
-    if (!GEditor) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("Editor not available"), nullptr,
-                             TEXT("EDITOR_NOT_AVAILABLE"));
-      return true;
-    }
-
-    FString DestPath = DestinationPath.IsEmpty()
-                           ? TEXT("/Game/Maps")
-                           : FPaths::GetPath(DestinationPath);
-    FString DestName = FPaths::GetBaseFilename(
-        DestinationPath.IsEmpty() ? SourcePath : DestinationPath);
-
-    TArray<FString> Files;
-    Files.Add(SourcePath);
-    // FEditorFileUtils::Import(DestPath, DestName); // Ambiguous/Removed
-    // Use GEditor->ImportMap or handle via AssetTools
-    // Simple fallback:
-    if (GEditor) {
-      // ImportMap is usually for T3D. If SourcePath is .umap, we should
-      // Copy/Load. Assuming T3D import or similar:
-      // GEditor->ImportMap(*DestPath, *DestName, *SourcePath);
-      // ImportMap is deprecated/removed. For .umap files, manual import or Copy
-      // is preferred.
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("Direct map file import not supported. Use "
-                                  "import_level with a package path to copy."),
-                             nullptr, TEXT("NOT_IMPLEMENTED"));
-      return true;
-    }
-    // Automation of Import is tricky without a factory wrapper.
-    // Use AssetTools Import.
-
-    SendAutomationResponse(
-        RequestingSocket, RequestId, false,
-        TEXT("File-based level import not fully automatic yet"), nullptr,
-        TEXT("NOT_IMPLEMENTED"));
-    return true;
-  }
-  if (EffectiveAction == TEXT("add_sublevel")) {
-    FString SubLevelPath;
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("subLevelPath"), SubLevelPath);
-    if (SubLevelPath.IsEmpty() && Payload.IsValid())
-      Payload->TryGetStringField(TEXT("levelPath"), SubLevelPath);
-
-    if (SubLevelPath.IsEmpty()) {
-      SendAutomationError(RequestingSocket, RequestId,
-                          TEXT("subLevelPath required"),
-                          TEXT("INVALID_ARGUMENT"));
-      return true;
-    }
-
-    // Robustness: Cleanup before adding
-    if (GEditor) {
-      GEditor->ForceGarbageCollection(true);
-    }
-
-    // Verify file existence (more robust than DoesPackageExist for new files)
-    FString Filename;
-    bool bFileFound = false;
-    if (FPackageName::TryConvertLongPackageNameToFilename(
-            SubLevelPath, Filename, FPackageName::GetMapPackageExtension())) {
-      if (IFileManager::Get().FileExists(*Filename)) {
-        bFileFound = true;
-      }
-    }
-
-    // Fallback: Check without conversion if it's already a file path?
-    if (!bFileFound && IFileManager::Get().FileExists(*SubLevelPath)) {
-      bFileFound = true;
-    }
-
-    if (!bFileFound) {
-      // Try checking DoesPackageExist as last resort
-      if (!FPackageName::DoesPackageExist(SubLevelPath)) {
-        SendAutomationResponse(
-            RequestingSocket, RequestId, false,
-            FString::Printf(TEXT("Level file not found: %s"), *SubLevelPath),
-            nullptr, TEXT("PACKAGE_NOT_FOUND"));
-        return true;
-      }
-    }
-
-    FString StreamingMethod = TEXT("Blueprint");
-    if (Payload.IsValid())
-      Payload->TryGetStringField(TEXT("streamingMethod"), StreamingMethod);
-
-    if (!GEditor) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("Editor unavailable"), nullptr,
-                             TEXT("NO_EDITOR"));
-      return true;
-    }
-
-    UWorld *World = GEditor->GetEditorWorldContext().World();
-    if (!World) {
-      SendAutomationResponse(RequestingSocket, RequestId, false,
-                             TEXT("No world loaded"), nullptr,
-                             TEXT("NO_WORLD"));
-      return true;
-    }
-
-    // Determine streaming class
-    UClass *StreamingClass = ULevelStreamingDynamic::StaticClass();
-    if (StreamingMethod.Equals(TEXT("AlwaysLoaded"), ESearchCase::IgnoreCase)) {
-      StreamingClass = ULevelStreamingAlwaysLoaded::StaticClass();
-    }
-
-    ULevelStreaming *NewLevel = UEditorLevelUtils::AddLevelToWorld(
-        World, *SubLevelPath, StreamingClass);
-    if (NewLevel) {
-      SendAutomationResponse(RequestingSocket, RequestId, true,
-                             TEXT("Sublevel added successfully"), nullptr);
-    } else {
-      // Did we fail because it's already there?
-      SendAutomationResponse(
-          RequestingSocket, RequestId, false,
-          FString::Printf(TEXT("Failed to add sublevel %s (Check logs)"),
-                          *SubLevelPath),
-          nullptr, TEXT("ADD_FAILED"));
-    }
-    return true;
-  }
-
-  return false;
-#else
-  SendAutomationResponse(RequestingSocket, RequestId, false,
-                         TEXT("Level actions require editor build."), nullptr,
-                         TEXT("NOT_IMPLEMENTED"));
-  return true;
-#endif
-}