can1357
4f03180ae6
- Moved the `INTENT_FIELD` constant from `@oh-my-pi/pi-agent-core` to the specialized `@oh-my-pi/pi-wire` package to permit broader usage across the monorepo. - Updated all references across `agent`, `ai`, `coding-agent`, `collab-web`, and `snapcompact` packages to import the constant from the new location. - Added `@oh-my-pi/pi-wire` as a dependency to all affected packages.
261 lines
8.2 KiB
TypeScript
261 lines
8.2 KiB
TypeScript
/**
|
|
* JSON tree rendering utilities shared across tool renderers.
|
|
*/
|
|
import { INTENT_FIELD } from "@oh-my-pi/pi-wire";
|
|
import type { Theme } from "../modes/theme/theme";
|
|
import { truncateToWidth } from "./render-utils";
|
|
|
|
/** Max depth for JSON tree rendering */
|
|
export const JSON_TREE_MAX_DEPTH_COLLAPSED = 2;
|
|
export const JSON_TREE_MAX_DEPTH_EXPANDED = 6;
|
|
export const JSON_TREE_MAX_LINES_COLLAPSED = 6;
|
|
export const JSON_TREE_MAX_LINES_EXPANDED = 200;
|
|
export const JSON_TREE_SCALAR_LEN_COLLAPSED = 60;
|
|
export const JSON_TREE_SCALAR_LEN_EXPANDED = 2000;
|
|
|
|
const HIDDEN_ARG_KEYS = { [INTENT_FIELD]: 1, __partialJson: 1 };
|
|
|
|
const ARGS_INLINE_PAIR_SEP = ", ";
|
|
const ARGS_INLINE_PAIR_SEP_WIDTH = Bun.stringWidth(ARGS_INLINE_PAIR_SEP);
|
|
const ARGS_INLINE_MORE = "…";
|
|
const ARGS_INLINE_MORE_WIDTH = Bun.stringWidth(ARGS_INLINE_MORE);
|
|
/** Minimal value footprint (quotes + a couple chars) reserved for each not-yet-rendered key. */
|
|
const ARGS_INLINE_TAIL_VALUE_RESERVE = 4;
|
|
|
|
function isRecord(value: unknown): value is Record<string, unknown> {
|
|
return !!value && typeof value === "object" && !Array.isArray(value);
|
|
}
|
|
|
|
/**
|
|
* Format a scalar value for inline display.
|
|
*/
|
|
export function formatScalar(value: unknown, maxLen: number): string {
|
|
if (value === null) return "null";
|
|
if (value === undefined) return "undefined";
|
|
if (typeof value === "boolean") return String(value);
|
|
if (typeof value === "number") return String(value);
|
|
if (typeof value === "string") {
|
|
const escaped = value.replace(/\n/g, "\\n").replace(/\t/g, "\\t");
|
|
const truncated = truncateToWidth(escaped, maxLen);
|
|
return `"${truncated}"`;
|
|
}
|
|
if (Array.isArray(value)) return `[${value.length} items]`;
|
|
if (typeof value === "object") {
|
|
const keys = Object.keys(value);
|
|
return `{${keys.length} keys}`;
|
|
}
|
|
return String(value);
|
|
}
|
|
|
|
/**
|
|
* Format args inline for collapsed view.
|
|
*/
|
|
export function formatArgsInline(args: Record<string, unknown>, maxWidth: number): string {
|
|
const keys: string[] = [];
|
|
for (const key in args) {
|
|
if (key in HIDDEN_ARG_KEYS) continue;
|
|
keys.push(key);
|
|
}
|
|
let result = "";
|
|
let width = 0;
|
|
for (let i = 0; i < keys.length; i++) {
|
|
const key = keys[i];
|
|
const value = args[key];
|
|
const sep = width > 0 ? ARGS_INLINE_PAIR_SEP : "";
|
|
const sepW = width > 0 ? ARGS_INLINE_PAIR_SEP_WIDTH : 0;
|
|
const current = width + sepW;
|
|
const cap = maxWidth - current - ARGS_INLINE_MORE_WIDTH;
|
|
if (cap <= 0) {
|
|
return `${result}${ARGS_INLINE_MORE}`;
|
|
}
|
|
// Reserve each still-pending key's minimal footprint (sep + name + `=` +
|
|
// a short value) so a long value can't starve the keys that follow it.
|
|
let tailReserve = 0;
|
|
for (let j = i + 1; j < keys.length; j++) {
|
|
tailReserve += ARGS_INLINE_PAIR_SEP_WIDTH + Bun.stringWidth(keys[j]) + 1 + ARGS_INLINE_TAIL_VALUE_RESERVE;
|
|
}
|
|
// Budget the whole `key=value` piece against the width left after the
|
|
// tail reserve, then back out the value's share. The last key reserves
|
|
// nothing and fills the line.
|
|
const pieceBudget = Math.min(cap, maxWidth - current - tailReserve);
|
|
const valueMaxLen = Math.max(1, pieceBudget - Bun.stringWidth(key) - 3);
|
|
const valueStr = formatScalar(value, valueMaxLen);
|
|
const piece = `${key}=${valueStr}`;
|
|
const pieceW = Bun.stringWidth(piece);
|
|
if (pieceW > pieceBudget) {
|
|
return `${result}${sep}${truncateToWidth(piece, cap)}`;
|
|
}
|
|
result += sep + piece;
|
|
width = current + pieceW;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
/**
|
|
* Build tree prefix for nested rendering.
|
|
*/
|
|
function buildTreePrefix(theme: Theme, ancestors: readonly boolean[]): string {
|
|
return ancestors.map(hasNext => (hasNext ? `${theme.tree.vertical} ` : " ")).join("");
|
|
}
|
|
|
|
/**
|
|
* Render a JSON value as tree lines.
|
|
*/
|
|
export function renderJsonTreeLines(
|
|
value: unknown,
|
|
theme: Theme,
|
|
maxDepth: number,
|
|
maxLines: number,
|
|
maxScalarLen: number,
|
|
): { lines: string[]; truncated: boolean } {
|
|
const lines: string[] = [];
|
|
let truncated = false;
|
|
|
|
const iconObject = theme.styledSymbol("icon.folder", "muted");
|
|
const iconArray = theme.styledSymbol("icon.package", "muted");
|
|
const iconScalar = theme.styledSymbol("icon.file", "muted");
|
|
|
|
const pushLine = (line: string): boolean => {
|
|
if (lines.length >= maxLines) {
|
|
truncated = true;
|
|
return false;
|
|
}
|
|
lines.push(line);
|
|
return true;
|
|
};
|
|
|
|
const renderNode = (val: unknown, key: string | undefined, ancestors: boolean[], isLast: boolean, depth: number) => {
|
|
if (lines.length >= maxLines) {
|
|
truncated = true;
|
|
return;
|
|
}
|
|
|
|
const connector = isLast ? theme.tree.last : theme.tree.branch;
|
|
const prefix = `${buildTreePrefix(theme, ancestors)}${theme.fg("dim", connector)} `;
|
|
|
|
ancestors.push(!isLast);
|
|
try {
|
|
// Handle scalars
|
|
if (val === null || val === undefined || typeof val !== "object") {
|
|
const label = key ? theme.fg("muted", key) : theme.fg("muted", "value");
|
|
|
|
// Special handling for multiline strings
|
|
if (typeof val === "string" && val.includes("\n")) {
|
|
const strLines = val.split("\n");
|
|
const maxStrLines = Math.min(strLines.length, Math.max(1, maxLines - lines.length - 1));
|
|
const continuePrefix = buildTreePrefix(theme, ancestors);
|
|
|
|
// First line with label
|
|
const firstLine = truncateToWidth(strLines[0], maxScalarLen);
|
|
pushLine(`${prefix}${iconScalar} ${label}: ${theme.fg("dim", `"${firstLine}`)}`);
|
|
|
|
// Subsequent lines indented
|
|
for (let i = 1; i < maxStrLines; i++) {
|
|
if (lines.length >= maxLines) {
|
|
truncated = true;
|
|
break;
|
|
}
|
|
const line = truncateToWidth(strLines[i], maxScalarLen);
|
|
pushLine(`${continuePrefix} ${theme.fg("dim", ` ${line}`)}`);
|
|
}
|
|
|
|
// Show truncation and closing quote
|
|
if (strLines.length > maxStrLines) {
|
|
truncated = true;
|
|
pushLine(
|
|
`${continuePrefix} ${theme.fg("dim", ` …(${strLines.length - maxStrLines} more lines)"`)}`,
|
|
);
|
|
} else {
|
|
// Add closing quote to last line - need to modify the last pushed line
|
|
const lastIdx = lines.length - 1;
|
|
lines[lastIdx] = `${lines[lastIdx]}${theme.fg("dim", '"')}`;
|
|
}
|
|
return;
|
|
}
|
|
|
|
const scalar = formatScalar(val, maxScalarLen);
|
|
pushLine(`${prefix}${iconScalar} ${label}: ${theme.fg("dim", scalar)}`);
|
|
return;
|
|
}
|
|
|
|
// Handle arrays
|
|
if (Array.isArray(val)) {
|
|
const header = key ? theme.fg("muted", key) : theme.fg("muted", "array");
|
|
pushLine(`${prefix}${iconArray} ${header}`);
|
|
if (val.length === 0) {
|
|
pushLine(
|
|
`${buildTreePrefix(theme, ancestors)}${theme.fg("dim", theme.tree.last)} ${theme.fg("dim", "[]")}`,
|
|
);
|
|
return;
|
|
}
|
|
if (depth >= maxDepth) {
|
|
pushLine(
|
|
`${buildTreePrefix(theme, ancestors)}${theme.fg("dim", theme.tree.last)} ${theme.fg("dim", "…")}`,
|
|
);
|
|
return;
|
|
}
|
|
for (let i = 0; i < val.length; i++) {
|
|
renderNode(val[i], `[${i}]`, ancestors, i === val.length - 1, depth + 1);
|
|
if (lines.length >= maxLines) {
|
|
truncated = true;
|
|
return;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
// Handle objects
|
|
if (!isRecord(val)) return;
|
|
|
|
const header = key ? theme.fg("muted", key) : theme.fg("muted", "object");
|
|
pushLine(`${prefix}${iconObject} ${header}`);
|
|
if (depth >= maxDepth) {
|
|
pushLine(`${buildTreePrefix(theme, ancestors)}${theme.fg("dim", theme.tree.last)} ${theme.fg("dim", "…")}`);
|
|
return;
|
|
}
|
|
const keys = Object.keys(val);
|
|
if (keys.length === 0) {
|
|
pushLine(
|
|
`${buildTreePrefix(theme, ancestors)}${theme.fg("dim", theme.tree.last)} ${theme.fg("dim", "{}")}`,
|
|
);
|
|
return;
|
|
}
|
|
for (let i = 0; i < keys.length; i++) {
|
|
const childKey = keys[i];
|
|
const child = val[childKey];
|
|
renderNode(child, childKey, ancestors, i === keys.length - 1, depth + 1);
|
|
if (lines.length >= maxLines) {
|
|
truncated = true;
|
|
return;
|
|
}
|
|
}
|
|
} finally {
|
|
ancestors.pop();
|
|
}
|
|
};
|
|
|
|
// Render root level
|
|
if (isRecord(value)) {
|
|
for (const key in value) {
|
|
if (key in HIDDEN_ARG_KEYS) continue;
|
|
renderNode(value[key], key, [], true, 1);
|
|
if (lines.length >= maxLines) {
|
|
truncated = true;
|
|
break;
|
|
}
|
|
}
|
|
} else if (Array.isArray(value)) {
|
|
for (let i = 0; i < value.length; i++) {
|
|
renderNode(value[i], `[${i}]`, [], i === value.length - 1, 1);
|
|
if (lines.length >= maxLines) {
|
|
truncated = true;
|
|
break;
|
|
}
|
|
}
|
|
} else {
|
|
renderNode(value, undefined, [], true, 0);
|
|
}
|
|
|
|
return { lines, truncated };
|
|
}
|