Update logger
This commit is contained in:
		
							parent
							
								
									dcc32d5581
								
							
						
					
					
						commit
						5b0a227382
					
				
					 3 changed files with 68 additions and 35 deletions
				
			
		
							
								
								
									
										51
									
								
								src/index.ts
									
										
									
									
									
								
							
							
						
						
									
										51
									
								
								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<void> {
 | 
			
		|||
 | 
			
		||||
	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<State> {
 | 
			
		||||
	let warn = false;
 | 
			
		||||
 | 
			
		||||
	log('Info', 'Welcome to Misskey!');
 | 
			
		||||
	log('Info', chalk.bold('Misskey Core <aoi>'));
 | 
			
		||||
	log('Info', 'Initializing...');
 | 
			
		||||
	Logger.info('Welcome to Misskey!');
 | 
			
		||||
	Logger.info(chalk.bold('Misskey Core <aoi>'));
 | 
			
		||||
	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');
 | 
			
		||||
});
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -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`);
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -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);
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue