pi-rtk-optimizer 0.3.3 → 0.5.0

package/src/techniques/test-output.ts

@@ -1,172 +1,172 @@
-import { matchesCommandPatterns } from "./command-detection.js";
-
-interface TestSummary {
-	passed: number;
-	failed: number;
-	skipped: number;
-	failures: string[];
-}
-
-const TEST_COMMAND_PATTERNS = [
-	/^npm\s+test\b/,
-	/^pnpm\s+test\b/,
-	/^yarn\s+test\b/,
-	/^bun\s+test\b/,
-	/^cargo\s+test\b/,
-	/^go\s+test\b/,
-	/^pytest\b/,
-	/^python\s+-m\s+pytest\b/,
-	/^(?:pnpm\s+)?(?:npx\s+)?vitest\b/,
-	/^(?:npx\s+)?jest\b/,
-	/^mocha\b/,
-	/^ava\b/,
-	/^tap\b/,
-] as const;
-
-const TEST_RESULT_PATTERNS = [
-	/test result:\s*(\w+)\.\s*(\d+)\s*passed;\s*(\d+)\s*failed;/,
-	/(\d+)\s*passed(?:,\s*(\d+)\s*failed)?(?:,\s*(\d+)\s*skipped)?/i,
-	/(\d+)\s*pass(?:,\s*(\d+)\s*fail)?(?:,\s*(\d+)\s*skip)?/i,
-	/tests?:\s*(\d+)\s*passed(?:,\s*(\d+)\s*failed)?(?:,\s*(\d+)\s*skipped)?/i,
-];
-
-const FAILURE_START_PATTERNS = [
-	/^FAIL\s+/,
-	/^FAILED\s+/,
-	/^\s*●\s+/,
-	/^\s*✕\s+/,
-	/test\s+\w+\s+\.\.\.\s*FAILED/,
-	/thread\s+'\w+'\s+panicked/,
-];
-
-function isFailureStart(line: string): boolean {
-	return FAILURE_START_PATTERNS.some((pattern) => pattern.test(line));
-}
-
-function extractTestStats(output: string): Partial<TestSummary> {
-	for (const pattern of TEST_RESULT_PATTERNS) {
-		const match = output.match(pattern);
-		if (!match) {
-			continue;
-		}
-		return {
-			passed: Number.parseInt(match[1] ?? "0", 10) || 0,
-			failed: Number.parseInt(match[2] ?? "0", 10) || 0,
-			skipped: Number.parseInt(match[3] ?? "0", 10) || 0,
-		};
-	}
-	return {};
-}
-
-export function isTestCommand(command: string | undefined | null): boolean {
-	return matchesCommandPatterns(command, TEST_COMMAND_PATTERNS);
-}
-
-export function aggregateTestOutput(output: string, command: string | undefined | null): string | null {
-	if (!isTestCommand(command)) {
-		return null;
-	}
-
-	const lines = output.split("\n");
-	const summary: TestSummary = {
-		passed: 0,
-		failed: 0,
-		skipped: 0,
-		failures: [],
-	};
-
-	const stats = extractTestStats(output);
-	summary.passed = stats.passed ?? 0;
-	summary.failed = stats.failed ?? 0;
-	summary.skipped = stats.skipped ?? 0;
-
-	if (summary.passed === 0 && summary.failed === 0) {
-		for (const line of lines) {
-			if (line.match(/\b(ok|PASS|✓|✔)\b/)) {
-				summary.passed++;
-			}
-			if (line.match(/\b(FAIL|fail|✗|✕)\b/)) {
-				summary.failed++;
-			}
-		}
-	}
-
-	if (summary.failed > 0) {
-		let inFailure = false;
-		let currentFailure: string[] = [];
-		let blankCount = 0;
-
-		for (const line of lines) {
-			if (isFailureStart(line)) {
-				if (inFailure && currentFailure.length > 0) {
-					summary.failures.push(currentFailure.join("\n"));
-				}
-				inFailure = true;
-				currentFailure = [line];
-				blankCount = 0;
-				continue;
-			}
-
-			if (!inFailure) {
-				continue;
-			}
-
-			if (line.trim() === "") {
-				blankCount++;
-				if (blankCount >= 2 && currentFailure.length > 3) {
-					summary.failures.push(currentFailure.join("\n"));
-					inFailure = false;
-					currentFailure = [];
-				} else {
-					currentFailure.push(line);
-				}
-				continue;
-			}
-
-			if (line.match(/^\s/) || line.match(/^-/)) {
-				currentFailure.push(line);
-				blankCount = 0;
-				continue;
-			}
-
-			summary.failures.push(currentFailure.join("\n"));
-			inFailure = false;
-			currentFailure = [];
-		}
-
-		if (inFailure && currentFailure.length > 0) {
-			summary.failures.push(currentFailure.join("\n"));
-		}
-	}
-
-	const result: string[] = ["📋 Test Results:"];
-	result.push(`   ✅ ${summary.passed} passed`);
-	if (summary.failed > 0) {
-		result.push(`   ❌ ${summary.failed} failed`);
-	}
-	if (summary.skipped > 0) {
-		result.push(`   ⏭️  ${summary.skipped} skipped`);
-	}
-
-	if (summary.failed > 0 && summary.failures.length > 0) {
-		result.push("\n   Failures:");
-		for (const failure of summary.failures.slice(0, 5)) {
-			const failureLines = failure.split("\n");
-			const firstLine = failureLines[0] ?? "";
-			result.push(`   • ${firstLine.slice(0, 70)}${firstLine.length > 70 ? "..." : ""}`);
-			for (const detailLine of failureLines.slice(1, 4)) {
-				if (detailLine.trim()) {
-					result.push(`     ${detailLine.slice(0, 65)}${detailLine.length > 65 ? "..." : ""}`);
-				}
-			}
-			if (failureLines.length > 4) {
-				result.push(`     ... (${failureLines.length - 4} more lines)`);
-			}
-		}
-		if (summary.failures.length > 5) {
-			result.push(`   ... and ${summary.failures.length - 5} more failures`);
-		}
-	}
-
-	return result.join("\n");
-}
+import { matchesCommandPatterns } from "./command-detection.js";
+
+interface TestSummary {
+	passed: number;
+	failed: number;
+	skipped: number;
+	failures: string[];
+}
+
+const TEST_COMMAND_PATTERNS = [
+	/^npm\s+test\b/,
+	/^pnpm\s+test\b/,
+	/^yarn\s+test\b/,
+	/^bun\s+test\b/,
+	/^cargo\s+test\b/,
+	/^go\s+test\b/,
+	/^pytest\b/,
+	/^python\s+-m\s+pytest\b/,
+	/^(?:pnpm\s+)?(?:npx\s+)?vitest\b/,
+	/^(?:npx\s+)?jest\b/,
+	/^mocha\b/,
+	/^ava\b/,
+	/^tap\b/,
+] as const;
+
+const TEST_RESULT_PATTERNS = [
+	/test result:\s*(\w+)\.\s*(\d+)\s*passed;\s*(\d+)\s*failed;/,
+	/(\d+)\s*passed(?:,\s*(\d+)\s*failed)?(?:,\s*(\d+)\s*skipped)?/i,
+	/(\d+)\s*pass(?:,\s*(\d+)\s*fail)?(?:,\s*(\d+)\s*skip)?/i,
+	/tests?:\s*(\d+)\s*passed(?:,\s*(\d+)\s*failed)?(?:,\s*(\d+)\s*skipped)?/i,
+];
+
+const FAILURE_START_PATTERNS = [
+	/^FAIL\s+/,
+	/^FAILED\s+/,
+	/^\s*●\s+/,
+	/^\s*✕\s+/,
+	/test\s+\w+\s+\.\.\.\s*FAILED/,
+	/thread\s+'\w+'\s+panicked/,
+];
+
+function isFailureStart(line: string): boolean {
+	return FAILURE_START_PATTERNS.some((pattern) => pattern.test(line));
+}
+
+function extractTestStats(output: string): Partial<TestSummary> {
+	for (const pattern of TEST_RESULT_PATTERNS) {
+		const match = output.match(pattern);
+		if (!match) {
+			continue;
+		}
+		return {
+			passed: Number.parseInt(match[1] ?? "0", 10) || 0,
+			failed: Number.parseInt(match[2] ?? "0", 10) || 0,
+			skipped: Number.parseInt(match[3] ?? "0", 10) || 0,
+		};
+	}
+	return {};
+}
+
+export function isTestCommand(command: string | undefined | null): boolean {
+	return matchesCommandPatterns(command, TEST_COMMAND_PATTERNS);
+}
+
+export function aggregateTestOutput(output: string, command: string | undefined | null): string | null {
+	if (!isTestCommand(command)) {
+		return null;
+	}
+
+	const lines = output.split("\n");
+	const summary: TestSummary = {
+		passed: 0,
+		failed: 0,
+		skipped: 0,
+		failures: [],
+	};
+
+	const stats = extractTestStats(output);
+	summary.passed = stats.passed ?? 0;
+	summary.failed = stats.failed ?? 0;
+	summary.skipped = stats.skipped ?? 0;
+
+	if (summary.passed === 0 && summary.failed === 0) {
+		for (const line of lines) {
+			if (line.match(/\b(ok|PASS|✓|✔)\b/)) {
+				summary.passed++;
+			}
+			if (line.match(/\b(FAIL|fail|✗|✕)\b/)) {
+				summary.failed++;
+			}
+		}
+	}
+
+	if (summary.failed > 0) {
+		let inFailure = false;
+		let currentFailure: string[] = [];
+		let blankCount = 0;
+
+		for (const line of lines) {
+			if (isFailureStart(line)) {
+				if (inFailure && currentFailure.length > 0) {
+					summary.failures.push(currentFailure.join("\n"));
+				}
+				inFailure = true;
+				currentFailure = [line];
+				blankCount = 0;
+				continue;
+			}
+
+			if (!inFailure) {
+				continue;
+			}
+
+			if (line.trim() === "") {
+				blankCount++;
+				if (blankCount >= 2 && currentFailure.length > 3) {
+					summary.failures.push(currentFailure.join("\n"));
+					inFailure = false;
+					currentFailure = [];
+				} else {
+					currentFailure.push(line);
+				}
+				continue;
+			}
+
+			if (line.match(/^\s/) || line.match(/^-/)) {
+				currentFailure.push(line);
+				blankCount = 0;
+				continue;
+			}
+
+			summary.failures.push(currentFailure.join("\n"));
+			inFailure = false;
+			currentFailure = [];
+		}
+
+		if (inFailure && currentFailure.length > 0) {
+			summary.failures.push(currentFailure.join("\n"));
+		}
+	}
+
+	const result: string[] = ["Test Results:"];
+	result.push(`   PASS: ${summary.passed} passed`);
+	if (summary.failed > 0) {
+		result.push(`   FAIL: ${summary.failed} failed`);
+	}
+	if (summary.skipped > 0) {
+		result.push(`   SKIP: ${summary.skipped} skipped`);
+	}
+
+	if (summary.failed > 0 && summary.failures.length > 0) {
+		result.push("\n   Failures:");
+		for (const failure of summary.failures.slice(0, 5)) {
+			const failureLines = failure.split("\n");
+			const firstLine = failureLines[0] ?? "";
+			result.push(`   - ${firstLine.slice(0, 70)}${firstLine.length > 70 ? "..." : ""}`);
+			for (const detailLine of failureLines.slice(1, 4)) {
+				if (detailLine.trim()) {
+					result.push(`     ${detailLine.slice(0, 65)}${detailLine.length > 65 ? "..." : ""}`);
+				}
+			}
+			if (failureLines.length > 4) {
+				result.push(`     ... (${failureLines.length - 4} more lines)`);
+			}
+		}
+		if (summary.failures.length > 5) {
+			result.push(`   ... and ${summary.failures.length - 5} more failures`);
+		}
+	}
+
+	return result.join("\n");
+}

