Disable sql prettifier by default

[github/Chocobozzz/PeerTube.git] / scripts / parse-log.ts

diff --git a/scripts/parse-log.ts b/scripts/parse-log.ts
index e2c42bf4c6ca3c8ec60afb4ec054966ad7c070e1..065fe7e82ff5e7b7c678f3c1400bec09f88c2609 100755 (executable)
--- a/scripts/parse-log.ts
+++ b/scripts/parse-log.ts
@@ -1,48 +1,118 @@
-import { createReadStream } from 'fs'
+import { registerTSPaths } from '../server/helpers/register-ts-paths'
+registerTSPaths()
+
+import * as program from 'commander'
+import { createReadStream, readdir } from 'fs-extra'
 import { join } from 'path'
 import { createInterface } from 'readline'
 import * as winston from 'winston'
-import { CONFIG } from '../server/initializers/constants'
+import { labelFormatter } from '../server/helpers/logger'
+import { CONFIG } from '../server/initializers/config'
+import { mtimeSortFilesDesc } from '../shared/core-utils/logs/logs'
+import { inspect } from 'util'
+
+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)')
+  .parse(process.argv)
+
+const excludedKeys = {
+  level: true,
+  message: true,
+  splat: true,
+  timestamp: true,
+  label: 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
 
-const label = CONFIG.WEBSERVER.HOSTNAME + ':' + CONFIG.WEBSERVER.PORT
+  return `[${info.label}] ${toTimeFormat(info.timestamp)} ${info.level}: ${info.message}${additionalInfos}`
+})
 
-const logger = new winston.Logger({
+const logger = winston.createLogger({
   transports: [
     new winston.transports.Console({
-      level: 'debug',
-      label: label,
-      handleExceptions: true,
-      humanReadableUnhandledException: true,
-      json: false,
-      colorize: true,
-      prettyPrint: true,
-      stderrLevels: []
+      level: program['level'] || 'debug',
+      stderrLevels: [],
+      format: winston.format.combine(
+        winston.format.splat(),
+        labelFormatter(),
+        winston.format.colorize(),
+        loggerFormat
+      )
     })
   ],
   exitOnError: true
 })
 
 const logLevels = {
-  error: logger.error,
-  warn: logger.warn,
-  info: logger.info,
-  debug: logger.debug
+  error: logger.error.bind(logger),
+  warn: logger.warn.bind(logger),
+  info: logger.info.bind(logger),
+  debug: logger.debug.bind(logger)
 }
 
-const path = join(CONFIG.STORAGE.LOG_DIR, 'all-logs.log')
-console.log('Opening %s.', path)
+run()
+  .then(() => process.exit(0))
+  .catch(err => console.error(err))
 
-const rl = createInterface({
-  input: createReadStream(path)
-})
+function run () {
+  return new Promise(async res => {
+    const files = await getFiles()
+
+    for (const file of files) {
+      console.log('Opening %s.', file)
+
+      const stream = createReadStream(file)
+
+      const rl = createInterface({
+        input: stream
+      })
+
+      rl.on('line', line => {
+        try {
+          const log = JSON.parse(line)
+          // Don't know why but loggerFormat does not remove splat key
+          Object.assign(log, { splat: undefined })
 
-rl.on('line', line => {
-  const log = JSON.parse(line)
-  const additionalInfo: any = {}
+          logLevels[log.level](log)
+        } catch (err) {
+          console.error('Cannot parse line.', inspect(line))
+          throw err
+        }
+      })
 
-  Object.keys(log).forEach(logKey => {
-    if (logKey !== 'message' && logKey !== 'level') additionalInfo[logKey] = log[logKey]
+      stream.once('close', () => res())
+    }
   })
+}
 
-  logLevels[log.level](log.message, additionalInfo)
-})
+// 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 (program['files']) return program['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'
+
+  return new Date(timestamp).toISOString()
+}