llama.cpp/tools/ui/src/lib/utils/path-display.ts

import { PATH_SEPARATOR } from '$lib/constants/mcp-resource';
import { TRAILING_SLASHES_REGEX } from '$lib/constants/url';
import {
	CWD_CHANGED_PREFIX,
	CWD_CLEARED_TEXT,
	CWD_LINK_REGEX,
	FILE_URI_PREFIX,
	HOME_TILDE,
	HOME_TILDE_PREFIX
} from '$lib/constants';

/**
 * Last non-empty slash-delimited segment of `path`, with trailing
 * slashes stripped. Returns the input unchanged when no `/` is present.
 */
export function lastPathSegment(p: string): string {
	const trimmed = p.replace(TRAILING_SLASHES_REGEX, '');
	const idx = trimmed.lastIndexOf(PATH_SEPARATOR);
	return idx === -1 ? trimmed : trimmed.slice(idx + 1);
}

/**
 * Abbreviate `path` to `~/...` when it sits under `home`, or to `~` when
 * it equals `home`. Falls back to `lastPathSegment(path)` when home is
 * unknown or the path is outside it. `~` semantics are reserved for the
 * home directory, mirroring how shells render it.
 */
export function abbreviateWorkingDir(
	path: string | null | undefined,
	home: string | null | undefined
): string {
	if (!path) return '';
	if (!home) return lastPathSegment(path);
	if (path === home) return HOME_TILDE;
	if (path.startsWith(home + PATH_SEPARATOR))
		return HOME_TILDE_PREFIX + path.slice(home.length + 1);
	return lastPathSegment(path);
}

/**
 * Replace a leading `home` prefix in `path` with `~`. Unlike
 * abbreviateWorkingDir, paths outside `home` (or an unknown home) are
 * returned unchanged - used for tool-call path displays where the full
 * path matters.
 */
export function abbreviateHome(path: string, home: string | null | undefined): string {
	if (!home) return path;
	if (path === home) return HOME_TILDE;
	if (path.startsWith(home + PATH_SEPARATOR))
		return HOME_TILDE_PREFIX + path.slice(home.length + 1);
	return path;
}

export { CWD_CHANGED_PREFIX, CWD_CLEARED_TEXT } from '$lib/constants';

export interface CwdMessageInfo {
	// absolute server-side path, null when the cwd was cleared
	path: string | null;
	// display form shown in the UI (e.g. ~/Documents)
	display: string;
}

/**
 * Format a synthetic cwd-change message. The text mirrors what the UI
 * renders for it; the path travels as `[file:///abs/path](display)` so
 * both the absolute and the short form are visible to the model and
 * parseable back by the UI.
 */
export function formatCwdMessage(cwd: string, home: string | null): string {
	const display = abbreviateWorkingDir(cwd, home);
	return `${CWD_CHANGED_PREFIX}[${FILE_URI_PREFIX}${cwd}](${display}).`;
}

/**
 * Parse a synthetic cwd message back into its parts. The caller must already
 * know the message is synthetic (via the persisted `isSynthetic` flag); this
 * only extracts the path from the message text. Returns null when `content`
 * is not a cwd message.
 */
export function parseCwdMessage(content: string): CwdMessageInfo | null {
	const trimmed = content.trim();
	if (trimmed === CWD_CLEARED_TEXT) {
		return { path: null, display: '' };
	}
	if (trimmed.startsWith(CWD_CHANGED_PREFIX)) {
		const rest = trimmed.slice(CWD_CHANGED_PREFIX.length);
		// not anchored to the end: guidance may follow the link
		const link = rest.match(CWD_LINK_REGEX);
		if (link) return { path: link[1], display: link[2] };
		return { path: rest, display: rest };
	}
	return null;
}