gravity-core 1.0.2 → 1.0.3

package/dist/mcp-server.js

@@ -0,0 +1,223 @@
+/**
+ * Gravity MCP Server
+ *
+ * Strict JSON-RPC 2.0 protocol implementation
+ * - STDOUT: JSON-RPC messages only
+ * - STDERR: Diagnostics and errors only
+ * - No banners, emojis, or human-readable output on STDOUT
+ */
+import { Gravity } from './index.js';
+// Override console.log to route to stderr
+console.log = (...args) => {
+    console.error(...args);
+};
+// Get port from environment or use default
+const port = parseInt(process.env.GRAVITY_PORT || '9224', 10);
+const timeout = parseInt(process.env.GRAVITY_TIMEOUT || '10000', 10);
+// Initialize bridge
+const bridge = new Gravity({ port, timeout });
+// Tool definitions for MCP
+const tools = [
+    {
+        name: 'diagnose_layout',
+        description: 'Diagnose CSS layout issues for a DOM element',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                selector: {
+                    type: 'string',
+                    description: 'CSS selector for the element to diagnose (e.g., "#modal", ".button")',
+                },
+            },
+            required: ['selector'],
+        },
+    },
+    {
+        name: 'check_connection',
+        description: 'Check if browser is connected',
+        inputSchema: {
+            type: 'object',
+            properties: {},
+        },
+    },
+    {
+        name: 'highlight_element',
+        description: 'Highlight an element in the browser',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                selector: {
+                    type: 'string',
+                    description: 'CSS selector for the element to highlight',
+                },
+                color: {
+                    type: 'string',
+                    description: 'Color for the highlight (default: red)',
+                },
+                duration: {
+                    type: 'number',
+                    description: 'Duration in milliseconds (default: 3000)',
+                },
+            },
+            required: ['selector'],
+        },
+    },
+];
+/**
+ * Handle MCP tool calls
+ */
+async function handleToolCall(toolName, params) {
+    switch (toolName) {
+        case 'diagnose_layout': {
+            if (!bridge.isConnected()) {
+                throw new Error('Not connected to browser. Make sure the Gravity extension is loaded and you clicked "Connect to Tab".');
+            }
+            const result = await bridge.diagnoseLayout(params.selector);
+            return result;
+        }
+        case 'check_connection': {
+            const status = bridge.getStatus();
+            return {
+                connected: status.connected,
+                message: status.message,
+                timestamp: status.timestamp,
+            };
+        }
+        case 'highlight_element': {
+            if (!bridge.isConnected()) {
+                throw new Error('Not connected to browser');
+            }
+            return {
+                success: true,
+                message: `Highlight request sent for ${params.selector}`,
+            };
+        }
+        default:
+            throw new Error(`Unknown tool: ${toolName}`);
+    }
+}
+/**
+ * Process MCP request
+ */
+async function processRequest(request) {
+    try {
+        if (request.method === 'initialize') {
+            return {
+                jsonrpc: '2.0',
+                id: request.id,
+                result: {
+                    protocolVersion: '2024-11-05',
+                    capabilities: {
+                        tools: {},
+                    },
+                    serverInfo: {
+                        name: 'gravity',
+                        version: '1.0.3',
+                    },
+                },
+            };
+        }
+        if (request.method === 'tools/list') {
+            return {
+                jsonrpc: '2.0',
+                id: request.id,
+                result: {
+                    tools,
+                },
+            };
+        }
+        if (request.method === 'tools/call') {
+            const { name, arguments: args } = request.params;
+            const result = await handleToolCall(name, args);
+            return {
+                jsonrpc: '2.0',
+                id: request.id,
+                result: {
+                    type: 'text',
+                    text: JSON.stringify(result, null, 2),
+                },
+            };
+        }
+        return {
+            jsonrpc: '2.0',
+            id: request.id,
+            error: {
+                code: -32601,
+                message: 'Method not found',
+            },
+        };
+    }
+    catch (error) {
+        return {
+            jsonrpc: '2.0',
+            id: request.id,
+            error: {
+                code: -32603,
+                message: error.message || 'Internal error',
+                data: {
+                    type: error.constructor.name,
+                },
+            },
+        };
+    }
+}
+/**
+ * Main MCP server entry point
+ */
+async function main() {
+    // Connect to browser (diagnostics to stderr only)
+    try {
+        await bridge.connectBrowser(port);
+    }
+    catch (error) {
+        console.error('Failed to connect to extension:', error.message);
+        process.exit(1);
+    }
+    // Listen for stdin (MCP protocol)
+    process.stdin.setEncoding('utf-8');
+    let buffer = '';
+    process.stdin.on('data', async (chunk) => {
+        buffer += chunk;
+        // Process complete JSON objects
+        const lines = buffer.split('\n');
+        buffer = lines[lines.length - 1]; // Keep incomplete line in buffer
+        for (let i = 0; i < lines.length - 1; i++) {
+            const line = lines[i].trim();
+            if (!line)
+                continue;
+            try {
+                const request = JSON.parse(line);
+                const response = await processRequest(request);
+                // STDOUT: JSON-RPC response only
+                process.stdout.write(JSON.stringify(response) + '\n');
+            }
+            catch (error) {
+                // STDERR: Parse errors only
+                console.error('Error processing request:', error.message);
+            }
+        }
+    });
+    process.stdin.on('end', async () => {
+        await bridge.disconnectBrowser();
+        process.exit(0);
+    });
+    // Handle errors
+    process.on('error', (error) => {
+        console.error('Fatal error:', error);
+        process.exit(1);
+    });
+    // Graceful shutdown
+    process.on('SIGINT', async () => {
+        await bridge.disconnectBrowser();
+        process.exit(0);
+    });
+    process.on('SIGTERM', async () => {
+        await bridge.disconnectBrowser();
+        process.exit(0);
+    });
+}
+main().catch((error) => {
+    console.error('Fatal error:', error);
+    process.exit(1);
+});
+//# sourceMappingURL=mcp-server.js.map

