pomelo-admin1312/lib/util/utils.js

`pomelo-admin1312` is an admin console library for [pomelo](https://github.com/NetEase/pomelo). It provides the a series of utilities to monitor the `pomelo` server clusters.

var crypto = require('crypto');
var util = require('util');
var path = require('path');
var fs = require('fs');

var utils = module.exports;

/**
 * Check and invoke callback
 */
utils.invokeCallback = function(cb) {
    if (!!cb && typeof cb === 'function') {
        cb.apply(null, Array.prototype.slice.call(arguments, 1));
    }
};

/*
 * Date format
 */
utils.format = function(date, format) {
    format = format || 'MM-dd-hhmm';
    var o = {
        "M+": date.getMonth() + 1, //month
        "d+": date.getDate(), //day
        "h+": date.getHours(), //hour
        "m+": date.getMinutes(), //minute
        "s+": date.getSeconds(), //second
        "q+": Math.floor((date.getMonth() + 3) / 3), //quarter
        "S": date.getMilliseconds() //millisecond
    };

    if (/(y+)/.test(format)) {
        format = format.replace(RegExp.$1, (date.getFullYear() + "").substr(4 - RegExp.$1.length));
    }
    for (var k in o) {
        if (new RegExp("(" + k + ")").test(format)) {
            format = format.replace(RegExp.$1,
                RegExp.$1.length === 1 ? o[k] :
                ("00" + o[k]).substr(("" + o[k]).length));
        }
    }

    return format;
};

utils.compareServer = function(server1, server2) {
    return (server1['host'] === server2['host']) &&
        (server1['port'] === server2['port']);
}

/**
 * Get the count of elements of object
 */
utils.size = function(obj, type) {
    var count = 0;
    for (var i in obj) {
        if (obj.hasOwnProperty(i) && typeof obj[i] !== 'function') {
            if (!type) {
                count++;
                continue;
            }

            if (type && type === obj[i]['type']) {
                count++;
            }
        }
    }
    return count;
};

utils.md5 = function(str) {
    var md5sum = crypto.createHash('md5');
    md5sum.update(str);
    str = md5sum.digest('hex');
    return str;
}

utils.defaultAuthUser = function(msg, env, cb) {
    var adminUser = null;
    var appBase = path.dirname(require.main.filename);
    var adminUserPath = path.join(appBase, '/config/adminUser.json');
    var presentPath = path.join(appBase, 'config', env, 'adminUser.json');
    if (fs.existsSync(adminUserPath)) {
        adminUser = require(adminUserPath);
    } else if (fs.existsSync(presentPath)) {
        adminUser = require(presentPath);
    } else {
        cb(null);
        return;
    }
    var username = msg['username'];
    var password = msg['password'];
    var md5 = msg['md5'];

    var len = adminUser.length;
    if (md5) {
        for (var i = 0; i < len; i++) {
            var user = adminUser[i];
            var p = "";
            if (user['username'] === username) {
                p = utils.md5(user['password']);
                if (password === p) {
                    cb(user);
                    return;
                }
            }
        }
    } else {
        for (var i = 0; i < len; i++) {
            var user = adminUser[i];
            if (user['username'] === username && user['password'] === password) {
                cb(user);
                return;
            }
        }
    }
    cb(null);
}

utils.defaultAuthServerMaster = function(msg, env, cb) {
    var id = msg['id'];
    var type = msg['serverType'];
    var token = msg['token'];
    if (type === 'master') {
        cb('ok');
        return;
    }

    var servers = null;
    var appBase = path.dirname(require.main.filename);
    var serverPath = path.join(appBase, '/config/adminServer.json');
    var presentPath = null;
    if (env) {
        presentPath = path.join(appBase, 'config', env, 'adminServer.json');
    }

    if (fs.existsSync(serverPath)) {
        servers = require(serverPath);
    } else if (fs.existsSync(presentPath)) {
        servers = require(presentPath);
    } else {
        cb('ok');
        return;
    }

    var len = servers.length;
    for (var i = 0; i < len; i++) {
        var server = servers[i];
        if (server['type'] === type && server['token'] === token) {
            cb('ok');
            return;
        }
    }
    cb('bad');
    return;
}

utils.defaultAuthServerMonitor = function(msg, env, cb) {
    var id = msg['id'];
    var type = msg['serverType'];

    var servers = null;
    var appBase = path.dirname(require.main.filename);
    var serverPath = path.join(appBase, '/config/adminServer.json');
    var presentPath = null;
    if (env) {
        presentPath = path.join(appBase, 'config', env, 'adminServer.json');
    }

    if (fs.existsSync(serverPath)) {
        servers = require(serverPath);
    } else if (fs.existsSync(presentPath)) {
        servers = require(presentPath);
    } else {
        cb('ok');
        return;
    }

    var len = servers.length;
    for (var i = 0; i < len; i++) {
        var server = servers[i];
        if (server['type'] === type) {
            cb(server['token']);
            return;
        }
    }
    cb(null);
    return;
}