pi-cc-extensions 1.0.5 → 1.0.7

package/README.en.md

@@ -28,7 +28,7 @@ Run `/reload` after installation.
 | Feature                     | Description                                                                                                                                  | Entry point                                     |
 | --------------------------- | -------------------------------------------------------------------------------------------------------------------------------------------- | ----------------------------------------------- |
 | Claude Code UI              | Tool summaries, expand/collapse, rich edit/write diffs, and `on` / `compact` / `off` modes                                                   | `/ccstyle`                                      |
-| Markdown enhancements       | Mermaid diagrams, admonitions, URL linking, and more                                                                                         | Automatic                                       |
+| Markdown enhancements       | Mermaid dialects and circled-number conversion                                                                                               | Automatic; turn off with `enableMarkdownEnhance: false` |
 | Fullscreen mode             | Tool card/group expand and collapse on click, previews, hover highlight, and a back-to-bottom button                                         | `TUIMODE=fullscreen` or `--tui-mode fullscreen` |
 | Settings panel              | `Style / Features / UI / Diff / Thinking / Footer` tabs                                                                                      | `/ccstyle`                                      |
 | Context inspection          | Usage breakdown and previews for the system prompt, memory, skills, tools definition, and messages                                           | `/context`                                      |
@@ -55,6 +55,7 @@ Renderer snapshots for `on` / `compact` modes: [default](./docs/tool-render-exam
   "enableAgentSummary": true,              // per-turn tool summary
   "enableWorkingMessage": true,            // Working... bottom token/elapsed
   "enableAliases": true,                   // /clear, /exit aliases
+  "enableMarkdownEnhance": true,             // markdown pass: Mermaid dialects and circled-number conversion; false hands it to other extensions
   "enableEditorPrefix": true,              // editor ❯ prefix
 
   // ui

package/README.md

@@ -28,7 +28,7 @@ pi install git:github.com/minuque/pi-cc-extensions
 | 功能                  | 说明                                                                            | 入口                                            |
 | --------------------- | ------------------------------------------------------------------------------- | ----------------------------------------------- |
 | Claude Code UI        | 工具摘要、折叠展开、rich edit/write diff，以及`on` / `compact` / `off` 三种模式 | `/ccstyle`                                      |
-| Markdown 增强         | Mermaid 图、提示框、URL 链接化等                                                        | 自动生效                                        |
+| Markdown 增强         | Mermaid 方言渲染、圈数字转半角括号                                                         | 自动生效；`enableMarkdownEnhance: false` 关闭   |
 | Fullscreen mode       | 工具卡/group 单击展开与收起、预览、hover 高亮、回到底部按钮                 | `TUIMODE=fullscreen` 或 `--tui-mode fullscreen` |
 | 配置面板              | `Style / Features / UI / Diff / Thinking / Footer` 六页签                       | `/ccstyle`                                      |
 | 上下文检查            | 查看上下文占用，并预览 System prompt、Memory、Skills、Tools definition 和消息内容 | `/context`                                      |
@@ -55,6 +55,7 @@ pi install git:github.com/minuque/pi-cc-extensions
   "enableAgentSummary": true,              // 每回合工具摘要
   "enableWorkingMessage": true,            // Working... 底部 token/耗时
   "enableAliases": true,                   // /clear、/exit 别名
+  "enableMarkdownEnhance": true,             // markdown 转换：Mermaid 方言渲染、圈数字转半角括号；false 让位给其他扩展
   "enableEditorPrefix": true,              // 输入框 ❯ 前缀
 
   // ui

package/extensions/config/config.ts

@@ -74,6 +74,8 @@ export type Config = {
 	enableAgentSummary: boolean;
 	enableWorkingMessage: boolean;
 	enableAliases: boolean;
+	/** markdown 转换：Mermaid 方言渲染、圈数字转半角括号。关掉让位给其他扩展。 */
+	enableMarkdownEnhance: boolean;
 	/** 输入框 `❯ ` 前缀（占左侧 2 列，颜色随 thinking/bash mode 联动）。 */
 	enableEditorPrefix: boolean;
 	enableCustomFooter: boolean;
@@ -157,6 +159,7 @@ export const DEFAULT_CONFIG: Config = {
 	enableAgentSummary: true,
 	enableWorkingMessage: true,
 	enableAliases: true,
+	enableMarkdownEnhance: true,
 	enableEditorPrefix: true,
 	enableCustomFooter: true,
 	footerTokenRate: true,
@@ -290,6 +293,7 @@ export function normalizeConfig(input: unknown): Config {
 		enableAgentSummary: source.enableAgentSummary !== false,
 		enableWorkingMessage: source.enableWorkingMessage !== false,
 		enableAliases: source.enableAliases !== false,
+		enableMarkdownEnhance: source.enableMarkdownEnhance !== false,
 		enableEditorPrefix: (source.enableEditorPrefix ?? source.enableInputPrompt) !== false, // 兼容旧配置键
 		enableCustomFooter: source.enableCustomFooter !== false,
 		footerTokenRate: source.footerTokenRate !== false,
@@ -361,6 +365,7 @@ export function formatConfigStatus(source: Config = config): string {
 		`agentSummary=${source.enableAgentSummary ? "on" : "off"}`,
 		`workingMsg=${source.enableWorkingMessage ? "on" : "off"}`,
 		`aliases=${source.enableAliases ? "on" : "off"}`,
+		`markdownEnhance=${source.enableMarkdownEnhance ? "on" : "off"}`,
 		`editorPrefix=${source.enableEditorPrefix ? "on" : "off"}`,
 		`footer=${source.enableCustomFooter ? "on" : "off"}`,
 		`footerTokenRate=${source.footerTokenRate ? "on" : "off"}`,

package/extensions/config/panel.ts

@@ -656,6 +656,13 @@ export async function showCcstylePanel(
 			"Aliases disabled.",
 			config.enableAliases,
 		);
+		const markdownEnhanceToggle = featureToggleSetting(
+			"enableMarkdownEnhance",
+			"Markdown enhance",
+			"Mermaid dialects and circled-number conversion. Next restart applies.",
+			"Markdown enhancements off; another extension can own the markdown pass.",
+			config.enableMarkdownEnhance,
+		);
 		const footerCurrencySetting = {
 			id: "footerCurrency",
 			label: "Cost currency",
@@ -798,6 +805,7 @@ export async function showCcstylePanel(
 			enableAgentSummary: agentSummaryToggle,
 			enableWorkingMessage: workingMessageToggle,
 			enableAliases: aliasesToggle,
+			enableMarkdownEnhance: markdownEnhanceToggle,
 		};
 
 		const onSettingChange = (id: string, value: string) => {
@@ -1049,6 +1057,7 @@ export async function showCcstylePanel(
 					agentSummaryToggle.setting,
 					workingMessageToggle.setting,
 					aliasesToggle.setting,
+					markdownEnhanceToggle.setting,
 				],
 			},
 			{

package/extensions/feature/reference/index.ts

@@ -417,9 +417,8 @@ export default function sessionReferenceExtension(pi: ExtensionAPI): void {
 
 		getAvailableReferences = getReferences;
 		if (ctx.mode === "tui") {
-			// 预取是 detached 的，必须兜住 rejection，否则 stale 错误会变成
-			// unhandled rejection 直接终止 Pi。
-			void getReferences().catch(() => {});
+			// 不在这里预取：SessionManager.listAll() 会读完整个会话库，大库要几秒，
+			// 而启动并不需要结果。首次 @ 输入由 getSuggestions 触发加载。
 			// Register after other session_start handlers. pi-fff claims every @
 			// prefix, so a provider installed before it would never see session mentions.
 			setTimeout(() => {

package/extensions/feature/shell/footer-layout.ts

@@ -40,10 +40,7 @@ const LEGACY_FIXED_FOOTER_KEYS = [
 	FOOTER_GIT_KEY,
 ] as const;
 
-const BUILTIN_FOOTER_KEYS = new Set<string>([
-	...DEFAULT_FOOTER_LINE1_KEYS,
-	...DEFAULT_FOOTER_LINE2_KEYS,
-]);
+const BUILTIN_FOOTER_KEYS = new Set<string>(LEGACY_FIXED_FOOTER_KEYS);
 
 export function isBuiltinFooterKey(key: string): boolean {
 	return BUILTIN_FOOTER_KEYS.has(key);

package/extensions/feature/shell/prompt-editor.ts

@@ -50,18 +50,37 @@ export class PromptEditor extends CustomEditor {
 	}
 }
 
-/** 应用/恢复输入框前缀。enableEditorPrefix=false 时不占扩展点。 */
+type EditorFactory = Parameters<ExtensionContext["ui"]["setEditorComponent"]>[0];
+
+const promptEditorFactory: EditorFactory = (tui, theme, keybindings) =>
+	new PromptEditor(tui, theme, keybindings, { embedWorkingStatus: true });
+
+/** 是否正占着插槽；占据前的工厂用于原样还回去。 */
+let active = false;
+let previousEditorFactory: EditorFactory | undefined;
+
+/** 撤销本扩展占据的插槽；从未占用就不碰插槽。 */
+function releasePromptEditor(ctx: ExtensionContext): void {
+	if (!active) return;
+	ctx.ui.setEditorComponent(previousEditorFactory);
+	active = false;
+	previousEditorFactory = undefined;
+}
+
+/** 应用输入框前缀；enableEditorPrefix=false 时只撤自己装的那层。 */
 export function applyPromptEditor(ctx: ExtensionContext): void {
 	if (!ctx?.hasUI || typeof ctx.ui?.setEditorComponent !== "function") return;
 	try {
-		if (config.enableEditorPrefix) {
-			ctx.ui.setEditorComponent(
-				(tui, theme, keybindings) =>
-					new PromptEditor(tui, theme, keybindings, { embedWorkingStatus: true }),
-			);
-		} else {
-			ctx.ui.setEditorComponent(undefined);
+		if (!config.enableEditorPrefix) {
+			releasePromptEditor(ctx);
+			return;
+		}
+		if (!active) {
+			// pi 的插槽是硬替换、没有栈，先记住是谁占着，撤销时还回去。
+			previousEditorFactory = ctx.ui.getEditorComponent?.();
+			active = true;
 		}
+		ctx.ui.setEditorComponent(promptEditorFactory);
 	} catch (err) {
 		ctx.ui.notify(
 			`prompt editor error: ${err instanceof Error ? err.message : String(err)}`,
@@ -70,8 +89,8 @@ export function applyPromptEditor(ctx: ExtensionContext): void {
 	}
 }
 
-/** 恢复 pi 原生编辑器。 */
+/** 恢复本扩展占据前的编辑器；未占用时不动作。 */
 export function clearPromptEditor(ctx: ExtensionContext): void {
 	if (!ctx?.hasUI || typeof ctx.ui?.setEditorComponent !== "function") return;
-	ctx.ui.setEditorComponent(undefined);
+	releasePromptEditor(ctx);
 }

package/extensions/feature/shell/token-rate.ts

@@ -3,8 +3,10 @@
  *
  * average：一次用户指令的墙钟平均（工具等待计入分母），显示 `avg`。
  * live：最近 1 秒的瞬时速率，生成期暂停时钟，显示 `live`；指令结束定格为整段平均。
- * token 与 working-message 同源：优先 provider `usage.output`，否则按 delta 增量估算
- * （CJK 1 token，其余 chars/4）。provider 值回填到本段消息窗口。
+ * token 与 working-message 同源：变大的 provider `usage.output` 覆盖估算，否则按 delta 累加
+ * （CJK 1 token，其余 chars/4）。没有变大的 usage 不当作累计值：Anthropic 的 message_start
+ * 会给出一个很小的 output_tokens，并在每个 partial 上重复，直到最后的 message_delta。
+ * 消息结束时用 message_end 上的 usage 定格本段总量。
  * Pi 的 turn 是单次 assistant 消息，工具调用结束会开新 turn。token 跨 turn 累加，
  * 计时与 TTFT 直到下一条用户指令才重置。
  * TTFT：用户指令开始到第一个内容块（text/thinking/toolcall）开始的毫秒数，
@@ -120,15 +122,29 @@ export function sampleTokenDelta(delta: string, now = Date.now()): void {
 }
 
 /**
- * 记录本段 provider `usage.output`。usage 按锚点摊回窗口：
+ * 记录本段 provider `usage.output`。变大的 usage 按锚点摊回窗口：
  * 首次报告从消息起点铺到现在，之后每次把增量摊在上次报告之后。
+ * 没有变大的报告直接忽略，避免 message_start 的占位值反复覆盖 delta 累计。
+ * `final` 用于消息结束：以本次 output 为本段总量，估算偏高偏低都按它修正。
+ * 结束值含未逐字流出的思考 token，它们不属于最近一秒，所以窗口只整体平移，不回填。
  */
-export function recordProviderUsage(output: number, now = Date.now()): void {
+export function recordProviderUsage(output: number, now = Date.now(), final = false): void {
 	if (!runStartTime || output <= 0) return;
+	if (final) {
+		const shift = output - currentTokens;
+		for (const sample of samples) sample.cum += shift;
+		countedUsage = output;
+		currentTokens = output;
+		usageAnchorTime = now;
+		usageAnchorTokens = output;
+		updateRates(now);
+		return;
+	}
+	if (output <= countedUsage) return;
 	const firstReport = countedUsage === 0;
 	const jump = output - countedUsage;
 	countedUsage = output;
-	// provider 是权威计数；首次报告以它为准，之后增量接管。
+	// 变大的 provider 计数是权威值，覆盖本段估算。
 	currentTokens = output;
 	if (firstReport) {
 		samples = samples.filter((sample) => sample.t < usageAnchorTime);

package/extensions/feature/shell/working-message.ts

@@ -49,9 +49,9 @@ export default function (pi: ExtensionAPI): void {
 		return totalTokens();
 	}
 
-	function updateProviderUsage(message: StreamMessage): void {
+	function updateProviderUsage(message: StreamMessage, final = false): void {
 		const output = outputUsage(message);
-		if (output > 0) recordProviderUsage(output);
+		if (output > 0) recordProviderUsage(output, Date.now(), final);
 	}
 
 	function buildWorkingMessage(): string {
@@ -171,13 +171,20 @@ export default function (pi: ExtensionAPI): void {
 			// 分段锚点只在 turn_start 重置。这里再重置会把累计 usage.output 当成新的首报。
 			markFirstToken();
 			updateProviderUsage(evt.partial);
-		} else if (evt.type === "thinking_delta" || evt.type === "text_delta") {
+		} else if (
+			evt.type === "thinking_delta" ||
+			evt.type === "text_delta" ||
+			evt.type === "toolcall_delta"
+		) {
+			// 工具参数走 toolcall_delta；不计入的话，写长文件这类调用期间 token 一直不涨。
 			if (typeof evt.delta === "string") sampleTokenDelta(evt.delta);
 			updateProviderUsage(evt.partial);
 		} else if (evt.type === "text_end" || evt.type === "thinking_end") {
 			updateProviderUsage(evt.partial);
 		} else if (evt.type === "done") {
-			updateProviderUsage(evt.message);
+			// Pi 目前不会把 done/error 转成 message_update，定格在下面的 message_end 里做。
+			// 这两个分支只在宿主直接转发 done/error 时生效；同一个值重复定格不改变结果。
+			updateProviderUsage(evt.message, true);
 		} else if (evt.type === "error") {
 			updateProviderUsage(evt.error);
 		} else {
@@ -188,6 +195,18 @@ export default function (pi: ExtensionAPI): void {
 		scheduleRefreshTick();
 	});
 
+	// message_delta 的累计 usage 只出现在 message_end 上，含未逐字流出的思考 token。
+	// message_end 在工具执行和 turn_end 之前，先于 carryTokenRate 结转。
+	pi.on("message_end", async (event, ctx) => {
+		const message = event?.message;
+		if (message?.role !== "assistant") return;
+		activeCtx = ctx;
+		// 出错/中断时 usage 可能还是 message_start 的占位值，只接受变大的值。
+		const failed = message.stopReason === "error" || message.stopReason === "aborted";
+		updateProviderUsage(message, !failed);
+		syncWorkingMessage();
+	});
+
 	pi.on("tool_execution_start", async () => {
 		pauseTokenRate();
 	});

package/extensions/index.ts

@@ -35,7 +35,8 @@ export default function (pi: ExtensionAPI): void {
 	});
 
 	// render stack：thinking controller 直接交给 style 作 query
-	markdownEnhance(pi);
+	// markdownEnhance 单独开关：其他扩展也注册 transformer 时会互相改写输出。
+	if (config.enableMarkdownEnhance) markdownEnhance(pi);
 	claudeCodeStyle(pi, undefined, installCompactThinking(pi, getCompactThinkingConfig()));
 
 	// features

package/extensions/renderer/markdown-enhance.ts

@@ -41,99 +41,6 @@ function collectFence(lines: string[], i: number): { diagram: string; next: numb
 	return null; // 未闭合：放弃转换，保持原文
 }
 
-// ============================================================================
-// GitHub 风格提示框（admonition）
-// ============================================================================
-
-const ADMONITION = /^>\s*\[\s*!(NOTE|TIP|IMPORTANT|WARNING|CAUTION)\s*\]\s*(.*)$/i;
-const ADMONITION_STYLE: Record<string, { icon: string; label: string }> = {
-	NOTE: { icon: "💡", label: "NOTE" },
-	TIP: { icon: "✅", label: "TIP" },
-	IMPORTANT: { icon: "❗", label: "IMPORTANT" },
-	WARNING: { icon: "⚠️", label: "WARNING" },
-	CAUTION: { icon: "⚠️", label: "CAUTION" },
-};
-
-/**
- * 把 > [!TYPE] 及其后续 > 行转成带加粗标签的引用块，视觉上成提示框。
- * 多行内容用空格合并为一行。返回 null 表示该行不是提示框。
- */
-function renderAdmonition(lines: string[], i: number): { output: string[]; next: number } | null {
-	const m = lines[i].match(ADMONITION);
-	if (!m) return null;
-	const style = ADMONITION_STYLE[m[1].toUpperCase()];
-	const body: string[] = [m[2]];
-	let j = i + 1;
-	while (j < lines.length && /^>\s?/.test(lines[j]) && !ADMONITION.test(lines[j])) {
-		body.push(lines[j].replace(/^>\s?/, ""));
-		j++;
-	}
-	while (body.length > 0 && body[body.length - 1].trim() === "") body.pop();
-	// 内容合并为一行，作为引用块正文
-	const content = body.join(" ").trim();
-	// 保留引用块形态（pi 渲染为左侧竖线），标签加粗
-	return {
-		output: [`> **${style.icon} ${style.label}**${content ? ` ${content}` : ""}`, ""],
-		next: j,
-	};
-}
-
-// ============================================================================
-// 裸 URL 转可点击超链接
-// ============================================================================
-
-const URL_RE = /(?<!<)(?<!\]\()https?:\/\/[^\s<>'"|，。；：！？、」』】（）【】《》『』「」]+/g;
-const TRIM_URL_RE = /[.,;:!?】」』"'》）}]+$/;
-
-/**
- * 去掉 URL 尾部标点；括号按平衡保留（如 Wikipedia 链接含括号、
- * IPv6 地址含 [::1]），尾部不成对的 ] 与 ) 会截掉。
- */
-function trimUrl(url: string): string {
-	let t = url.replace(TRIM_URL_RE, "");
-	for (;;) {
-		if (t.endsWith(")") && (t.match(/\(/g)?.length ?? 0) < (t.match(/\)/g)?.length ?? 0)) {
-			t = t.slice(0, -1);
-		} else if (t.endsWith("]") && (t.match(/\[/g)?.length ?? 0) < (t.match(/\]/g)?.length ?? 0)) {
-			t = t.slice(0, -1);
-		} else {
-			break;
-		}
-	}
-	return t;
-}
-
-/**
- * 把代码块和行内代码外的裸 URL 转成 markdown 链接。
- * 已有的 [text](url) / ![alt](url) / <url> 自动链接用 lookbehind 排除。
- */
-function linkifyUrls(markdown: string): string {
-	const lines = markdown.split("\n");
-	const out: string[] = [];
-	let fence: MarkdownFence | undefined;
-	for (const line of lines) {
-		const scanned = scanMarkdownFence(line, fence);
-		fence = scanned.fence;
-		if (scanned.protected) {
-			out.push(line);
-			continue;
-		}
-		// 行内代码（`...`）按反引号分片，只处理偶数片（代码外）
-		const parts = line.split("`");
-		for (let p = 0; p < parts.length; p++) {
-			if (p % 2 === 0) {
-				parts[p] = parts[p].replace(URL_RE, (url) => {
-					const trimmed = trimUrl(url);
-					if (trimmed.length === 0) return url;
-					return `[${trimmed}](${trimmed})`;
-				});
-			}
-		}
-		out.push(parts.join("`"));
-	}
-	return out.join("\n");
-}
-
 // ============================================================================
 // 圈数字 → 半角括号
 // ============================================================================
@@ -231,7 +138,9 @@ function normalizeMultilineLinks(markdown: string): string {
 
 export default function (pi: ExtensionAPI): void {
 	// 注意：pi 每个扩展只有一个 markdownTransformer 槽位，多次注册会互相覆盖，
-	// 所以三个转换合并为一次注册，内部按序链式执行。
+	// 所以剩下的转换合并为一次注册，内部按序链式执行。
+	// 提示框和裸 URL 不接管：pi 的 marked 已自带裸 URL 链接化（带尾标点回退，
+	// 且在链接文本内不再嵌套），提示框交给默认引用块风格。
 	pi.registerMarkdownTransformer((markdown, context) => {
 		const { messageType, isStreaming = false } = context ?? {};
 		// thinking 保持原文；普通回复即使流式中也先关闭跨行链接的空白点击区。
@@ -241,11 +150,7 @@ export default function (pi: ExtensionAPI): void {
 		// 0. 圈数字转半角括号（Nerd Font 补丁字形缺陷规避）
 		markdown = deCircled(markdown);
 		// 1. Mermaid 方言渲染
-		markdown = renderDiagrams(markdown, context);
-		// 2. GitHub 风格提示框
-		markdown = renderAdmonitions(markdown);
-		// 3. 裸 URL 转超链接
-		return linkifyUrls(markdown);
+		return renderDiagrams(markdown, context);
 	});
 }
 
@@ -307,32 +212,3 @@ function renderDiagrams(
 	}
 	return out.join("\n");
 }
-
-/** GitHub 风格提示框 → 加粗标签引用块（跳过代码块）。 */
-function renderAdmonitions(markdown: string): string {
-	const lines = markdown.split("\n");
-	const out: string[] = [];
-	let fence: MarkdownFence | undefined;
-	let i = 0;
-	while (i < lines.length) {
-		const line = lines[i];
-		const scanned = scanMarkdownFence(line, fence);
-		fence = scanned.fence;
-		if (scanned.protected) {
-			out.push(line);
-			i++;
-			continue;
-		}
-		if (/^>\s*\[\s*!/i.test(line)) {
-			const result = renderAdmonition(lines, i);
-			if (result) {
-				out.push(...result.output);
-				i = result.next;
-				continue;
-			}
-		}
-		out.push(line);
-		i++;
-	}
-	return out.join("\n");
-}

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "pi-cc-extensions",
-	"version": "1.0.5",
+	"version": "1.0.7",
 	"license": "MIT",
 	"description": "Claude Code-style TUI output with some personal touches and handy utilities.",
 	"type": "module",