prepper/lib/Logger.js at 421025bd1967e5a6fe53fa5857eebf2a1010a8d2 · onebeyond/prepper · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
var EventEmitter = require('events').EventEmitter
var util = require('util')
var R = require('ramda')
var isString = R.compose(R.equals('String'), R.type)
var isObject = R.compose(R.equals('Object'), R.type)
var _ = require('lodash')
var Sequence = require('./handlers/Sequence')

function Logger(_options) {

    var levels = ['trace', 'debug', 'info', 'warn', 'error', 'fatal']
    var handlers = _options && _options.handlers
    var sequence = _options && _options.sequence || new Sequence(handlers)
    var options = _.merge({}, { message: '', level: 'info', maxListeners: 100, sequence: sequence }, _options)
    var self = this

    self.child = function(_options) {
        var sequence = _options.sequence ? _options.sequence : new Sequence(_options.handlers)
        var child = new Logger(_.merge({}, { sequence: sequence }, _options))
        child.on('message', self.log)
        return child
    }

    self.log = function log(arg1, arg2, arg3) {
        self._logEvent.apply(self, toArray(arguments))
        return self
    }

    levels.forEach(function(level) {
        self[level] = function(arg1, arg2, arg3) {
            self._logEvent.apply(self, toArray(arguments).concat(level))
            return self
        }
    })

    self._logEvent = function(arg1, arg2, arg3, arg4) {
        // message, error, context, level
        if (arguments.length === 4) return self.emit('internal-message', _.merge(
            {},
            { message: options.message, level: options.level },
            arg3,
            toError(arg2),
            toMessage(arg1),
            toLevel(arg1, arg2, arg3, arg4)
        ))

        // message, error, level
        if (arguments.length === 3 && isString(arg1) && arg2 instanceof Error && isString(arg3)) return self._logEvent(arg1, arg2, null, arg3)
        // message, error, context
        if (arguments.length === 3 && isString(arg1) && arg2 instanceof Error && isObject(arg3)) return self._logEvent(arg1, arg2, arg3, null)
        // message, context, level
        if (arguments.length === 3 && isString(arg1) && isObject(arg2) && isString(arg3)) return self._logEvent(arg1, null, arg2, arg3)
        // error, context, level
        if (arguments.length === 3 && arg1 instanceof Error && isObject(arg2) && isString(arg3)) return self._logEvent(null, arg1, arg2, arg3)
        // message, level
        if (arguments.length === 2 && isString(arg1) && isString(arg2)) return self._logEvent(arg1, null, null, arg2)
        // message, error
        if (arguments.length === 2 && isString(arg1) && arg2 instanceof Error) return self._logEvent(arg1, arg2, null, null)
        // message, context
        if (arguments.length === 2 && isString(arg1) && isObject(arg2)) return self._logEvent(arg1, null, arg2, null)
        // error, level
        if (arguments.length === 2 && arg1 instanceof Error && isString(arg2)) return self._logEvent(null, arg1, null, arg2)
        // error, context
        if (arguments.length === 2 && arg1 instanceof Error && isObject(arg2)) return self._logEvent(null, arg1, arg2, null)
        // context, level
        if (arguments.length === 2 && isObject(arg1) && isString(arg2)) return self._logEvent(null, null, arg1, arg2)
        // message
        if (arguments.length === 1 && isString(arg1)) return self._logEvent(arg1, null, null, null)
        // error
        if (arguments.length === 1 && arg1 instanceof Error) return self._logEvent(null, arg1, null, null)
        // context
        if (arguments.length === 1 && isObject(arg1)) return self._logEvent(null, null, arg1, null)
        // usage error
        return self._logEvent(null, toError(new Error('Logger usage error')), { arguments: toArray(arguments) }, 'error')
    }

    function toMessage(message) {
        return message ? { message: message } : {}
    }

    function toError(error) {
        return error ? {
            message: error.message && error.message !== 'null' ? error.message : 'An unspecified error occurred',
            level: error.level || 'error',
            error: R.pipe(R.pick(Object.getOwnPropertyNames(error)), R.omit('level'))(error)
        } : {}
    }

    function toLevel(message, error, context, level) {
        if (level) return { level: level }
        if (error && error.level) return { level: error.level }
        if (context && context.level) return { level: context.level }
        if (error) return { level: 'error' }
        return {}
    }

    function toArray(args) {
        return Array.prototype.slice.call(args)
    }

    EventEmitter.call(self);
    self.setMaxListeners(options.maxListeners)
    self.on('internal-message', options.sequence.handle)
    options.sequence.on('message', function(event) {
        self.emit('message', event)
    })
}

util.inherits(Logger, EventEmitter)

module.exports = Logger