@zhin.js/client 1.1.4 → 1.1.6

package/client/console-utils/remoteApi.test.ts

@@ -0,0 +1,91 @@
+import { afterEach, describe, expect, it, vi } from "vitest";
+
+import { apiFetch, consoleRpc, getToken } from "./remoteApi.js";
+
+function mockStorage(map: Record<string, string> = {}): Storage {
+  return {
+    getItem: (key: string) => map[key] ?? null,
+    setItem: (key: string, value: string) => {
+      map[key] = value;
+    },
+    removeItem: (key: string) => {
+      delete map[key];
+    },
+    clear: () => {
+      for (const key of Object.keys(map)) delete map[key];
+    },
+    key: () => null,
+    length: 0,
+  };
+}
+
+function installBrowserGlobals(runtimeToken?: string) {
+  const win = {
+    location: { origin: "http://localhost:5173", protocol: "http:", host: "localhost:5173" },
+    __ZHIN_API_TOKEN: runtimeToken as string | undefined,
+    dispatchEvent: vi.fn(),
+  };
+  vi.stubGlobal("window", win);
+  vi.stubGlobal("localStorage", mockStorage());
+  vi.stubGlobal("sessionStorage", mockStorage());
+  return win;
+}
+
+function fakeResponse(status: number): Response {
+  return { ok: status >= 200 && status < 300, status } as Response;
+}
+
+describe("apiFetch 401 handling", () => {
+  afterEach(() => {
+    vi.unstubAllGlobals();
+    vi.restoreAllMocks();
+  });
+
+  it("invalidates an expired runtime token on 401 so it cannot 401-loop forever", async () => {
+    // 回归：runtime token（Demo 预置）优先级最高，而旧实现 401 只清 local/sessionStorage，
+    // 过期 demo token 会让后续每个请求都 401。
+    const win = installBrowserGlobals("expired-demo-token");
+    vi.stubGlobal("fetch", vi.fn(async () => fakeResponse(401)));
+
+    expect(getToken()).toBe("expired-demo-token");
+    await apiFetch("/api/system/status");
+
+    expect(win.__ZHIN_API_TOKEN).toBeUndefined();
+    expect(getToken()).toBeNull();
+    expect(win.dispatchEvent).toHaveBeenCalledTimes(1);
+  });
+
+  it("keeps the runtime token when the response is not 401", async () => {
+    const win = installBrowserGlobals("valid-demo-token");
+    vi.stubGlobal("fetch", vi.fn(async () => fakeResponse(200)));
+
+    await apiFetch("/api/system/status");
+
+    expect(win.__ZHIN_API_TOKEN).toBe("valid-demo-token");
+    expect(getToken()).toBe("valid-demo-token");
+    expect(win.dispatchEvent).not.toHaveBeenCalled();
+  });
+
+  it("sends canonical Console RPC envelopes and returns typed data", async () => {
+    installBrowserGlobals("sponsor-token");
+    const fetchMock = vi.fn(async (_url: unknown, init?: RequestInit) => ({
+      ok: true,
+      status: 200,
+      json: async () => ({ success: true, data: { ready: true } }),
+      request: init,
+    }));
+    vi.stubGlobal("fetch", fetchMock);
+
+    await expect(consoleRpc<{ ready: boolean }>("workroom.profile.status", {
+      projectId: "alpha",
+    })).resolves.toEqual({ ready: true });
+    expect(fetchMock).toHaveBeenCalledWith(
+      "http://localhost:5173/api/console/request",
+      expect.objectContaining({ method: "POST" }),
+    );
+    const init = fetchMock.mock.calls[0]![1] as RequestInit;
+    expect(JSON.parse(String(init.body))).toMatchObject({
+      type: "workroom.profile.status", projectId: "alpha",
+    });
+  });
+});

package/client/console-utils/remoteApi.ts

