Update controller.js to async await

Author: uplift

lib/controllers.js

@@ -1,57 +1,59 @@
 'use strict';
 
-var groups = require.main.require('./src/groups');
+const groups = require.main.require('./src/groups');
 
-var Controllers = {};
+const Controllers = {};
 
 Controllers.renderAdminPage = async (req, res) => {
 	const groupData = await groups.getGroupsFromSet('groups:visible:createtime', 0, -1);
 	res.render('admin/plugins/session-sharing', { groups: groupData });
 };
 
-Controllers.retrieveUser = function (req, res) {
+Controllers.retrieveUser = async (req, res) {
 	const main = module.parent.exports;
 	const remoteId = req.query.id;
-
-	if (remoteId) {
-		main.getUser(remoteId, function (err, userObj) {
-			if (err) {
-				res.status(500).json({
-					error: err.message,
-				});
-			} else if (userObj) {
-				res.status(200).json(userObj);
-			} else {
-				res.sendStatus(404);
-			}
-		});
-	} else {
-		res.status(400).json({
+	
+	if (!remoteId) {
+		return res.status(400).json({
 			error: 'no-id-supplied',
 		});
 	}
+	
+	try {
+		const userObj = await main.getUser(remoteId);
+
+		if (!userObj) {
+			return res.sendStatus(404);
+		}
+
+		return res.status(200).json(userObj);
+	} catch (error) {
+		return res.status(500).json({
+			error: error.message,
+		});
+	}
 };
 
-Controllers.process = function (req, res) {
+Controllers.process = async (req, res) {
 	const main = module.parent.exports;
 
 	if (!req.body || !req.body.token) {
 		return res.status(400).json({
 			error: 'no-token-provided',
 		});
 	}
+	
+	try {
+		const uid = await main.process(req.body.token);
 
-	main.process(req.body.token, (err, uid) => {
-		if (err) {
-			return res.status(500).json({
-				error: err.message,
-			});
-		}
-
-		res.status(200).json({
+		return res.status(200).json({
 			uid,
 		});
-	});
+	} catch (error) {
+		return res.status(500).json({
+			error: error.message,
+		});
+	}
 };
 
 module.exports = Controllers;