can1357
508a9f543e
- Removed render_mermaid from tool discovery, task definitions, and registries. - Removed renderMermaid setting and prompt/docs references tied to the deleted tool. - Added maxWidth and theme color options to Mermaid ASCII resolution in markdown flow. - Re-rendered Mermaid ASCII in both directions and clipped output to available width.
93 lines
3.3 KiB
TypeScript
93 lines
3.3 KiB
TypeScript
import { type MermaidAsciiRenderOptions, renderMermaidAsciiSafe } from "@oh-my-pi/pi-utils";
|
|
|
|
/**
|
|
* Options controlling how fenced Mermaid source is resolved to terminal ASCII.
|
|
* Extends the raw render options (theme, color mode, spacing, `useAscii`) with a
|
|
* viewport-fitting hint.
|
|
*/
|
|
export interface MermaidResolveOptions extends MermaidAsciiRenderOptions {
|
|
/**
|
|
* Maximum display width (terminal columns) the diagram should occupy. A
|
|
* layout that overflows this width is re-rendered in the perpendicular
|
|
* orientation — a wide horizontal chain collapses to a tall vertical column
|
|
* (which the terminal can scroll), and a wide vertical fan-out collapses to a
|
|
* tall horizontal column. Omit to keep the source's own layout regardless of
|
|
* width.
|
|
*/
|
|
maxWidth?: number;
|
|
}
|
|
|
|
// Memoizes rendered ASCII (and failures) keyed on the render options + the
|
|
// layout-direction variant + source. Width selection happens per call against
|
|
// the cached renders, so a terminal resize re-decides without re-rendering.
|
|
const cache = new Map<string, string | null>();
|
|
|
|
/** Widest rendered row in display columns (ANSI- and CJK-aware). */
|
|
function asciiDisplayWidth(ascii: string): number {
|
|
let max = 0;
|
|
for (const line of ascii.split("\n")) {
|
|
const width = Bun.stringWidth(line);
|
|
if (width > max) max = width;
|
|
}
|
|
return max;
|
|
}
|
|
|
|
function renderVariant(
|
|
source: string,
|
|
baseOptions: MermaidAsciiRenderOptions,
|
|
baseKey: string,
|
|
direction: "TD" | "LR" | null,
|
|
): string | null {
|
|
const key = `${baseKey}\x00${direction ?? ""}\x00${source}`;
|
|
const cached = cache.get(key);
|
|
if (cached !== undefined) return cached;
|
|
|
|
const ascii = renderMermaidAsciiSafe(source, direction ? { ...baseOptions, direction } : baseOptions);
|
|
cache.set(key, ascii);
|
|
return ascii;
|
|
}
|
|
|
|
/**
|
|
* Resolve mermaid ASCII from fenced block source text.
|
|
* Returns null when rendering fails, while memoizing failures to avoid repeated work.
|
|
*/
|
|
export function resolveMermaidAscii(source: string, options?: MermaidResolveOptions): string | null {
|
|
const normalizedSource = source.replace(/\r\n?/g, "\n").trim();
|
|
if (!normalizedSource) return null;
|
|
|
|
const { maxWidth, ...rest } = options ?? {};
|
|
// Default to uncolored output; callers opt into a themed palette explicitly.
|
|
const baseOptions: MermaidAsciiRenderOptions = { colorMode: "none", ...rest };
|
|
const baseKey = JSON.stringify(baseOptions);
|
|
|
|
const base = renderVariant(normalizedSource, baseOptions, baseKey, null);
|
|
if (base === null) return null;
|
|
if (maxWidth === undefined) return base;
|
|
|
|
let best = base;
|
|
let bestWidth = asciiDisplayWidth(base);
|
|
if (bestWidth <= maxWidth) return base;
|
|
|
|
// The as-authored layout overflows. Render both forced orientations and keep
|
|
// the narrowest (clipping at the call site handles any residual overflow).
|
|
// Re-rendering the already-authored orientation is a cache hit, so this stays
|
|
// cheap, and one of the two will be the perpendicular fit.
|
|
for (const direction of ["TD", "LR"] as const) {
|
|
const variant = renderVariant(normalizedSource, baseOptions, baseKey, direction);
|
|
if (variant === null) continue;
|
|
const variantWidth = asciiDisplayWidth(variant);
|
|
if (variantWidth < bestWidth) {
|
|
best = variant;
|
|
bestWidth = variantWidth;
|
|
}
|
|
}
|
|
return best;
|
|
}
|
|
|
|
/**
|
|
* Clear the mermaid cache.
|
|
*/
|
|
export function clearMermaidCache(): void {
|
|
cache.clear();
|
|
}
|