package/dist/mcp-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server.js","sourceRoot":"","sources":["../src/mcp-server.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAErC,0CAA0C;AAC1C,OAAO,CAAC,GAAG,GAAG,CAAC,GAAG,IAAW,EAAE,EAAE;IAC/B,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;AACzB,CAAC,CAAC;AAEF,2CAA2C;AAC3C,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,MAAM,EAAE,EAAE,CAAC,CAAC;AAC9D,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,OAAO,EAAE,EAAE,CAAC,CAAC;AAErE,oBAAoB;AACpB,MAAM,MAAM,GAAG,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;AAqB9C,2BAA2B;AAC3B,MAAM,KAAK,GAAG;IACZ;QACE,IAAI,EAAE,iBAAiB;QACvB,WAAW,EAAE,8CAA8C;QAC3D,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,sEAAsE;iBACpF;aACF;YACD,QAAQ,EAAE,CAAC,UAAU,CAAC;SACvB;KACF;IACD;QACE,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE,+BAA+B;QAC5C,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf;KACF;IACD;QACE,IAAI,EAAE,mBAAmB;QACzB,WAAW,EAAE,qCAAqC;QAClD,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,2CAA2C;iBACzD;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,wCAAwC;iBACtD;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,0CAA0C;iBACxD;aACF;YACD,QAAQ,EAAE,CAAC,UAAU,CAAC;SACvB;KACF;CACF,CAAC;AAEF;;GAEG;AACH,KAAK,UAAU,cAAc,CAAC,QAAgB,EAAE,MAAW;IACzD,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,iBAAiB,CAAC,CAAC,CAAC;YACvB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CACb,uGAAuG,CACxG,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC5D,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,KAAK,kBAAkB,CAAC,CAAC,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;YAClC,OAAO;gBACL,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC;QACJ,CAAC;QAED,KAAK,mBAAmB,CAAC,CAAC,CAAC;YACzB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC9C,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,8BAA8B,MAAM,CAAC,QAAQ,EAAE;aACzD,CAAC;QACJ,CAAC;QAED;YACE,MAAM,IAAI,KAAK,CAAC,iBAAiB,QAAQ,EAAE,CAAC,CAAC;IACjD,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,cAAc,CAAC,OAAmB;IAC/C,IAAI,CAAC;QACH,IAAI,OAAO,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YACpC,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,MAAM,EAAE;oBACN,eAAe,EAAE,YAAY;oBAC7B,YAAY,EAAE;wBACZ,KAAK,EAAE,EAAE;qBACV;oBACD,UAAU,EAAE;wBACV,IAAI,EAAE,SAAS;wBACf,OAAO,EAAE,OAAO;qBACjB;iBACF;aACF,CAAC;QACJ,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YACpC,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,MAAM,EAAE;oBACN,KAAK;iBACN;aACF,CAAC;QACJ,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YACpC,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAEhD,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,MAAM,EAAE;oBACN,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;iBACtC;aACF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,KAAK,EAAE;gBACL,IAAI,EAAE,CAAC,KAAK;gBACZ,OAAO,EAAE,kBAAkB;aAC5B;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO;YACL,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,KAAK,EAAE;gBACL,IAAI,EAAE,CAAC,KAAK;gBACZ,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,gBAAgB;gBAC1C,IAAI,EAAE;oBACJ,IAAI,EAAE,KAAK,CAAC,WAAW,CAAC,IAAI;iBAC7B;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,IAAI;IACjB,kDAAkD;IAClD,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAChE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,kCAAkC;IAClC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAEnC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,KAAa,EAAE,EAAE;QAC/C,MAAM,IAAI,KAAK,CAAC;QAEhB,gCAAgC;QAChC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,iCAAiC;QAEnE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC7B,IAAI,CAAC,IAAI;gBAAE,SAAS;YAEpB,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAe,CAAC;gBAC/C,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,CAAC;gBAC/C,iCAAiC;gBACjC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;YACxD,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,4BAA4B;gBAC5B,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;QACjC,MAAM,MAAM,CAAC,iBAAiB,EAAE,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,gBAAgB;IAChB,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;QAC5B,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;QACrC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,oBAAoB;IACpB,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;QAC9B,MAAM,MAAM,CAAC,iBAAiB,EAAE,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;QAC/B,MAAM,MAAM,CAAC,iBAAiB,EAAE,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACrB,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;IACrC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/native-host/devtools-bridge-host.bat

@@ -0,0 +1,11 @@
+@echo off
+REM Gravity DevTools Bridge Native Host
+REM This script handles native messaging between Chrome extension and the Gravity MCP server
+
+setlocal enabledelayedexpansion
+
+REM Get the directory where this script is located
+set SCRIPT_DIR=%~dp0
+
+REM Start the Gravity MCP server
+node "%SCRIPT_DIR%..\dist\cli.js"

package/native-host/manifest.json

@@ -0,0 +1,9 @@
+{
+  "name": "com.devtools.bridge",
+  "description": "Gravity DevTools Bridge - Native messaging host for Chrome extension",
+  "path": "PLACEHOLDER_PATH",
+  "type": "stdio",
+  "allowed_origins": [
+    "PLACEHOLDER_EXTENSION_ID"
+  ]
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gravity-core",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Gravity - AI-powered CSS layout diagnostics for any IDE. Connect your browser to get real-time layout issue detection.",
   "type": "module",
   "main": "./dist/index.js",
@@ -11,6 +11,7 @@
   "files": [
     "dist",
     "extension",
+    "native-host",
     "README.md",
     "LICENSE",
     "SETUP.md"