@blade-hq/agent-kit 1.0.29 → 1.0.31

package/dist/{chunk-YXHPJGXR.js → chunk-XUWAM6MF.js}

@@ -1,16 +1,17 @@
 import {
   getSessionFilePath,
   resolveSessionFilePreviewTarget
-} from "./chunk-PYYDXV2A.js";
+} from "./chunk-O32JOWYM.js";
 import {
   CardJSON,
   cardRegistry
 } from "./chunk-2UP7MG3J.js";
 import {
+  apiFetch,
   formatToolName,
   getAuthedUrl,
   useUiStore
-} from "./chunk-UWNQ4PXK.js";
+} from "./chunk-3PX2W7II.js";
 import {
   cn,
   copyToClipboard
@@ -376,6 +377,7 @@ function CardCodeBlock({ className, children, node, ...props }) {
 
 // src/react/components/markdown/MarkdownContent.tsx
 import { mermaid } from "@streamdown/mermaid";
+import { useQuery } from "@tanstack/react-query";
 import { Check as Check2, Copy as Copy2, Download as Download2, X as X2 } from "lucide-react";
 import {
   useEffect as useEffect3,
@@ -387,11 +389,11 @@ import { createPortal as createPortal2 } from "react-dom";
 import { Streamdown, defaultRehypePlugins } from "streamdown";
 
 // src/react/components/markdown/ExternalLinkDialog.tsx
-import { Check, Copy, ExternalLink, X } from "lucide-react";
+import { Check, Copy, ExternalLink, Monitor, X } from "lucide-react";
 import { useEffect as useEffect2, useState as useState2 } from "react";
 import { createPortal } from "react-dom";
 import { jsx as jsx4, jsxs as jsxs3 } from "react/jsx-runtime";
-function ExternalLinkDialog({ isOpen, onClose, onConfirm, url }) {
+function ExternalLinkDialog({ isOpen, onClose, onConfirm, url, onOpenInternal }) {
   const [copied, setCopied] = useState2(false);
   useEffect2(() => {
     if (!isOpen) return;
@@ -471,6 +473,21 @@ function ExternalLinkDialog({ isOpen, onClose, onConfirm, url }) {
                     ]
                   }
                 ),
+                onOpenInternal && /* @__PURE__ */ jsxs3(
+                  "button",
+                  {
+                    type: "button",
+                    onClick: () => {
+                      void onOpenInternal();
+                      onClose();
+                    },
+                    className: "inline-flex items-center gap-1.5 rounded-md border border-[hsl(var(--border))] bg-[hsl(var(--background))] px-3 py-1.5 text-sm font-medium text-[hsl(var(--foreground))] transition-colors hover:bg-[hsl(var(--accent))]",
+                    children: [
+                      /* @__PURE__ */ jsx4(Monitor, { size: 14 }),
+                      /* @__PURE__ */ jsx4("span", { children: "\u5728\u6D4F\u89C8\u5668\u4E2D\u6253\u5F00" })
+                    ]
+                  }
+                ),
                 /* @__PURE__ */ jsxs3(
                   "button",
                   {
@@ -647,33 +664,80 @@ var LINK_SAFETY_CONFIG = {
   enabled: true,
   renderModal: (props) => /* @__PURE__ */ jsx6(ExternalLinkDialog, { ...props })
 };
-var hardenFn = Array.isArray(defaultRehypePlugins.harden) ? defaultRehypePlugins.harden[0] : defaultRehypePlugins.harden;
-var REHYPE_PLUGINS = Object.values({
-  ...defaultRehypePlugins,
-  harden: [
-    hardenFn,
-    {
-      allowedImagePrefixes: ["*"],
-      allowedLinkPrefixes: ["*"],
-      allowedProtocols: ["*"],
-      defaultOrigin: "http://localhost",
-      allowDataImages: true
-    }
-  ]
-});
+async function openInSandboxBrowser(sessionId, url) {
+  try {
+    await apiFetch(`/api/sessions/${encodeURIComponent(sessionId)}/browser/tabs`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ url })
+    });
+    const uiStore = useUiStore.getState();
+    uiStore.clearRightPanelTabUnread("browser");
+    uiStore.setActiveRightTab("browser");
+  } catch (error) {
+    console.warn("open in sandbox browser failed", error);
+  }
+}
 var STREAMDOWN_PLUGINS = { mermaid };
 var SYSTEM_REMINDER_TAG = "system-reminder";
 var TOOL_CALL_TAG = "blade-malformed-tool-call";
 var FILE_CARD_TAG = "blade-file-card";
 var AGENT_WORKSPACE_DIR_NAME = "\u667A\u80FD\u52A9\u624B\u5DE5\u4F5C\u7A7A\u95F4";
-var HIDDEN_SYSTEM_REMINDER_TAGS = {
-  [SYSTEM_REMINDER_TAG]: []
+var CUSTOM_ALLOWED_TAGS = {
+  [SYSTEM_REMINDER_TAG]: [],
+  [TOOL_CALL_TAG]: ["data-key", "data-raw"],
+  [FILE_CARD_TAG]: ["data-rel-path", "data-name"]
 };
 var HIDDEN_SYSTEM_REMINDER_COMPONENTS = {
   [SYSTEM_REMINDER_TAG]: () => null
 };
 var TOOL_CALL_BLOCK_RE = /<tool_call>\s*[\s\S]*?\s*<\/tool_call>/gi;
 var malformedToolCallContent = /* @__PURE__ */ new Map();
+var sanitizePlugin = defaultRehypePlugins.sanitize;
+var sanitizeFn = Array.isArray(sanitizePlugin) ? sanitizePlugin[0] : sanitizePlugin;
+var sanitizeSchema = Array.isArray(sanitizePlugin) ? sanitizePlugin[1] : {};
+var hardenFn = Array.isArray(defaultRehypePlugins.harden) ? defaultRehypePlugins.harden[0] : defaultRehypePlugins.harden;
+function buildSanitizeAttributes(allowedTags) {
+  return Object.fromEntries(
+    Object.entries(allowedTags).map(([tag, attributes]) => [
+      tag,
+      [
+        ...new Set(
+          attributes.flatMap((attribute) => [
+            attribute,
+            attribute.replace(/-([a-z])/g, (_, letter) => letter.toUpperCase())
+          ])
+        )
+      ]
+    ])
+  );
+}
+function buildRehypePlugins(allowedTags) {
+  return Object.values({
+    ...defaultRehypePlugins,
+    sanitize: [
+      sanitizeFn,
+      {
+        ...sanitizeSchema,
+        tagNames: [...sanitizeSchema.tagNames ?? [], ...Object.keys(allowedTags)],
+        attributes: {
+          ...sanitizeSchema.attributes ?? {},
+          ...buildSanitizeAttributes(allowedTags)
+        }
+      }
+    ],
+    harden: [
+      hardenFn,
+      {
+        allowedImagePrefixes: ["*"],
+        allowedLinkPrefixes: ["*"],
+        allowedProtocols: ["*"],
+        defaultOrigin: "http://localhost",
+        allowDataImages: true
+      }
+    ]
+  });
+}
 function InlineToolCallBlock({
   children,
   node,
@@ -788,8 +852,11 @@ function CodeBlockPre({ children, node: _node, ...props }) {
   const preRef = useRef(null);
   const [copied, setCopied] = useState4(false);
   const [hasCodeNode, setHasCodeNode] = useState4(false);
+  const [language, setLanguage] = useState4("");
   useEffect3(() => {
-    setHasCodeNode(!!preRef.current?.querySelector("code"));
+    const codeEl = preRef.current?.querySelector("code");
+    setHasCodeNode(!!codeEl);
+    setLanguage(codeEl?.className.match(/language-(\S+)/)?.[1] ?? "");
   }, []);
   const getRawCode = () => preRef.current?.querySelector("code")?.textContent ?? "";
   const handleCopy = async () => {
@@ -811,37 +878,55 @@ function CodeBlockPre({ children, node: _node, ...props }) {
     a.click();
     URL.revokeObjectURL(url);
   };
-  return /* @__PURE__ */ jsxs5("div", { className: "relative group", children: [
-    /* @__PURE__ */ jsx6("pre", { ref: preRef, ...props, children }),
-    hasCodeNode && /* @__PURE__ */ jsxs5("div", { className: "absolute top-2 right-2 flex gap-1 opacity-0 group-hover:opacity-100 transition-opacity", children: [
-      /* @__PURE__ */ jsxs5(
-        "button",
-        {
-          type: "button",
-          onClick: handleCopy,
-          className: cn(
-            "flex items-center gap-1 rounded-md px-1.5 py-0.5 text-[11px] transition-colors",
-            copied ? "text-[hsl(var(--primary))]" : "text-[hsl(var(--muted-foreground))] hover:text-[hsl(var(--foreground))] hover:bg-[hsl(var(--accent))]"
-          ),
-          children: [
-            copied ? /* @__PURE__ */ jsx6(Check2, { size: 12 }) : /* @__PURE__ */ jsx6(Copy2, { size: 12 }),
-            /* @__PURE__ */ jsx6("span", { children: copied ? "\u5DF2\u590D\u5236" : "\u590D\u5236" })
-          ]
-        }
-      ),
-      /* @__PURE__ */ jsxs5(
-        "button",
-        {
-          type: "button",
-          onClick: handleDownload,
-          className: "flex items-center gap-1 rounded-md px-1.5 py-0.5 text-[11px] transition-colors text-[hsl(var(--muted-foreground))] hover:text-[hsl(var(--foreground))] hover:bg-[hsl(var(--accent))]",
-          children: [
-            /* @__PURE__ */ jsx6(Download2, { size: 12 }),
-            /* @__PURE__ */ jsx6("span", { children: "\u4E0B\u8F7D" })
-          ]
-        }
-      )
-    ] })
+  if (!hasCodeNode) {
+    return /* @__PURE__ */ jsx6("div", { className: "relative", children: /* @__PURE__ */ jsx6("pre", { ref: preRef, ...props, children }) });
+  }
+  return /* @__PURE__ */ jsxs5("div", { className: "not-prose my-3 overflow-hidden rounded-xl border border-[hsl(var(--border))]", children: [
+    /* @__PURE__ */ jsxs5("div", { className: "flex h-[34px] items-center justify-between border-b border-[hsl(var(--border))] bg-[hsl(var(--muted))/0.5] pl-3.5 pr-1.5", children: [
+      /* @__PURE__ */ jsx6("span", { className: "font-mono text-[12px] text-[hsl(var(--muted-foreground))]", children: language || "code" }),
+      /* @__PURE__ */ jsxs5("div", { className: "flex items-center gap-0.5", children: [
+        /* @__PURE__ */ jsxs5(
+          "button",
+          {
+            type: "button",
+            onClick: handleCopy,
+            className: cn(
+              "flex h-6 items-center gap-1 rounded-md px-1.5 text-[11px] transition-colors",
+              copied ? "text-[hsl(var(--primary))]" : "text-[hsl(var(--muted-foreground))] hover:bg-[hsl(var(--accent))] hover:text-[hsl(var(--foreground))]"
+            ),
+            children: [
+              copied ? /* @__PURE__ */ jsx6(Check2, { size: 12 }) : /* @__PURE__ */ jsx6(Copy2, { size: 12 }),
+              /* @__PURE__ */ jsx6("span", { children: copied ? "\u5DF2\u590D\u5236" : "\u590D\u5236" })
+            ]
+          }
+        ),
+        /* @__PURE__ */ jsxs5(
+          "button",
+          {
+            type: "button",
+            onClick: handleDownload,
+            className: "flex h-6 items-center gap-1 rounded-md px-1.5 text-[11px] text-[hsl(var(--muted-foreground))] transition-colors hover:bg-[hsl(var(--accent))] hover:text-[hsl(var(--foreground))]",
+            children: [
+              /* @__PURE__ */ jsx6(Download2, { size: 12 }),
+              /* @__PURE__ */ jsx6("span", { children: "\u4E0B\u8F7D" })
+            ]
+          }
+        )
+      ] })
+    ] }),
+    /* @__PURE__ */ jsx6(
+      "pre",
+      {
+        ref: preRef,
+        ...props,
+        className: cn(
+          props.className,
+          // 原型：13.5px/1.7 等宽、浅底；覆盖 prose 的继承字号与主题里手写的 pre 样式。
+          "m-0! overflow-x-auto! rounded-none! border-0! bg-[hsl(var(--muted))/0.25]! px-4! py-3.5! font-mono text-[13.5px]! leading-[1.7]! [&_code]:bg-transparent! [&_code]:p-0! [&_code]:font-mono [&_code]:text-[13.5px]! [&_code]:text-[hsl(var(--foreground))]"
+        ),
+        children
+      }
+    )
   ] });
 }
 function isExternalImageSrc(src) {
@@ -906,7 +991,7 @@ function collapseMediaTags(children, sessionId) {
       return match;
     }
     const name = basename2(relPath);
-    return `<${FILE_CARD_TAG} data-rel-path="${encodeURIComponent(relPath)}" data-name="${escapeAttribute(name)}">${encodeURIComponent(relPath)}</${FILE_CARD_TAG}>`;
+    return `<${FILE_CARD_TAG} data-rel-path="${encodeURIComponent(relPath)}" data-name="${escapeAttribute(name)}">${escapeAttribute(name)}</${FILE_CARD_TAG}>`;
   });
 }
 function collapseInlineToolCalls(children) {
@@ -955,9 +1040,7 @@ function MarkdownContent({
   const resolvedAllowedTags = useMemo2(
     () => ({
       ...allowedTags ?? {},
-      ...HIDDEN_SYSTEM_REMINDER_TAGS,
-      [TOOL_CALL_TAG]: ["data-key", "data-raw"],
-      [FILE_CARD_TAG]: ["data-rel-path", "data-name"]
+      ...CUSTOM_ALLOWED_TAGS
     }),
     [allowedTags]
   );
@@ -972,6 +1055,10 @@ function MarkdownContent({
     }),
     [components]
   );
+  const resolvedRehypePlugins = useMemo2(
+    () => buildRehypePlugins(resolvedAllowedTags),
+    [resolvedAllowedTags]
+  );
   const resolvedPlugins = useMemo2(
     () => ({
       ...STREAMDOWN_PLUGINS,
@@ -979,14 +1066,41 @@ function MarkdownContent({
     }),
     [plugins]
   );
+  const { data: hasBrowserMirror = false } = useQuery({
+    queryKey: ["browser-mirror-ready", sessionId],
+    queryFn: async () => {
+      const resp = await apiFetch(`/api/sessions/${sessionId}/browser/health`, {
+        method: "GET"
+      }).then(
+        () => true,
+        () => false
+      );
+      return resp;
+    },
+    enabled: Boolean(sessionId),
+    staleTime: 6e4
+  });
+  const linkSafety = useMemo2(
+    () => sessionId && hasBrowserMirror ? {
+      enabled: true,
+      renderModal: (modalProps) => /* @__PURE__ */ jsx6(
+        ExternalLinkDialog,
+        {
+          ...modalProps,
+          onOpenInternal: () => openInSandboxBrowser(sessionId, modalProps.url)
+        }
+      )
+    } : LINK_SAFETY_CONFIG,
+    [sessionId, hasBrowserMirror]
+  );
   return /* @__PURE__ */ jsx6(
     Streamdown,
     {
       ...props,
       className: cn("break-words [&_button[data-streamdown=link]]:max-w-full", props.className),
-      linkSafety: LINK_SAFETY_CONFIG,
+      linkSafety,
       mode: mode ?? "static",
-      rehypePlugins: REHYPE_PLUGINS,
+      rehypePlugins: resolvedRehypePlugins,
       allowedTags: resolvedAllowedTags,
       components: resolvedComponents,
       plugins: resolvedPlugins,
@@ -1951,4 +2065,4 @@ export {
   PlanSummaryCard,
   extractLatestPlanMessages
 };
-//# sourceMappingURL=chunk-YXHPJGXR.js.map
+//# sourceMappingURL=chunk-XUWAM6MF.js.map