PeerTube/server/scripts/parse-log.ts
2024-02-22 10:32:28 +01:00

162 lines
4.1 KiB
TypeScript
Executable File

import { program } from 'commander'
import { createReadStream } from 'fs'
import { readdir } from 'fs/promises'
import { join } from 'path'
import { stdin } from 'process'
import { createInterface } from 'readline'
import { format as sqlFormat } from 'sql-formatter'
import { inspect } from 'util'
import * as winston from 'winston'
import { labelFormatter, mtimeSortFilesDesc } from '@server/helpers/logger.js'
import { CONFIG } from '@server/initializers/config.js'
program
.option('-l, --level [level]', 'Level log (debug/info/warn/error)')
.option('-f, --files [file...]', 'Files to parse. If not provided, the script will parse the latest log file from config)')
.option('-t, --tags [tags...]', 'Display only lines with these tags')
.option('-nt, --not-tags [tags...]', 'Do not display lines containing these tags')
.parse(process.argv)
const options = program.opts()
const excludedKeys = {
level: true,
message: true,
splat: true,
timestamp: true,
tags: true,
label: true,
sql: true
}
function keysExcluder (key, value) {
return excludedKeys[key] === true ? undefined : value
}
const loggerFormat = winston.format.printf((info) => {
let additionalInfos = JSON.stringify(info, keysExcluder, 2)
if (additionalInfos === '{}') additionalInfos = ''
else additionalInfos = ' ' + additionalInfos
if (info.sql) {
if (CONFIG.LOG.PRETTIFY_SQL) {
additionalInfos += '\n' + sqlFormat(info.sql, {
language: 'sql',
tabWidth: 2
})
} else {
additionalInfos += ' - ' + info.sql
}
}
return `[${info.label}] ${toTimeFormat(info.timestamp)} ${info.level}: ${info.message}${additionalInfos}`
})
const logger = winston.createLogger({
transports: [
new winston.transports.Console({
level: options.level || 'debug',
stderrLevels: [],
format: winston.format.combine(
winston.format.splat(),
labelFormatter(),
winston.format.colorize(),
loggerFormat
)
})
],
exitOnError: true
})
const logLevels = {
error: logger.error.bind(logger),
warn: logger.warn.bind(logger),
info: logger.info.bind(logger),
debug: logger.debug.bind(logger)
}
run()
.then(() => process.exit(0))
.catch(err => console.error(err))
async function run () {
const files = await getFiles()
for (const file of files) {
if (file === 'peertube-audit.log') continue
await readFile(file)
}
}
function readFile (file: string) {
console.log('Opening %s.', file)
const stream = file === '-' ? stdin : createReadStream(file)
const rl = createInterface({
input: stream
})
return new Promise<void>(res => {
rl.on('line', line => {
try {
const log = JSON.parse(line)
if (options.tags && !containsTags(log.tags, options.tags)) {
return
}
if (options.notTags && containsTags(log.tags, options.notTags)) {
return
}
// Don't know why but loggerFormat does not remove splat key
Object.assign(log, { splat: undefined })
logLevels[log.level](log)
} catch (err) {
console.error('Cannot parse line.', inspect(line))
throw err
}
})
stream.once('end', () => res())
})
}
// Thanks: https://stackoverflow.com/a/37014317
async function getNewestFile (files: string[], basePath: string) {
const sorted = await mtimeSortFilesDesc(files, basePath)
return (sorted.length > 0) ? sorted[0].file : ''
}
async function getFiles () {
if (options.files) return options.files
const logFiles = await readdir(CONFIG.STORAGE.LOG_DIR)
const filename = await getNewestFile(logFiles, CONFIG.STORAGE.LOG_DIR)
return [ join(CONFIG.STORAGE.LOG_DIR, filename) ]
}
function toTimeFormat (time: string) {
const timestamp = Date.parse(time)
if (isNaN(timestamp) === true) return 'Unknown date'
const d = new Date(timestamp)
return d.toLocaleString() + `.${d.getMilliseconds()}`
}
function containsTags (loggerTags: string[], optionsTags: string[]) {
if (!loggerTags) return false
for (const lt of loggerTags) {
for (const ot of optionsTags) {
if (lt === ot) return true
}
}
return false
}