openclaw/src/auto-reply/tool-meta.ts

144 lines
3.6 KiB
TypeScript

import { formatToolSummary, resolveToolDisplay } from "../agents/tool-display.js";
import { shortenHomeInString, shortenHomePath } from "../utils.js";
type ToolAggregateOptions = {
markdown?: boolean;
};
export function shortenPath(p: string): string {
return shortenHomePath(p);
}
export function shortenMeta(meta: string): string {
if (!meta) {
return meta;
}
return shortenHomeInString(meta);
}
export function formatToolAggregate(
toolName?: string,
metas?: string[],
options?: ToolAggregateOptions,
): string {
const filtered = (metas ?? []).filter(Boolean).map(shortenMeta);
const display = resolveToolDisplay({ name: toolName });
const prefix = `${display.emoji} ${display.label}`;
if (!filtered.length) {
return prefix;
}
const rawSegments: string[] = [];
// Group by directory and brace-collapse filenames
const grouped: Record<string, string[]> = {};
for (const m of filtered) {
if (!isPathLike(m)) {
rawSegments.push(m);
continue;
}
if (m.includes("→")) {
rawSegments.push(m);
continue;
}
const parts = m.split("/");
if (parts.length > 1) {
const dir = parts.slice(0, -1).join("/");
const base = parts.at(-1) ?? m;
if (!grouped[dir]) {
grouped[dir] = [];
}
grouped[dir].push(base);
} else {
if (!grouped["."]) {
grouped["."] = [];
}
grouped["."].push(m);
}
}
const segments = Object.entries(grouped).map(([dir, files]) => {
const brace = files.length > 1 ? `{${files.join(", ")}}` : files[0];
if (dir === ".") {
return brace;
}
return `${dir}/${brace}`;
});
const allSegments = [...rawSegments, ...segments];
const meta = allSegments.join("; ");
return `${prefix}: ${formatMetaForDisplay(toolName, meta, options?.markdown)}`;
}
export function formatToolPrefix(toolName?: string, meta?: string) {
const extra = meta?.trim() ? shortenMeta(meta) : undefined;
const display = resolveToolDisplay({ name: toolName, meta: extra });
return formatToolSummary(display);
}
function formatMetaForDisplay(
toolName: string | undefined,
meta: string,
markdown?: boolean,
): string {
const normalized = (toolName ?? "").trim().toLowerCase();
if (normalized === "exec" || normalized === "bash") {
const { flags, body } = splitExecFlags(meta);
if (flags.length > 0) {
if (!body) {
return flags.join(" · ");
}
return `${flags.join(" · ")} · ${maybeWrapMarkdown(body, markdown)}`;
}
}
return maybeWrapMarkdown(meta, markdown);
}
function splitExecFlags(meta: string): { flags: string[]; body: string } {
const parts = meta
.split(" · ")
.map((part) => part.trim())
.filter(Boolean);
if (parts.length === 0) {
return { flags: [], body: "" };
}
const flags: string[] = [];
const bodyParts: string[] = [];
for (const part of parts) {
if (part === "elevated" || part === "pty") {
flags.push(part);
continue;
}
bodyParts.push(part);
}
return { flags, body: bodyParts.join(" · ") };
}
function isPathLike(value: string): boolean {
if (!value) {
return false;
}
if (value.includes(" ")) {
return false;
}
if (value.includes("://")) {
return false;
}
if (value.includes("·")) {
return false;
}
if (value.includes("&&") || value.includes("||")) {
return false;
}
return /^~?(\/[^\s]+)+$/.test(value);
}
function maybeWrapMarkdown(value: string, markdown?: boolean): string {
if (!markdown) {
return value;
}
if (value.includes("`")) {
return value;
}
return `\`${value}\``;
}