@heyhuynhgiabuu/pi-pretty 0.1.4 → 0.1.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@heyhuynhgiabuu/pi-pretty",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "Pretty terminal output for pi — syntax-highlighted file reads, colored bash output, tree-view directory listings, and more.",
   "author": "huynhgiabuu",
   "license": "MIT",

package/src/index.ts

@@ -155,6 +155,51 @@ function lang(fp: string): BundledLanguage | undefined {
 	return EXT_LANG[extname(fp).slice(1).toLowerCase()];
 }
 
+// ---------------------------------------------------------------------------
+// Terminal image rendering (iTerm2 / Kitty / WezTerm inline image protocol)
+// ---------------------------------------------------------------------------
+
+function supportsInlineImages(): boolean {
+	const term = process.env.TERM_PROGRAM ?? "";
+	// iTerm2, WezTerm, Mintty support the iTerm2 protocol
+	if (["iTerm.app", "WezTerm", "mintty"].includes(term)) return true;
+	// Kitty has its own protocol but also supports iTerm2 compat in newer versions
+	if (term === "kitty") return true;
+	// TERM_PROGRAM_VERSION for iTerm2 check
+	if (process.env.LC_TERMINAL === "iTerm2") return true;
+	return false;
+}
+
+/**
+ * Render base64 image inline using iTerm2 inline image protocol.
+ * Protocol: ESC ] 1337 ; File=[args] : base64data BEL
+ * Supported by: iTerm2, WezTerm, Mintty, Kitty (compat mode)
+ */
+function renderInlineImage(
+	base64Data: string,
+	opts: { width?: string; height?: string; preserveAspect?: boolean; name?: string } = {},
+): string {
+	const args: string[] = ["inline=1"];
+	if (opts.width) args.push(`width=${opts.width}`);
+	if (opts.height) args.push(`height=${opts.height}`);
+	if (opts.preserveAspect !== false) args.push("preserveAspectRatio=1");
+	if (opts.name) args.push(`name=${Buffer.from(opts.name).toString("base64")}`);
+	// Calculate size for the size= param
+	const byteSize = Math.ceil(base64Data.length * 3 / 4);
+	args.push(`size=${byteSize}`);
+
+	return `\x1b]1337;File=${args.join(";")}:${base64Data}\x07`;
+}
+
+/**
+ * Get human-readable file size
+ */
+function humanSize(bytes: number): string {
+	if (bytes < 1024) return `${bytes}B`;
+	if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)}KB`;
+	return `${(bytes / (1024 * 1024)).toFixed(1)}MB`;
+}
+
 // ---------------------------------------------------------------------------
 // File-type icons — Nerd Font glyphs (Seti-UI + Devicons, stable in NF v3+)
 //
@@ -576,6 +621,18 @@ export default function piPrettyExtension(pi: any): void {
 			const fp = params.path ?? "";
 			const offset = params.offset ?? 1;
 
+			// Check for image content
+			const imageBlock = result.content?.find((c: any) => c.type === "image");
+			if (imageBlock) {
+				(result as any).details = {
+					_type: "readImage",
+					filePath: fp,
+					data: imageBlock.data,
+					mimeType: imageBlock.mimeType ?? "image/png",
+				};
+				return result;
+			}
+
 			// Extract text content for rendering
 			const textContent = result.content
 				?.filter((c: any) => c.type === "text")
@@ -620,6 +677,36 @@ export default function piPrettyExtension(pi: any): void {
 			}
 
 			const d = result.details;
+
+			// Image rendering
+			if (d?._type === "readImage") {
+				const tw = termW();
+				const out: string[] = [];
+				const fname = basename(d.filePath);
+				const byteSize = Math.ceil((d.data as string).length * 3 / 4);
+				const sizeStr = humanSize(byteSize);
+				const mimeStr = d.mimeType ?? "image";
+
+				out.push(`  ${fileIcon(d.filePath)}${FG_DIM}${mimeStr} · ${sizeStr}${RST}`);
+				out.push(rule(tw));
+
+				if (supportsInlineImages()) {
+					// Max width: use ~80 columns or terminal width, capped for readability
+					const imgWidth = Math.min(tw - 4, 80);
+					out.push(renderInlineImage(d.data, {
+						width: `${imgWidth}`,
+						name: fname,
+					}));
+				} else {
+					// Fallback: show metadata only
+					out.push(`  ${FG_DIM}(Inline image preview requires iTerm2, WezTerm, or Kitty)${RST}`);
+				}
+
+				out.push(rule(tw));
+				text.setText(out.join("\n"));
+				return text;
+			}
+
 			if (d?._type === "readFile" && d.content) {
 				const key = `read:${d.filePath}:${d.offset}:${d.lineCount}:${termW()}`;
 				if (ctx.state._rk !== key) {