@effect-tui/react 0.1.3 → 0.1.4

package/src/console/ConsoleCapture.ts

@@ -0,0 +1,272 @@
+// Console capture singleton for TUI debugging
+// Intercepts console.log/info/warn/error/debug and stores entries for display
+
+import { EventEmitter } from "events"
+import { Console } from "node:console"
+import { Writable } from "node:stream"
+import * as util from "node:util"
+
+// ─────────────────────────────────────────────────────────────
+// Types
+// ─────────────────────────────────────────────────────────────
+
+export type LogLevel = "LOG" | "INFO" | "WARN" | "ERROR" | "DEBUG"
+
+export interface SourceLocation {
+	file: string
+	line: number
+	column?: number
+}
+
+export interface LogEntry {
+	timestamp: Date
+	level: LogLevel
+	message: string
+	args: unknown[]
+	location?: SourceLocation
+}
+
+// ─────────────────────────────────────────────────────────────
+// Location capture utilities
+// ─────────────────────────────────────────────────────────────
+
+const cwd = process.cwd()
+
+function isAppPath(file: string): boolean {
+	if (file.includes("/node_modules/")) return false
+	if (file.includes("/.bun/")) return false
+	if (file.includes("/ConsoleCapture")) return false
+	return file.startsWith(cwd)
+}
+
+function captureCallerLocation(): SourceLocation | undefined {
+	const err = new Error()
+	const stack = err.stack
+	if (!stack) return
+
+	const lines = stack.split("\n").slice(1)
+
+	// Skip internal frames
+	const skipPatterns = [/ConsoleCapture/, /captureCallerLocation/, /appendToConsole/, /node:console/, /node:internal/]
+
+	for (const line of lines) {
+		const trimmed = line.trim()
+
+		// Check if line should be skipped
+		if (skipPatterns.some((p) => p.test(trimmed))) continue
+
+		// Parse stack frame: "at functionName (file:line:column)" or "at file:line:column"
+		const re = /at\s+(?:.+?\s+\()?((?:file:\/\/)?[^):]+):(\d+):(\d+)\)?$/
+		const match = re.exec(trimmed)
+
+		if (match) {
+			let file = match[1]!
+			if (file.startsWith("file://")) {
+				file = file.slice(7)
+			}
+
+			// Only return app paths (not node_modules)
+			if (isAppPath(file)) {
+				return {
+					file,
+					line: Number(match[2]),
+					column: Number(match[3]),
+				}
+			}
+		}
+	}
+}
+
+// ─────────────────────────────────────────────────────────────
+// Mock writable stream for console capture
+// ─────────────────────────────────────────────────────────────
+
+class CapturedWritableStream extends Writable {
+	isTTY = true
+	columns = process.stdout.columns || 80
+	rows = process.stdout.rows || 24
+
+	override _write(_chunk: Buffer, _encoding: BufferEncoding, callback: (error?: Error | null) => void): void {
+		// We capture through console method overrides instead
+		callback()
+	}
+
+	getColorDepth(): number {
+		return process.stdout.getColorDepth?.() || 8
+	}
+}
+
+// ─────────────────────────────────────────────────────────────
+// Console capture singleton
+// ─────────────────────────────────────────────────────────────
+
+export interface ConsoleCaptureEvents {
+	entry: (entry: LogEntry) => void
+	error: (entry: LogEntry) => void
+}
+
+export class ConsoleCapture extends EventEmitter {
+	private entries: LogEntry[] = []
+	private originalConsole: typeof console | null = null
+	private active = false
+	private maxEntries: number
+
+	constructor(options?: { maxEntries?: number }) {
+		super()
+		this.maxEntries = options?.maxEntries ?? 1000
+	}
+
+	get isActive(): boolean {
+		return this.active
+	}
+
+	getEntries(): LogEntry[] {
+		return [...this.entries]
+	}
+
+	clear(): void {
+		this.entries = []
+	}
+
+	activate(): void {
+		if (this.active) return
+
+		// Store original console
+		this.originalConsole = globalThis.console
+
+		// Create mock streams for the new console
+		const mockStdout = new CapturedWritableStream()
+		const mockStderr = new CapturedWritableStream()
+
+		// Create a new Console with mock streams
+		// This properly handles util.format for objects, etc.
+		const mockConsole = new Console({
+			stdout: mockStdout,
+			stderr: mockStderr,
+			colorMode: true,
+			inspectOptions: {
+				compact: false,
+				breakLength: 80,
+				depth: 4,
+			},
+		}) as typeof console
+
+		// Copy over all methods from original console that we don't override
+		// This ensures things like timeStamp, time, timeEnd, trace, etc. still work
+		const methodsToCopy = [
+			"assert",
+			"clear",
+			"count",
+			"countReset",
+			"dir",
+			"dirxml",
+			"group",
+			"groupCollapsed",
+			"groupEnd",
+			"table",
+			"time",
+			"timeEnd",
+			"timeLog",
+			"timeStamp",
+			"trace",
+			"profile",
+			"profileEnd",
+		] as const
+
+		for (const method of methodsToCopy) {
+			if (method in this.originalConsole) {
+				;(mockConsole as unknown as Record<string, unknown>)[method] = (
+					this.originalConsole as unknown as Record<string, unknown>
+				)[method]
+			}
+		}
+
+		// Override console methods to capture output
+		globalThis.console = mockConsole
+
+		// Attach our capture handlers
+		console.log = (...args: unknown[]) => this.appendToConsole("LOG", args)
+		console.info = (...args: unknown[]) => this.appendToConsole("INFO", args)
+		console.warn = (...args: unknown[]) => this.appendToConsole("WARN", args)
+		console.error = (...args: unknown[]) => this.appendToConsole("ERROR", args)
+		console.debug = (...args: unknown[]) => this.appendToConsole("DEBUG", args)
+
+		this.active = true
+	}
+
+	deactivate(): void {
+		if (!this.active || !this.originalConsole) return
+
+		globalThis.console = this.originalConsole
+		this.originalConsole = null
+		this.active = false
+	}
+
+	private appendToConsole(level: LogLevel, args: unknown[]): void {
+		const location = captureCallerLocation()
+		const message = util.format(...args)
+
+		const entry: LogEntry = {
+			timestamp: new Date(),
+			level,
+			message,
+			args,
+			location,
+		}
+
+		// Enforce max entries
+		if (this.entries.length >= this.maxEntries) {
+			this.entries.shift()
+		}
+		this.entries.push(entry)
+
+		// Emit events
+		this.emit("entry", entry)
+		// Only emit error event if there are listeners (avoid unhandled error exception)
+		if (level === "ERROR" && this.listenerCount("error") > 0) {
+			this.emit("error", entry)
+		}
+	}
+
+	// Also expose write to original console for debugging the capture itself
+	writeToOriginal(level: LogLevel, ...args: unknown[]): void {
+		if (!this.originalConsole) return
+		switch (level) {
+			case "LOG":
+				this.originalConsole.log(...args)
+				break
+			case "INFO":
+				this.originalConsole.info(...args)
+				break
+			case "WARN":
+				this.originalConsole.warn(...args)
+				break
+			case "ERROR":
+				this.originalConsole.error(...args)
+				break
+			case "DEBUG":
+				this.originalConsole.debug(...args)
+				break
+		}
+	}
+}
+
+// ─────────────────────────────────────────────────────────────
+// Singleton instance
+// ─────────────────────────────────────────────────────────────
+
+let instance: ConsoleCapture | null = null
+
+export function getConsoleCapture(): ConsoleCapture {
+	if (!instance) {
+		instance = new ConsoleCapture()
+	}
+	return instance
+}
+
+// Clean up on process exit
+process.on("exit", () => {
+	if (instance?.isActive) {
+		instance.deactivate()
+	}
+})