import {Console as NodeConsole} from 'console'; // eslint-disable-line node/prefer-global/console import {initSetUp} from './default-util'; import {sentryInit, captureException} from './sentry-util'; import fs from 'fs'; import os from 'os'; import isDev from 'electron-is-dev'; import electron from 'electron'; // This interface adds [key: string]: any so // we can do console[type] later on in the code interface PatchedConsole extends Console { [key: string]: any; } interface LoggerOptions { timestamp?: true | (() => string); file?: string; level?: boolean; logInDevMode?: boolean; } initSetUp(); let app: Electron.App = null; let reportErrors = true; if (process.type === 'renderer') { app = electron.remote.app; // Report Errors to Sentry only if it is enabled in settings // Gets the value of reportErrors from config-util for renderer process // For main process, sentryInit() is handled in index.js const {ipcRenderer} = electron; ipcRenderer.send('error-reporting'); ipcRenderer.on('error-reporting-val', (_event: any, errorReporting: boolean) => { reportErrors = errorReporting; if (reportErrors) { sentryInit(); } }); } else { app = electron.app; } const browserConsole: PatchedConsole = console; const logDir = `${app.getPath('userData')}/Logs`; export default class Logger { [key: string]: any; nodeConsole: PatchedConsole; timestamp?: () => string; level: boolean; logInDevMode: boolean; constructor(options: LoggerOptions = {}) { let { timestamp = true, file = 'console.log', level = true, logInDevMode = false } = options; file = `${logDir}/${file}`; if (timestamp === true) { timestamp = this.getTimestamp; } // Trim log according to type of process if (process.type === 'renderer') { requestIdleCallback(() => this.trimLog(file)); } else { process.nextTick(() => this.trimLog(file)); } const fileStream = fs.createWriteStream(file, {flags: 'a'}); const nodeConsole = new NodeConsole(fileStream); this.nodeConsole = nodeConsole; this.timestamp = timestamp; this.level = level; this.logInDevMode = logInDevMode; this.setUpConsole(); } _log(type: string, ...args: any[]): void { const { nodeConsole, timestamp, level, logInDevMode } = this; let nodeConsoleLog; /* eslint-disable no-fallthrough */ switch (true) { case typeof timestamp === 'function': args.unshift(timestamp() + ' |\t'); case (level): args.unshift(type.toUpperCase() + ' |'); case isDev || logInDevMode: nodeConsoleLog = nodeConsole[type] || nodeConsole.log; nodeConsoleLog.apply(null, args); // eslint-disable-line prefer-spread default: break; } /* eslint-enable no-fallthrough */ browserConsole[type].apply(null, args); } setUpConsole(): void { for (const type of Object.keys(browserConsole)) { this.setupConsoleMethod(type); } } setupConsoleMethod(type: string): void { this[type] = (...args: any[]) => { const log = this._log.bind(this, type, ...args); log(); }; } getTimestamp(): string { const date = new Date(); const timestamp = `${date.getMonth()}/${date.getDate()} ` + `${date.getMinutes()}:${date.getSeconds()}`; return timestamp; } reportSentry(err: string): void { if (reportErrors) { captureException(err); } } trimLog(file: string): void{ fs.readFile(file, 'utf8', (err, data) => { if (err) { throw err; } const MAX_LOG_FILE_LINES = 500; const logs = data.split(os.EOL); const logLength = logs.length - 1; // Keep bottom MAX_LOG_FILE_LINES of each log instance if (logLength > MAX_LOG_FILE_LINES) { const trimmedLogs = logs.slice(logLength - MAX_LOG_FILE_LINES); const toWrite = trimmedLogs.join(os.EOL); fs.writeFileSync(file, toWrite); } }); } }