@mehmoodqureshi/chrome-mcp 0.5.2 → 0.6.0

package/dist/src/mcp/tools.js

@@ -24,6 +24,8 @@ const policy_1 = require("../security/policy");
 const envelopes_1 = require("./envelopes");
 const batch_1 = require("./batch");
 const helpers_1 = require("./helpers");
+const tasks_1 = require("../bridge/tasks");
+const workspace_1 = require("../bridge/workspace");
 const validators_1 = require("./validators");
 const TARGET_PROPS = {
     selector: { type: 'string', description: 'CSS selector (exactly one of selector|ref)' },
@@ -64,6 +66,10 @@ exports.TOOL_DEFINITIONS = [
     { name: 'download_file', description: 'Download a file by URL or from a link element.', inputSchema: obj({ url: { type: 'string' }, ...TARGET_PROPS, suggestedName: { type: 'string' }, tabId: { type: 'string' } }) },
     { name: 'upload_file', description: 'Set local file(s) on a file <input> (target by selector or ref) — uploads without the OS dialog. Requires --enable-uploads. `files` are absolute local paths.', inputSchema: obj({ ...TARGET_PROPS, files: { type: 'array', items: { type: 'string' } }, tabId: { type: 'string' } }, ['files']) },
     { name: 'chrome_status', description: 'Report backend/session status.', inputSchema: obj({}) },
+    { name: 'profile_use', description: 'Switch the active browser profile (identity). Subsequent downloads, results, screenshots, and the action log are stored under profiles/<name>/. Resets the active task to "default" unless you then call task_new.', inputSchema: obj({ name: { type: 'string', description: 'Profile name (becomes a folder; sanitized to a safe path segment).' } }, ['name']) },
+    { name: 'task_new', description: 'Start a new task (run) under the active profile. Creates profiles/<profile>/tasks/<name>/ with downloads/, results/, screenshots/ and makes it the active task so all captured artifacts land there.', inputSchema: obj({ name: { type: 'string', description: 'Task name (becomes a folder; sanitized to a safe path segment).' } }, ['name']) },
+    { name: 'tasks_list', description: 'List every task across all profiles under the data dir, with sizes and download counts.', inputSchema: obj({}) },
+    { name: 'task_status', description: 'Report the active profile/task and the folder paths where this run\'s artifacts are stored.', inputSchema: obj({}) },
     {
         name: 'batch',
         description: 'Run multiple tool calls in one request — parallel (default) or serial. Each op is { tool, args } and goes through the same policy gate, rate limit, and error handling as a direct call. In parallel mode, tab-scoped ops MUST pass an explicit tabId (the active-tab default is unsafe under concurrency). Use to drive several tabs at once (e.g. open tabs, then batch get_text across them). Cannot be nested.',
@@ -99,7 +105,24 @@ const waitUntil = (args) => (0, validators_1.optionalString)(args, 'waitUntil');
 /** Tools that don't act on a single tab (so `tabId` is irrelevant) — exempt from
  *  the parallel-batch explicit-tabId requirement. Everything else falls back to
  *  the active tab when `tabId` is omitted, which races under concurrency. */
-const PARALLEL_TAB_EXEMPT = new Set(['tabs_list', 'tab_new', 'chrome_status', 'batch']);
+const PARALLEL_TAB_EXEMPT = new Set([
+    'tabs_list', 'tab_new', 'chrome_status', 'batch',
+    'profile_use', 'task_new', 'tasks_list', 'task_status',
+]);
+/** Server-side tools that manage the task workspace and need no browser backend. */
+const NO_BACKEND_TOOLS = new Set(['profile_use', 'task_new', 'tasks_list', 'task_status']);
+/** Project a Workspace to the path fields worth returning to the caller. */
+function workspaceView(w) {
+    return {
+        profile: w.profile,
+        task: w.task,
+        taskDir: w.taskDir,
+        downloadDir: w.downloadDir,
+        resultsDir: w.resultsDir,
+        screenshotsDir: w.screenshotsDir,
+        historyPath: w.historyPath,
+    };
+}
 /** A known tool that operates on a specific tab — needs an explicit tabId in a parallel batch. */
 function requiresExplicitTab(tool) {
     return tool in exports.TOOL_HANDLERS && !PARALLEL_TAB_EXEMPT.has(tool);
@@ -194,12 +217,15 @@ exports.TOOL_HANDLERS = {
             fullPage: (0, validators_1.optionalBoolean)(a, 'fullPage'),
             target: (0, validators_1.optionalTarget)(a),
         });
+        (0, workspace_1.saveScreenshot)(shot.dataBase64);
         const caption = shot.truncated ? `(truncated; full height ${shot.fullHeight}px)` : undefined;
         return (0, envelopes_1.imageResult)(shot.dataBase64, shot.mimeType, caption);
     },
     get_text: async (a, ctx) => {
         await gate(ctx, 'get_text');
-        return (0, envelopes_1.jsonResult)(await ctx.ex.getText((0, validators_1.optionalTarget)(a), { tabId: tabId(a) }));
+        const res = await ctx.ex.getText((0, validators_1.optionalTarget)(a), { tabId: tabId(a) });
+        (0, workspace_1.saveResult)('get_text', 'json', JSON.stringify(res, null, 2));
+        return (0, envelopes_1.jsonResult)(res);
     },
     get_html: async (a, ctx) => {
         await gate(ctx, 'get_html');
@@ -251,15 +277,19 @@ exports.TOOL_HANDLERS = {
     },
     extract_links: async (a, ctx) => {
         await gate(ctx, 'get_text'); // read of page content
-        return (0, envelopes_1.jsonResult)(await (0, helpers_1.extractLinks)(ctx.ex, {
+        const res = await (0, helpers_1.extractLinks)(ctx.ex, {
             selector: (0, validators_1.optionalString)(a, 'selector'),
             sameOriginOnly: (0, validators_1.optionalBoolean)(a, 'sameOriginOnly'),
             tabId: tabId(a),
-        }));
+        });
+        (0, workspace_1.saveResult)('extract_links', 'json', JSON.stringify(res, null, 2));
+        return (0, envelopes_1.jsonResult)(res);
     },
     read_as_markdown: async (a, ctx) => {
         await gate(ctx, 'get_text');
-        return (0, envelopes_1.textResult)(await (0, helpers_1.readAsMarkdown)(ctx.ex, { selector: (0, validators_1.optionalString)(a, 'selector'), tabId: tabId(a) }));
+        const md = await (0, helpers_1.readAsMarkdown)(ctx.ex, { selector: (0, validators_1.optionalString)(a, 'selector'), tabId: tabId(a) });
+        (0, workspace_1.saveResult)('read_as_markdown', 'md', md);
+        return (0, envelopes_1.textResult)(md);
     },
     fill_form: async (a, ctx) => {
         await gate(ctx, 'type'); // mutating
@@ -308,6 +338,11 @@ exports.TOOL_HANDLERS = {
         return (0, envelopes_1.jsonResult)(await ctx.ex.uploadFile(t, files, { tabId: tabId(a) }));
     },
     chrome_status: async (_a, ctx) => (0, envelopes_1.jsonResult)(ctx.ex.status()),
+    // --- task workspace management (server-side; no browser needed) ---
+    profile_use: async (a) => (0, envelopes_1.jsonResult)(workspaceView((0, workspace_1.switchWorkspace)({ profile: (0, validators_1.requireString)(a, 'name') }))),
+    task_new: async (a) => (0, envelopes_1.jsonResult)(workspaceView((0, workspace_1.switchWorkspace)({ task: (0, validators_1.requireString)(a, 'name') }))),
+    task_status: async () => (0, envelopes_1.jsonResult)(workspaceView((0, workspace_1.getActiveWorkspace)())),
+    tasks_list: async () => (0, envelopes_1.jsonResult)((0, tasks_1.listTasks)((0, workspace_1.getActiveWorkspace)().dataDir)),
     // Fan-out: each sub-op is routed back through `dispatchToolCall`, so it gets
     // the same policy gate, rate limit, and never-throw handling as a direct call.
     batch: async (a) => (0, batch_1.runBatch)(a, { dispatch: dispatchToolCall, requiresExplicitTab }),
@@ -351,6 +386,24 @@ function allowCall(now) {
     rateWindow.push(now);
     return true;
 }
+/** A compact, length-bounded summary of a call's args for the history log. */
+function summarizeArgs(rawArgs) {
+    if (typeof rawArgs !== 'object' || rawArgs === null)
+        return undefined;
+    const a = rawArgs;
+    const out = {};
+    for (const k of ['url', 'selector', 'ref', 'name', 'text', 'key', 'op', 'tabId']) {
+        const v = a[k];
+        if (v === undefined)
+            continue;
+        out[k] = typeof v === 'string' && v.length > 120 ? `${v.slice(0, 120)}…` : v;
+    }
+    return Object.keys(out).length ? out : undefined;
+}
+/** Append one action record to the active task's history.jsonl (best-effort). */
+function recordHistory(tool, rawArgs, ok, error) {
+    (0, workspace_1.appendHistory)({ ts: new Date().toISOString(), tool, args: summarizeArgs(rawArgs), ok, ...(error ? { error } : {}) });
+}
 async function dispatchToolCall(name, rawArgs) {
     const handler = exports.TOOL_HANDLERS[name];
     if (!handler)
@@ -359,11 +412,17 @@ async function dispatchToolCall(name, rawArgs) {
         return (0, envelopes_1.errorResult)('rate limit exceeded; slow down');
     try {
         const mgr = (0, manager_1.getManager)();
-        const ex = await mgr.ensureReady();
-        return await handler((0, validators_1.asArgs)(rawArgs), { ex, policy: mgr.policy });
+        // Workspace-management tools run server-side and must work even with no
+        // browser paired, so they skip the executor readiness check.
+        const ex = NO_BACKEND_TOOLS.has(name) ? null : await mgr.ensureReady();
+        const result = await handler((0, validators_1.asArgs)(rawArgs), { ex, policy: mgr.policy });
+        recordHistory(name, rawArgs, !result.isError);
+        return result;
     }
     catch (err) {
-        return (0, envelopes_1.errorResult)(errMessage(err));
+        const message = errMessage(err);
+        recordHistory(name, rawArgs, false, message);
+        return (0, envelopes_1.errorResult)(message);
     }
 }
 /** Assert the catalog and the dispatch table describe the same tool set. */

package/extension-dist/background.js

@@ -44,7 +44,7 @@
     isConnected() {
       return this.state === "connected" && this.ws?.readyState === WebSocket.OPEN;
     }
-    connect(port, token) {
+    connect(port, token, profile) {
       if (this.ws && (this.ws.readyState === WebSocket.OPEN || this.ws.readyState === WebSocket.CONNECTING)) {
         return;
       }
@@ -62,7 +62,8 @@
           type: "hello",
           v: PROTOCOL_VERSION,
           token,
-          ext: { id: chrome.runtime.id, version: chrome.runtime.getManifest().version, chrome: chromeVersion() }
+          ext: { id: chrome.runtime.id, version: chrome.runtime.getManifest().version, chrome: chromeVersion() },
+          profile: profile && profile.trim() ? profile.trim() : void 0
         };
         ws2.send(JSON.stringify(hello));
       };
@@ -424,6 +425,36 @@
       this.code = code;
     }
   };
+  var DOWNLOAD_TIMEOUT_MS = 12e4;
+  function waitForDownloadComplete(id) {
+    return new Promise((resolve, reject) => {
+      let settled = false;
+      const settle = (fn) => {
+        if (settled) return;
+        settled = true;
+        chrome.downloads.onChanged.removeListener(onChanged);
+        clearTimeout(timer);
+        fn();
+      };
+      const timer = setTimeout(
+        () => settle(() => reject(new CmdError("DOWNLOAD_FAILED", "download timed out"))),
+        DOWNLOAD_TIMEOUT_MS
+      );
+      const check = async () => {
+        const [item] = await chrome.downloads.search({ id });
+        if (!item) return;
+        if (item.state === "complete") settle(() => resolve(item));
+        else if (item.state === "interrupted") {
+          settle(() => reject(new CmdError("DOWNLOAD_FAILED", `download interrupted: ${item.error ?? "unknown"}`)));
+        }
+      };
+      const onChanged = (delta) => {
+        if (delta.id === id) void check();
+      };
+      chrome.downloads.onChanged.addListener(onChanged);
+      void check();
+    });
+  }
   var SESSION = crypto.randomUUID();
   var CONTENT_SCHEME = /^(https?|file):/i;
   var delay = (ms) => new Promise((r) => setTimeout(r, ms));
@@ -1023,7 +1054,9 @@
           );
           return { matched: matched === true, waitedMs: Date.now() - start };
         }
-        // -- download (user's Downloads dir) --
+        // -- download (saved to the user's Downloads dir; the server then moves it
+        //    into the active task's downloads/). We wait for completion and report
+        //    the absolute on-disk path so the server can relocate it. --
         case "download_file": {
           const url = typeof cmd.params.url === "string" ? cmd.params.url : void 0;
           if (!url) throw new CmdError("DOWNLOAD_FAILED", "the extension download path requires a url");
@@ -1031,7 +1064,16 @@
             typeof cmd.params.suggestedName === "string" ? cmd.params.suggestedName : void 0
           );
           const downloadId = await chrome.downloads.download({ url, filename: name });
-          return { path: `(downloads)/${name}`, backend: "extension", bytes: 0, suggestedName: name };
+          const item = await waitForDownloadComplete(downloadId);
+          const bytes = item.fileSize > 0 ? item.fileSize : item.bytesReceived;
+          return {
+            path: item.filename,
+            sourcePath: item.filename,
+            downloadId,
+            backend: "extension",
+            bytes,
+            suggestedName: name
+          };
         }
         // -- upload: set local file(s) on a file <input> via CDP DOM.setFileInputFiles --
         case "upload_file": {
@@ -1117,9 +1159,10 @@
     log: (m) => console.debug("[chrome-mcp]", m)
   });
   async function getConfig() {
-    const { wsPort, token } = await chrome.storage.local.get(["wsPort", "token"]);
+    const { wsPort, token, profile } = await chrome.storage.local.get(["wsPort", "token", "profile"]);
     if (typeof wsPort === "number" && wsPort > 0 && typeof token === "string" && token.length > 0) {
-      return { wsPort, token };
+      const name = typeof profile === "string" && profile.trim() ? profile.trim() : "default";
+      return { wsPort, token, profile: name };
     }
     return null;
   }
