Merge pull request #834 from Lemoncode/refactor/vscode-extension-mcp-restructure

Refactor/vscode-extension-mcp-restructure

Author: nasdan

packages/vscode-extension/src/commands/index.ts

@@ -0,0 +1,2 @@
+export * from './new-wireframe';
+export * from './register';

packages/vscode-extension/src/commands/new-wireframe.ts

@@ -0,0 +1,11 @@
+import * as vscode from 'vscode';
+
+export const registerNewWireframeCommand = (
+  context: vscode.ExtensionContext
+): void => {
+  context.subscriptions.push(
+    vscode.commands.registerCommand('quickmock.newWireframe', () => {
+      vscode.window.showInformationMessage('New wireframe coming soon'); // TODO: Implement the actual functionality for creating a new wireframe
+    })
+  );
+};

packages/vscode-extension/src/commands/register.ts

@@ -0,0 +1,10 @@
+import * as vscode from 'vscode';
+import { registerNewWireframeCommand } from './new-wireframe';
+
+/**
+ * Registers all VS Code commands exposed by the extension.
+ * @param context The VS Code extension context.
+ */
+export const registerCommands = (context: vscode.ExtensionContext): void => {
+  registerNewWireframeCommand(context);
+};

packages/vscode-extension/src/core/index.ts

@@ -2,3 +2,4 @@ export * from './config';
 export * from './document-registry';
 export * from './logger';
 export * from './paths';
+export * from './workspace';

packages/vscode-extension/src/core/workspace.ts

@@ -0,0 +1,5 @@
+import * as vscode from 'vscode';
+
+export const getPrimaryWorkspaceFolder = ():
+  | vscode.WorkspaceFolder
+  | undefined => vscode.workspace.workspaceFolders?.[0];

packages/vscode-extension/src/index.ts

@@ -1,34 +1,15 @@
-import { logError, onAppUrlChange, syncAppUrlFile } from '#core';
+import { registerCommands } from '#commands';
+import { onAppUrlChange, syncAppUrlFile } from '#core';
 import { QuickMockEditorProvider } from '#editor';
-import {
-  registerMcpServer,
-  registerQuickMockMcpServerProvider,
-  RegistryServer,
-} from '#mcp';
+import { setupMcp } from '#mcp';
 import * as vscode from 'vscode';
 
 export const activate = (context: vscode.ExtensionContext) => {
   syncAppUrlFile();
   context.subscriptions.push(onAppUrlChange(syncAppUrlFile));
-
   context.subscriptions.push(QuickMockEditorProvider.register(context));
-
-  const registryServer = new RegistryServer();
-  registryServer
-    .start(context)
-    .catch(err => logError('Failed to start MCP registry server:', err));
-
-  context.subscriptions.push(registerQuickMockMcpServerProvider(context));
-
-  registerMcpServer(context).catch(err =>
-    logError('Failed to register MCP server:', err)
-  );
-
-  context.subscriptions.push(
-    vscode.commands.registerCommand('quickmock.newWireframe', () => {
-      vscode.window.showInformationMessage('New wireframe coming soon');
-    })
-  );
+  setupMcp(context);
+  registerCommands(context);
 };
 
 export const deactivate = () => {};

packages/vscode-extension/src/mcp/document-bridge/constants.ts

@@ -0,0 +1,2 @@
+export const TOKEN_BYTE_LENGTH = 32;
+export const PORT_FILE_MODE = 0o600;

packages/vscode-extension/src/mcp/document-bridge/index.ts

@@ -0,0 +1 @@
+export * from './server';

packages/vscode-extension/src/mcp/document-bridge/server.ts

@@ -0,0 +1,94 @@
+import { documentRegistry, getPrimaryWorkspaceFolder } from '#core';
+import {
+  buildPortFilePath,
+  DOCUMENT_ROUTE,
+  encodePortFile,
+  LOOPBACK_HOST,
+  TOKEN_HEADER,
+} from '@lemoncode/quickmock-registry-protocol';
+import { randomBytes } from 'node:crypto';
+import { unlinkSync, writeFileSync } from 'node:fs';
+import {
+  createServer,
+  type IncomingMessage,
+  type ServerResponse,
+} from 'node:http';
+import * as vscode from 'vscode';
+import { PORT_FILE_MODE, TOKEN_BYTE_LENGTH } from './constants';
+
+/**
+ * Starts the MCP document bridge server, which serves the content of documents open in the editor to the MCP server.
+ * @param context The VS Code extension context.
+ * @returns A promise that resolves when the server has started.
+ */
+export const startDocumentBridge = async (
+  context: vscode.ExtensionContext
+): Promise<void> => {
+  const workspaceRoot = getPrimaryWorkspaceFolder()?.uri.fsPath;
+  if (!workspaceRoot) return;
+
+  const portFile = buildPortFilePath(workspaceRoot);
+  const token = randomBytes(TOKEN_BYTE_LENGTH).toString('hex');
+
+  const handleRequest = (req: IncomingMessage, res: ServerResponse): void => {
+    if (req.headers[TOKEN_HEADER] !== token) {
+      res.writeHead(401);
+      res.end();
+      return;
+    }
+
+    const url = new URL(req.url ?? '/', 'http://localhost');
+
+    if (url.pathname !== DOCUMENT_ROUTE) {
+      res.writeHead(404);
+      res.end();
+      return;
+    }
+
+    const path = url.searchParams.get('path');
+    if (!path) {
+      res.writeHead(400);
+      res.end('Missing path parameter');
+      return;
+    }
+
+    const content = documentRegistry.get(path);
+    if (content === undefined) {
+      res.writeHead(404);
+      res.end('Document not open in editor');
+      return;
+    }
+
+    res.writeHead(200, { 'Content-Type': 'text/plain; charset=utf-8' });
+    res.end(content);
+  };
+
+  const server = createServer(handleRequest);
+
+  await new Promise<void>((resolve, reject) => {
+    server.on('error', reject);
+    // Get the assigned port and write it to the port file
+    server.listen(0, LOOPBACK_HOST, () => {
+      const { port } = server.address() as { port: number };
+      try {
+        writeFileSync(portFile, encodePortFile(port, token), {
+          mode: PORT_FILE_MODE,
+        });
+      } catch (err) {
+        reject(err);
+        return;
+      }
+      resolve();
+    });
+  });
+
+  context.subscriptions.push({
+    dispose: () => {
+      server.closeAllConnections();
+      server.close();
+      try {
+        unlinkSync(portFile);
+      } catch {}
+    },
+  });
+};

packages/vscode-extension/src/mcp/external-clients/clients/claude-code.ts

@@ -0,0 +1,8 @@
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import type { ExternalMcpClient } from '../model';
+
+export const claudeCode: ExternalMcpClient = {
+  label: 'Claude Code',
+  getConfigPath: () => join(homedir(), '.claude.json'),
+};