@akiojin/unity-editor-mcp/src/handlers/script/BuildCodeIndexToolHandler.js

MCP server for Unity Editor integration - enables AI assistants to control Unity Editor

import { BaseToolHandler } from '../base/BaseToolHandler.js';
import { RoslynCliUtils } from '../roslyn/RoslynCliUtils.js';
import { CodeIndex } from '../../core/codeIndex.js';

export class BuildCodeIndexToolHandler extends BaseToolHandler {
  constructor(unityConnection) {
    super(
      'build_code_index',
      'Build a persistent SQLite symbol index using roslyn-cli scan-symbols. Stores DB under Library/UnityMCP/CodeIndex/code-index.db.',
      {
        type: 'object',
        properties: {},
        required: []
      }
    );
    this.unityConnection = unityConnection;
    this.roslyn = new RoslynCliUtils(unityConnection);
    this.index = new CodeIndex(unityConnection);
  }

  async execute() {
    try {
      const args = ['scan-symbols'];
      args.push(...(await this.roslyn.getSolutionOrProjectArgs()));
      const res = await this.roslyn.runCli(args);
      const symbols = res.results || [];
      const r = await this.index.clearAndLoad(symbols);
      const stats = await this.index.getStats();
      return { success: true, inserted: r.total, total: stats.total, lastIndexedAt: stats.lastIndexedAt };
    } catch (e) {
      return {
        success: false,
        error: 'build_index_failed',
        message: e.message,
        hint: 'roslyn-cli may be missing. Run scripts/install-roslyn-cli.sh (or .ps1) to provision .tools/roslyn-cli/<rid>/roslyn-cli.'
      };
    }
  }
}