pi-squeeze 0.0.0-stage → 0.2.0

package/extensions/core/xcode.ts

@@ -0,0 +1,123 @@
+/**
+ * `xcodebuild` and `swift build` / `swift test`:
+ *
+ * * build steps (`CompileSwift …`, `Ld …`, `CodeSign …` with their indented command lines)
+ *   are counted, not listed; so are progress lines (`[12/40] Compiling …`);
+ * * warnings are grouped by message, one location each, without their source frame;
+ * * errors keep their frame and their notes;
+ * * passing tests (XCTest and Swift Testing) are counted; failures, issues and totals stay;
+ * * the `** BUILD SUCCEEDED **` / `** TEST FAILED **` result stays.
+ */
+import { capLines, cutLine, DEFAULT_LIMITS, type Limits, type Squeezed } from "./elide.ts";
+import { stripAnsi } from "./generic.ts";
+import { renderDiags, type Diag } from "./lint.ts";
+
+const DIAG = /^(.+?):(\d+)(?::(\d+))?: (error|warning|note): (.*)$/;
+/** `CompileSwift normal arm64 …`, `WriteAuxiliaryFile …`: a CamelCase verb, or a one-word one. */
+const ACTION = /^(?:[A-Z][a-z]+[A-Z]\w*|Ld|Copy|Touch|Libtool|Strip|Lipo|Ditto|Validate)(?=[ \\]|$)/;
+const DROP = [
+	// `2026-11-02 01:38:45.358 xcodebuild[40118:44955631] …`: tool chatter.
+	/^\d{4}-\d\d-\d\d [\d:.]+ [\w.-]+\[\d+:\d+\] /,
+	/^(Details|Object|Method|Thread): {1,3}\S/,
+	/^Please file a bug at https:\/\/feedbackassistant/,
+	/^\{ platform:/,
+	/^(Command line invocation|Resolved source packages):$/,
+	/^(Resolve Package Graph|Prepare packages|Testing started)$/,
+	/^Build description (signature|path): /,
+	/^Building for (debugging|production)\.\.\.$/,
+	/^\[Planning deferred tasks\]$/,
+	// SwiftPM progress; the spaces are narrow no-break ones.
+	/^\[\d+\s?\/\s?\d+\]/,
+	/^(Fetching|Fetched|Computing version for|Computed|Creating working copy for|Working copy of|Updating|Updated) /,
+	/^note: (Building targets|Target dependency graph|Run script build phase|Using codesigning|Removed stale)/,
+	/^Test Case '.*' started\.$/,
+	/^Test Suite '.*' (started|passed|failed) at /,
+	/^◇ /,
+	/^↳ (Testing Library Version|Target Platform): /,
+	/^\[#[\w-]+\]: <https?:/,
+];
+/** A source frame line under a diagnostic. */
+const FRAME = /^\s*(\d+ )?\|/;
+
+export function compactXcode(output: string, limits: Limits = DEFAULT_LIMITS): Squeezed | undefined {
+	const lines = stripAnsi(output).replace(/\u200b|XCTestOutputBarrier/g, "").split("\n");
+	while (lines.length && lines.at(-1) === "") lines.pop();
+	if (!lines.some((l) => /^\*\* \w+ (SUCCEEDED|FAILED) \*\*$|^Build complete!|^error: (Build failed|fatalError)|^Test Suite '|^✘ Test run|^✔ Test run/.test(l))) return undefined;
+
+	const actions = new Map<string, number>();
+	const warnings: Diag[] = [];
+	const seen = new Set<string>();
+	const out: string[] = [];
+	let executed: string | undefined;
+	let passed = 0;
+	let indented = false;
+	let last: "error" | "warning" | "other" = "other";
+	let skipNext = false;
+	for (const l of lines) {
+		if (skipNext) {
+			skipNext = false;
+			continue;
+		}
+		// Each suite repeats the totals; the last one (all tests) is the one that counts.
+		if (/^\s+Executed \d+ tests?/.test(l)) {
+			executed = l.trim();
+			continue;
+		}
+		if (indented && /^\s+\S/.test(l)) continue;
+		indented = false;
+		const action = ACTION.exec(l);
+		if (action) {
+			actions.set(action[0], (actions.get(action[0]) ?? 0) + 1);
+			indented = true;
+			continue;
+		}
+		if (/^(Command line invocation|Resolved source packages):$|^error: .* Command line:/.test(l)) indented = true;
+		if (DROP.some((d) => d.test(l))) {
+			indented = true;
+			continue;
+		}
+		if (/^Test Case '.*' passed \(/.test(l) || /^✔ Test /.test(l)) {
+			passed++;
+			continue;
+		}
+		const d = DIAG.exec(l);
+		if (d) {
+			const severity = d[4]!;
+			if (severity === "warning") {
+				const key = `${d[1]}:${d[2]}:${d[3]}:${d[5]}`;
+				if (!seen.has(key)) warnings.push({ file: d[1]!, line: d[2]!, col: d[3], severity, message: d[5]! });
+				seen.add(key);
+				last = "warning";
+				continue;
+			}
+			if (severity === "note" && last !== "error") continue;
+			if (severity === "error") last = "error";
+			out.push(cutLine(l, limits.lineChars));
+			continue;
+		}
+		if (FRAME.test(l) || /^\s+`- /.test(l)) {
+			if (last === "error") out.push(cutLine(l, limits.lineChars));
+			continue;
+		}
+		last = "other";
+		if (l === "Failed frontend command:") skipNext = true;
+		if (l.trim() === "" && (out.length === 0 || out.at(-1) === "")) continue;
+		out.push(cutLine(l.trim() === "" ? "" : l, limits.lineChars));
+	}
+	while (out.at(-1) === "") out.pop();
+	if (executed) out.push(executed);
+
+	const steps = [...actions].sort((a, b) => b[1] - a[1]);
+	const total = steps.reduce((n, [, c]) => n + c, 0);
+	const head: string[] = [];
+	if (total) head.push(`build: ${total} steps (${steps.slice(0, 4).map(([k, v]) => `${v} ${k}`).join(", ")}${steps.length > 4 ? ", …" : ""})`);
+	if (passed) head.push(`tests: ${passed} passed (not listed)`);
+	const warned = warnings.length ? ["", ...renderDiags({ tool: "swift", diags: warnings, notes: [] }, limits).text.split("\n")] : [];
+	const capped = capLines([...head, ...out, ...warned], limits.total);
+	return { text: capped.text, truncated: capped.truncated || warnings.length > 0 };
+}
+
+/** For sniffing: only with xcodebuild's own `** BUILD SUCCEEDED **`-style result line. */
+export function compactXcodeSniffed(output: string, limits: Limits = DEFAULT_LIMITS): Squeezed | undefined {
+	return /^\*\* [A-Z ]+ (SUCCEEDED|FAILED) \*\*$/m.test(output) ? compactXcode(output, limits) : undefined;
+}

package/extensions/squeeze.ts

@@ -0,0 +1,122 @@
+/**
+ * pi-squeeze: Squeezes tool output so the agent's context stays small.
+ *
+ * Pure logic lives in core/ and command/ (tested directly); this file only wires it to pi.
+ * Output is only ever replaced when the result is clearly smaller, and a truncated result
+ * says where the full output was saved. When pi itself cut a long output to its tail, the
+ * full output pi saved is compacted instead (a build's tail alone often misses its errors). Settings persist in <agent dir>/squeeze.json.
+ */
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { completions, parseCommand } from "./command/parse.ts";
+import { compact, KINDS } from "./core/compact.ts";
+import { type Config, DEFAULT_CONFIG, optionsOf, readConfig, saveConfig } from "./core/config.ts";
+import { rewrite } from "./core/rtk.ts";
+import { fileIn, readFullOutput } from "./core/files.ts";
+import { saveFull } from "./core/save.ts";
+import { emptyStats, record, saved, summary, tokens } from "./core/stats.ts";
+
+type StatusUi = { ui: { setStatus(key: string, text: string): void } };
+
+export const settingsText = (c: Config): string =>
+	[c.enabled ? "on" : "off", `rtk ${c.rtk ? "on" : "off"}`, ...KINDS.map((k) => `${k} ${c.kinds[k] ? "on" : "off"}`), `min ${c.minChars} chars`].join(" · ");
+
+export default function (pi: ExtensionAPI) {
+	let config: Config = { ...DEFAULT_CONFIG, kinds: { ...DEFAULT_CONFIG.kinds } };
+	const stats = emptyStats();
+	if (process.env.PI_SQUEEZE === "off") config.enabled = false;
+
+	const showStatus = (ctx: StatusUi) => {
+		const n = tokens(saved(stats));
+		ctx.ui.setStatus("squeeze", config.enabled && config.showStatus && n > 0 ? `squeeze: ~${n >= 1000 ? `${(n / 1000).toFixed(1)}k` : n} tokens saved` : "");
+	};
+
+	pi.on("session_start", async (_event, ctx) => {
+		const { config: loaded, problem } = await readConfig();
+		config = loaded;
+		if (process.env.PI_SQUEEZE === "off") config.enabled = false;
+		if (problem) ctx.ui.notify(`squeeze: ${problem}; using the defaults`, "warning");
+		showStatus(ctx);
+	});
+
+	pi.on("tool_call", async (event) => {
+		if (!config.enabled || !config.rtk || event.toolName !== "bash") return;
+		const input = event.input as { command?: unknown };
+		if (typeof input.command !== "string") return;
+		const rewritten = rewrite(input.command);
+		if (rewritten) input.command = rewritten;
+	});
+
+	pi.on("tool_result", async (event, ctx) => {
+		if (!config.enabled || event.toolName !== "bash") return;
+		const command = (event.input as { command?: unknown }).command;
+		if (typeof command !== "string") return;
+		const blocks = event.content;
+		if (blocks.length !== 1 || blocks[0]!.type !== "text") return;
+		const shown = blocks[0]!.text;
+		// pi cut this one to its tail and saved the whole of it: compact the whole, not the tail.
+		const details = event.details as { truncation?: unknown; fullOutputPath?: unknown } | undefined;
+		const fullPath = details?.truncation ? details.fullOutputPath : undefined;
+		let original = shown;
+		if (details?.truncation) {
+			const full = readFullOutput(fullPath);
+			if (full === undefined) return;
+			// The exit status pi appends to a failed run's text.
+			original = full + (/\n\nCommand exited with code \d+$/.exec(shown)?.[0] ?? "");
+		}
+
+		const r = compact(command, original, { ...optionsOf(config), failed: event.isError, isFile: fileIn(ctx.cwd) });
+		// Never more than pi would have shown.
+		if (!r || r.text.length >= shown.length) return;
+		record(stats, { ...r, before: shown.length });
+		showStatus(ctx);
+		const where = fullPath ?? (r.truncated ? saveFull(original) : undefined);
+		const text = where ? `${r.text}\n[squeeze: output shortened; full output: ${where}]` : r.text;
+		return { content: [{ type: "text", text }] };
+	});
+
+	pi.registerCommand("squeeze", {
+		description: "pi-squeeze: status, on, off, <compactor>|rtk on|off, min <chars>",
+		getArgumentCompletions: (prefix: string) => completions(prefix),
+		handler: async (args, ctx) => {
+			const cmd = parseCommand(args);
+			// The setting applies to this session either way; a failed save only means it is not remembered.
+			const persist = async () => {
+				try {
+					await saveConfig(config);
+				} catch (error) {
+					ctx.ui.notify(`squeeze: could not save the setting (${(error as Error)?.message ?? String(error)}); it applies to this session only`, "warning");
+				}
+				showStatus(ctx);
+			};
+			switch (cmd.kind) {
+				case "status":
+					ctx.ui.notify(`squeeze ${settingsText(config)}\n${summary(stats)}`, "info");
+					return;
+				case "on":
+				case "off":
+					config.enabled = cmd.kind === "on";
+					await persist();
+					ctx.ui.notify(`squeeze ${cmd.kind}`, "info");
+					return;
+				case "rtk":
+					config.rtk = cmd.on;
+					await persist();
+					ctx.ui.notify(`squeeze: rtk ${cmd.on ? "on" : "off"}`, "info");
+					return;
+				case "compactor":
+					config.kinds[cmd.compactor] = cmd.on;
+					await persist();
+					ctx.ui.notify(`squeeze: ${cmd.compactor} ${cmd.on ? "on" : "off"}`, "info");
+					return;
+				case "min":
+					config.minChars = cmd.chars;
+					await persist();
+					ctx.ui.notify(`squeeze: output under ${cmd.chars} characters is left alone`, "info");
+					return;
+				case "invalid":
+					ctx.ui.notify(cmd.message, "warning");
+					return;
+			}
+		},
+	});
+}

package/package.json

@@ -1,6 +1,34 @@
 {
   "name": "pi-squeeze",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "version": "0.2.0",
+  "description": "Shrinks bash output before it reaches the agent's context: native compactors for git, search, listings, tests, builds (cargo, gradle, Maven, javac, Xcode, tsc, docker), Android (adb, lint, detekt), linters, package installs and logs, plus rtk when installed. Full output stays one temp file away.",
+  "keywords": ["pi-package", "pi-extension", "pi", "tokens", "context", "compaction", "rtk"],
+  "license": "MIT",
+  "author": "Sorin Albu-Irimies",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/sorinirimies/pi-squeeze.git"
+  },
+  "homepage": "https://github.com/sorinirimies/pi-squeeze#readme",
+  "files": ["extensions", "README.md", "LICENSE"],
+  "pi": {
+    "extensions": ["./extensions/squeeze.ts"]
+  },
+  "scripts": {
+    "test": "bun test",
+    "typecheck": "tsc --noEmit",
+    "check": "nu scripts/ci/quality_gate.nu"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  },
+  "overrides": {
+    "undici": "8.12.3",
+    "minimatch": "11.0.3"
+  },
+  "devDependencies": {
+    "@earendil-works/pi-coding-agent": "^1.4.0",
+    "@types/bun": "^1.4.4",
+    "typescript": "^7.0.2"
+  }
+}