Cleanup client manager functions

This commit is contained in:
Pavel Djundik 2017-10-15 19:05:19 +03:00
parent 69ef6831b9
commit 5c45321cca
2 changed files with 71 additions and 80 deletions

View File

@ -1,10 +1,11 @@
"use strict"; "use strict";
var _ = require("lodash"); const _ = require("lodash");
var colors = require("colors/safe"); const colors = require("colors/safe");
var fs = require("fs"); const fs = require("fs");
var Client = require("./client"); const path = require("path");
var Helper = require("./helper"); const Client = require("./client");
const Helper = require("./helper");
const WebPush = require("./plugins/webpush"); const WebPush = require("./plugins/webpush");
module.exports = ClientManager; module.exports = ClientManager;
@ -19,6 +20,7 @@ ClientManager.prototype.init = function(identHandler, sockets) {
this.webPush = new WebPush(); this.webPush = new WebPush();
if (!Helper.config.public && !Helper.config.ldap.enable) { if (!Helper.config.public && !Helper.config.ldap.enable) {
// TODO: Remove deprecated warning in v3.0.0
if ("autoload" in Helper.config) { if ("autoload" in Helper.config) {
log.warn(`Autoloading users is now always enabled. Please remove the ${colors.yellow("autoload")} option from your configuration file.`); log.warn(`Autoloading users is now always enabled. Please remove the ${colors.yellow("autoload")} option from your configuration file.`);
} }
@ -35,12 +37,6 @@ ClientManager.prototype.autoloadUsers = function() {
const users = this.getUsers(); const users = this.getUsers();
const noUsersWarning = `There are currently no users. Create one with ${colors.bold("lounge add <name>")}.`; const noUsersWarning = `There are currently no users. Create one with ${colors.bold("lounge add <name>")}.`;
// There was an error, already logged, but we have to crash the server as
// user directory could not be accessed
if (users === undefined) {
process.exit(1);
}
if (users.length === 0) { if (users.length === 0) {
log.info(noUsersWarning); log.info(noUsersWarning);
} }
@ -64,84 +60,77 @@ ClientManager.prototype.autoloadUsers = function() {
if (client) { if (client) {
client.quit(true); client.quit(true);
this.clients = _.without(this.clients, client); this.clients = _.without(this.clients, client);
log.info(`User ${colors.bold(name)} disconnected and removed`); log.info(`User ${colors.bold(name)} disconnected and removed.`);
} }
}); });
}, 1000, {maxWait: 10000})); }, 1000, {maxWait: 10000}));
}; };
ClientManager.prototype.loadUser = function(name) { ClientManager.prototype.loadUser = function(name) {
let json; const user = readUserConfig(name);
try {
json = this.readUserConfig(name); if (!user) {
} catch (e) {
log.error("Failed to read user config", e);
return; return;
} }
if (!this.findClient(name)) {
this.clients.push(new Client( let client = this.findClient(name);
this,
name, if (client) {
json log.warn(`Tried to load user ${colors.bold(name)}, which is already loaded.`);
)); return client;
} }
client = new Client(this, name, user);
this.clients.push(client);
return client;
}; };
ClientManager.prototype.getUsers = function() { ClientManager.prototype.getUsers = function() {
var users = []; return fs
try { .readdirSync(Helper.USERS_PATH)
var files = fs.readdirSync(Helper.USERS_PATH); .filter((file) => file.endsWith(".json"))
files.forEach((file) => { .map((file) => file.slice(0, -5));
if (file.indexOf(".json") !== -1) {
users.push(file.replace(".json", ""));
}
});
} catch (e) {
log.error(`Failed to get users (${e})`);
return;
}
return users;
}; };
ClientManager.prototype.addUser = function(name, password, enableLog) { ClientManager.prototype.addUser = function(name, password, enableLog) {
var users = this.getUsers(); if (path.basename(name) !== name) {
if (users.indexOf(name) !== -1) { throw new Error(`${name} is an invalid username.`);
}
const userPath = Helper.getUserConfigPath(name);
if (fs.existsSync(userPath)) {
log.error(`User ${colors.green(name)} already exists.`);
return false; return false;
} }
try {
if (require("path").basename(name) !== name) {
throw new Error(name + " is an invalid username.");
}
var user = { const user = {
user: name, password: password || "",
password: password || "", log: enableLog || false,
log: enableLog, awayMessage: "",
awayMessage: "", networks: [],
networks: [], sessions: {},
sessions: {}, };
};
fs.writeFileSync( try {
Helper.getUserConfigPath(name), fs.writeFileSync(userPath, JSON.stringify(user, null, "\t"));
JSON.stringify(user, null, "\t")
);
} catch (e) { } catch (e) {
log.error("Failed to add user " + name, e); log.error(`Failed to create user ${colors.green(name)} (${e})`);
throw e; throw e;
} }
return true; return true;
}; };
ClientManager.prototype.updateUser = function(name, opts, callback) { ClientManager.prototype.updateUser = function(name, opts, callback) {
const users = this.getUsers(); const user = readUserConfig(name);
if (users.indexOf(name) === -1) {
return false; if (!user) {
} log.error(`Tried to update invalid user ${colors.green(name)}. This is most likely a bug.`);
if (typeof opts === "undefined") {
return false; return false;
} }
const user = this.readUserConfig(name);
const currentUser = JSON.stringify(user, null, "\t"); const currentUser = JSON.stringify(user, null, "\t");
_.assign(user, opts); _.assign(user, opts);
const newUser = JSON.stringify(user, null, "\t"); const newUser = JSON.stringify(user, null, "\t");
@ -153,7 +142,7 @@ ClientManager.prototype.updateUser = function(name, opts, callback) {
fs.writeFile(Helper.getUserConfigPath(name), newUser, (err) => { fs.writeFile(Helper.getUserConfigPath(name), newUser, (err) => {
if (err) { if (err) {
log.error(`Failed to update user ${colors.green(name)} (${err})`); log.error(`Failed to update user ${colors.green(name)}. (${err})`);
} }
if (callback) { if (callback) {
@ -162,24 +151,27 @@ ClientManager.prototype.updateUser = function(name, opts, callback) {
}); });
}; };
ClientManager.prototype.readUserConfig = function(name) {
var users = this.getUsers();
if (users.indexOf(name) === -1) {
return false;
}
var data = fs.readFileSync(Helper.getUserConfigPath(name), "utf-8");
return JSON.parse(data);
};
ClientManager.prototype.removeUser = function(name) { ClientManager.prototype.removeUser = function(name) {
var users = this.getUsers(); const userPath = Helper.getUserConfigPath(name);
if (users.indexOf(name) === -1) {
if (!fs.existsSync(userPath)) {
log.error(`Tried to remove non-existing user ${colors.green(name)}.`);
return false; return false;
} }
try {
fs.unlinkSync(Helper.getUserConfigPath(name)); fs.unlinkSync(userPath);
} catch (e) {
throw e;
}
return true; return true;
}; };
function readUserConfig(name) {
const userPath = Helper.getUserConfigPath(name);
if (!fs.existsSync(userPath)) {
log.error(`Tried to read non-existing user ${colors.green(name)}`);
return false;
}
const data = fs.readFileSync(userPath, "utf-8");
return JSON.parse(data);
}

View File

@ -510,8 +510,7 @@ function performAuthentication(data) {
// If authorization succeeded but there is no loaded user, // If authorization succeeded but there is no loaded user,
// load it and find the user again (this happens with LDAP) // load it and find the user again (this happens with LDAP)
if (!client) { if (!client) {
manager.loadUser(data.user); client = manager.loadUser(data.user);
client = manager.findClient(data.user);
} }
initClient(); initClient();