pi-rtk-optimizer 0.3.3 → 0.5.0

package/src/runtime-guard-test.ts

@@ -1,50 +1,42 @@
-import assert from "node:assert/strict";
-
-import {
-	shouldRequireRtkAvailabilityForCommandHandling,
-	shouldSkipCommandHandlingWhenRtkMissing,
-} from "./runtime-guard.ts";
-import { DEFAULT_RTK_INTEGRATION_CONFIG, type RtkIntegrationConfig, type RuntimeStatus } from "./types.ts";
-
-function runTest(name: string, testFn: () => void): void {
-	testFn();
-	console.log(`[PASS] ${name}`);
-}
-
-function cloneConfig(): RtkIntegrationConfig {
-	return structuredClone(DEFAULT_RTK_INTEGRATION_CONFIG);
-}
-
-function runtimeStatus(rtkAvailable: boolean): RuntimeStatus {
-	return { rtkAvailable };
-}
-
-runTest("rewrite mode still requires RTK availability when guard is enabled", () => {
-	const config = cloneConfig();
-	config.mode = "rewrite";
-	config.guardWhenRtkMissing = true;
-
-	assert.equal(shouldRequireRtkAvailabilityForCommandHandling(config), true);
-	assert.equal(shouldSkipCommandHandlingWhenRtkMissing(config, runtimeStatus(false)), true);
-	assert.equal(shouldSkipCommandHandlingWhenRtkMissing(config, runtimeStatus(true)), false);
-});
-
-runTest("suggest mode does not suppress suggestions when RTK is missing", () => {
-	const config = cloneConfig();
-	config.mode = "suggest";
-	config.guardWhenRtkMissing = true;
-
-	assert.equal(shouldRequireRtkAvailabilityForCommandHandling(config), false);
-	assert.equal(shouldSkipCommandHandlingWhenRtkMissing(config, runtimeStatus(false)), false);
-});
-
-runTest("guard disabled never blocks command handling", () => {
-	const config = cloneConfig();
-	config.mode = "rewrite";
-	config.guardWhenRtkMissing = false;
-
-	assert.equal(shouldRequireRtkAvailabilityForCommandHandling(config), false);
-	assert.equal(shouldSkipCommandHandlingWhenRtkMissing(config, runtimeStatus(false)), false);
-});
-
-console.log("All runtime-guard tests passed.");
+import assert from "node:assert/strict";
+
+import {
+	shouldRequireRtkAvailabilityForCommandHandling,
+	shouldSkipCommandHandlingWhenRtkMissing,
+} from "./runtime-guard.ts";
+import { cloneDefaultConfig, runTest } from "./test-helpers.ts";
+import type { RuntimeStatus } from "./types.ts";
+
+function runtimeStatus(rtkAvailable: boolean): RuntimeStatus {
+	return { rtkAvailable };
+}
+
+runTest("rewrite mode still requires RTK availability when guard is enabled", () => {
+	const config = cloneDefaultConfig();
+	config.mode = "rewrite";
+	config.guardWhenRtkMissing = true;
+
+	assert.equal(shouldRequireRtkAvailabilityForCommandHandling(config), true);
+	assert.equal(shouldSkipCommandHandlingWhenRtkMissing(config, runtimeStatus(false)), true);
+	assert.equal(shouldSkipCommandHandlingWhenRtkMissing(config, runtimeStatus(true)), false);
+});
+
+runTest("suggest mode does not suppress suggestions when RTK is missing", () => {
+	const config = cloneDefaultConfig();
+	config.mode = "suggest";
+	config.guardWhenRtkMissing = true;
+
+	assert.equal(shouldRequireRtkAvailabilityForCommandHandling(config), false);
+	assert.equal(shouldSkipCommandHandlingWhenRtkMissing(config, runtimeStatus(false)), false);
+});
+
+runTest("guard disabled never blocks command handling", () => {
+	const config = cloneDefaultConfig();
+	config.mode = "rewrite";
+	config.guardWhenRtkMissing = false;
+
+	assert.equal(shouldRequireRtkAvailabilityForCommandHandling(config), false);
+	assert.equal(shouldSkipCommandHandlingWhenRtkMissing(config, runtimeStatus(false)), false);
+});
+
+console.log("All runtime-guard tests passed.");

