unreal-engine-mcp-server 0.5.4 → 0.5.6

package/src/resources/assets.ts

@@ -1,286 +0,0 @@
-import { BaseTool } from '../tools/base-tool.js';
-import { IAssetResources } from '../types/tool-interfaces.js';
-import { coerceString } from '../utils/result-helpers.js';
-import { AutomationResponse } from '../types/automation-responses.js';
-
-export class AssetResources extends BaseTool implements IAssetResources {
-  // Simple in-memory cache for asset listing
-  private cache = new Map<string, { timestamp: number; data: any }>();
-  private get ttlMs(): number { return Number(process.env.ASSET_LIST_TTL_MS || 10000); }
-  private makeKey(dir: string, recursive: boolean, page?: number) {
-    return page !== undefined ? `${dir}::${recursive ? 1 : 0}::${page}` : `${dir}::${recursive ? 1 : 0}`;
-  }
-
-  // Normalize UE content paths:
-  // - Map '/Content' -> '/Game'
-  // - Ensure forward slashes
-  private normalizeDir(dir: string): string {
-    try {
-      if (!dir || typeof dir !== 'string') return '/Game';
-      let d = dir.replace(/\\/g, '/');
-      if (!d.startsWith('/')) d = '/' + d;
-      if (d.toLowerCase().startsWith('/content')) {
-        d = '/Game' + d.substring('/Content'.length);
-      }
-      // Collapse multiple slashes
-      d = d.replace(/\/+/g, '/');
-      // Remove trailing slash except root
-      if (d.length > 1) d = d.replace(/\/$/, '');
-      return d;
-    } catch {
-      return '/Game';
-    }
-  }
-
-  clearCache(dir?: string) {
-    if (!dir) {
-      this.cache.clear();
-      return;
-    }
-
-    const normalized = this.normalizeDir(dir);
-    for (const key of Array.from(this.cache.keys())) {
-      if (key.startsWith(`${normalized}::`)) {
-        this.cache.delete(key);
-      }
-    }
-  }
-
-  invalidateAssetPaths(paths: string[]) {
-    if (!Array.isArray(paths) || paths.length === 0) {
-      return;
-    }
-
-    const dirs = new Set<string>();
-    for (const rawPath of paths) {
-      if (typeof rawPath !== 'string' || rawPath.trim().length === 0) {
-        continue;
-      }
-      const normalized = this.normalizeDir(rawPath);
-      dirs.add(normalized);
-      const parent = this.parentDirectory(normalized);
-      if (parent) {
-        dirs.add(parent);
-      }
-    }
-
-    for (const dir of dirs) {
-      this.clearCache(dir);
-    }
-  }
-
-  async list(dir = '/Game', _recursive = false, limit = 50) {
-    // ALWAYS use non-recursive listing to show only immediate children
-    // This prevents timeouts and makes navigation clearer
-    _recursive = false; // Force non-recursive
-
-    // Normalize directory first
-    dir = this.normalizeDir(dir);
-
-    // Cache fast-path
-    try {
-      const key = this.makeKey(dir, false);
-      const entry = this.cache.get(key);
-      const now = Date.now();
-      if (entry && (now - entry.timestamp) < this.ttlMs) {
-        return { success: true, ...entry.data };
-      }
-    } catch { }
-
-    // Check if bridge is connected
-    if (!this.bridge.isConnected) {
-      return {
-        success: false,
-        assets: [],
-        warning: 'Unreal Engine is not connected. Please ensure Unreal Engine is running with the MCP server enabled.',
-        connectionStatus: 'disconnected'
-      };
-    }
-
-    // Always use directory-only listing (immediate children)
-    const listed = await this.listDirectoryOnly(dir, false, limit);
-    // Ensure a success flag is present so downstream evaluators don't assume success implicitly
-    return { ...listed, success: listed && (listed as any).success === false ? false : true };
-  }
-
-  /**
-   * List assets with pagination support
-   * @param dir Directory to list assets from
-   * @param page Page number (0-based)
-   * @param pageSize Number of assets per page (max 50 to avoid socket failures)
-   */
-  async listPaged(dir = '/Game', page = 0, pageSize = 30, recursive = false) {
-    // Ensure pageSize doesn't exceed safe limit
-    const safePageSize = Math.min(pageSize, 50);
-    const offset = page * safePageSize;
-
-    // Normalize directory and check cache for this specific page
-    dir = this.normalizeDir(dir);
-    const cacheKey = this.makeKey(dir, recursive, page);
-    const cached = this.cache.get(cacheKey);
-    if (cached && (Date.now() - cached.timestamp) < this.ttlMs) {
-      return cached.data;
-    }
-
-    if (!this.bridge.isConnected) {
-      return {
-        assets: [],
-        page,
-        pageSize: safePageSize,
-        warning: 'Unreal Engine is not connected.',
-        connectionStatus: 'disconnected'
-      };
-    }
-
-    try {
-      // Use search API with pagination
-      // Use the same directory listing approach but with pagination
-      const allAssets = await this.listDirectoryOnly(dir, false, 1000);
-
-      // Paginate the results
-      const start = offset;
-      const end = offset + safePageSize;
-      const pagedAssets = allAssets.assets ? allAssets.assets.slice(start, end) : [];
-
-      const result = {
-        assets: pagedAssets,
-        page,
-        pageSize: safePageSize,
-        count: pagedAssets.length,
-        totalCount: allAssets.assets ? allAssets.assets.length : 0,
-        hasMore: end < (allAssets.assets ? allAssets.assets.length : 0),
-        method: 'directory_listing_paged'
-      };
-
-      this.cache.set(cacheKey, { timestamp: Date.now(), data: result });
-      return result;
-    } catch (err: any) {
-      console.warn(`Asset listing page ${page} failed:`, err.message);
-    }
-
-    return {
-      assets: [],
-      page,
-      pageSize: safePageSize,
-      error: 'Failed to fetch page'
-    };
-  }
-
-  /**
-   * Directory-based listing of immediate children using AssetRegistry.
-   * Returns both subfolders and assets at the given path.
-   */
-  private async listDirectoryOnly(dir: string, _recursive: boolean, limit: number) {
-    // Always return only immediate children to avoid timeout and improve navigation
-    try {
-      // Use the native C++ plugin's list action instead of Python
-      try {
-        const normalizedDir = this.normalizeDir(dir);
-        const response = await this.sendAutomationRequest<AutomationResponse>(
-          'list',
-          { directory: normalizedDir, limit, recursive: false },
-          { timeoutMs: 30000 }
-        );
-
-        if (response.success !== false && response.result) {
-          const payload = response.result as any;
-
-          const foldersArr = Array.isArray(payload.folders_list)
-            ? payload.folders_list.map((f: any) => ({
-              Name: coerceString(f?.n ?? f?.Name ?? f?.name) ?? '',
-              Path: coerceString(f?.p ?? f?.Path ?? f?.path) ?? '',
-              Class: 'Folder',
-              isFolder: true
-            }))
-            : [];
-
-          const assetsArr = Array.isArray(payload.assets)
-            ? payload.assets.map((a: any) => ({
-              Name: coerceString(a?.n ?? a?.Name ?? a?.name) ?? '',
-              Path: coerceString(a?.p ?? a?.Path ?? a?.path) ?? '',
-              Class: coerceString(a?.c ?? a?.Class ?? a?.class) ?? 'Object'
-            }))
-            : [];
-
-          const result = {
-            success: true,
-            assets: [...foldersArr, ...assetsArr],
-            count: foldersArr.length + assetsArr.length,
-            folders: foldersArr.length,
-            files: assetsArr.length,
-            path: normalizedDir,
-            recursive: false,
-            method: 'automation_bridge',
-            cached: false
-          };
-
-          const key = this.makeKey(dir, false);
-          this.cache.set(key, { timestamp: Date.now(), data: result });
-          return result;
-        }
-      } catch { }
-
-      // No fallback available
-    } catch (err: any) {
-      const errorMessage = err?.message ? String(err.message) : 'Asset registry request failed';
-      console.warn('Engine asset listing failed:', errorMessage);
-      return {
-        success: false,
-        path: this.normalizeDir(dir),
-        summary: { total: 0, folders: 0, assets: 0 },
-        foldersList: [],
-        assets: [],
-        error: errorMessage,
-        warning: 'AssetRegistry query failed. Ensure the MCP Automation Bridge is connected.',
-        transport: 'automation_bridge',
-        method: 'asset_registry_alternate'
-      };
-    }
-
-    return {
-      success: false,
-      path: this.normalizeDir(dir),
-      summary: { total: 0, folders: 0, assets: 0 },
-      foldersList: [],
-      assets: [],
-      error: 'Asset registry returned no payload.',
-      warning: 'No items returned from AssetRegistry request.',
-      transport: 'automation_bridge',
-      method: 'asset_registry_empty'
-    };
-  }
-
-  async find(assetPath: string) {
-    // Guard against invalid paths (trailing slash, empty, whitespace)
-    if (!assetPath || typeof assetPath !== 'string' || assetPath.trim() === '' || assetPath.endsWith('/')) {
-      return false;
-    }
-
-    try {
-      const normalizedPath = this.normalizeDir(assetPath);
-      const response = await this.sendAutomationRequest<AutomationResponse>(
-        'asset_exists',
-        { asset_path: normalizedPath }
-      );
-
-      return response?.success !== false && (response?.result as any)?.exists === true;
-    } catch {
-      return false;
-    }
-  }
-
-  private parentDirectory(path: string): string | null {
-    if (!path || typeof path !== 'string') {
-      return null;
-    }
-
-    const normalized = this.normalizeDir(path);
-    const lastSlash = normalized.lastIndexOf('/');
-    if (lastSlash <= 0) {
-      return normalized === '/' ? '/' : null;
-    }
-
-    const parent = normalized.substring(0, lastSlash);
-    return parent.length > 0 ? parent : '/';
-  }
-}