@@ -3,13 +3,23 @@ const API_BASE_KEY = "zhin_api_base";
 
 export function getApiBase(): string {
   const stored = localStorage.getItem(API_BASE_KEY)?.trim();
-  if (stored) return stored.replace(/\/$/, "");
+  if (stored) return trimTrailingSlash(stored);
   if (typeof window !== "undefined") return window.location.origin;
   return "";
 }
 
 export function getToken(): string | null {
-  return localStorage.getItem(TOKEN_KEY);
+  const runtime = getRuntimeToken();
+  if (runtime) return runtime;
+  if (typeof localStorage !== "undefined") {
+    const local = localStorage.getItem(TOKEN_KEY)?.trim();
+    if (local) return local;
+  }
+  if (typeof sessionStorage !== "undefined") {
+    const session = sessionStorage.getItem(TOKEN_KEY)?.trim();
+    if (session) return session;
+  }
+  return null;
 }
 
 export function resolveApiUrl(path: string): string {
@@ -18,13 +28,6 @@ export function resolveApiUrl(path: string): string {
   return `${base}${p}`;
 }
 
-export function resolveWebSocketUrl(path: string): string {
-  const httpUrl = resolveApiUrl(path);
-  const url = new URL(httpUrl);
-  url.protocol = url.protocol === "https:" ? "wss:" : "ws:";
-  return url.href;
-}
-
 /** fetch with Bearer token; relative paths resolve against {@link getApiBase}. */
 export async function apiFetch(
   input: RequestInfo | URL,
@@ -41,8 +44,43 @@ export async function apiFetch(
 
   const res = await fetch(url, { ...init, headers });
   if (res.status === 401) {
+    // runtime token（Demo 预置）过期后也必须失效，否则 getToken() 永远优先命中它，
+    // local/sessionStorage 清理不掉，造成 401 死循环。
+    clearRuntimeToken();
+    sessionStorage.removeItem(TOKEN_KEY);
     localStorage.removeItem(TOKEN_KEY);
     window.dispatchEvent(new CustomEvent("zhin:auth-required"));
   }
   return res;
 }
+
+/** Typed POST adapter for the canonical Plugin Runtime Console RPC endpoint. */
+export async function consoleRpc<T>(
+  type: string,
+  data: Readonly<Record<string, unknown>> = {},
+): Promise<T> {
+  const requestId = Date.now();
+  const res = await apiFetch('/api/console/request', {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ type, requestId, ...data }),
+  });
+  const body = await res.json() as { success?: boolean; data?: T; error?: string };
+  if (!res.ok || !body.success) throw new Error(body.error ?? `HTTP ${res.status}`);
+  return body.data as T;
+}
+
+function trimTrailingSlash(value: string): string {
+  return value.endsWith("/") ? value.slice(0, -1) : value;
+}
+
+function getRuntimeToken(): string | null {
+  if (typeof window === "undefined") return null;
+  const token = (window as unknown as { __ZHIN_API_TOKEN?: string }).__ZHIN_API_TOKEN;
+  return token?.trim() || null;
+}
+
+function clearRuntimeToken(): void {
+  if (typeof window === "undefined") return;
+  delete (window as unknown as { __ZHIN_API_TOKEN?: string }).__ZHIN_API_TOKEN;
+}

package/client/index.ts

@@ -1,43 +1,96 @@
 // Types
 export * from "./types";
+export {
+  ENDPOINT_RPC,
+  CONSOLE_EVENT_RECOVERY_GAP_EVENT,
+  ENDPOINT_MANAGEMENT_CAPABILITIES,
+  INBOX_RPC,
+  SIDE_EVENT_PUSH,
+  SIDE_EVENT_RPC,
+  parseConsoleInboxEvent,
+  type ConsoleInboxEvent,
+  type ConsoleInboxEventKind,
+  type ConsoleInboxMessageRow,
+  type ConsoleInboxNoticeRow,
+  type ConsoleInboxNoticesQuery,
+  type ConsoleInboxNoticesResult,
+  type ConsoleInboxRequestRow,
+  type ConsoleEndpointPhase,
+  type ConsoleEndpointSummary,
+  type ConsoleEventActor,
+  type ConsoleEventChannel,
+  type ConsoleEventParent,
+  type EndpointManagementCapability,
+  parseConsoleSseFrame,
+  type ConsoleEventDelivery,
+  type ConsoleEventData,
+  type ConsoleEndpointEventData,
+  type ConsoleEventEnvelope,
+  type ConsoleEventHistoryPage,
+  type ConsoleEventHistoryQuery,
+  type ConsoleEventPayloadMap,
+  type ConsoleMessageEventData,
+  type ConsoleNoticeEventData,
+  type ConsoleRequestEventData,
+  type KnownConsoleEventType,
+  type KnownConsoleEventEnvelope,
+  type ParsedConsoleSseFrame,
+} from "@zhin.js/console-protocol";
+
+export {
+  fetchConsoleEventHistory,
+  type FetchConsoleEventHistoryOptions,
+} from './console-events.js';
 
 // Media URL resolution
 export { resolveMediaSrc, pickMediaRawUrl, type MediaKind } from "./mediaSrc";
 
