feat: widgets/index.js

enable widget-essentials in tests
fix widget test

src/widgets/index.js

@@ -1,104 +1,84 @@
 'use strict';
 
-var async = require('async');
+const async = require('async');
-var winston = require('winston');
+const winston = require('winston');
-var _ = require('lodash');
+const _ = require('lodash');
-var Benchpress = require('benchpressjs');
+const Benchpress = require('benchpressjs');
-
+const util = require('util');
-var plugins = require('../plugins');
+
-var groups = require('../groups');
+const plugins = require('../plugins');
-var translator = require('../translator');
+const groups = require('../groups');
-var db = require('../database');
+const translator = require('../translator');
-var apiController = require('../controllers/api');
+const db = require('../database');
-var meta = require('../meta');
+const apiController = require('../controllers/api');
-
+const loadConfigAsync = util.promisify(apiController.loadConfig);
-var widgets = module.exports;
+const meta = require('../meta');
-
+
-widgets.render = function (uid, options, callback) {
+const widgets = module.exports;
+
+widgets.render = async function (uid, options) {
 	if (!options.template) {
-		return callback(new Error('[[error:invalid-data]]'));
+		throw new Error('[[error:invalid-data]]');
 	}
+	const data = await widgets.getWidgetDataForTemplates(['global', options.template]);
+	delete data.global.drafts;
 
-	async.waterfall([
+	const locations = _.uniq(Object.keys(data.global).concat(Object.keys(data[options.template])));
-		function (next) {
-			widgets.getWidgetDataForTemplates(['global', options.template], next);
-		},
-		function (data, next) {
-			var widgetsByLocation = {};
 
-			delete data.global.drafts;
+	const widgetData = await Promise.all(locations.map(location => renderLocation(location, data, uid, options)));
 
-			var locations = _.uniq(Object.keys(data.global).concat(Object.keys(data[options.template])));
+	const returnData = {};
+	locations.forEach(function (location, i) {
+		if (Array.isArray(widgetData[i]) && widgetData[i].length) {
+			returnData[location] = widgetData[i].filter(Boolean);
+		}
+	});
 
-			var returnData = {};
+	return returnData;
+};
 
-			async.each(locations, function (location, done) {
+async function renderLocation(location, data, uid, options) {
-				widgetsByLocation[location] = (data[options.template][location] || []).concat(data.global[location] || []);
+	const widgetsAtLocation = (data[options.template][location] || []).concat(data.global[location] || []);
 
-				if (!widgetsByLocation[location].length) {
+	if (!widgetsAtLocation.length) {
-					return done(null, { location: location, widgets: [] });
+		return [];
 	}
 
-				async.map(widgetsByLocation[location], function (widget, next) {
+	const renderedWidgets = await Promise.all(widgetsAtLocation.map(widget => renderWidget(widget, uid, options)));
-					renderWidget(widget, uid, options, next);
+	return renderedWidgets;
-				}, function (err, renderedWidgets) {
-					if (err) {
-						return done(err);
 }
-					renderedWidgets = renderedWidgets.filter(Boolean);
-					returnData[location] = renderedWidgets.length ? renderedWidgets : undefined;
-
-					done();
-				});
-			}, function (err) {
-				next(err, returnData);
-			});
-		},
-	], callback);
-};
-
-function renderWidget(widget, uid, options, callback) {
-	var userLang;
 
+async function renderWidget(widget, uid, options) {
 	if (!widget || !widget.data || (!!widget.data['hide-mobile'] && options.req.useragent.isMobile)) {
-		return setImmediate(callback);
+		return;
 	}
-
+	let isVisible = true;
-	async.waterfall([
+	if (widget.data.groups.length) {
-		function (next) {
+		isVisible = await groups.isMemberOfAny(uid, widget.data.groups);
-			if (!widget.data.groups.length) {
-				return next(null, true);
 	}
-			groups.isMemberOfAny(uid, widget.data.groups, next);
-		},
-		function (isVisible, next) {
 	if (!isVisible) {
-				return callback();
+		return;
 	}
-
+	let config = options.res.locals.config || {};
 	if (options.res.locals.isAPI) {
-				apiController.loadConfig(options.req, next);
+		config = await loadConfigAsync(options.req);
-			} else {
-				next(null, options.res.locals.config || {});
 	}
-		},
+
-		function (config, next) {
+	const userLang = config.userLang || meta.config.defaultLang || 'en-GB';
-			userLang = config.userLang || meta.config.defaultLang || 'en-GB';
+	const templateData = _.assign({ }, options.templateData, { config: config });
-			var templateData = _.assign({ }, options.templateData, { config: config });
+	const data = await plugins.fireHook('filter:widget.render:' + widget.widget, {
-			plugins.fireHook('filter:widget.render:' + widget.widget, {
 		uid: uid,
 		area: options,
 		templateData: templateData,
 		data: widget.data,
 		req: options.req,
 		res: options.res,
-			}, next);
+	});
-		},
+
-		function (data, next) {
 	if (!data) {
-				return callback();
+		return;
 	}
-			var html = data;
+	let html = data;
 	if (typeof html !== 'string') {
 		html = data.html;
 	} else {
@@ -106,37 +86,31 @@ function renderWidget(widget, uid, options, callback) {
 	}
 
 	if (widget.data.container && widget.data.container.match('{body}')) {
-				Benchpress.compileParse(widget.data.container, {
+		html = await Benchpress.compileRender(widget.data.container, {
 			title: widget.data.title,
 			body: html,
 			template: data.templateData && data.templateData.template,
-				}, next);
-			} else {
-				next(null, html);
-			}
-		},
-		function (html, next) {
-			translator.translate(html, userLang, function (translatedHtml) {
-				next(null, { html: translatedHtml });
 		});
-		},
-	], callback);
 	}
 
-widgets.getWidgetDataForTemplates = function (templates, callback) {
+	if (html !== undefined) {
-	async.waterfall([
+		html = await translator.translate(html, userLang);
-		function (next) {
+	}
+
+	return { html: html };
+}
+
+widgets.getWidgetDataForTemplates = async function (templates) {
 	const keys = templates.map(tpl => 'widgets:' + tpl);
-			db.getObjects(keys, next);
+	const data = await db.getObjects(keys);
-		},
+
-		function (data, next) {
+	const returnData = {};
-			var returnData = {};
 
 	templates.forEach(function (template, index) {
 		returnData[template] = returnData[template] || {};
 
-				var templateWidgetData = data[index] || {};
+		const templateWidgetData = data[index] || {};
-				var locations = Object.keys(templateWidgetData);
+		const locations = Object.keys(templateWidgetData);
 
 		locations.forEach(function (location) {
 			if (templateWidgetData && templateWidgetData[location]) {
@@ -152,29 +126,19 @@ widgets.getWidgetDataForTemplates = function (templates, callback) {
 		});
 	});
 
-			next(null, returnData);
+	return returnData;
-		},
-	], callback);
 };
 
-widgets.getArea = function (template, location, callback) {
+widgets.getArea = async function (template, location) {
-	async.waterfall([
+	const result = await db.getObjectField('widgets:' + template, location);
-		function (next) {
-			db.getObjectField('widgets:' + template, location, next);
-		},
-		function (result, next) {
 	if (!result) {
-				return callback(null, []);
+		return [];
 	}
 	try {
-				result = parseWidgetData(result);
+		return parseWidgetData(result);
 	} catch (err) {
-				return callback(err);
+		throw err;
 	}
-
-			next(null, result);
-		},
-	], callback);
 };
 
 function parseWidgetData(data) {
@@ -190,84 +154,62 @@ function parseWidgetData(data) {
 	return widgets;
 }
 
-widgets.setArea = function (area, callback) {
+widgets.setArea = async function (area) {
 	if (!area.location || !area.template) {
-		return callback(new Error('Missing location and template data'));
+		throw new Error('Missing location and template data');
 	}
 
-	db.setObjectField('widgets:' + area.template, area.location, JSON.stringify(area.widgets), callback);
+	await db.setObjectField('widgets:' + area.template, area.location, JSON.stringify(area.widgets));
 };
 
-widgets.reset = function (callback) {
+widgets.reset = async function () {
-	var defaultAreas = [
+	const defaultAreas = [
 		{ name: 'Draft Zone', template: 'global', location: 'header' },
 		{ name: 'Draft Zone', template: 'global', location: 'footer' },
 		{ name: 'Draft Zone', template: 'global', location: 'sidebar' },
 	];
-	var drafts;
+
-	async.waterfall([
+	const [areas, drafts] = await Promise.all([
-		function (next) {
+		plugins.fireHook('filter:widgets.getAreas', defaultAreas),
-			async.parallel({
+		widgets.getArea('global', 'drafts'),
-				areas: function (next) {
+	]);
-					plugins.fireHook('filter:widgets.getAreas', defaultAreas, next);
+
-				},
+	let saveDrafts = drafts || [];
-				drafts: function (next) {
+	for (const area of areas) {
-					widgets.getArea('global', 'drafts', next);
+		/* eslint-disable no-await-in-loop */
-				},
+		const areaData = await widgets.getArea(area.template, area.location);
-			}, next);
+		saveDrafts = saveDrafts.concat(areaData);
-		},
-		function (results, next) {
-			drafts = results.drafts || [];
-
-			async.eachSeries(results.areas, function (area, next) {
-				async.waterfall([
-					function (next) {
-						widgets.getArea(area.template, area.location, next);
-					},
-					function (areaData, next) {
-						drafts = drafts.concat(areaData);
 		area.widgets = [];
-						widgets.setArea(area, next);
+		await widgets.setArea(area);
-					},
+	}
-				], next);
+
-			}, next);
+	await widgets.setArea({
-		},
-		function (next) {
-			widgets.setArea({
 		template: 'global',
 		location: 'drafts',
-				widgets: drafts,
+		widgets: saveDrafts,
-			}, next);
+	});
-		},
-	], callback);
 };
 
-widgets.resetTemplate = function (template, callback) {
+widgets.resetTemplate = async function (template) {
-	var toBeDrafted = [];
+	let toBeDrafted = [];
-	async.waterfall([
+	const area = await db.getObject('widgets:' + template + '.tpl');
-		function (next) {
-			db.getObject('widgets:' + template + '.tpl', next);
-		},
-		function (area, next) {
 	for (var location in area) {
 		if (area.hasOwnProperty(location)) {
 			toBeDrafted = toBeDrafted.concat(JSON.parse(area[location]));
 		}
 	}
-			db.delete('widgets:' + template + '.tpl', next);
+	await db.delete('widgets:' + template + '.tpl');
-		},
+	let draftWidgets = await db.getObjectField('widgets:global', 'drafts');
-		function (next) {
-			db.getObjectField('widgets:global', 'drafts', next);
-		},
-		function (draftWidgets, next) {
 	draftWidgets = JSON.parse(draftWidgets).concat(toBeDrafted);
-			db.setObjectField('widgets:global', 'drafts', JSON.stringify(draftWidgets), next);
+	await db.setObjectField('widgets:global', 'drafts', JSON.stringify(draftWidgets));
-		},
-	], callback);
 };
 
-widgets.resetTemplates = function (templates, callback) {
+widgets.resetTemplates = async function (templates) {
-	async.eachSeries(templates, widgets.resetTemplate, callback);
+	async.eachSeries(templates, widgets.resetTemplate);
+	for (const template of templates) {
+		/* eslint-disable no-await-in-loop */
+		await widgets.resetTemplate(template);
+	}
 };
 
 require('../promisify')(widgets);

test/controllers.js

@@ -885,15 +885,12 @@ describe('Controllers', function () {
 						location: 'sidebar',
 						widgets: [
 							{
-								widget: 'html',
-								data: [{
 								widget: 'html',
 								data: {
 									html: 'test',
 									title: '',
 									container: '',
 								},
-								}],
 							},
 						],
 					};
@@ -920,6 +917,7 @@ describe('Controllers', function () {
 				assert.equal(res.statusCode, 200);
 				assert(body.widgets);
 				assert(body.widgets.sidebar);
+				assert.equal(body.widgets.sidebar[0].html, 'test');
 				done();
 			});
 		});

test/mocks/databasemock.js

@@ -257,6 +257,7 @@ function enableDefaultPlugins(callback) {
 	var defaultEnabled = [
 		'nodebb-plugin-dbsearch',
 		'nodebb-plugin-soundpack-default',
+		'nodebb-widget-essentials',
 	];
 
 	winston.info('[install/enableDefaultPlugins] activating default plugins', defaultEnabled);