ES6 menu stack

core/menu_stack.js

@@ -2,93 +2,96 @@
 'use strict';
 
 //	ENiGMA½
-var loadMenu	= require('./menu_util.js').loadMenu;
+const loadMenu	= require('./menu_util.js').loadMenu;
 
-var _			= require('lodash');
-var assert		= require('assert');
+//	deps
+const _			= require('lodash');
+const assert	= require('assert');
 
-module.exports	= MenuStack;
+//	:TODO: Stack is backwards.... top should be most recent! :)
 
-function MenuStack(client) {
+module.exports = class MenuStack {
+	constructor(client) {
 		this.client	= client;
 		this.stack	= [];
-
-	var self	= this;
-
-	this.push = function(moduleInfo) {
-		return self.stack.push(moduleInfo);
-	};
-
-	this.pop = function() {
-		return self.stack.pop();
-	};
-
-	this.peekPrev = function() {
-		if(this.stackSize() > 1) {
-			return self.stack[self.stack.length - 2];
-		}
-	};
-
-	this.top = function() {
-		if(self.stackSize() > 0) {
-			return self.stack[self.stack.length - 1];
-		}
-	};
-
-	this.stackSize = function() {
-		return self.stack.length;
-	};
 	}
 
-MenuStack.prototype.next = function(cb) {
-	var currentModuleInfo = this.top();
+	push(moduleInfo) {
+		return this.stack.push(moduleInfo);
+	}
+
+	pop() {
+		return this.stack.pop();
+	}
+
+	peekPrev() {
+		if(this.stackSize > 1) {
+			return this.stack[this.stack.length - 2];
+		}
+	}
+
+	top() {
+		if(this.stackSize > 0) {
+			return this.stack[this.stack.length - 1];
+		}
+	}
+
+	get stackSize() {
+		return this.stack.length;
+	}
+
+	get currentModule() {
+		const top = this.top();
+		if(top) {
+			return top.instance;
+		}
+	}
+
+	next(cb) {
+		const currentModuleInfo = this.top();
 		assert(currentModuleInfo, 'Empty menu stack!');
 
-	var menuConfig = currentModuleInfo.instance.menuConfig;
-	var next;
+		const menuConfig = currentModuleInfo.instance.menuConfig;
+		let nextMenu;
 
 		if(_.isArray(menuConfig.next)) {
-		next = this.client.acs.getConditionalValue(menuConfig.next, 'next');
-		if(!next) {
-			cb(new Error('No matching condition for \'next\'!'));
-			return;
+			nextMenu = this.client.acs.getConditionalValue(menuConfig.next, 'next');
+			if(!nextMenu) {
+				return cb(new Error('No matching condition for \'next\'!'));
 			}
 		} else if(_.isString(menuConfig.next)) {
-		next = menuConfig.next;
+			nextMenu = menuConfig.next;
 		} else {
-		cb(new Error('Invalid or missing \'next\' member in menu config!'));
-		return;
+			return cb(new Error('Invalid or missing \'next\' member in menu config!'));
 		}
 
-	if(next === currentModuleInfo.name) {
-		cb(new Error('Menu config \'next\' specifies current menu!'));
-		return;
+		if(nextMenu === currentModuleInfo.name) {
+			return cb(new Error('Menu config \'next\' specifies current menu!'));
 		}
 
-	this.goto(next, { }, cb);
-};
+		this.goto(nextMenu, { }, cb);
+	}
 
-MenuStack.prototype.prev = function(cb) {
+	prev(cb) {
+		//	:TODO: leave() should really take a cb...
 		this.pop().instance.leave();	//	leave & remove current
 	
 		const previousModuleInfo = this.pop();	//	get previous
 
 		if(previousModuleInfo) {
-		this.goto(
-			previousModuleInfo.name, 
-			{ 
+			const opts = {
 				extraArgs	: previousModuleInfo.extraArgs, 
 				savedState	: previousModuleInfo.savedState
-			}, 
-			cb
-		);
-	} else {
-		cb(new Error('No previous menu available!'));
-	}
 			};
 
-MenuStack.prototype.goto = function(name, options, cb) {
-	var currentModuleInfo = this.top();
+			return this.goto(previousModuleInfo.name, opts, cb);
+		}
+		
+		return cb(new Error('No previous menu available!'));		
+	}
+
+	goto(name, options, cb) {
+		const currentModuleInfo = this.top();
 
 		if(!cb && _.isFunction(options)) {
 			cb = options;
@@ -112,12 +115,12 @@ MenuStack.prototype.goto = function(name, options, cb) {
 			loadOpts.extraArgs = options.extraArgs;
 		}
 
-	loadMenu(loadOpts, function menuLoaded(err, modInst) {
+		loadMenu(loadOpts, (err, modInst) => {
 			if(err) {
+				//	:TODO: probably should just require a cb...
 				const errCb = cb || self.client.defaultHandlerMissingMod();
 				errCb(err);
 			} else {
-			//	:TODO: Move this log to caller
 				self.client.log.debug( { menuName : name }, 'Goto menu module');
 
 				if(currentModuleInfo) {
@@ -141,7 +144,7 @@ MenuStack.prototype.goto = function(name, options, cb) {
 				modInst.enter();
 
 				self.client.log.trace(
-				{ stack : _.map(self.stack, function(si) { return si.name; } ) },
+					{ stack : _.map(self.stack, stackEntry => stackEntry.name) },
 					'Updated menu stack');
 
 				if(cb) {
@@ -149,11 +152,5 @@ MenuStack.prototype.goto = function(name, options, cb) {
 				}
 			}
 		});
-};
-
-MenuStack.prototype.getCurrentModule = function() {
-	var top = this.top();
-	if(top) {
-		return top.instance;
 	}
 };