marble-headed-mcp 0.1.38 → 0.1.39

package/dist/index.js

@@ -7,6 +7,11 @@ import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { CallToolRequestSchema, ListToolsRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
 const DEFAULT_BASE_URL = 'http://localhost:4000';
+const ENVIRONMENT_BASE_URLS = {
+    localhost: 'http://localhost:3000',
+    dev: 'https://dev.withmarble.ai',
+    production: 'https://withmarble.ai',
+};
 const WEBAPP_LOG_PATH = '/tmp/webapp';
 const DEFAULT_DEV_CONTAINER_PATH = '/Users/akilanbabu/code/marble-container';
 const VSCODE_REPO_PATH = '/Users/akilanbabu/code/vscode-source-marble/code-server-7/lib/vscode';
@@ -15,6 +20,7 @@ const DEFAULT_CODEX_LOGGER_PATH = path.join(process.env.HOME || '/Users/akilanba
 const CODEX_LOGGER_PATH_ENV = 'CODEX_LOGGER_PATH';
 const CODEX_LOG_PREFIX = 'codex-run-';
 const execFileAsync = promisify(execFile);
+let selectedEnvironment = null;
 function parseCodexJsonOutput(output) {
     const result = {};
     if (!output)
@@ -50,6 +56,15 @@ function parseCodexJsonOutput(output) {
 function normalizeBaseUrl(input) {
     return input.replace(/\/+$/, '');
 }
+function parseEnvironment(input) {
+    if (typeof input !== 'string')
+        return null;
+    const normalized = input.trim().toLowerCase();
+    if (normalized === 'localhost' || normalized === 'dev' || normalized === 'production') {
+        return normalized;
+    }
+    return null;
+}
 function shellQuote(value) {
     return `'${value.replace(/'/g, `'\"'\"'`)}'`;
 }
@@ -111,6 +126,9 @@ async function readCodexThreadIdFromLog(filePath) {
     }
 }
 function resolveRawBaseUrl() {
+    if (selectedEnvironment) {
+        return normalizeBaseUrl(ENVIRONMENT_BASE_URLS[selectedEnvironment]);
+    }
     const base = process.env.HEADED_SERVER_BASE_URL || DEFAULT_BASE_URL;
     return normalizeBaseUrl(base);
 }
@@ -398,6 +416,22 @@ async function saveImageFromUrl({ url, filename }) {
     return { ok: true, path: targetPath, bytes: buffer.length };
 }
 const TOOLS = [
+    {
+        name: 'set_environment',
+        description: 'Set the base URL for HTTP-backed actions. localhost -> http://localhost:3000, dev -> https://dev.withmarble.ai, production -> https://withmarble.ai.',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                environment: {
+                    type: 'string',
+                    enum: ['localhost', 'dev', 'production'],
+                    description: 'Environment to target.',
+                },
+            },
+            required: ['environment'],
+            additionalProperties: false,
+        },
+    },
     {
         name: 'headed_start_session',
         description: 'Start a headed browser session (if email or password are not provided, they default to akilan@withmarble.ai and marbledebug123, respectively). Remember to call `headed_end_session` when finished to avoid leaving sessions open.',
@@ -716,6 +750,38 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
     const { name, arguments: args } = request.params;
     try {
         switch (name) {
+            case 'set_environment': {
+                const environment = parseEnvironment(args?.environment);
+                if (!environment) {
+                    return {
+                        content: [
+                            {
+                                type: 'text',
+                                text: JSON.stringify({
+                                    ok: false,
+                                    error: 'environment must be one of: localhost, dev, production.',
+                                }, null, 2),
+                            },
+                        ],
+                    };
+                }
+                selectedEnvironment = environment;
+                const appBaseUrl = resolveAppBaseUrl();
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: JSON.stringify({
+                                ok: true,
+                                environment,
+                                appBaseUrl,
+                                apiBaseUrl: `${appBaseUrl}/api`,
+                                headedApiBaseUrl: `${appBaseUrl}/api/headed`,
+                            }, null, 2),
+                        },
+                    ],
+                };
+            }
             case 'headed_start_session': {
                 const result = await postHeadedJson('/start_session', args);
                 return { content: [{ type: 'text', text: JSON.stringify(result.json || { status: result.status, body: result.text }, null, 2) }] };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "marble-headed-mcp",
-  "version": "0.1.38",
+  "version": "0.1.39",
   "description": "MCP server for Marble headed automation endpoints",
   "main": "dist/index.js",
   "type": "module",