diff --git a/src/index.ts b/src/index.ts index 0e5a49c78..eccc9f586 100644 --- a/src/index.ts +++ b/src/index.ts @@ -11,7 +11,7 @@ import * as fs from 'fs'; import * as os from 'os'; import * as cluster from 'cluster'; const prominence = require('prominence'); -import { log } from './utils/logger'; +import Logger from './utils/logger'; import * as chalk from 'chalk'; const git = require('git-last-commit'); const portUsed = require('tcp-port-used'); @@ -72,14 +72,14 @@ async function master(): Promise { switch (state) { case State.failed: - log('Error', chalk.red('Fatal error occurred :(')); + Logger.error(chalk.red('Fatal error occurred :(')); process.exit(); return; case State.warn: - log('Warn', chalk.yellow('Some problem(s) :|')); + Logger.warn(chalk.yellow('Some problem(s) :|')); break; case State.success: - log('Info', chalk.green('OK :)')); + Logger.info(chalk.green('OK :)')); break; } @@ -132,59 +132,64 @@ function worker(): void { async function init(): Promise { let warn = false; - log('Info', 'Welcome to Misskey!'); - log('Info', chalk.bold('Misskey Core ')); - log('Info', 'Initializing...'); + Logger.info('Welcome to Misskey!'); + Logger.info(chalk.bold('Misskey Core ')); + Logger.info('Initializing...'); // Get commit info + let lastCommitLogger = new Logger('LastCommit'); try { const commit = await prominence(git).getLastCommit(); const shortHash: string = commit.shortHash; const hash: string = commit.hash; const commitDate = new Date(parseInt(commit.committedOn, 10) * 1000).toLocaleDateString('ja-JP'); const commitTime = new Date(parseInt(commit.committedOn, 10) * 1000).toLocaleTimeString('ja-JP'); - log('Info', `${shortHash}${chalk.gray(hash.substr(shortHash.length))}`, 'LastCommit'); - log('Info', `${commit.subject} ${chalk.green(`(${commitDate} ${commitTime})`)} ${chalk.blue(`<${commit.author.name}>`)}`, 'LastCommit'); + lastCommitLogger.info(`${shortHash}${chalk.gray(hash.substr(shortHash.length))}`); + lastCommitLogger.info(`${commit.subject} ${chalk.green(`(${commitDate} ${commitTime})`)} ${chalk.blue(`<${commit.author.name}>`)}`); } catch (e) { - log('Info', `No commit information found`, 'LastCommit'); + lastCommitLogger.info('No commit information found') } - log('Info', typeof env == 'undefined' ? 'NODE_ENV is not set' : `NODE_ENV: ${env}`, 'Env'); + let envLogger = new Logger('Env'); + envLogger.info(typeof env == 'undefined' ? 'NODE_ENV is not set' : `NODE_ENV: ${env}`); if (IS_DEBUG) { - log('Warn', 'The environment is not in production mode', 'Env'); - log('Warn', 'Do not use for production purpose', 'Env'); + envLogger.warn('The environment is not in production mode'); + envLogger.warn('Do not use for production purpose'); } // Get machine info const totalmem = (os.totalmem() / 1024 / 1024 / 1024).toFixed(1); const freemem = (os.freemem() / 1024 / 1024 / 1024).toFixed(1); - log('Info', `${os.hostname()}`, 'Machine'); - log('Info', `CPU: ${os.cpus().length}core`, 'Machine'); - log('Info', `MEM: ${totalmem}GB (available: ${freemem}GB)`, 'Machine'); + let machineLogger = new Logger('Machine'); + machineLogger.info(os.hostname()); + machineLogger.info(`CPU: ${os.cpus().length}core`); + machineLogger.info(`MEM: ${totalmem}GB (available: ${freemem}GB)`); if (!fs.existsSync(`${__dirname}/../.config/config.yml`)) { - log('Error', 'Configuration not found'); + Logger.error('Configuration not found'); return State.failed; } - log('Info', 'Successfully loaded', 'Config'); - log('Info', `maintainer: ${config.maintainer}`, 'Config'); + let configLogger = new Logger('Config'); + configLogger.info('Successfully loaded'); + configLogger.info(`maintainer: ${config.maintainer}`); checkDependencies(); // Check if a port is being used if (await portUsed.check(config.port)) { - log('Error', `Port: ${config.port} is already used!`); + Logger.error(`Port: ${config.port} is already used!`); return State.failed; } // Try to connect to MongoDB + let mongoDBLogger = new Logger('MongoDB'); try { const db = await initdb(config); - log('Info', 'Successfully connected', 'MongoDB'); + mongoDBLogger.info('Successfully connected'); db.close(); } catch (e) { - log('Error', `${e}`, 'MongoDB'); + mongoDBLogger.error(`${e}`); return State.failed; } @@ -218,5 +223,5 @@ function spawn(callback: any): void { // Dying away... process.on('exit', () => { - log('Info', 'Misskey is going down'); + Logger.info('Misskey is going down'); }); diff --git a/src/utils/check-dependencies.ts b/src/utils/check-dependencies.ts index 77665e5ac..2378041e2 100644 --- a/src/utils/check-dependencies.ts +++ b/src/utils/check-dependencies.ts @@ -1,4 +1,4 @@ -import { log } from './logger'; +import Logger from './logger'; import { exec } from 'shelljs'; export default function(): void { @@ -14,9 +14,10 @@ function checkDependency(serviceName: string, command: string, transform: (x: st notFound: 127 }; const x = exec(command, { silent: true }) as any; + let depsLogger = new Logger('Deps'); if (x.code === code.success) { - log('Info', `${serviceName} ${transform(x.stdout)} found`, 'Deps'); + depsLogger.info(`${serviceName} ${transform(x.stdout)} found`); } else if (x.code === code.notFound) { - log('Warn', `${serviceName} not found`, 'Deps'); + depsLogger.warn(`${serviceName} not found`); } } diff --git a/src/utils/logger.ts b/src/utils/logger.ts index 495035d47..ebfa3c34f 100644 --- a/src/utils/logger.ts +++ b/src/utils/logger.ts @@ -10,17 +10,44 @@ function toLevelColor(level: LogLevel): chalk.ChalkStyle { } } -export function log(message: string): void; -export function log(level: LogLevel, message: string): void; -export function log(level: LogLevel, message: string, domain: string): void; -export function log(x: string | LogLevel, message?: string, domain?: string): void { - const level = typeof message == 'undefined' ? 'Info' : x as LogLevel; - message = typeof message == 'undefined' ? x : message; - if (typeof domain == 'string') { - log(level, `[${domain}] ${message}`); - } else { +export default class Logger { + domain: string; + + static log(level: LogLevel, message: string): void { let color = toLevelColor(level); let time = (new Date()).toLocaleTimeString('ja-JP'); console.log(`[${time} ${color.bold(level.toUpperCase())}]: ${message}`); } + + static error(message: string): void { + Logger.log('Error', message); + } + + static warn(message: string): void { + Logger.log('Warn', message); + } + + static info(message: string): void { + Logger.log('Info', message); + } + + constructor(domain: string) { + this.domain = domain; + } + + log(level: LogLevel, message: string): void { + Logger.log(level, `[${this.domain}] ${message}`); + } + + error(message: string): void { + this.log('Error', message); + } + + warn(message: string): void { + this.log('Warn', message); + } + + info(message: string): void { + this.log('Info', message); + } }