feat: enable mcp to communicate with IDE via JSON-RPC server

.nx/workflows/windows-distribution-config.yaml

@@ -2,3 +2,9 @@ distribute-on:
   small-changeset: 3 windows-medium-js
   medium-changeset: 4 windows-medium-js
   large-changeset: 6 windows-medium-js
+
+assignment-rules:
+  - project: intellij
+    target: build
+    runs-on:
+      - windows-medium-js

apps/nx-mcp-e2e/src/tools.test.ts

@@ -1,11 +1,15 @@
 import {
+  cleanupNxWorkspace,
   createInvokeMCPInspectorCLI,
+  defaultVersion,
   e2eCwd,
+  logWindowsFileLocks,
   newWorkspace,
   simpleReactWorkspaceOptions,
   uniq,
 } from '@nx-console/shared-e2e-utils';
 import { rmSync } from 'node:fs';
+import { platform } from 'node:os';
 import { join } from 'node:path';
 
 describe('tools', () => {
@@ -26,7 +30,13 @@ describe('tools', () => {
     );
   });
 
-  afterAll(() => {
+  afterAll(async () => {
+    // Clean up Nx workspace processes before attempting to remove directory
+    await cleanupNxWorkspace(testWorkspacePath, defaultVersion);
+
+    if (platform() === 'win32') {
+      logWindowsFileLocks(testWorkspacePath);
+    }
     rmSync(testWorkspacePath, { recursive: true, force: true });
   });
 
@@ -39,8 +49,8 @@ describe('tools', () => {
     expect(toolNames).toEqual([
       'nx_docs',
       'nx_available_plugins',
-      'nx_workspace_path',
       'nx_workspace',
+      'nx_workspace_path',
       'nx_project_details',
       'nx_generators',
       'nx_generator_schema',

apps/nx-mcp-e2e/src/workspace_detection.test.ts

@@ -1,11 +1,15 @@
 import {
+  cleanupNxWorkspace,
   createInvokeMCPInspectorCLI,
+  defaultVersion,
   e2eCwd,
+  logWindowsFileLocks,
   newWorkspace,
   simpleReactWorkspaceOptions,
   uniq,
 } from '@nx-console/shared-e2e-utils';
 import { mkdirSync, rmSync, writeFileSync } from 'node:fs';
+import { platform } from 'node:os';
 import { join } from 'node:path';
 
 describe('workspace detection', () => {
@@ -28,7 +32,11 @@ describe('workspace detection', () => {
       );
     });
 
-    afterAll(() => {
+    afterAll(async () => {
+      if (platform() === 'win32') {
+        logWindowsFileLocks(testWorkspacePath);
+      }
+      await cleanupNxWorkspace(testWorkspacePath, defaultVersion);
       rmSync(testWorkspacePath, { recursive: true, force: true });
     });
 

apps/nx-mcp-e2e/src/workspace_path.test.ts

@@ -1,11 +1,15 @@
 import {
+  cleanupNxWorkspace,
   createInvokeMCPInspectorCLI,
+  defaultVersion,
   e2eCwd,
+  logWindowsFileLocks,
   newWorkspace,
   simpleReactWorkspaceOptions,
   uniq,
 } from '@nx-console/shared-e2e-utils';
 import { rmSync } from 'node:fs';
+import { platform } from 'node:os';
 import { join } from 'node:path';
 
 describe('workspace path', () => {
@@ -26,7 +30,13 @@ describe('workspace path', () => {
     );
   });
 
-  afterAll(() => {
+  afterAll(async () => {
+    // Clean up Nx workspace processes before attempting to remove directory
+    await cleanupNxWorkspace(testWorkspacePath, defaultVersion);
+
+    if (platform() === 'win32') {
+      logWindowsFileLocks(testWorkspacePath);
+    }
     rmSync(testWorkspacePath, { recursive: true, force: true });
   });
 

apps/nx-mcp-e2e/tsconfig.json

@@ -3,6 +3,9 @@
   "files": [],
   "include": [],
   "references": [
+    {
+      "path": "../../libs/shared/utils"
+    },
     {
       "path": "../../libs/shared/e2e-utils"
     }

apps/nx-mcp/src/ensureOnlyJsonRpcStdout.ts

@@ -0,0 +1,22 @@
+export function ensureOnlyJsonRpcStdout() {
+  process.stdout.write = ((
+    chunk: any,
+    encodingOrCallback?: any,
+    callback?: any,
+  ) => {
+    const message = chunk.toString();
+
+    if (
+      !message.startsWith('Content-Length:') &&
+      (!message.startsWith('{') || !message.includes('"jsonrpc":'))
+    ) {
+      process.stderr.write(message);
+      return;
+    }
+
+    const originalWrite = process.stdout.constructor.prototype.write.bind(
+      process.stdout,
+    );
+    return originalWrite(chunk, encodingOrCallback, callback);
+  }) as typeof process.stdout.write;
+}

apps/nx-mcp/src/ide-client/create-ide-client.ts

@@ -0,0 +1,32 @@
+import { IIdeJsonRpcClient } from '@nx-console/shared-types';
+import { Logger, testIdeConnection } from '@nx-console/shared-utils';
+import { IdeJsonRpcClient } from './json-rpc-client';
+
+export async function createIdeClient(
+  workspacePath: string,
+  logger?: Logger,
+): Promise<{ client: IIdeJsonRpcClient | undefined; available: boolean }> {
+  try {
+    const ideListening = await testIdeConnection(workspacePath);
+
+    if (!ideListening) {
+      return { client: undefined, available: false };
+    }
+
+    const client = new IdeJsonRpcClient(workspacePath, logger);
+
+    await client.connect();
+
+    // Verify connection is actually working
+    const status = client.getStatus();
+    if (status !== 'connected') {
+      client.disconnect();
+      return { client: undefined, available: false };
+    }
+
+    return { client, available: true };
+  } catch (error) {
+    logger?.log('Failed to create IDE client:', error);
+    return { client: undefined, available: false };
+  }
+}