@mcp-use/cli 3.2.0-canary.7 → 3.2.0-canary.9

package/dist/index.js

@@ -3227,6 +3227,31 @@ function resolveChromePath() {
 }
 
 // src/commands/screenshot.ts
+function parseHeaderArg(raw) {
+  const idx = raw.indexOf(":");
+  if (idx === -1) {
+    throw new Error(
+      `Invalid --header value "${raw}". Expected "Key: Value" (e.g. "Authorization: Bearer xyz").`
+    );
+  }
+  const key = raw.slice(0, idx).trim();
+  const value = raw.slice(idx + 1).trim();
+  if (!key) {
+    throw new Error(`Invalid --header value "${raw}". Header name is empty.`);
+  }
+  return [key, value];
+}
+function parseHeaderArgs(args) {
+  const headers = {};
+  for (const raw of args) {
+    const [key, value] = parseHeaderArg(raw);
+    headers[key] = value;
+  }
+  return headers;
+}
+function collectHeader(value, previous = []) {
+  return previous.concat([value]);
+}
 function detectToolResourceUri(tool) {
   if (!tool) return null;
   const meta = tool._meta;
@@ -3422,7 +3447,7 @@ function parseDimension(raw, name) {
   return n;
 }
 var AD_HOC_SESSION_NAME = "__screenshot_ad_hoc__";
-async function resolveSessionForScreenshot(options) {
+async function resolveSessionForScreenshot(options, headers) {
   if (options.session) {
     const result2 = await getOrRestoreSession(options.session);
     return result2?.session ?? null;
@@ -3431,6 +3456,7 @@ async function resolveSessionForScreenshot(options) {
     const client = new MCPClient2();
     client.addServer(AD_HOC_SESSION_NAME, {
       url: options.mcp,
+      ...headers ? { headers } : {},
       clientInfo: getCliClientInfo()
     });
     try {
@@ -3458,6 +3484,27 @@ async function screenshotCommand(options, argsList) {
       exitCode = 1;
       return;
     }
+    let headers;
+    if (options.header && options.header.length > 0) {
+      if (!options.mcp) {
+        console.error(
+          formatError(
+            "--header is only supported with --mcp <url>. Saved sessions (use --session) carry their own auth from `mcp-use client connect`."
+          )
+        );
+        exitCode = 1;
+        return;
+      }
+      try {
+        headers = parseHeaderArgs(options.header);
+      } catch (err) {
+        console.error(
+          formatError(err instanceof Error ? err.message : String(err))
+        );
+        exitCode = 1;
+        return;
+      }
+    }
     try {
       resolveChromePath();
     } catch (err) {
@@ -3471,7 +3518,7 @@ async function screenshotCommand(options, argsList) {
     const height = parseDimension(options.height, "height");
     const navTimeout = parseInt(options.timeout, 10) || 3e4;
     const delayMs = options.delay ? parseInt(options.delay, 10) : 0;
-    const session = await resolveSessionForScreenshot(options);
+    const session = await resolveSessionForScreenshot(options, headers);
     if (!session) {
       exitCode = 1;
       return;
@@ -3569,7 +3616,12 @@ function createScreenshotCommand() {
     "Saved session name (from `mcp-use client connect`). Defaults to the active session."
   ).option(
     "--mcp <url>",
-    "Ad-hoc MCP server URL (escape hatch). Used only when no --session and no active saved session. No authentication."
+    "Ad-hoc MCP server URL (escape hatch). Used only when no --session and no active saved session. No authentication unless --header is supplied."
+  ).option(
+    "-H, --header <header>",
+    'HTTP header to send to the --mcp <url> server, formatted "Key: Value". Repeatable. Use to pass an Authorization bearer token or other auth headers when screenshotting an authenticated MCP server.',
+    collectHeader,
+    []
   ).option(
     "--theme <light|dark>",
     "Color scheme to render the view in.",