mirror of https://github.com/Chocobozzz/PeerTube
				
				
				
			
		
			
				
	
	
		
			132 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			132 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			TypeScript
		
	
	
| // Thanks http://tostring.it/2014/06/23/advanced-logging-with-nodejs/
 | |
| import { mkdirpSync } from 'fs-extra'
 | |
| import * as path from 'path'
 | |
| import * as winston from 'winston'
 | |
| import { FileTransportOptions } from 'winston/lib/winston/transports'
 | |
| import { CONFIG } from '../initializers/config'
 | |
| import { omit } from 'lodash'
 | |
| import { LOG_FILENAME } from '@server/initializers/constants'
 | |
| 
 | |
| const label = CONFIG.WEBSERVER.HOSTNAME + ':' + CONFIG.WEBSERVER.PORT
 | |
| 
 | |
| // Create the directory if it does not exist
 | |
| // FIXME: use async
 | |
| mkdirpSync(CONFIG.STORAGE.LOG_DIR)
 | |
| 
 | |
| function getLoggerReplacer () {
 | |
|   const seen = new WeakSet()
 | |
| 
 | |
|   // Thanks: https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Errors/Cyclic_object_value#Examples
 | |
|   return (key: string, value: any) => {
 | |
|     if (typeof value === 'object' && value !== null) {
 | |
|       if (seen.has(value)) return
 | |
| 
 | |
|       seen.add(value)
 | |
|     }
 | |
| 
 | |
|     if (value instanceof Error) {
 | |
|       const error = {}
 | |
| 
 | |
|       Object.getOwnPropertyNames(value).forEach(key => error[ key ] = value[ key ])
 | |
| 
 | |
|       return error
 | |
|     }
 | |
| 
 | |
|     return value
 | |
|   }
 | |
| }
 | |
| 
 | |
| const consoleLoggerFormat = winston.format.printf(info => {
 | |
|   const obj = omit(info, 'label', 'timestamp', 'level', 'message')
 | |
| 
 | |
|   let additionalInfos = JSON.stringify(obj, getLoggerReplacer(), 2)
 | |
| 
 | |
|   if (additionalInfos === undefined || additionalInfos === '{}') additionalInfos = ''
 | |
|   else additionalInfos = ' ' + additionalInfos
 | |
| 
 | |
|   return `[${info.label}] ${info.timestamp} ${info.level}: ${info.message}${additionalInfos}`
 | |
| })
 | |
| 
 | |
| const jsonLoggerFormat = winston.format.printf(info => {
 | |
|   return JSON.stringify(info, getLoggerReplacer())
 | |
| })
 | |
| 
 | |
| const timestampFormatter = winston.format.timestamp({
 | |
|   format: 'YYYY-MM-DD HH:mm:ss.SSS'
 | |
| })
 | |
| const labelFormatter = winston.format.label({
 | |
|   label
 | |
| })
 | |
| 
 | |
| const fileLoggerOptions: FileTransportOptions = {
 | |
|   filename: path.join(CONFIG.STORAGE.LOG_DIR, LOG_FILENAME),
 | |
|   handleExceptions: true,
 | |
|   format: winston.format.combine(
 | |
|     winston.format.timestamp(),
 | |
|     jsonLoggerFormat
 | |
|   )
 | |
| }
 | |
| 
 | |
| if (CONFIG.LOG.ROTATION.ENABLED) {
 | |
|   fileLoggerOptions.maxsize = CONFIG.LOG.ROTATION.MAX_FILE_SIZE
 | |
|   fileLoggerOptions.maxFiles = CONFIG.LOG.ROTATION.MAX_FILES
 | |
| }
 | |
| 
 | |
| const logger = winston.createLogger({
 | |
|   level: CONFIG.LOG.LEVEL,
 | |
|   format: winston.format.combine(
 | |
|     labelFormatter,
 | |
|     winston.format.splat()
 | |
|   ),
 | |
|   transports: [
 | |
|     new winston.transports.File(fileLoggerOptions),
 | |
|     new winston.transports.Console({
 | |
|       handleExceptions: true,
 | |
|       format: winston.format.combine(
 | |
|         timestampFormatter,
 | |
|         winston.format.colorize(),
 | |
|         consoleLoggerFormat
 | |
|       )
 | |
|     })
 | |
|   ],
 | |
|   exitOnError: true
 | |
| })
 | |
| 
 | |
| function bunyanLogFactory (level: string) {
 | |
|   return function () {
 | |
|     let meta = null
 | |
|     let args: any[] = []
 | |
|     args.concat(arguments)
 | |
| 
 | |
|     if (arguments[ 0 ] instanceof Error) {
 | |
|       meta = arguments[ 0 ].toString()
 | |
|       args = Array.prototype.slice.call(arguments, 1)
 | |
|       args.push(meta)
 | |
|     } else if (typeof (args[ 0 ]) !== 'string') {
 | |
|       meta = arguments[ 0 ]
 | |
|       args = Array.prototype.slice.call(arguments, 1)
 | |
|       args.push(meta)
 | |
|     }
 | |
| 
 | |
|     logger[ level ].apply(logger, args)
 | |
|   }
 | |
| }
 | |
| const bunyanLogger = {
 | |
|   trace: bunyanLogFactory('debug'),
 | |
|   debug: bunyanLogFactory('debug'),
 | |
|   info: bunyanLogFactory('info'),
 | |
|   warn: bunyanLogFactory('warn'),
 | |
|   error: bunyanLogFactory('error'),
 | |
|   fatal: bunyanLogFactory('error')
 | |
| }
 | |
| // ---------------------------------------------------------------------------
 | |
| 
 | |
| export {
 | |
|   timestampFormatter,
 | |
|   labelFormatter,
 | |
|   consoleLoggerFormat,
 | |
|   jsonLoggerFormat,
 | |
|   logger,
 | |
|   bunyanLogger
 | |
| }
 |