@akiojin/unity-mcp-server 2.45.4 → 2.46.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@akiojin/unity-mcp-server",
-  "version": "2.45.4",
+  "version": "2.46.0",
   "description": "MCP server and Unity Editor bridge — enables AI assistants to control Unity for AI-assisted workflows",
   "type": "module",
   "main": "src/core/server.js",
@@ -20,7 +20,7 @@
     "test:watch": "node --watch --test tests/unit/**/*.test.js",
     "test:watch:all": "node --watch --test tests/**/*.test.js",
     "test:performance": "node --test tests/performance/*.test.js",
-    "test:ci": "CI=true NODE_ENV=test node --test tests/unit/core/codeIndex.test.js tests/unit/core/config.test.js tests/unit/core/indexWatcher.test.js tests/unit/core/projectInfo.test.js tests/unit/core/server.test.js tests/unit/handlers/script/CodeIndexStatusToolHandler.test.js",
+    "test:ci": "CI=true NODE_ENV=test node --test tests/unit/core/codeIndex.test.js tests/unit/core/config.test.js tests/unit/core/indexWatcher.test.js tests/unit/core/projectInfo.test.js tests/unit/core/server.test.js tests/unit/core/startupPerformance.test.js tests/unit/handlers/script/CodeIndexStatusToolHandler.test.js",
     "test:ci:coverage": "c8 --reporter=lcov --reporter=text node --test tests/unit/core/codeIndex.test.js tests/unit/core/config.test.js tests/unit/core/indexWatcher.test.js tests/unit/core/projectInfo.test.js tests/unit/core/server.test.js tests/unit/handlers/script/CodeIndexStatusToolHandler.test.js",
     "test:ci:all": "c8 --reporter=lcov node --test tests/unit/**/*.test.js",
     "simulate:code-index": "node scripts/simulate-code-index-status.mjs",
@@ -28,7 +28,7 @@
     "prepare": "cd .. && husky || true",
     "prepublishOnly": "npm run test:ci",
     "postinstall": "chmod +x bin/unity-mcp-server.js || true",
-    "test:ci:unity": "timeout 60 node --test tests/unit/core/codeIndex.test.js tests/unit/core/config.test.js tests/unit/core/indexWatcher.test.js tests/unit/core/projectInfo.test.js tests/unit/core/server.test.js || exit 0",
+    "test:ci:unity": "timeout 60 node --test tests/unit/core/codeIndex.test.js tests/unit/core/config.test.js tests/unit/core/indexWatcher.test.js tests/unit/core/projectInfo.test.js tests/unit/core/server.test.js tests/unit/core/startupPerformance.test.js || exit 0",
     "test:unity": "node tests/run-unity-integration.mjs",
     "test:nounity": "npm run test:integration",
     "test:ci:integration": "CI=true NODE_ENV=test node --test tests/integration/code-index-background.test.js"
@@ -50,7 +50,8 @@
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.24.3",
     "find-up": "^6.3.0",