package/src/resources/levels.ts

@@ -1,68 +0,0 @@
-import { UnrealBridge } from '../unreal-bridge.js';
-import { coerceString } from '../utils/result-helpers.js';
-
-export class LevelResources {
-  private automationBridge: any;
-
-  constructor(bridge: UnrealBridge) {
-    this.automationBridge = (bridge as any).automationBridge;
-  }
-
-  async getCurrentLevel() {
-    try {
-      if (!this.automationBridge || typeof this.automationBridge.sendAutomationRequest !== 'function') {
-        return { success: false, error: 'Automation bridge is not available' };
-      }
-
-      const resp: any = await this.automationBridge.sendAutomationRequest('get_level', { action: 'get_current' });
-      if (resp && resp.success !== false && resp.result) {
-        return {
-          success: true,
-          name: coerceString(resp.result.name) ?? 'None',
-          path: coerceString(resp.result.path) ?? 'None'
-        };
-      }
-
-      return { success: false, error: 'Failed to get current level' };
-    } catch (err) {
-      return { error: `Failed to get current level: ${err}`, success: false };
-    }
-  }
-
-  async getLevelName() {
-    try {
-      if (!this.automationBridge || typeof this.automationBridge.sendAutomationRequest !== 'function') {
-        return { success: false, error: 'Automation bridge is not available' };
-      }
-
-      const resp: any = await this.automationBridge.sendAutomationRequest('get_level', { action: 'get_name' });
-      if (resp && resp.success !== false && resp.result) {
-        return {
-          success: true,
-          path: coerceString(resp.result.path) ?? ''
-        };
-      }
-
-      return { success: false, error: 'Failed to get level name' };
-    } catch (err) {
-      return { error: `Failed to get level name: ${err}`, success: false };
-    }
-  }
-
-  async saveCurrentLevel() {
-    try {
-      if (!this.automationBridge || typeof this.automationBridge.sendAutomationRequest !== 'function') {
-        return { success: false, error: 'Automation bridge is not available' };
-      }
-
-      const resp: any = await this.automationBridge.sendAutomationRequest('save_level', {});
-      if (resp && resp.success !== false) {
-        return { success: true, message: 'Level saved' };
-      }
-
-      return { success: false, error: 'Failed to save level' };
-    } catch (err) {
-      return { error: `Failed to save level: ${err}`, success: false };
-    }
-  }
-}

