|
|
/*jshint esversion: 6 */
|
|
|
const BatchRead = require("../batch-read");
|
|
|
const nThen = require("nthen");
|
|
|
const getFolderSize = require("get-folder-size");
|
|
|
var Fs = require("fs");
|
|
|
|
|
|
var Admin = module.exports;
|
|
|
|
|
|
var getActiveSessions = function (Env, ctx, cb) {
|
|
|
var total = ctx.users ? Object.keys(ctx.users).length : '?';
|
|
|
|
|
|
var ips = [];
|
|
|
Object.keys(ctx.users).forEach(function (u) {
|
|
|
var user = ctx.users[u];
|
|
|
var socket = user.socket;
|
|
|
var req = socket.upgradeReq;
|
|
|
var conn = req && req.connection;
|
|
|
var ip = (req && req.headers && req.headers['x-forwarded-for']) || (conn && conn.remoteAddress);
|
|
|
if (ip && ips.indexOf(ip) === -1) {
|
|
|
ips.push(ip);
|
|
|
}
|
|
|
});
|
|
|
|
|
|
cb (void 0, [total, ips.length]);
|
|
|
};
|
|
|
|
|
|
var shutdown = function (Env, ctx, cb) {
|
|
|
return void cb('E_NOT_IMPLEMENTED');
|
|
|
//clearInterval(Env.sessionExpirationInterval);
|
|
|
// XXX set a flag to prevent incoming database writes
|
|
|
// XXX disconnect all users and reject new connections
|
|
|
// XXX wait until all pending writes are complete
|
|
|
// then process.exit(0);
|
|
|
// and allow system functionality to restart the server
|
|
|
};
|
|
|
|
|
|
const batchRegisteredUsers = BatchRead("GET_REGISTERED_USERS");
|
|
|
var getRegisteredUsers = function (Env, cb) {
|
|
|
batchRegisteredUsers('', cb, function (done) {
|
|
|
var dir = Env.paths.pin;
|
|
|
var folders;
|
|
|
var users = 0;
|
|
|
nThen(function (waitFor) {
|
|
|
Fs.readdir(dir, waitFor(function (err, list) {
|
|
|
if (err) {
|
|
|
waitFor.abort();
|
|
|
return void done(err);
|
|
|
}
|
|
|
folders = list;
|
|
|
}));
|
|
|
}).nThen(function (waitFor) {
|
|
|
folders.forEach(function (f) {
|
|
|
var dir = Env.paths.pin + '/' + f;
|
|
|
Fs.readdir(dir, waitFor(function (err, list) {
|
|
|
if (err) { return; }
|
|
|
users += list.length;
|
|
|
}));
|
|
|
});
|
|
|
}).nThen(function () {
|
|
|
done(void 0, users);
|
|
|
});
|
|
|
});
|
|
|
};
|
|
|
|
|
|
const batchDiskUsage = BatchRead("GET_DISK_USAGE");
|
|
|
var getDiskUsage = function (Env, cb) {
|
|
|
batchDiskUsage('', cb, function (done) {
|
|
|
var data = {};
|
|
|
nThen(function (waitFor) {
|
|
|
getFolderSize('./', waitFor(function(err, info) {
|
|
|
data.total = info;
|
|
|
}));
|
|
|
getFolderSize(Env.paths.pin, waitFor(function(err, info) {
|
|
|
data.pin = info;
|
|
|
}));
|
|
|
getFolderSize(Env.paths.blob, waitFor(function(err, info) {
|
|
|
data.blob = info;
|
|
|
}));
|
|
|
getFolderSize(Env.paths.staging, waitFor(function(err, info) {
|
|
|
data.blobstage = info;
|
|
|
}));
|
|
|
getFolderSize(Env.paths.block, waitFor(function(err, info) {
|
|
|
data.block = info;
|
|
|
}));
|
|
|
getFolderSize(Env.paths.data, waitFor(function(err, info) {
|
|
|
data.datastore = info;
|
|
|
}));
|
|
|
}).nThen(function () {
|
|
|
done(void 0, data);
|
|
|
});
|
|
|
});
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
Admin.command = function (Env, ctx, publicKey, data, cb) {
|
|
|
var admins = Env.admins;
|
|
|
if (admins.indexOf(publicKey) === -1) {
|
|
|
return void cb("FORBIDDEN");
|
|
|
}
|
|
|
// Handle commands here
|
|
|
switch (data[0]) {
|
|
|
case 'ACTIVE_SESSIONS':
|
|
|
return getActiveSessions(Env, ctx, cb);
|
|
|
case 'ACTIVE_PADS':
|
|
|
return cb(void 0, ctx.channels ? Object.keys(ctx.channels).length : '?');
|
|
|
case 'REGISTERED_USERS':
|
|
|
return getRegisteredUsers(Env, cb);
|
|
|
case 'DISK_USAGE':
|
|
|
return getDiskUsage(Env, cb);
|
|
|
case 'FLUSH_CACHE':
|
|
|
Env.flushCache();
|
|
|
return cb(void 0, true);
|
|
|
case 'SHUTDOWN':
|
|
|
return shutdown(Env, ctx, cb);
|
|
|
default:
|
|
|
return cb('UNHANDLED_ADMIN_COMMAND');
|
|
|
}
|
|
|
};
|
|
|
|
|
|
|