pi-rtk-optimizer 0.3.2 → 0.4.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/types-shims.d.ts

@@ -1,131 +1,189 @@
-declare module "@mariozechner/pi-tui" {
-	export interface SettingItem {
-		id: string;
-		label: string;
-		description: string;
-		currentValue: string;
-		values: string[];
-	}
-
-	export interface AutocompleteItem {
-		value: string;
-		label: string;
-		description?: string;
-	}
-}
-
-declare module "@mariozechner/pi-coding-agent" {
-	interface UiLike {
-		notify(message: string, level: "info" | "warning" | "error"): void;
-		custom<T>(
-			renderer: (
-				tui: { requestRender(): void },
-				theme: unknown,
-				keybindings: unknown,
-				done: () => void,
-			) => {
-				render(width: number): string[];
-				invalidate?(): void;
-				handleInput(data: string): void;
-			},
-			options?: Record<string, unknown>,
-		): Promise<T>;
-	}
-
-	export interface ExtensionContext {
-		hasUI: boolean;
-		cwd?: string;
-		ui: UiLike;
-	}
-
-	export interface ExtensionCommandContext extends ExtensionContext {}
-
-	export interface ToolResultEvent {
-		toolName: string;
-		input: Record<string, unknown>;
-		content: Array<Record<string, unknown>>;
-		details?: unknown;
-	}
-
-	export interface BashToolCallEvent {
-		toolName: "bash";
-		input: { command: string } & Record<string, unknown>;
-	}
-
-	type MaybePromise<T> = T | Promise<T>;
-
-	export interface ExtensionAPI {
-		exec(
-			command: string,
-			args: string[],
-			options?: { timeout?: number },
-		): Promise<{ code: number; stdout: string; stderr: string }>;
-
-		on(
-			eventName: "tool_call",
-			handler: (
-				event: Record<string, unknown>,
-				ctx: ExtensionContext,
-			) => MaybePromise<Record<string, unknown> | void>,
-		): void;
-
-		on(
-			eventName: "tool_result",
-			handler: (
-				event: ToolResultEvent,
-				ctx: ExtensionContext,
-			) => MaybePromise<Record<string, unknown> | void>,
-		): void;
-
-		on(
-			eventName: "before_agent_start",
-			handler: (
-				event: { systemPrompt: string },
-				ctx: ExtensionContext,
-			) => MaybePromise<{ systemPrompt: string } | Record<string, unknown> | void>,
-		): void;
-
-		on(
-			eventName: string,
-			handler: (event: Record<string, unknown>, ctx: ExtensionContext) => MaybePromise<Record<string, unknown> | void>,
-		): void;
-
-		registerCommand(
-			name: string,
-			definition: {
-				description: string;
-				getArgumentCompletions?: (argumentPrefix: string) => Array<{ value: string; label: string; description?: string }> | null;
-				handler: (args: string, ctx: ExtensionCommandContext) => MaybePromise<void>;
-			},
-		): void;
-	}
-
-	export function isToolCallEventType(
-		toolName: "bash",
-		event: Record<string, unknown>,
-	): event is BashToolCallEvent;
-
-	export function isToolCallEventType(
-		toolName: string,
-		event: Record<string, unknown>,
-	): boolean;
-}
-
-
-declare module "node:os" {
-	export function homedir(): string;
-}
-
-declare module "node:path" {
-	export function join(...segments: string[]): string;
-	export function dirname(path: string): string;
-}
-
-declare module "node:fs" {
-	export function existsSync(path: string): boolean;
-	export function mkdirSync(path: string, options?: { recursive?: boolean }): void;
-	export function readFileSync(path: string, encoding: "utf-8"): string;
-	export function renameSync(oldPath: string, newPath: string): void;
-	export function unlinkSync(path: string): void;
-	export function writeFileSync(path: string, data: string, encoding: "utf-8"): void;
-}
+declare module "@mariozechner/pi-tui" {
+	export interface SettingItem {
+		id: string;
+		label: string;
+		description: string;
+		currentValue: string;
+		values: string[];
+	}
+
+	export interface AutocompleteItem {
+		value: string;
+		label: string;
+		description?: string;
+	}
+
+	export class Box {
+		constructor(...args: unknown[]);
+		addChild(child: unknown): void;
+	}
+
+	export class Container {
+		constructor(...args: unknown[]);
+		addChild(child: unknown): void;
+		render(width: number): string[];
+		invalidate(): void;
+	}
+
+	export class SettingsList {
+		constructor(...args: unknown[]);
+		handleInput(data: string): void;
+		updateValue(id: string, value: string): void;
+	}
+
+	export class Spacer {
+		constructor(...args: unknown[]);
+	}
+
+	export class Text {
+		constructor(...args: unknown[]);
+	}
+
+	export function truncateToWidth(text: string, width: number, suffix?: string, pad?: boolean): string;
+	export function visibleWidth(text: string): number;
+}
+
+declare module "@mariozechner/pi-coding-agent" {
+	interface UiLike {
+		notify(message: string, level: "info" | "warning" | "error"): void;
+		custom<T>(
+			renderer: (
+				tui: { requestRender(): void },
+				theme: Theme,
+				keybindings: unknown,
+				done: () => void,
+			) => {
+				render(width: number): string[];
+				invalidate?(): void;
+				handleInput(data: string): void;
+			},
+			options?: Record<string, unknown>,
+		): Promise<T>;
+	}
+
+	export interface ExtensionContext {
+		hasUI: boolean;
+		cwd?: string;
+		ui: UiLike;
+	}
+
+	export interface ExtensionCommandContext extends ExtensionContext {}
+
+	export interface ToolResultEvent {
+		toolName: string;
+		input: Record<string, unknown>;
+		content: Array<Record<string, unknown>>;
+		details?: unknown;
+	}
+
+	export interface BashToolCallEvent {
+		toolName: "bash";
+		input: { command: string } & Record<string, unknown>;
+	}
+
+	type MaybePromise<T> = T | Promise<T>;
+
+	export interface Theme {
+		fg(color: string, text: string): string;
+		bold(text: string): string;
+		getFgAnsi?(name: string): string;
+	}
+
+	export function getSettingsListTheme(): unknown;
+
+	export interface ExtensionAPI {
+		exec(
+			command: string,
+			args: string[],
+			options?: { timeout?: number },
+		): Promise<{ code: number; stdout: string; stderr: string }>;
+
+		on(
+			eventName: "tool_call",
+			handler: (
+				event: Record<string, unknown>,
+				ctx: ExtensionContext,
+			) => MaybePromise<Record<string, unknown> | void>,
+		): void;
+
+		on(
+			eventName: "tool_result",
+			handler: (
+				event: ToolResultEvent,
+				ctx: ExtensionContext,
+			) => MaybePromise<Record<string, unknown> | void>,
+		): void;
+
+		on(
+			eventName: "before_agent_start",
+			handler: (
+				event: { systemPrompt: string },
+				ctx: ExtensionContext,
+			) => MaybePromise<{ systemPrompt: string } | Record<string, unknown> | void>,
+		): void;
+
+		on(
+			eventName: string,
+			handler: (event: Record<string, unknown>, ctx: ExtensionContext) => MaybePromise<Record<string, unknown> | void>,
+		): void;
+
+		registerCommand(
+			name: string,
+			definition: {
+				description: string;
+				getArgumentCompletions?: (argumentPrefix: string) => Array<{ value: string; label: string; description?: string }> | null;
+				handler: (args: string, ctx: ExtensionCommandContext) => MaybePromise<void>;
+			},
+		): void;
+	}
+
+	export function isToolCallEventType(
+		toolName: "bash",
+		event: Record<string, unknown>,
+	): event is BashToolCallEvent;
+
+	export function isToolCallEventType(
+		toolName: string,
+		event: Record<string, unknown>,
+	): boolean;
+}
+
+
+declare module "node:assert/strict" {
+	const assert: {
+		equal(actual: unknown, expected: unknown, message?: string): void;
+		deepEqual(actual: unknown, expected: unknown, message?: string): void;
+		ok(value: unknown, message?: string): void;
+	};
+
+	export default assert;
+}
+
+declare module "bun:test" {
+	export const mock: {
+		module(specifier: string, factory: () => Record<string, unknown>): void;
+	};
+}
+
+declare const process: {
+	platform: string;
+	env: Record<string, string | undefined>;
+};
+
+declare module "node:os" {
+	export function homedir(): string;
+}
+
+declare module "node:path" {
+	export function join(...segments: string[]): string;
+	export function dirname(path: string): string;
+}
+
+declare module "node:fs" {
+	export function existsSync(path: string): boolean;
+	export function mkdirSync(path: string, options?: { recursive?: boolean }): void;
+	export function readFileSync(path: string, encoding: "utf-8"): string;
+	export function renameSync(oldPath: string, newPath: string): void;
+	export function unlinkSync(path: string): void;
+	export function writeFileSync(path: string, data: string, encoding: "utf-8"): void;
+}