TOTVSTEC/cloudbridge-cli

'use strict';

let Logging = module.exports,
    colors = require('colors'),
    winston = require('winston');

Logging.winston = winston;

let messageFormat = winston.format.printf(function(info) {
    return info.message;
});

Logging.logger = winston.createLogger({
    exitOnError: false,
    transports: [
        new (winston.transports.Console)({
            name: 'console',
            format: messageFormat
        })
    ]
});

// To be used by helpers createDefaultLogger and createLoggerWithFile
Logging.createLogger = function createLogger(transports, level) {
    level = level || 'info';

    Logging.logger = winston.createLogger({
        exitOnError: false,
        level: level,
        transports: transports
    });

    return Logging.logger;
};

Logging.createDefaultLogger = function createDefaultLogger(level) {
    level = level || 'info';
    let transports = [
        new (winston.transports.Console)({
            name: 'console',
            format: messageFormat
        })
    ];
    return Logging.createLogger(transports, level);
};

Logging.createLoggerWithFile = function createLoggerWithFile(logFilePath, level) {
    let transports = [
        new (winston.transports.File)({
            filename: logFilePath,
            name: 'file'
        })
    ];
    return Logging.createLogger(transports, level);
};

Logging.queryLogs = function queryLogs(searchText, timestamp) {
    let searchDate = new Date(timestamp) || new Date();
    let options = {
        from: new Date() - (24 * 60 * 60 * 1000),
        until: searchDate,
        limit: 10,
        start: 0,
        order: 'desc',
        fields: ['message', 'level']
    };

    //
    // Find items logged between today and yesterday.
    //
    winston.query(options, function(err, results) {
        if (err) {
            throw err;
        }

        console.log(results);
    });
};

Logging.setUpConsoleLoggingHelpers = function setUpConsoleLoggingHelpers() {
    colors.setTheme({
        silly: 'rainbow',
        input: 'grey',
        small: 'grey',
        verbose: 'cyan',
        prompt: ['yellow', 'bold'],
        info: 'white',
        data: 'grey',
        help: 'cyan',
        warn: 'yellow',
        debug: 'blue',
        error: 'red'
    });

    let consoleInfo = console.info;
    console.info = function() {
        if (arguments.length === 1 && !arguments[0])
            return;

        let msg = '';
        for (let n in arguments) {
            msg += arguments[n] + ' ';
        }

        consoleInfo.call(console, msg.blue.bold);
    };

    let consoleError = console.error;
    console.error = function() {
        if (arguments.length === 1 && !arguments[0])
            return;

        let msg = ' ✗';

        for (let n in arguments) {
            msg += ' ' + arguments[n];
        }

        consoleError.call(console, msg.red.bold);
    };

    console.success = function() {
        if (arguments.length === 1 && !arguments[0])
            return;

        let msg = ' ✓';
        for (let n in arguments) {
            msg += ' ' + arguments[n];
        }

        console.log(msg.green.bold);
    };

    //Default level is set to 'info'
    Logging.createDefaultLogger();
};