Khaushik-keygraph
56 lines
1.7 KiB
JavaScript
56 lines
1.7 KiB
JavaScript
import { fs } from 'zx';
|
|
import { path } from 'zx';
|
|
|
|
/**
|
|
* Sets up logging to capture all stdout and stderr to a file
|
|
* @param {string} logFilePath - Path to the log file
|
|
* @returns {Promise<Function>} Cleanup function to restore original streams
|
|
*/
|
|
export async function setupLogging(logFilePath) {
|
|
// Resolve to absolute path
|
|
const absoluteLogPath = path.isAbsolute(logFilePath)
|
|
? logFilePath
|
|
: path.join(process.cwd(), logFilePath);
|
|
|
|
// Ensure the directory exists
|
|
await fs.ensureDir(path.dirname(absoluteLogPath));
|
|
|
|
// Create write stream for the log file
|
|
const logStream = fs.createWriteStream(absoluteLogPath, { flags: 'a' });
|
|
|
|
// Store original stdout/stderr write functions
|
|
const originalStdoutWrite = process.stdout.write.bind(process.stdout);
|
|
const originalStderrWrite = process.stderr.write.bind(process.stderr);
|
|
|
|
// Override stdout
|
|
process.stdout.write = function(chunk, encoding, callback) {
|
|
// Write to both original stdout and log file
|
|
originalStdoutWrite(chunk, encoding, callback);
|
|
logStream.write(chunk, encoding);
|
|
return true;
|
|
};
|
|
|
|
// Override stderr
|
|
process.stderr.write = function(chunk, encoding, callback) {
|
|
// Write to both original stderr and log file
|
|
originalStderrWrite(chunk, encoding, callback);
|
|
logStream.write(chunk, encoding);
|
|
return true;
|
|
};
|
|
|
|
// Return cleanup function
|
|
return async function cleanup() {
|
|
// Restore original streams
|
|
process.stdout.write = originalStdoutWrite;
|
|
process.stderr.write = originalStderrWrite;
|
|
|
|
// Close the log stream
|
|
return new Promise((resolve, reject) => {
|
|
logStream.end((err) => {
|
|
if (err) reject(err);
|
|
else resolve();
|
|
});
|
|
});
|
|
};
|
|
}
|