package/src/runtime-guard.ts

@@ -1,14 +1,14 @@
-import type { RtkIntegrationConfig, RuntimeStatus } from "./types.js";
-
-export function shouldRequireRtkAvailabilityForCommandHandling(
-	config: Pick<RtkIntegrationConfig, "mode" | "guardWhenRtkMissing">,
-): boolean {
-	return config.mode === "rewrite" && config.guardWhenRtkMissing;
-}
-
-export function shouldSkipCommandHandlingWhenRtkMissing(
-	config: Pick<RtkIntegrationConfig, "mode" | "guardWhenRtkMissing">,
-	runtimeStatus: Pick<RuntimeStatus, "rtkAvailable">,
-): boolean {
-	return shouldRequireRtkAvailabilityForCommandHandling(config) && !runtimeStatus.rtkAvailable;
-}
+import type { RtkIntegrationConfig, RuntimeStatus } from "./types.js";
+
+export function shouldRequireRtkAvailabilityForCommandHandling(
+	config: Pick<RtkIntegrationConfig, "mode" | "guardWhenRtkMissing">,
+): boolean {
+	return config.mode === "rewrite" && config.guardWhenRtkMissing;
+}
+
+export function shouldSkipCommandHandlingWhenRtkMissing(
+	config: Pick<RtkIntegrationConfig, "mode" | "guardWhenRtkMissing">,
+	runtimeStatus: Pick<RuntimeStatus, "rtkAvailable">,
+): boolean {
+	return shouldRequireRtkAvailabilityForCommandHandling(config) && !runtimeStatus.rtkAvailable;
+}

package/src/techniques/build.ts

