@oh-my-pi/pi-tui 16.3.6 → 16.3.8

package/CHANGELOG.md

@@ -2,6 +2,13 @@
 
 ## [Unreleased]
 
+## [16.3.7] - 2026-07-05
+
+### Fixed
+
+- Fixed an issue where `@` file-reference tokens in slash-command arguments incorrectly triggered prompt-composer autocompletion when the command did not define argument completions.
+- Fixed a memory leak caused by unbounded map growth in the image budget cache.
+
 ## [16.3.6] - 2026-07-04
 
 ### Added

package/dist/types/autocomplete.d.ts

@@ -19,6 +19,8 @@ export interface SlashCommand {
     aliases?: string[];
     description?: string;
     argumentHint?: string;
+    /** Whether the command consumes argument text after the command name. False means the full input stays normal prompt text once args are present. */
+    allowArgs?: boolean;
     /** Dynamic display-only description for slash-command autocomplete. Must be synchronous and side-effect free. */
     getAutocompleteDescription?: () => string | undefined;
     getArgumentCompletions?(argumentPrefix: string): Awaitable<AutocompleteItem[] | null>;

package/dist/types/components/box.d.ts

@@ -1,4 +1,4 @@
-import type { Component } from "../tui";
+import type { Component } from "../tui.js";
 /** Box-drawing glyphs plus an optional colorizer for an outline drawn around a {@link Box}. */
 export interface BoxBorder {
     chars: {

package/dist/types/components/cancellable-loader.d.ts

@@ -1,4 +1,4 @@
-import { Loader } from "./loader";
+import { Loader } from "./loader.js";
 /**
  * Loader that can be cancelled with Escape.
  * Extends Loader with an AbortSignal for cancelling async operations.

package/dist/types/components/editor.d.ts

@@ -1,7 +1,7 @@
-import { type AutocompleteProvider } from "../autocomplete";
-import type { SymbolTheme } from "../symbols";
-import { type Component, type Focusable } from "../tui";
-import { type SelectListTheme } from "./select-list";
+import { type AutocompleteProvider } from "../autocomplete.js";
+import type { SymbolTheme } from "../symbols.js";
+import { type Component, type Focusable } from "../tui.js";
+import { type SelectListTheme } from "./select-list.js";
 export interface EditorTheme {
     borderColor: (str: string) => string;
     selectList: SelectListTheme;

package/dist/types/components/image.d.ts

@@ -1,5 +1,5 @@
-import { type ImageDimensions } from "../terminal-capabilities";
-import type { Component } from "../tui";
+import { type ImageDimensions } from "../terminal-capabilities.js";
+import type { Component } from "../tui.js";
 export interface ImageTheme {
     fallbackColor: (str: string) => string;
 }

package/dist/types/components/input.d.ts

@@ -1,4 +1,4 @@
-import { type Component, type Focusable } from "../tui";
+import { type Component, type Focusable } from "../tui.js";
 /**
  * Input component - single-line text input with horizontal scrolling
  */

package/dist/types/components/loader.d.ts

@@ -1,5 +1,5 @@
-import type { TUI } from "../tui";
-import { Text } from "./text";
+import type { TUI } from "../tui.js";
+import { Text } from "./text.js";
 type ColorFn = (str: string) => string;
 export type LoaderMessageColorFn = ColorFn & {
     readonly animated?: true;

package/dist/types/components/markdown.d.ts

@@ -1,5 +1,5 @@
-import type { SymbolTheme } from "../symbols";
-import type { Component } from "../tui";
+import type { SymbolTheme } from "../symbols.js";
+import type { Component } from "../tui.js";
 /** Drop all L2 cache entries. Call on theme change to prevent stale styled output. */
 export declare function clearRenderCache(): void;
 /**

package/dist/types/components/scroll-view.d.ts

@@ -1,5 +1,5 @@
-import type { Component } from "../tui";
-import { Ellipsis } from "../utils";
+import type { Component } from "../tui.js";
+import { Ellipsis } from "../utils.js";
 type ScrollbarMode = "auto" | "always" | "never";
 export interface ScrollViewTheme {
     track?: (text: string) => string;

package/dist/types/components/select-list.d.ts

@@ -1,6 +1,6 @@
-import { type MouseRoutable, type SgrMouseEvent } from "../mouse";
-import type { SymbolTheme } from "../symbols";
-import type { Component } from "../tui";
+import { type MouseRoutable, type SgrMouseEvent } from "../mouse.js";
+import type { SymbolTheme } from "../symbols.js";
+import type { Component } from "../tui.js";
 export interface SelectItem {
     value: string;
     label: string;

package/dist/types/components/settings-list.d.ts

@@ -1,5 +1,5 @@
-import type { SgrMouseEvent } from "../mouse";
-import type { Component } from "../tui";
+import type { SgrMouseEvent } from "../mouse.js";
+import type { Component } from "../tui.js";
 export interface SettingItem {
     /** Unique identifier for this setting */
     id: string;

package/dist/types/components/spacer.d.ts

@@ -1,4 +1,4 @@
-import type { Component } from "../tui";
+import type { Component } from "../tui.js";
 /**
  * Spacer component that renders empty lines
  */

package/dist/types/components/tab-bar.d.ts

@@ -1,4 +1,4 @@
-import type { Component } from "../tui";
+import type { Component } from "../tui.js";
 /** Tab definition */
 export interface Tab {
     /** Unique identifier for the tab */

package/dist/types/components/text.d.ts

@@ -1,4 +1,4 @@
-import type { Component } from "../tui";
+import type { Component } from "../tui.js";
 /**
  * Text component - displays multi-line text with word wrapping
  */

package/dist/types/components/truncated-text.d.ts

@@ -1,4 +1,4 @@
-import type { Component } from "../tui";
+import type { Component } from "../tui.js";
 /**
  * Text component that truncates to fit viewport width
  */

package/dist/types/desktop-notify.d.ts

@@ -1,4 +1,4 @@
-import type { TerminalId, TerminalNotification } from "./terminal-capabilities";
+import type { TerminalId, TerminalNotification } from "./terminal-capabilities.js";
 /** Resolved notifier binary used to fan a notification out to D-Bus. */
 export type DesktopNotifierKind = "notify-send" | "gdbus";
 export interface DesktopNotifier {

package/dist/types/editor-component.d.ts

@@ -1,5 +1,5 @@
-import type { AutocompleteProvider } from "./autocomplete";
-import type { Component } from "./tui";
+import type { AutocompleteProvider } from "./autocomplete.js";
+import type { Component } from "./tui.js";
 /**
  * Interface for custom editor components.
  *

package/dist/types/index.d.ts

@@ -1,32 +1,32 @@
-export * from "./autocomplete";
-export * from "./components/box";
-export * from "./components/cancellable-loader";
-export * from "./components/editor";
-export * from "./components/image";
-export * from "./components/input";
-export * from "./components/loader";
-export * from "./components/markdown";
-export * from "./components/scroll-view";
-export * from "./components/select-list";
-export * from "./components/settings-list";
-export * from "./components/spacer";
-export * from "./components/tab-bar";
-export * from "./components/text";
-export * from "./components/truncated-text";
-export * from "./deccara";
-export * from "./desktop-notify";
-export type * from "./editor-component";
-export * from "./fuzzy";
-export * from "./keybindings";
-export * from "./keys";
-export * from "./kitty-graphics";
-export * from "./latex-block";
-export * from "./latex-to-unicode";
-export * from "./mouse";
-export * from "./stdin-buffer";
-export type * from "./symbols";
-export * from "./terminal";
-export * from "./terminal-capabilities";
-export * from "./ttyid";
-export * from "./tui";
-export * from "./utils";
+export * from "./autocomplete.js";
+export * from "./components/box.js";
+export * from "./components/cancellable-loader.js";
+export * from "./components/editor.js";
+export * from "./components/image.js";
+export * from "./components/input.js";
+export * from "./components/loader.js";
+export * from "./components/markdown.js";
+export * from "./components/scroll-view.js";
+export * from "./components/select-list.js";
+export * from "./components/settings-list.js";
+export * from "./components/spacer.js";
+export * from "./components/tab-bar.js";
+export * from "./components/text.js";
+export * from "./components/truncated-text.js";
+export * from "./deccara.js";
+export * from "./desktop-notify.js";
+export type * from "./editor-component.js";
+export * from "./fuzzy.js";
+export * from "./keybindings.js";
+export * from "./keys.js";
+export * from "./kitty-graphics.js";
+export * from "./latex-block.js";
+export * from "./latex-to-unicode.js";
+export * from "./mouse.js";
+export * from "./stdin-buffer.js";
+export type * from "./symbols.js";
+export * from "./terminal.js";
+export * from "./terminal-capabilities.js";
+export * from "./ttyid.js";
+export * from "./tui.js";
+export * from "./utils.js";

package/dist/types/keybindings.d.ts

@@ -1,4 +1,4 @@
-import { type KeyId } from "./keys";
+import { type KeyId } from "./keys.js";
 /**
  * Global keybinding registry.
  * Downstream packages can add keybindings via declaration merging.

package/dist/types/terminal-capabilities.d.ts

@@ -42,6 +42,12 @@ export declare class TerminalInfo {
 export declare function isInsideTmux(env?: NodeJS.ProcessEnv): boolean;
 /** Detect terminal multiplexers where scrollback clearing and height-change redraws are hostile. */
 export declare function isInsideTerminalMultiplexer(env?: NodeJS.ProcessEnv): boolean;
+/**
+ * Whether the agent process is running inside a Zellij session. Read fresh on
+ * each call (like {@link isInsideTmux}) so a session attached/detached mid-run
+ * is observed and tests can toggle `Bun.env.ZELLIJ` per case.
+ */
+export declare function isInsideZellij(env?: NodeJS.ProcessEnv): boolean;
 /**
  * Wrap a control-sequence payload in tmux's DCS passthrough envelope. Each
  * ESC byte inside `payload` is doubled per tmux's escape rules. tmux strips

package/dist/types/terminal.d.ts

@@ -1,4 +1,4 @@
-import { type HangulCompatibilityJamoWidth } from "./utils";
+import { type HangulCompatibilityJamoWidth } from "./utils.js";
 export declare function resolveHangulCompatibilityJamoWidthFromTerminalIdentity(env?: NodeJS.ProcessEnv): HangulCompatibilityJamoWidth;
 /**
  * Split `data` into chunks whose encoded UTF-8 byte length is no greater than

package/dist/types/tui.d.ts

@@ -1,6 +1,6 @@
-import { ImageBudget } from "./components/image";
-import { type Terminal } from "./terminal";
-import { visibleWidth } from "./utils";
+import { ImageBudget } from "./components/image.js";
+import { type Terminal } from "./terminal.js";
+import { visibleWidth } from "./utils.js";
 type InputListenerResult = {
     consume?: boolean;
     data?: string;

package/package.json

@@ -1,7 +1,7 @@
 {
 	"type": "module",
 	"name": "@oh-my-pi/pi-tui",
-	"version": "16.3.6",
+	"version": "16.3.8",
 	"description": "Terminal User Interface library with differential rendering for efficient text-based applications",
 	"homepage": "https://omp.sh",
 	"author": "Can Boluk",
@@ -37,8 +37,8 @@
 		"fmt": "biome format --write ."
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-natives": "16.3.6",
-		"@oh-my-pi/pi-utils": "16.3.6",
+		"@oh-my-pi/pi-natives": "16.3.8",
+		"@oh-my-pi/pi-utils": "16.3.8",
 		"lru-cache": "11.5.1",
 		"marked": "^18.0.5"
 	},

package/src/autocomplete.ts

@@ -182,6 +182,8 @@ export interface SlashCommand {
 	aliases?: string[];
 	description?: string;
 	argumentHint?: string;
+	/** Whether the command consumes argument text after the command name. False means the full input stays normal prompt text once args are present. */
+	allowArgs?: boolean;
 	/** Dynamic display-only description for slash-command autocomplete. Must be synchronous and side-effect free. */
 	getAutocompleteDescription?: () => string | undefined;
 	// Function to get argument completions for this command
@@ -387,44 +389,18 @@ export class CombinedAutocompleteProvider implements AutocompleteProvider {
 		const currentLine = lines[cursorLine] || "";
 		const textBeforeCursor = currentLine.slice(0, cursorCol);
 
-		// Check for @ file reference (fuzzy search) - must be after a delimiter or at start
-		const atPrefix = this.#extractAtPrefix(textBeforeCursor);
-		if (atPrefix) {
-			const { rawPrefix, isQuotedPrefix } = parsePathPrefix(atPrefix);
-			// Recursive fuzzy walks rooted outside the project (e.g. `@../`,
-			// `@~/`, `@/abs`) can be huge — a parent dir full of sibling
-			// projects blows past several seconds of latency. Outside cwd,
-			// fall back to plain prefix listing of the immediate directory
-			// (matches Claude Code's behavior). Inside cwd we keep the
-			// fuzzy-then-prefix flow.
-			if (rawPrefix.length > 0 && this.#isOutsideCwd(rawPrefix)) {
-				const items = await this.#getFileSuggestions(atPrefix);
-				if (items.length === 0) return null;
-				return { items, prefix: atPrefix };
-			}
-			const suggestions =
-				rawPrefix.length > 0
-					? await this.#getFuzzyFileSuggestions(rawPrefix, { isQuotedPrefix })
-					: await this.#getFileSuggestions("@");
-			if (suggestions.length === 0 && rawPrefix.length > 0) {
-				const fallback = await this.#getFileSuggestions(atPrefix);
-				if (fallback.length === 0) return null;
-				return { items: fallback, prefix: atPrefix };
-			}
-			if (suggestions.length === 0) return null;
-
-			return {
-				items: suggestions,
-				prefix: atPrefix,
-			};
-		}
-
 		const leadingSlashStart = findLeadingSlashCommandStart(textBeforeCursor);
 		const trailingSlashStart = findTrailingSlashCommandStart(textBeforeCursor);
 		const hasPromptTextBeforeTrailingSlash =
 			trailingSlashStart !== null &&
 			hasPromptTextBeforeSlash(lines, cursorLine, textBeforeCursor, trailingSlashStart);
-		const slashStart = hasPromptTextBeforeTrailingSlash ? trailingSlashStart : leadingSlashStart;
+		const hasPromptTextBeforeLeadingSlash =
+			leadingSlashStart !== null && hasPromptTextBeforeSlash(lines, cursorLine, textBeforeCursor, leadingSlashStart);
+		const slashStart = hasPromptTextBeforeTrailingSlash
+			? trailingSlashStart
+			: hasPromptTextBeforeLeadingSlash
+				? null
+				: leadingSlashStart;
 		if (slashStart !== null) {
 			const commandText = textBeforeCursor.slice(slashStart);
 			const spaceIndex = commandText.indexOf(" ");
@@ -453,26 +429,62 @@ export class CombinedAutocompleteProvider implements AutocompleteProvider {
 				// A mid-prompt slash token with no matching skill may still be an
 				// absolute path (`see /tmp`); fall through to file-path completion.
 			} else if (!isMidPromptSkillLookup) {
-				// Space found - complete command arguments
+				// Submitted slash commands own their argument text only when the
+				// matched command accepts args. No-arg slash-looking prompts such
+				// as `/settings @file` still fall through to prompt-composer
+				// completions because submit treats them as normal prompt text.
 				const commandName = commandText.slice(1, spaceIndex); // Command without "/"
 				const argumentText = commandText.slice(spaceIndex + 1); // Text after space
 
 				const command = this.#commands.find(cmd => commandMatchesNameOrAlias(cmd, commandName));
-				if (!command || !("getArgumentCompletions" in command) || !command.getArgumentCompletions) {
-					return null; // No argument completion for this command
-				}
+				if (command && (!("allowArgs" in command) || command.allowArgs !== false)) {
+					if (!("getArgumentCompletions" in command) || !command.getArgumentCompletions) {
+						return null; // No argument completion for this command
+					}
+
+					const argumentSuggestions = await command.getArgumentCompletions(argumentText);
+					if (!Array.isArray(argumentSuggestions) || argumentSuggestions.length === 0) {
+						return null;
+					}
 
-				const argumentSuggestions = await command.getArgumentCompletions(argumentText);
-				if (!Array.isArray(argumentSuggestions) || argumentSuggestions.length === 0) {
-					return null;
+					return {
+						items: argumentSuggestions,
+						prefix: argumentText,
+					};
 				}
+			}
+		}
 
-				return {
-					items: argumentSuggestions,
-					prefix: argumentText,
-				};
+		// Check for @ file reference (fuzzy search) - must be after a delimiter or at start
+		const atPrefix = this.#extractAtPrefix(textBeforeCursor);
+		if (atPrefix) {
+			const { rawPrefix, isQuotedPrefix } = parsePathPrefix(atPrefix);
+			// Recursive fuzzy walks rooted outside the project (e.g. `@../`,
+			// `@~/`, `@/abs`) can be huge — a parent dir full of sibling
+			// projects blows past several seconds of latency. Outside cwd,
+			// fall back to plain prefix listing of the immediate directory
+			// (matches Claude Code's behavior). Inside cwd we keep the
+			// fuzzy-then-prefix flow.
+			if (rawPrefix.length > 0 && this.#isOutsideCwd(rawPrefix)) {
+				const items = await this.#getFileSuggestions(atPrefix);
+				if (items.length === 0) return null;
+				return { items, prefix: atPrefix };
 			}
-			if (!isMidPromptSkillLookup) return null;
+			const suggestions =
+				rawPrefix.length > 0
+					? await this.#getFuzzyFileSuggestions(rawPrefix, { isQuotedPrefix })
+					: await this.#getFileSuggestions("@");
+			if (suggestions.length === 0 && rawPrefix.length > 0) {
+				const fallback = await this.#getFileSuggestions(atPrefix);
+				if (fallback.length === 0) return null;
+				return { items: fallback, prefix: atPrefix };
+			}
+			if (suggestions.length === 0) return null;
+
+			return {
+				items: suggestions,
+				prefix: atPrefix,
+			};
 		}
 
 		// Check for file paths - triggered by Tab or if we detect a path pattern

package/src/components/image.ts

@@ -65,6 +65,7 @@ export class ImageBudget {
 	#requestRender: () => void;
 	#nextId = nextImageIdSeed();
 	#keyToId = new Map<string, number>();
+	#idToKey = new Map<number, string>();
 	/** Display-order image ids observed during the in-flight pass. */
 	#passIds: number[] = [];
 	/**
@@ -132,6 +133,7 @@ export class ImageBudget {
 			const id = this.#nextId;
 			this.#nextId = (this.#nextId + 1) & 0xffffff || 1;
 			this.#keyToId.set(key, id);
+			this.#idToKey.set(id, key);
 			return id;
 		}
 		const id = this.#nextId;
@@ -163,11 +165,15 @@ export class ImageBudget {
 	 */
 	observe(imageId: number): boolean {
 		if (this.#stablePass) {
-			return this.#cap > 0 && this.#suppressedIds.has(imageId);
+			const suppressed = this.#cap > 0 && this.#suppressedIds.has(imageId);
+			if (suppressed) this.#forgetKeyForId(imageId);
+			return suppressed;
 		}
 		const index = this.#passIds.length;
 		this.#passIds.push(imageId);
-		return this.#cap > 0 && index < this.#planned;
+		const suppressed = this.#cap > 0 && index < this.#planned;
+		if (suppressed) this.#forgetKeyForId(imageId);
+		return suppressed;
 	}
 
 	/**
@@ -185,6 +191,7 @@ export class ImageBudget {
 				this.#purgeIds.push(id);
 				// d=I frees the data too, so the image must re-transmit if it returns.
 				this.#transmitted.delete(id);
+				this.#forgetKeyForId(id);
 			}
 			this.#onTerminal = this.#planned;
 			this.#applyingReset = false;
@@ -214,6 +221,8 @@ export class ImageBudget {
 		this.#transmitted.clear();
 		this.#purgeIds = [];
 		this.#pendingTransmits = [];
+		this.#keyToId.clear();
+		this.#idToKey.clear();
 		return ids;
 	}
 
@@ -274,6 +283,13 @@ export class ImageBudget {
 		this.#pendingTransmits = [];
 	}
 
+	#forgetKeyForId(id: number): void {
+		const key = this.#idToKey.get(id);
+		if (key === undefined) return;
+		this.#idToKey.delete(id);
+		if (this.#keyToId.get(key) === id) this.#keyToId.delete(key);
+	}
+
 	#reconcile(total: number): void {
 		const desired = this.#cap > 0 ? Math.max(0, total - this.#cap) : 0;
 		if (desired === this.#planned) {

package/src/terminal-capabilities.ts

@@ -122,6 +122,13 @@ export class TerminalInfo {
 			process.stdout.write(`${wrapTmuxPassthrough(formatted)}\x07`);
 			return;
 		}
+		// Zellij drops OSC 9/99 and has no DCS passthrough envelope, but raises its
+		// `[!]` bell flag on a bare BEL — the same backgrounded-pane signal tmux
+		// users get. So follow the (Zellij-swallowed) OSC with a plain BEL.
+		if (this.notifyProtocol !== NotifyProtocol.Bell && isInsideZellij()) {
+			process.stdout.write(`${formatted}\x07`);
+			return;
+		}
 		process.stdout.write(formatted);
 		// VTE-family terminals (Ptyxis, GNOME Terminal, Tilix, …) plus Alacritty
 		// and bare xterm-on-Wayland have no in-band escape that surfaces an
@@ -156,6 +163,15 @@ export function isInsideTerminalMultiplexer(env: NodeJS.ProcessEnv = Bun.env): b
 	return term.startsWith("tmux") || term.startsWith("screen");
 }
 
+/**
+ * Whether the agent process is running inside a Zellij session. Read fresh on
+ * each call (like {@link isInsideTmux}) so a session attached/detached mid-run
+ * is observed and tests can toggle `Bun.env.ZELLIJ` per case.
+ */
+export function isInsideZellij(env: NodeJS.ProcessEnv = Bun.env): boolean {
+	return Boolean(env.ZELLIJ);
+}
+
 /**
  * Wrap a control-sequence payload in tmux's DCS passthrough envelope. Each
  * ESC byte inside `payload` is doubled per tmux's escape rules. tmux strips