-// Console app singleton (page-manager style)
+// Safe rich message rendering (Markdown + code blocks; no raw HTML execution)
+export {
+  MarkdownContent,
+  CodeBlock,
+  type MarkdownContentProps,
+} from "./message-content/MarkdownContent.js";
+export {
+  parseMarkdown,
+  parseMarkdownInline,
+  highlightCodeLine,
+  isSafeMarkdownHref,
+  type MarkdownBlock,
+  type MarkdownInline,
+  type CodeToken,
+  type CodeTokenKind,
+} from "./message-content/markdown.js";
+
+// Segment IM visibility (inbox vs agent panel)
+export { segmentsForImDelivery, segmentsForAgentPanel } from "./segments.js";
+
+// Owner-scoped Console client and application registry
 export {
-  app,
-  type ConsoleApp,
+  ConsoleApp,
+  createConsoleApp,
   type AddRouteInput,
   type AddToolInput,
   type ConsoleRouteRecord,
   type RouteTreeNode,
   type ToolTreeNode,
+  type SidebarRenderer,
+  type ToolbarRenderer,
   type ConsoleRouteRenderer,
-} from "./app";
-
-// WebSocket (business data only)
-export * from "./websocket";
-
-// Re-export console-types
-export type {
-  PluginRegisterHostApi,
-  PluginAddRouteInput,
-  PluginAddToolInput,
-  ConsolePluginRegister,
-  ConsoleClientEntry,
-  ConsoleEntriesResponse,
-  ConsoleEntry,
-  ConsoleFileAddEntryInput,
-  RuntimeEnv,
-} from "@zhin.js/console-types";
-
+} from "./app.js";
+export {
+  ConsoleClient,
+  createConsoleClient,
+  type ConsoleClientOptions,
+} from './console-client.js';
 export {
-  DEFAULT_CONSOLE_BASE_PATH,
-  CONSOLE_HOST_REACT_NAMESPACE_KEY,
-} from "@zhin.js/console-types";
+  ConsoleClientProvider,
+  useConsoleClient,
+  type ConsoleClientProviderProps,
+} from './console-client-context.js';
 