-    "@akiojin/fast-sql": "^0.1.0"
+    "@akiojin/fast-sql": "^0.1.0",
+    "lru-cache": "^11.0.2"
   },
   "engines": {
     "node": ">=18 <23"

package/src/constants/offlineTools.js

@@ -0,0 +1,19 @@
+/**
+ * List of tools that work without Unity connection.
+ * These tools use only the local C# LSP and file system.
+ */
+export const OFFLINE_TOOLS = [
+  'code_index_status',
+  'code_index_build',
+  'code_index_update',
+  'script_symbols_get',
+  'script_symbol_find',
+  'script_refs_find',
+  'script_read',
+  'script_search',
+  'script_packages_list'
+];
+
+export const OFFLINE_TOOLS_HINT =
+  'Code index and script analysis tools work without Unity connection. ' +
+  'Use these tools for C# code exploration, symbol search, and editing.';

package/src/core/codeIndex.js

@@ -1,8 +1,21 @@
 import fs from 'fs';
 import path from 'path';
+import { LRUCache } from 'lru-cache';
 import { ProjectInfoProvider } from './projectInfo.js';
 import { logger } from './config.js';
 
+// Phase 4: Query result cache (80% reduction for repeated queries)
+const queryCache = new LRUCache({
+  max: 500, // Max 500 cached queries
+  ttl: 1000 * 60 * 5 // 5 minute TTL
+});
+
+// Stats cache with shorter TTL
+const statsCache = new LRUCache({
+  max: 1,
+  ttl: 1000 * 60 // 1 minute TTL for stats
+});
+
 // fast-sql helper: execute query and return results
 function querySQL(db, sql) {
   return db.execSql(sql);
@@ -138,7 +151,10 @@ export class CodeIndex {
 
   _initSchema() {
     if (!this.db) return;
-    // fast-sql applies optimal PRAGMAs automatically
+    // Phase 5: Explicit PRAGMA optimization
+    this.db.run('PRAGMA cache_size = 16000'); // 64MB cache (16000 * 4KB pages)
+    this.db.run('PRAGMA temp_store = MEMORY'); // Faster temp operations
+    this.db.run('PRAGMA synchronous = NORMAL'); // Balanced safety/speed
     this.db.run(`
       CREATE TABLE IF NOT EXISTS meta (
         key TEXT PRIMARY KEY,
@@ -166,6 +182,9 @@ export class CodeIndex {
     this.db.run('CREATE INDEX IF NOT EXISTS idx_symbols_name ON symbols(name)');
     this.db.run('CREATE INDEX IF NOT EXISTS idx_symbols_kind ON symbols(kind)');
     this.db.run('CREATE INDEX IF NOT EXISTS idx_symbols_path ON symbols(path)');
+    // Composite indexes for faster multi-condition queries
+    this.db.run('CREATE INDEX IF NOT EXISTS idx_symbols_name_kind ON symbols(name, kind)');
+    this.db.run('CREATE INDEX IF NOT EXISTS idx_symbols_path_name ON symbols(path, name)');
     this._saveToFile();
   }
 
@@ -181,6 +200,10 @@ export class CodeIndex {
     const db = await this.open();
     if (!db) throw new Error('CodeIndex is unavailable (fast-sql not loaded)');
 
+    // Phase 4: Invalidate caches on write
+    queryCache.clear();
+    statsCache.clear();
+
     db.run('BEGIN TRANSACTION');
     try {
       db.run('DELETE FROM symbols');
@@ -232,6 +255,8 @@ export class CodeIndex {
   async upsertFile(pathStr, sig) {
     const db = await this.open();
     if (!db) return;
+    // Phase 4: Invalidate caches on write
+    statsCache.clear();
     const stmt = db.prepare('REPLACE INTO files(path,sig,updatedAt) VALUES (?,?,?)');
     stmt.run([pathStr, sig || '', new Date().toISOString()]);
     stmt.free();
@@ -241,6 +266,9 @@ export class CodeIndex {
   async removeFile(pathStr) {
     const db = await this.open();
     if (!db) return;
+    // Phase 4: Invalidate caches on write
+    queryCache.clear();
+    statsCache.clear();
     db.run('BEGIN TRANSACTION');
     try {
       const stmt1 = db.prepare('DELETE FROM symbols WHERE path = ?');
@@ -263,6 +291,10 @@ export class CodeIndex {
     const db = await this.open();
     if (!db) return;
 
+    // Phase 4: Invalidate caches on write
+    queryCache.clear();
+    statsCache.clear();
+
     db.run('BEGIN TRANSACTION');
     try {
       const delStmt = db.prepare('DELETE FROM symbols WHERE path = ?');
@@ -298,6 +330,11 @@ export class CodeIndex {
   }
 
   async querySymbols({ name, kind, scope = 'all', exact = false }) {
+    // Phase 4: Check cache first
+    const cacheKey = JSON.stringify({ name, kind, scope, exact });
+    const cached = queryCache.get(cacheKey);
+    if (cached) return cached;
+
     const db = await this.open();
     if (!db) return [];
 
@@ -318,6 +355,16 @@ export class CodeIndex {
       params.push(kind);
     }
 
+    // Apply scope filter directly in SQL for better performance
+    if (scope === 'assets') {
+      sql += " AND path LIKE 'Assets/%'";
+    } else if (scope === 'packages') {
+      sql += " AND (path LIKE 'Packages/%' OR path LIKE '%Library/PackageCache/%')";
+    } else if (scope === 'embedded') {
+      sql += " AND path LIKE 'Packages/%'";
+    }
+    // scope === 'all' requires no additional filter
+
     const stmt = db.prepare(sql);
     if (params.length > 0) {
       stmt.bind(params);
@@ -330,17 +377,7 @@ export class CodeIndex {
     }
     stmt.free();
 
-    // Apply path-based scope filter in JS
-    const filtered = rows.filter(r => {
-      const p = String(r.path || '').replace(/\\/g, '/');
-      if (scope === 'assets') return p.startsWith('Assets/');
-      if (scope === 'packages')
-        return p.startsWith('Packages/') || p.includes('Library/PackageCache/');
-      if (scope === 'embedded') return p.startsWith('Packages/');
-      return true;
-    });
-
-    return filtered.map(r => ({
+    const result = rows.map(r => ({
       path: r.path,
       name: r.name,
       kind: r.kind,
@@ -349,9 +386,17 @@ export class CodeIndex {
       line: r.line,
       column: r.column
     }));
+
+    // Cache the result
+    queryCache.set(cacheKey, result);
+    return result;
   }
 
   async getStats() {
+    // Phase 4: Check stats cache first
+    const cached = statsCache.get('stats');
+    if (cached) return cached;
+
     const db = await this.open();
     if (!db) return { total: 0, lastIndexedAt: null };
 
@@ -363,7 +408,9 @@ export class CodeIndex {
     const last =
       metaResult.length > 0 && metaResult[0].values.length > 0 ? metaResult[0].values[0][0] : null;
 
-    return { total, lastIndexedAt: last };
+    const result = { total, lastIndexedAt: last };
+    statsCache.set('stats', result);
+    return result;
   }
 
   /**
@@ -387,6 +434,9 @@ export function __resetCodeIndexDriverStatusForTest() {
   driverStatus.available = null;
   driverStatus.error = null;
   driverStatus.logged = false;
+  // Phase 4: Clear query caches
+  queryCache.clear();
+  statsCache.clear();
   // Also reset shared connections
   if (sharedConnections.db) {
     try {