import type { ResearchProgress } from './deep-research.js';
import { writeFileSync } from 'fs';
import { generateProgressBar, TERMINAL_CONTROLS } from './terminal-utils.js';
export class OutputManager {
private progressLines = 4;
private progressArea: string[] = [];
private initialized = false;
private lastLogMessage = ''; // Store the last log message
private logQueue: string[] = [];
private logTimer?: NodeJS.Timeout;
constructor() {
// Initialize terminal
if (process.stdout.isTTY) {
process.stdout.write('\n'.repeat(this.progressLines));
this.initialized = true;
} else {
this.initialized = false;
console.warn('Not running in a TTY environment. Progress updates will be disabled.');
}
}
private formatLogEntry(message: string, metadata?: Record<string, unknown>): string {
const timestamp = new Date().toISOString();
const metaStr = metadata ? JSON.stringify(metadata) : '';
return `[${timestamp}] ${message} ${metaStr}`;
}
log(message: string, metadata?: Record<string, unknown>) {
this.enqueueLog(this.formatLogEntry(message, metadata));
this.scheduleFlush();
}
private enqueueLog(entry: string) {
this.logQueue.push(entry);
if (this.logQueue.length > 100) {
this.flushLogs(); // Prevent memory leaks
}
}
private scheduleFlush() {
if (!this.logTimer) {
this.logTimer = setTimeout(() => this.flushLogs(), 50);
}
}
private flushLogs() {
try {
process.stdout.write(TERMINAL_CONTROLS.savePos);
process.stdout.write(this.logQueue.join('\n') + '\n');
this.logQueue = [];
} finally {
if (this.logTimer) {
clearTimeout(this.logTimer);
}
process.stdout.write(TERMINAL_CONTROLS.restorePos);
}
}
static logCacheEviction(value: unknown) {
console.log(`Cache evicted: ${JSON.stringify(value)}`);
}
updateProgress(progress: ResearchProgress) {
if (!this.initialized) {
return;
}
this.progressArea = [
`Depth: [${this.getProgressBar(progress.totalDepth - progress.currentDepth, progress.totalDepth)}] ${Math.round((progress.totalDepth - progress.currentDepth) / progress.totalDepth * 100)}%`,
`Breadth: [${this.getProgressBar(progress.totalBreadth - progress.currentBreadth, progress.totalBreadth)}] ${Math.round((progress.totalBreadth - progress.currentBreadth) / progress.totalBreadth * 100)}%`,
`Queries: [${this.getProgressBar(progress.completedQueries, progress.totalQueries)}] ${Math.round(progress.completedQueries / progress.totalQueries * 100)}%`,
progress.currentQuery ? `Current: ${progress.currentQuery}` : ''
];
this.drawProgress();
}
private getProgressBar(value: number, total: number): string {
const width = process.stdout.columns ? Math.min(30, process.stdout.columns - 20) : 30;
const filled = Math.round((width * value) / total);
return '█'.repeat(filled) + ' '.repeat(width - filled);
}
private drawProgress() {
if (!this.initialized || this.progressArea.length === 0) {
return;
}
// Save cursor position
process.stdout.write('\x1b[s');
// Move cursor to the beginning of the progress area
const terminalHeight = process.stdout.rows || 24;
process.stdout.write(`\x1B[${terminalHeight - this.progressLines};1H`);
// Clear the progress area
for (let i = 0; i < this.progressLines; i++) {
process.stdout.write('\x1B[2K\r'); // Clear the entire line
if (i < this.progressLines - 1) {
process.stdout.write('\x1B[1B'); // Move to the next line
}
}
// Move cursor back to the beginning of the progress area
process.stdout.write(`\x1B[${terminalHeight - this.progressLines};1H`);
// Draw progress bars
process.stdout.write(this.progressArea.join('\n'));
// Restore cursor position
process.stdout.write('\x1b[u');
}
saveResearchReport(reportContent: string) {
const filename = 'output.md';
try {
writeFileSync(filename, reportContent);
this.log(`Final report saved to ${filename}`);
} catch (error) {
this.log(`Error saving report to ${filename}:`, { error: error instanceof Error ? error.message : String(error) });
}
}
}