-export { configureConsole, getRuntimeEnv } from "./runtime/index.js";
+// Owner-scoped REST/SSE transport
+export * from "./transport/index.js";
 
 export {
   createRegistryStore,
@@ -52,16 +105,27 @@ export {
   getApiBase,
   getToken,
   resolveApiUrl,
-  resolveWebSocketUrl,
 } from "./console-utils/remoteApi.js";
 
 export {
   fetchConsoleEntries,
-  createPluginRegisterHostApi,
   getRegisterFn,
   loadConsoleEntries,
   registerConsolePluginsFromEntries,
-  type CreatePluginRegisterHostApiOptions,
+  resolveEntryRegister,
   type FetchConsoleEntriesOptions,
   type LoadConsoleEntriesOptions,
 } from "./bootstrap/loadConsoleEntries.js";
+
+export { registerWorkroomConsole } from "./workroom/registerWorkroomConsole.js";
+export { default as WorkroomRunsPage } from "./workroom/WorkroomRunsPage.js";
+
+export {
+  startAgentSession,
+  continueAgentSession,
+  subscribeAgentStream,
+  iterateAgentStreamNdjson,
+  foldAgentStreamNdjson,
+  type AgentStreamClientOptions,
+  type SubscribeAgentStreamOptions,
+} from "./agent-stream.js";

package/client/mediaSrc.test.ts

@@ -0,0 +1,49 @@
+import { describe, expect, it } from 'vitest';
+import { pickMediaRawUrl, resolveMediaSrc } from '../client/mediaSrc.js';
+
+describe('pickMediaRawUrl', () => {
+  it('prefers canonical media.value / url / file', () => {
+    expect(pickMediaRawUrl({ media: { value: 'https://a/b.png' } })).toBe('https://a/b.png');
+    expect(pickMediaRawUrl({ url: 'https://a/b.png' })).toBe('https://a/b.png');
+    expect(pickMediaRawUrl({ file: '/tmp/a.png' })).toBe('/tmp/a.png');
+  });
+
+  it('reads agent outbound image segment (base64 + mime, no url)', () => {
+    // media-publisher 出站形态：{ type:'image', data:{ base64, mime, name } }
+    const raw = pickMediaRawUrl({ base64: 'aGVsbG8=', mime: 'image/png', name: 'system-status.png' });
+    expect(raw).toBe('data:image/png;base64,aGVsbG8=');
+    expect(resolveMediaSrc(raw, 'image')).toBe('data:image/png;base64,aGVsbG8=');
+  });
+
+  it('falls back to base64:// so resolveMediaSrc applies the kind default MIME', () => {
+    const raw = pickMediaRawUrl({ base64: 'aGVsbG8=' });
+    expect(raw).toBe('base64://aGVsbG8=');
+    expect(resolveMediaSrc(raw, 'image')).toBe('data:image/png;base64,aGVsbG8=');
+    expect(resolveMediaSrc(raw, 'audio')).toBe('data:audio/mpeg;base64,aGVsbG8=');
+  });
+
+  it('passes through base64 values that already carry a scheme', () => {
+    expect(pickMediaRawUrl({ base64: 'base64://aGVsbG8=' })).toBe('base64://aGVsbG8=');
+    expect(pickMediaRawUrl({ base64: 'data:image/png;base64,aGVsbG8=' })).toBe('data:image/png;base64,aGVsbG8=');
+  });
+
+  it('returns undefined when no media field exists', () => {
+    expect(pickMediaRawUrl(undefined)).toBeUndefined();
+    expect(pickMediaRawUrl({})).toBeUndefined();
+  });
+
+  it('rejects executable data URLs, credentials, and malformed base64', () => {
+    expect(resolveMediaSrc('data:text/html;base64,PHNjcmlwdD4=', 'image')).toBeUndefined();
+    expect(resolveMediaSrc('data:image/svg+xml;base64,PHN2Zz4=', 'image')).toBeUndefined();
+    expect(resolveMediaSrc('https://user:secret@example.com/a.png', 'image')).toBeUndefined();
+    expect(resolveMediaSrc('base64://not:base64', 'image')).toBeUndefined();
+  });
+
+  it('accepts only data MIME types matching the rendered media kind', () => {
+    expect(resolveMediaSrc('data:image/png;base64,aGVsbG8=', 'image')).toBe(
+      'data:image/png;base64,aGVsbG8=',
+    );
+    expect(resolveMediaSrc('data:image/png;base64,aGVsbG8=', 'video')).toBeUndefined();
+    expect(resolveMediaSrc('javascript:alert(1)', 'image')).toBeUndefined();
+  });
+});

package/client/mediaSrc.ts

@@ -3,6 +3,56 @@ export type MediaKind = 'image' | 'video' | 'audio'
 
 const BASE64_PROTO = 'base64://'
 
+const SAFE_DATA_MIME: Readonly<Record<MediaKind, ReadonlySet<string>>> = Object.freeze({
+  image: new Set(['image/avif', 'image/gif', 'image/jpeg', 'image/png', 'image/webp']),
+  video: new Set(['video/mp4', 'video/ogg', 'video/webm']),
+  audio: new Set(['audio/mpeg', 'audio/mp4', 'audio/ogg', 'audio/wav', 'audio/webm']),
+})
+
+function resolveSafeRemoteUrl(value: string): string | undefined {
+  try {
+    const parsed = new URL(value)
+    if (parsed.protocol !== 'http:' && parsed.protocol !== 'https:' && parsed.protocol !== 'blob:') {
+      return undefined
+    }
+    if (parsed.username || parsed.password) return undefined
+    return parsed.href
+  } catch {
+    return undefined
+  }
+}
+
+function resolveSafeDataUrl(value: string, kind: MediaKind): string | undefined {
+  if (!value.startsWith('data:')) return undefined
+  const separator = value.indexOf(',')
+  if (separator < 0) return undefined
+  const header = value.slice('data:'.length, separator).toLowerCase()
+  if (!header.endsWith(';base64')) return undefined
+  const mime = header.slice(0, -';base64'.length)
+  if (!SAFE_DATA_MIME[kind].has(mime)) return undefined
+  const payload = compactBase64(value.slice(separator + 1))
+  return payload ? `data:${mime};base64,${payload}` : undefined
+}
+
+function compactBase64(value: string): string | undefined {
+  let compact = ''
+  for (const character of value) {
+    if (character === ' ' || character === '\n' || character === '\r' || character === '\t') continue
+    const code = character.charCodeAt(0)
+    const allowed = (code >= 48 && code <= 57)
+      || (code >= 65 && code <= 90)
+      || (code >= 97 && code <= 122)
+      || character === '+' || character === '/' || character === '='
+    if (!allowed) return undefined
+    compact += character
+  }
+  const padding = compact.indexOf('=')
+  if (padding >= 0 && (compact.length - padding > 2 || compact.slice(padding).replace(/=/g, '') !== '')) {
+    return undefined
+  }
+  return compact || undefined
+}
+
 /**
  * 将消息段中的 url/file 等转为浏览器可用的媒体地址。
  *
@@ -21,31 +71,54 @@ export function resolveMediaSrc(
   const s = raw.trim()
   if (!s) return undefined
 
-  if (s.startsWith('data:') || s.startsWith('blob:') || s.startsWith('http://') || s.startsWith('https://')) {
-    return s
-  }
+  const dataUrl = resolveSafeDataUrl(s, kind)
+  if (dataUrl) return dataUrl
+
+  const remoteUrl = resolveSafeRemoteUrl(s)
+  if (remoteUrl) return remoteUrl
 
   if (!s.startsWith(BASE64_PROTO)) {
-    return s
+    return undefined
   }
 
-  const payload = s.slice(BASE64_PROTO.length).replace(/^\s+/, '')
+  const payload = s.slice(BASE64_PROTO.length).trimStart()
   if (!payload) return undefined
 
   // 已是 type/subtype;base64, 片段（不含 data: 前缀）
-  if (/^[\w+.-]+\/[\w+.-]+;base64,/i.test(payload)) {
-    return `data:${payload}`
+  const declared = resolveSafeDataUrl(`data:${payload}`, kind)
+  if (declared) {
+    return declared
   }
 
   const defaultMime =
     kind === 'image' ? 'image/png' : kind === 'video' ? 'video/mp4' : 'audio/mpeg'
-  const b64 = payload.replace(/\s/g, '')
-  return `data:${defaultMime};base64,${b64}`
+  const b64 = compactBase64(payload)
+  return b64 ? `data:${defaultMime};base64,${b64}` : undefined
 }
 
-/** 从 segment.data 取常见媒体字段 */
+/** 从 segment.data 取常见媒体字段（优先 canonical MediaRef） */
 export function pickMediaRawUrl(data: Record<string, unknown> | undefined): string | undefined {
   if (!data) return undefined
+  const media = data.media
+  if (media && typeof media === 'object' && media !== null) {
+    const value = (media as { value?: unknown }).value
+    if (typeof value === 'string' && value.trim()) return value.trim()
+  }
   const v = data.url ?? data.file ?? data.src ?? data.href
-  return typeof v === 'string' && v.trim() ? v.trim() : undefined
+  if (typeof v === 'string' && v.trim()) return v.trim()
+  // agent 出站媒体段（media-publisher）直接携带纯 base64，无 url/file；
+  // 有 mime 时拼成 data: URL 透传，否则走 base64:// 让 resolveMediaSrc 按 kind 补默认 MIME
+  const b64 = data.base64
+  if (typeof b64 === 'string' && b64.trim()) {
+    const payload = b64.trim()
+    if (payload.startsWith('data:') || payload.startsWith('base64://')
+      || payload.startsWith('http://') || payload.startsWith('https://')) {
+      return payload
+    }
+    const mime = typeof data.mime === 'string' && data.mime.trim() ? data.mime.trim() : ''
+    return mime
+      ? `data:${mime};base64,${payload.replace(/\s/g, '')}`
+      : `base64://${payload}`
+  }
+  return undefined
 }

package/client/message-content/MarkdownContent.tsx

@@ -0,0 +1,128 @@
+import { Fragment, useMemo, useState, type ReactNode } from 'react';
+import {
+  highlightCodeLine,
+  parseMarkdown,
+  type MarkdownInline,
+} from './markdown.js';
+
+export interface MarkdownContentProps {
+  readonly text: string;
+  readonly className?: string;
+}
+
+function InlineContent({ content }: { readonly content: readonly MarkdownInline[] }) {
+  const nodes: ReactNode[] = [];
+  content.forEach((token, index) => {
+    const key = `${token.type}-${index}`;
+    const value = token.value.split('\n').map((line, lineIndex, lines) => (
+      <Fragment key={`${key}-line-${lineIndex}`}>
+        {line}
+        {lineIndex < lines.length - 1 ? <br /> : null}
+      </Fragment>
+    ));
+    if (token.type === 'strong') nodes.push(<strong key={key}>{value}</strong>);
+    else if (token.type === 'emphasis') nodes.push(<em key={key}>{value}</em>);
+    else if (token.type === 'strike') nodes.push(<del key={key}>{value}</del>);
+    else if (token.type === 'code') nodes.push(<code key={key} className="im-inline-code">{token.value}</code>);
+    else if (token.type === 'link') nodes.push(
+      <a key={key} href={token.href} target={token.href.startsWith('#') ? undefined : '_blank'} rel="noreferrer" className="im-message-link">{value}</a>,
+    );
+    else nodes.push(<Fragment key={key}>{value}</Fragment>);
+  });
+  return <>{nodes}</>;
+}
+
+export function CodeBlock({ code, language = '', closed = true }: { readonly code: string; readonly language?: string; readonly closed?: boolean }) {
+  const [copied, setCopied] = useState(false);
+  const [wrapped, setWrapped] = useState(false);
+  const [expanded, setExpanded] = useState(false);
+  const lines = useMemo(() => code.split('\n'), [code]);
+  const isLong = lines.length > 18 || code.length > 1800;
+  const visibleLines = isLong && !expanded ? lines.slice(0, 18) : lines;
+
+  const copy = async () => {
+    try {
+      await navigator.clipboard.writeText(code);
+      setCopied(true);
+      window.setTimeout(() => setCopied(false), 1600);
+    } catch {
+      setCopied(false);
+    }
+  };
+
+  return (
+    <figure className="zhin-code-block">
+      <figcaption className="zhin-code-toolbar">
+        <span className="zhin-code-language">{language || 'plain text'}</span>
+        {!closed ? <span className="zhin-code-streaming">生成中</span> : null}
+        <span className="zhin-code-actions">
+          <button type="button" onClick={() => setWrapped((value) => !value)} aria-pressed={wrapped}>{wrapped ? '不折行' : '折行'}</button>
+          <button type="button" onClick={() => void copy()}>{copied ? '已复制' : '复制'}</button>
+        </span>
+      </figcaption>
+      <pre className={wrapped ? 'zhin-code-pre zhin-code-pre--wrap' : 'zhin-code-pre'}>
+        <code>
+          {visibleLines.map((line, lineIndex) => (
+            <span className="zhin-code-line" key={`${lineIndex}-${line}`}>
+              <span className="zhin-code-line-number" aria-hidden="true">{lineIndex + 1}</span>
+              <span className="zhin-code-line-content">
+                {highlightCodeLine(line, language).map((token, tokenIndex) => (
+                  <span key={`${token.kind}-${tokenIndex}`} className={`zhin-code-token zhin-code-token--${token.kind}`}>{token.value}</span>
+                ))}
+                {'\n'}
+              </span>
+            </span>
+          ))}
+        </code>
+      </pre>
+      {isLong ? (
+        <button type="button" className="zhin-code-expand" onClick={() => setExpanded((value) => !value)}>
+          {expanded ? '收起代码' : `展开其余 ${lines.length - visibleLines.length} 行`}
+        </button>
+      ) : null}
+    </figure>
+  );
+}
+
+export function MarkdownContent({ text, className }: MarkdownContentProps) {
+  const blocks = useMemo(() => parseMarkdown(text), [text]);
+
+  return (
+    <div className={['im-markdown-body', 'zhin-markdown', className].filter(Boolean).join(' ')}>
+      {blocks.map((block, index) => {
+        if (block.type === 'heading') {
+          const Heading = `h${block.level}` as keyof React.JSX.IntrinsicElements;
+          return <Heading key={index} className={`im-markdown-heading im-markdown-heading-${block.level}`}><InlineContent content={block.content} /></Heading>;
+        }
+        if (block.type === 'paragraph') return <p key={index} className="im-markdown-paragraph"><InlineContent content={block.content} /></p>;
+        if (block.type === 'quote') return <blockquote key={index} className="im-markdown-quote"><InlineContent content={block.content} /></blockquote>;
+        if (block.type === 'rule') return <hr key={index} className="zhin-markdown-rule" />;
+        if (block.type === 'code') return <CodeBlock key={index} code={block.value} language={block.language} closed={block.closed} />;
+        if (block.type === 'list') {
+          const List = block.ordered ? 'ol' : 'ul';
+          return (
+            <List key={index} className={`im-markdown-list${block.ordered ? ' im-markdown-list-ordered' : ''}`}>
+              {block.items.map((item, itemIndex) => (
+                <li key={itemIndex} className={item.checked === undefined ? undefined : 'zhin-markdown-task'}>
+                  {item.checked === undefined ? null : <input type="checkbox" checked={item.checked} readOnly aria-label={item.checked ? '已完成' : '未完成'} />}
+                  <InlineContent content={item.content} />
+                </li>
+              ))}
+            </List>
+          );
+        }
+        if (block.type === 'table') {
+          return (
+            <div className="zhin-markdown-table-wrap" key={index}>
+              <table className="zhin-markdown-table">
+                <thead><tr>{block.header.map((cell, cellIndex) => <th key={cellIndex}><InlineContent content={cell} /></th>)}</tr></thead>
+                <tbody>{block.rows.map((row, rowIndex) => <tr key={rowIndex}>{row.map((cell, cellIndex) => <td key={cellIndex}><InlineContent content={cell} /></td>)}</tr>)}</tbody>
+              </table>
+            </div>
+          );
+        }
+        return null;
+      })}
+    </div>
+  );
+}

package/client/message-content/markdown.test.ts

@@ -0,0 +1,48 @@
+import { highlightCodeLine, isSafeMarkdownHref, parseMarkdown, parseMarkdownInline } from './markdown.js';
+
+describe('message markdown', () => {
+  it('parses rich blocks including tables and tasks', () => {
+    const blocks = parseMarkdown(`# Result
+
+- [x] inspect runtime
+- [ ] publish patch
+
+| field | value |
+| --- | --- |
+| status | **ready** |`);
+
+    expect(blocks.map((block) => block.type)).toEqual(['heading', 'list', 'table']);
+    expect(blocks[1]).toMatchObject({ type: 'list', items: [{ checked: true }, { checked: false }] });
+  });
+
+  it('keeps incomplete streaming fences renderable', () => {
+    expect(parseMarkdown('```ts\nconst ready = true')).toEqual([
+      { type: 'code', language: 'ts', value: 'const ready = true', closed: false },
+    ]);
+  });
+
+  it('rejects unsafe markdown links without emitting active links', () => {
+    expect(isSafeMarkdownHref('javascript:alert(1)')).toBe(false);
+    expect(isSafeMarkdownHref('https://zhin.dev/docs')).toBe(true);
+    expect(parseMarkdownInline('[run](javascript:alert(1))')).toEqual([
+      { type: 'text', value: '[run](javascript:alert(1))' },
+    ]);
+  });
+
+  it('treats repeated unmatched link openers as plain text', () => {
+    const value = '['.repeat(20_000);
+    expect(parseMarkdownInline(value)).toEqual([{ type: 'text', value }]);
+  });
+
+  it('tokenizes common code without producing html', () => {
+    expect(highlightCodeLine('const answer = "<safe>" // note', 'ts')).toEqual([
+      { kind: 'keyword', value: 'const' },
+      { kind: 'plain', value: ' ' },
+      { kind: 'plain', value: 'answer' },
+      { kind: 'plain', value: ' = ' },
+      { kind: 'string', value: '"<safe>"' },
+      { kind: 'plain', value: ' ' },
+      { kind: 'comment', value: '// note' },
+    ]);
+  });
+});