package/src/test-helpers.ts

@@ -0,0 +1,10 @@
+import { DEFAULT_RTK_INTEGRATION_CONFIG, type RtkIntegrationConfig } from "./types.ts";
+
+export function runTest(name: string, testFn: () => void): void {
+	testFn();
+	console.log(`[PASS] ${name}`);
+}
+
+export function cloneDefaultConfig(): RtkIntegrationConfig {
+	return structuredClone(DEFAULT_RTK_INTEGRATION_CONFIG);
+}

package/src/tool-execution-sanitizer.ts

@@ -0,0 +1,69 @@
+import { toRecord } from "./record-utils.js";
+import { sanitizeRtkEmojiOutput, stripAnsiFast, stripRtkHookWarnings } from "./techniques/index.js";
+
+interface ToolResultTextBlock {
+	type: string;
+	text?: string;
+	[key: string]: unknown;
+}
+
+function sanitizeStreamingBashText(text: string, command: string | undefined | null): string {
+	let nextText = stripAnsiFast(text);
+
+	const withoutRtkHookWarnings = stripRtkHookWarnings(nextText, command);
+	if (withoutRtkHookWarnings !== null) {
+		nextText = withoutRtkHookWarnings;
+	}
+
+	const withoutRtkEmoji = sanitizeRtkEmojiOutput(nextText, command);
+	if (withoutRtkEmoji !== null) {
+		nextText = withoutRtkEmoji;
+	}
+
+	return nextText;
+}
+
+/**
+ * Sanitizes streamed bash result blocks before the TUI renders them so RTK
+ * self-diagnostics never flash in partial or final tool output.
+ */
+export function sanitizeStreamingBashExecutionResult(
+	result: unknown,
+	command: string | undefined | null,
+): boolean {
+	const resultRecord = toRecord(result);
+	const sourceContent = Array.isArray(resultRecord.content) ? resultRecord.content : null;
+	if (!sourceContent || sourceContent.length === 0) {
+		return false;
+	}
+
+	let changed = false;
+	const nextContent = sourceContent.map((block) => {
+		if (!block || typeof block !== "object" || Array.isArray(block)) {
+			return block;
+		}
+
+		const contentBlock = block as ToolResultTextBlock;
+		if (contentBlock.type !== "text" || typeof contentBlock.text !== "string") {
+			return block;
+		}
+
+		const sanitizedText = sanitizeStreamingBashText(contentBlock.text, command);
+		if (sanitizedText === contentBlock.text) {
+			return block;
+		}
+
+		changed = true;
+		return {
+			...contentBlock,
+			text: sanitizedText,
+		};
+	});
+
+	if (!changed) {
+		return false;
+	}
+
+	resultRecord.content = nextContent;
+	return true;
+}