package/src/server/resource-registry.ts

@@ -1,47 +0,0 @@
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { ListResourcesRequestSchema } from '@modelcontextprotocol/sdk/types.js';
-import { UnrealBridge } from '../unreal-bridge.js';
-import { AutomationBridge } from '../automation/index.js';
-import { HealthMonitor } from '../services/health-monitor.js';
-import { ResourceHandler } from '../handlers/resource-handlers.js';
-import { AssetResources } from '../resources/assets.js';
-import { ActorResources } from '../resources/actors.js';
-import { LevelResources } from '../resources/levels.js';
-
-export class ResourceRegistry {
-    constructor(
-        private server: Server,
-        private bridge: UnrealBridge,
-        private automationBridge: AutomationBridge,
-        private assetResources: AssetResources,
-        private actorResources: ActorResources,
-        private levelResources: LevelResources,
-        private healthMonitor: HealthMonitor,
-        private ensureConnected: () => Promise<boolean>
-    ) { }
-
-    register() {
-        this.server.setRequestHandler(ListResourcesRequestSchema, async () => {
-            return {
-                resources: [
-                    { uri: 'ue://assets', name: 'Assets', description: 'Project assets', mimeType: 'application/json' },
-                    { uri: 'ue://health', name: 'Health Status', description: 'Server health and performance metrics', mimeType: 'application/json' },
-                    { uri: 'ue://automation-bridge', name: 'Automation Bridge', description: 'Automation bridge diagnostics and recent activity', mimeType: 'application/json' },
-                    { uri: 'ue://version', name: 'Engine Version', description: 'Unreal Engine version and compatibility info', mimeType: 'application/json' }
-                ]
-            };
-        });
-
-        const resourceHandler = new ResourceHandler(
-            this.server,
-            this.bridge,
-            this.automationBridge,
-            this.assetResources,
-            this.actorResources,
-            this.levelResources,
-            this.healthMonitor,
-            this.ensureConnected
-        );
-        resourceHandler.registerHandlers();
-    }
-}