@@ -1146,7 +1189,7 @@
     if (ws.isConnected() || ws.state === "unauthorized") return;
     const cfg = await getConfig();
     if (!cfg) return;
-    ws.connect(cfg.wsPort, cfg.token);
+    ws.connect(cfg.wsPort, cfg.token, cfg.profile);
   }
   async function keepalivePulse() {
     await chrome.storage.local.get("connState");
@@ -1158,8 +1201,9 @@
     if (alarm.name === KEEPALIVE_ALARM) void keepalivePulse();
   });
   chrome.storage.onChanged.addListener((changes, area) => {
-    if (area === "local" && (changes.wsPort || changes.token)) {
+    if (area === "local" && (changes.wsPort || changes.token || changes.profile)) {
       if (ws.state === "unauthorized") ws.state = "idle";
+      if (changes.profile) ws.close();
       void ensureConnected();
     }
   });

package/extension-dist/options.html

@@ -25,6 +25,13 @@
     <input id="port" type="number" inputmode="numeric" placeholder="38017" />
     <label for="token">Token</label>
     <input id="token" type="password" placeholder="paste the handshake token" />
+    <label for="profile">Profile</label>
+    <input id="profile" type="text" placeholder="default" />
+    <p class="hint">
+      A name for this browser (e.g. <code>mehmood</code>). Several browsers can pair to the
+      same port &amp; token at once, each with its own profile — switch between them with
+      <code>profile_use</code>. Leave blank for <code>default</code>.
+    </p>
     <button id="save">Save & connect</button>
     <div class="status" id="status">Status: unknown</div>
     <script src="options.js"></script>

package/extension-dist/options.js

@@ -6,11 +6,13 @@
   // extension/src/options/options.ts
   var portEl = document.getElementById("port");
   var tokenEl = document.getElementById("token");
+  var profileEl = document.getElementById("profile");
   var saveEl = document.getElementById("save");
   var statusEl = document.getElementById("status");
   async function loadExisting() {
-    const { wsPort, connState } = await chrome.storage.local.get(["wsPort", "connState"]);
+    const { wsPort, profile, connState } = await chrome.storage.local.get(["wsPort", "profile", "connState"]);
     portEl.value = typeof wsPort === "number" && wsPort > 0 ? String(wsPort) : String(DEFAULT_WS_PORT);
+    profileEl.value = typeof profile === "string" ? profile : "";
     render(typeof connState === "string" ? connState : "idle");
   }
   function render(state) {
@@ -25,11 +27,12 @@
   saveEl.addEventListener("click", async () => {
     const wsPort = Number(portEl.value);
     const token = tokenEl.value.trim();
+    const profile = profileEl.value.trim() || "default";
     if (!Number.isInteger(wsPort) || wsPort <= 0 || !token) {
       statusEl.textContent = "Status: enter a valid port (> 0) and token";
       return;
     }
-    await chrome.storage.local.set({ wsPort, token });
+    await chrome.storage.local.set({ wsPort, token, profile });
     await chrome.runtime.sendMessage({ type: "reconnect" }).catch(() => void 0);
     statusEl.textContent = "Status: \u2026 connecting";
   });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mehmoodqureshi/chrome-mcp",
-  "version": "0.5.2",
+  "version": "0.6.0",
   "description": "Drive a real Chrome browser over MCP. A stdio MCP server (CLI) plus an MV3 extension, behind one pluggable Executor (extension via chrome.scripting, or a Playwright CDP fallback).",
   "author": "Mehmood Ur Rehman Qureshi",
   "license": "MIT",