1 files changed, 67 insertions, 37 deletions

diff --git a/bin/manage_users b/bin/manage_users
index 74ac1812..8c356200 100755
--- a/bin/manage_users
+++ b/bin/manage_users
@@ -8,7 +8,8 @@ const models = require("../lib/models/");
 const readline = require("readline-sync");
 const minimist = require("minimist");
 
-var usage = `
+function showUsage(tips) {
+	console.log(`${tips}
 
 Command-line utility to create users for email-signin.
 
@@ -16,74 +17,103 @@ Usage: bin/manage_users [--pass password] (--add | --del) user-email
 	Options:
 		--add 	Add user with the specified user-email
 		--del 	Delete user with specified user-email
+		--reset Reset user password with specified user-email
 		--pass	Use password from cmdline rather than prompting
-`
+`);
+	process.exit(1);
+}
+
+function getPass(argv, action) {
+	// Find whether we use cmdline or prompt password
+	if(typeof argv["pass"] !== 'string') {
+		return readline.question(`Password for ${argv[action]}:`, {hideEchoBack: true});
+	}
+	console.log("Using password from commandline...");
+	return argv["pass"];
+}
 
 // Using an async function to be able to use await inside
 async function createUser(argv) {
-	var existing_user = await models.User.findOne({where: {email: argv["add"]}});
+	const existing_user = await models.User.findOne({where: {email: argv["add"]}});
 	// Cannot create already-existing users
 	if(existing_user != undefined) {
-		console.log("User with e-mail "+existing_user.email+" already exists! Aborting ...");
+		console.log(`User with e-mail ${existing_user.email} already exists! Aborting ...`);
 		process.exit(1);
 	}
 
-	// Find whether we use cmdline or prompt password
-	if(argv["pass"] == undefined) {
-		var pass = readline.question("Password for "+argv["add"]+":", {hideEchoBack: true});
-	} else {
-		console.log("Using password from commandline...");
-		var pass = "" + argv["pass"];
-	}
+	const pass = getPass(argv, "add");
+
 
 	// Lets try to create, and check success
-	var ref = await models.User.create({email: argv["add"], password: pass});
+	const ref = await models.User.create({email: argv["add"], password: pass});
 	if(ref == undefined) {
-		console.log("Could not create user with email "+argv["add"]);
+		console.log(`Could not create user with email ${argv["add"]}`);
 		process.exit(1);
 	} else
-		console.log("Created user with email "+argv["add"]);
+		console.log(`Created user with email ${argv["add"]}`);
 }
 
 // Using an async function to be able to use await inside
 async function deleteUser(argv) {
 	// Cannot delete non-existing users
-	var existing_user = await models.User.findOne({where: {email: argv["del"]}});
-	if(existing_user == undefined) {
-		console.log("User with e-mail "+argv["del"]+" does not exist, cannot delete");
+	const existing_user = await models.User.findOne({where: {email: argv["del"]}});
+	if(existing_user === undefined) {
+		console.log(`User with e-mail ${argv["del"]} does not exist, cannot delete`);
 		process.exit(1);
 	}
 
 	// Sadly .destroy() does not return any success value with all
 	// backends. See sequelize #4124
 	await existing_user.destroy();
-	console.log("Deleted user "+argv["del"]+" ...");
+	console.log(`Deleted user ${argv["del"]} ...`);
 }
 
+
+// Using an async function to be able to use await inside
+async function resetUser(argv) {
+	const existing_user = await models.User.findOne({where: {email: argv["reset"]}});
+	// Cannot reset non-existing users
+	if(existing_user == undefined) {
+		console.log(`User with e-mail ${argv["reset"]} does not exist, cannot reset`);
+		process.exit(1);
+	}
+
+	const pass = getPass(argv, "reset");
+
+	// set password and save
+	existing_user.password = pass;
+	await existing_user.save();
+	console.log(`User with email ${argv["reset"]} password has been reset`);
+}
+
+const options = {
+	add: createUser,
+	del: deleteUser,
+	reset: resetUser,
+};
+
 // Perform commandline-parsing
-var argv = minimist(process.argv.slice(2));
+const argv = minimist(process.argv.slice(2));
 
-// Check for add/delete missing
-if (argv["add"] == undefined && argv["del"] == undefined) {
-	console.log("You did not specify either --add or --del!");
-	console.log(usage);
-	process.exit(1);
+const keys = Object.keys(options);
+const opts = keys.filter((key) => argv[key] !== undefined);
+const action = opts[0];
+
+// Check for options missing
+if (opts.length === 0) {
+	showUsage(`You did not specify either ${keys.map((key) => `--${key}`).join(' or ')}!`);
 }
 
 // Check if both are specified
-if (argv["add"] != undefined && argv["del"] != undefined) {
-	console.log("You cannot add and delete at the same time!");
-	console.log(usage);
-	process.exit(1);
+if (opts.length > 1) {
+	showUsage(`You cannot ${action.join(' and ')} at the same time!`);
+}
+// Check if not string
+if (typeof argv[action] !== 'string') {
+	showUsage(`You must follow an email after --${action}`);
 }
 
 // Call respective processing functions
-if (argv["add"] != undefined) {
-	createUser(argv).then(function() {
-		process.exit(0);
-	});
-} else if (argv["del"] != undefined) {
-	deleteUser(argv).then(function() {
-		process.exit(0);
-	})
-}
+options[action](argv).then(function() {
+  process.exit(0);
+});