@@ -1,155 +1,155 @@
-import { matchesCommandPatterns } from "./command-detection.js";
-
-interface BuildStats {
-	compiled: number;
-	errors: string[][];
-	warnings: string[];
-}
-
-const BUILD_COMMAND_PATTERNS = [
-	/^cargo\s+(build|check)\b/,
-	/^bun\s+build\b/,
-	/^npm\s+run\s+build\b/,
-	/^yarn\s+build\b/,
-	/^pnpm\s+build\b/,
-	/^(?:npx\s+)?tsc\b/,
-	/^make\b/,
-	/^cmake\b/,
-	/^gradle\b/,
-	/^mvn\b/,
-	/^go\s+(build|install)\b/,
-	/^python\s+setup\.py\s+build\b/,
-	/^pip\s+install\b/,
-] as const;
-
-const SKIP_PATTERNS = [
-	/^\s*Compiling\s+/,
-	/^\s*Checking\s+/,
-	/^\s*Downloading\s+/,
-	/^\s*Downloaded\s+/,
-	/^\s*Fetching\s+/,
-	/^\s*Fetched\s+/,
-	/^\s*Updating\s+/,
-	/^\s*Updated\s+/,
-	/^\s*Building\s+/,
-	/^\s*Generated\s+/,
-	/^\s*Creating\s+/,
-	/^\s*Running\s+/,
-];
-
-const ERROR_START_PATTERNS = [/^error\[/, /^error:/, /^\[ERROR\]/, /^FAIL/];
-const WARNING_PATTERNS = [/^warning:/, /^\[WARNING\]/, /^warn:/];
-
-function isSkipLine(line: string): boolean {
-	return SKIP_PATTERNS.some((pattern) => pattern.test(line));
-}
-
-function isErrorStart(line: string): boolean {
-	return ERROR_START_PATTERNS.some((pattern) => pattern.test(line));
-}
-
-function isWarning(line: string): boolean {
-	return WARNING_PATTERNS.some((pattern) => pattern.test(line));
-}
-
-export function isBuildCommand(command: string | undefined | null): boolean {
-	return matchesCommandPatterns(command, BUILD_COMMAND_PATTERNS);
-}
-
-export function filterBuildOutput(output: string, command: string | undefined | null): string | null {
-	if (!isBuildCommand(command)) {
-		return null;
-	}
-
-	const lines = output.split("\n");
-	const stats: BuildStats = {
-		compiled: 0,
-		errors: [],
-		warnings: [],
-	};
-
-	let inErrorBlock = false;
-	let currentError: string[] = [];
-	let blankCount = 0;
-
-	for (const line of lines) {
-		if (line.match(/^\s*(Compiling|Checking|Building)\s+/)) {
-			stats.compiled++;
-			continue;
-		}
-
-		if (isSkipLine(line)) {
-			continue;
-		}
-
-		if (isErrorStart(line)) {
-			if (inErrorBlock && currentError.length > 0) {
-				stats.errors.push([...currentError]);
-			}
-			inErrorBlock = true;
-			currentError = [line];
-			blankCount = 0;
-			continue;
-		}
-
-		if (isWarning(line)) {
-			stats.warnings.push(line);
-			continue;
-		}
-
-		if (!inErrorBlock) {
-			continue;
-		}
-
-		if (line.trim() === "") {
-			blankCount++;
-			if (blankCount >= 2 && currentError.length > 3) {
-				stats.errors.push([...currentError]);
-				inErrorBlock = false;
-				currentError = [];
-			} else {
-				currentError.push(line);
-			}
-			continue;
-		}
-
-		if (line.match(/^\s/) || line.match(/^-->/)) {
-			currentError.push(line);
-			blankCount = 0;
-			continue;
-		}
-
-		stats.errors.push([...currentError]);
-		inErrorBlock = false;
-		currentError = [];
-	}
-
-	if (inErrorBlock && currentError.length > 0) {
-		stats.errors.push(currentError);
-	}
-
-	if (stats.errors.length === 0 && stats.warnings.length === 0) {
-		return `✓ Build successful (${stats.compiled} units compiled)`;
-	}
-
-	const result: string[] = [];
-
-	if (stats.errors.length > 0) {
-		result.push(`❌ ${stats.errors.length} error(s):`);
-		for (const error of stats.errors.slice(0, 5)) {
-			result.push(...error.slice(0, 10));
-			if (error.length > 10) {
-				result.push("  ...");
-			}
-		}
-		if (stats.errors.length > 5) {
-			result.push(`... and ${stats.errors.length - 5} more errors`);
-		}
-	}
-
-	if (stats.warnings.length > 0) {
-		result.push(`\n⚠️  ${stats.warnings.length} warning(s)`);
-	}
-
-	return result.join("\n");
-}
+import { matchesCommandPatterns } from "./command-detection.js";
+
+interface BuildStats {
+	compiled: number;
+	errors: string[][];
+	warnings: string[];
+}
+
+const BUILD_COMMAND_PATTERNS = [
+	/^cargo\s+(build|check)\b/,
+	/^bun\s+build\b/,
+	/^npm\s+run\s+build\b/,
+	/^yarn\s+build\b/,
+	/^pnpm\s+build\b/,
+	/^(?:npx\s+)?tsc\b/,
+	/^make\b/,
+	/^cmake\b/,
+	/^gradle\b/,
+	/^mvn\b/,
+	/^go\s+(build|install)\b/,
+	/^python\s+setup\.py\s+build\b/,
+	/^pip\s+install\b/,
+] as const;
+
+const SKIP_PATTERNS = [
+	/^\s*Compiling\s+/,
+	/^\s*Checking\s+/,
+	/^\s*Downloading\s+/,
+	/^\s*Downloaded\s+/,
+	/^\s*Fetching\s+/,
+	/^\s*Fetched\s+/,
+	/^\s*Updating\s+/,
+	/^\s*Updated\s+/,
+	/^\s*Building\s+/,
+	/^\s*Generated\s+/,
+	/^\s*Creating\s+/,
+	/^\s*Running\s+/,
+];
+
+const ERROR_START_PATTERNS = [/^error\[/, /^error:/, /^\[ERROR\]/, /^FAIL/];
+const WARNING_PATTERNS = [/^warning:/, /^\[WARNING\]/, /^warn:/];
+
+function isSkipLine(line: string): boolean {
+	return SKIP_PATTERNS.some((pattern) => pattern.test(line));
+}
+
+function isErrorStart(line: string): boolean {
+	return ERROR_START_PATTERNS.some((pattern) => pattern.test(line));
+}
+
+function isWarning(line: string): boolean {
+	return WARNING_PATTERNS.some((pattern) => pattern.test(line));
+}
+
+export function isBuildCommand(command: string | undefined | null): boolean {
+	return matchesCommandPatterns(command, BUILD_COMMAND_PATTERNS);
+}
+
+export function filterBuildOutput(output: string, command: string | undefined | null): string | null {
+	if (!isBuildCommand(command)) {
+		return null;
+	}
+
+	const lines = output.split("\n");
+	const stats: BuildStats = {
+		compiled: 0,
+		errors: [],
+		warnings: [],
+	};
+
+	let inErrorBlock = false;
+	let currentError: string[] = [];
+	let blankCount = 0;
+
+	for (const line of lines) {
+		if (line.match(/^\s*(Compiling|Checking|Building)\s+/)) {
+			stats.compiled++;
+			continue;
+		}
+
+		if (isSkipLine(line)) {
+			continue;
+		}
+
+		if (isErrorStart(line)) {
+			if (inErrorBlock && currentError.length > 0) {
+				stats.errors.push([...currentError]);
+			}
+			inErrorBlock = true;
+			currentError = [line];
+			blankCount = 0;
+			continue;
+		}
+
+		if (isWarning(line)) {
+			stats.warnings.push(line);
+			continue;
+		}
+
+		if (!inErrorBlock) {
+			continue;
+		}
+
+		if (line.trim() === "") {
+			blankCount++;
+			if (blankCount >= 2 && currentError.length > 3) {
+				stats.errors.push([...currentError]);
+				inErrorBlock = false;
+				currentError = [];
+			} else {
+				currentError.push(line);
+			}
+			continue;
+		}
+
+		if (line.match(/^\s/) || line.match(/^-->/)) {
+			currentError.push(line);
+			blankCount = 0;
+			continue;
+		}
+
+		stats.errors.push([...currentError]);
+		inErrorBlock = false;
+		currentError = [];
+	}
+
+	if (inErrorBlock && currentError.length > 0) {
+		stats.errors.push(currentError);
+	}
+
+	if (stats.errors.length === 0 && stats.warnings.length === 0) {
+		return `[OK] Build successful (${stats.compiled} units compiled)`;
+	}
+
+	const result: string[] = [];
+
+	if (stats.errors.length > 0) {
+		result.push(`[ERROR] ${stats.errors.length} error(s):`);
+		for (const error of stats.errors.slice(0, 5)) {
+			result.push(...error.slice(0, 10));
+			if (error.length > 10) {
+				result.push("  ...");
+			}
+		}
+		if (stats.errors.length > 5) {
+			result.push(`... and ${stats.errors.length - 5} more errors`);
+		}
+	}
+
+	if (stats.warnings.length > 0) {
+		result.push(`\n[WARN] ${stats.warnings.length} warning(s)`);
+	}
+
+	return result.join("\n");
+}

package/src/techniques/emoji.ts

@@ -0,0 +1,91 @@
+const RTK_COMMAND_PATTERN = /^\s*rtk(?:\.exe)?(?:\s|$)/;
+const RTK_OUTPUT_SIGNATURE_PATTERNS = [
+	/^📂 PATH Variables:/m,
+	/^🔧 Language\/Runtime:/m,
+	/^☁️?\s+Cloud\/Services:/m,
+	/^🛠️?\s+Tools:/m,
+	/^📋 Other:/m,
+	/^📊 Total:/m,
+	/^📊\s+.+\s+→\s+.+/m,
+	/^📌\s+/m,
+	/^✅ Files are identical$/m,
+	/^✅ Staged:/m,
+	/^📝 Modified:/m,
+	/^❓ Untracked:/m,
+	/^⚠️?\s+Conflicts:/m,
+	/^🔍 CI Checks Summary:/m,
+	/^🔍\s+\d+\s+in\s+\d+F:/m,
+	/^--- Changes ---$/m,
+	/^📄\s+.+$/m,
+	/^📁\s+\d+F\s+\d+D:/m,
+	/^☸️?\s+\d+\s+pods:/m,
+	/^📦\s+/m,
+] as const;
+
+const LINE_PREFIX_REPLACEMENTS: Array<{ pattern: RegExp; replacement: string }> = [
+	{ pattern: /^🔍\s+/gm, replacement: "" },
+	{ pattern: /^📄\s+/gm, replacement: "> " },
+	{ pattern: /^📂\s+/gm, replacement: "" },
+	{ pattern: /^🔧\s+/gm, replacement: "" },
+	{ pattern: /^☁️?\s+/gm, replacement: "" },
+	{ pattern: /^🛠️?\s+/gm, replacement: "" },
+	{ pattern: /^📋\s+/gm, replacement: "" },
+	{ pattern: /^📊\s+/gm, replacement: "" },
+	{ pattern: /^📌\s+/gm, replacement: "Branch: " },
+	{ pattern: /^📝\s+/gm, replacement: "" },
+	{ pattern: /^📦\s+/gm, replacement: "" },
+	{ pattern: /^📁\s+/gm, replacement: "" },
+	{ pattern: /^☸️?\s+/gm, replacement: "" },
+] as const;
+
+const INLINE_REPLACEMENTS: Array<{ pattern: RegExp; replacement: string }> = [
+	{ pattern: /✅|✓|✔/g, replacement: "[OK]" },
+	{ pattern: /❌|✗|✕/g, replacement: "[ERROR]" },
+	{ pattern: /⚠️|⚠/g, replacement: "[WARN]" },
+	{ pattern: /❓/g, replacement: "[INFO]" },
+	{ pattern: /⏭️|⏭/g, replacement: "[SKIP]" },
+	{ pattern: /⏳/g, replacement: "Pending" },
+	{ pattern: /⬆️|⬆/g, replacement: "up" },
+	{ pattern: /→/g, replacement: "->" },
+	{ pattern: /•/g, replacement: "-" },
+] as const;
+
+const REMAINING_EMOJI_PATTERN = /\p{Extended_Pictographic}/gu;
+const EMOJI_VARIATION_SELECTOR_PATTERN = /\uFE0F/g;
+const INLINE_LABEL_SPACING_PATTERN = /(\[[A-Z]+\])(\S)/g;
+
+function isRtkCommand(command: string | undefined | null): boolean {
+	return typeof command === "string" && RTK_COMMAND_PATTERN.test(command);
+}
+
+function looksLikeRtkStyledOutput(output: string): boolean {
+	return RTK_OUTPUT_SIGNATURE_PATTERNS.some((pattern) => pattern.test(output));
+}
+
+/**
+ * RTK emits emoji-heavy presentation in several command outputs. Pi should
+ * present tool results as plain text, so normalize RTK output markers before
+ * the agent consumes them. We apply this to explicit `rtk ...` commands and to
+ * recognizable RTK-shaped output that may have been prefixed by another layer.
+ */
+export function sanitizeRtkEmojiOutput(output: string, command: string | undefined | null): string | null {
+	if (!isRtkCommand(command) && !looksLikeRtkStyledOutput(output)) {
+		return null;
+	}
+
+	let nextText = output;
+
+	for (const { pattern, replacement } of LINE_PREFIX_REPLACEMENTS) {
+		nextText = nextText.replace(pattern, replacement);
+	}
+
+	for (const { pattern, replacement } of INLINE_REPLACEMENTS) {
+		nextText = nextText.replace(pattern, replacement);
+	}
+
+	nextText = nextText.replace(REMAINING_EMOJI_PATTERN, "");
+	nextText = nextText.replace(EMOJI_VARIATION_SELECTOR_PATTERN, "");
+	nextText = nextText.replace(INLINE_LABEL_SPACING_PATTERN, "$1 $2");
+
+	return nextText === output ? null : nextText;
+}