forked from FoundKeyGang/FoundKey
143 lines
4.7 KiB
TypeScript
143 lines
4.7 KiB
TypeScript
import * as cluster from 'cluster';
|
|
import * as os from 'os';
|
|
import * as chalk from 'chalk';
|
|
import * as dateformat from 'dateformat';
|
|
import { program } from '../argv';
|
|
import { getRepository } from 'typeorm';
|
|
import { Log } from '../models/entities/log';
|
|
import { genId } from '@/misc/gen-id';
|
|
import config from '@/config';
|
|
|
|
const SyslogPro = require('syslog-pro');
|
|
|
|
type Domain = {
|
|
name: string;
|
|
color?: string;
|
|
};
|
|
|
|
type Level = 'error' | 'success' | 'warning' | 'debug' | 'info';
|
|
|
|
export default class Logger {
|
|
private domain: Domain;
|
|
private parentLogger: Logger | null = null;
|
|
private store: boolean;
|
|
private syslogClient: any | null = null;
|
|
|
|
constructor(domain: string, color?: string, store = true) {
|
|
this.domain = {
|
|
name: domain,
|
|
color: color,
|
|
};
|
|
this.store = store;
|
|
|
|
if (config.syslog) {
|
|
this.syslogClient = new SyslogPro.RFC5424({
|
|
applacationName: 'Misskey',
|
|
timestamp: true,
|
|
encludeStructuredData: true,
|
|
color: true,
|
|
extendedColor: true,
|
|
server: {
|
|
target: config.syslog.host,
|
|
port: config.syslog.port,
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
public createSubLogger(domain: string, color?: string, store = true): Logger {
|
|
const logger = new Logger(domain, color, store);
|
|
logger.parentLogger = this;
|
|
return logger;
|
|
}
|
|
|
|
private log(level: Level, message: string, data?: Record<string, any> | null, important = false, subDomains: Domain[] = [], store = true): void {
|
|
if (program.quiet) return;
|
|
if (!this.store) store = false;
|
|
if (level === 'debug') store = false;
|
|
|
|
if (this.parentLogger) {
|
|
this.parentLogger.log(level, message, data, important, [this.domain].concat(subDomains), store);
|
|
return;
|
|
}
|
|
|
|
const time = dateformat(new Date(), 'HH:MM:ss');
|
|
const worker = cluster.isMaster ? '*' : cluster.worker.id;
|
|
const l =
|
|
level === 'error' ? important ? chalk.bgRed.white('ERR ') : chalk.red('ERR ') :
|
|
level === 'warning' ? chalk.yellow('WARN') :
|
|
level === 'success' ? important ? chalk.bgGreen.white('DONE') : chalk.green('DONE') :
|
|
level === 'debug' ? chalk.gray('VERB') :
|
|
level === 'info' ? chalk.blue('INFO') :
|
|
null;
|
|
const domains = [this.domain].concat(subDomains).map(d => d.color ? chalk.keyword(d.color)(d.name) : chalk.white(d.name));
|
|
const m =
|
|
level === 'error' ? chalk.red(message) :
|
|
level === 'warning' ? chalk.yellow(message) :
|
|
level === 'success' ? chalk.green(message) :
|
|
level === 'debug' ? chalk.gray(message) :
|
|
level === 'info' ? message :
|
|
null;
|
|
|
|
let log = `${l} ${worker}\t[${domains.join(' ')}]\t${m}`;
|
|
if (program.withLogTime) log = chalk.gray(time) + ' ' + log;
|
|
|
|
console.log(important ? chalk.bold(log) : log);
|
|
|
|
if (store) {
|
|
if (this.syslogClient) {
|
|
const send =
|
|
level === 'error' ? this.syslogClient.error :
|
|
level === 'warning' ? this.syslogClient.warning :
|
|
level === 'success' ? this.syslogClient.info :
|
|
level === 'debug' ? this.syslogClient.info :
|
|
level === 'info' ? this.syslogClient.info :
|
|
null as never;
|
|
|
|
send.bind(this.syslogClient)(message).catch(() => {});
|
|
} else {
|
|
const Logs = getRepository(Log);
|
|
Logs.insert({
|
|
id: genId(),
|
|
createdAt: new Date(),
|
|
machine: os.hostname(),
|
|
worker: worker.toString(),
|
|
domain: [this.domain].concat(subDomains).map(d => d.name),
|
|
level: level,
|
|
message: message.substr(0, 1000), // 1024を超えるとログが挿入できずエラーになり無限ループする
|
|
data: data,
|
|
} as Log).catch(() => {});
|
|
}
|
|
}
|
|
}
|
|
|
|
public error(x: string | Error, data?: Record<string, any> | null, important = false): void { // 実行を継続できない状況で使う
|
|
if (x instanceof Error) {
|
|
data = data || {};
|
|
data.e = x;
|
|
this.log('error', x.toString(), data, important);
|
|
} else if (typeof x === 'object') {
|
|
this.log('error', `${(x as any).message || (x as any).name || x}`, data, important);
|
|
} else {
|
|
this.log('error', `${x}`, data, important);
|
|
}
|
|
}
|
|
|
|
public warn(message: string, data?: Record<string, any> | null, important = false): void { // 実行を継続できるが改善すべき状況で使う
|
|
this.log('warning', message, data, important);
|
|
}
|
|
|
|
public succ(message: string, data?: Record<string, any> | null, important = false): void { // 何かに成功した状況で使う
|
|
this.log('success', message, data, important);
|
|
}
|
|
|
|
public debug(message: string, data?: Record<string, any> | null, important = false): void { // デバッグ用に使う(開発者に必要だが利用者に不要な情報)
|
|
if (process.env.NODE_ENV != 'production' || program.verbose) {
|
|
this.log('debug', message, data, important);
|
|
}
|
|
}
|
|
|
|
public info(message: string, data?: Record<string, any> | null, important = false): void { // それ以外
|
|
this.log('info', message, data, important);
|
|
}
|
|
}
|