From 2ddcaa9ce12c1a8d7108f2a5a8924963f0fd8874 Mon Sep 17 00:00:00 2001
From: Julian Lam <julian@nodebb.org>
Date: Thu, 14 Apr 2016 11:16:12 -0400
Subject: [PATCH] added new hook for hotswap preparation

---
 src/routes/index.js | 26 +++++++++++++++++++-------
 src/webserver.js    |  5 +++--
 2 files changed, 22 insertions(+), 9 deletions(-)

diff --git a/src/routes/index.js b/src/routes/index.js
index a80eb2e46a..526d9e697c 100644
--- a/src/routes/index.js
+++ b/src/routes/index.js
@@ -80,13 +80,25 @@ function groupRoutes(app, middleware, controllers) {
 	setupPageRoute(app, '/groups/:slug/members', middleware, middlewares, controllers.groups.members);
 }
 
-module.exports = function(app, middleware) {
-	var router = express.Router(),
-		pluginRouter = express.Router(),
-		authRouter = express.Router(),
+module.exports = function(app, middleware, hotswapIds) {
+	var routers = [
+			express.Router(),	// plugin router
+			express.Router(),	// main app router
+			express.Router()	// auth router
+		],
+		router = routers[1],
+		pluginRouter = routers[0],
+		authRouter = routers[2],
 		relativePath = nconf.get('relative_path'),
 		ensureLoggedIn = require('connect-ensure-login');
 
+	if (Array.isArray(hotswapIds) && hotswapIds.length) {
+		for(var idx,x=0;x<hotswapIds.length;x++) {
+			idx = routers.push(express.Router()) - 1;
+			routers[idx].hotswapId = hotswapIds[x];
+		}
+	}
+
 	pluginRouter.render = function() {
 		app.render.apply(app, arguments);
 	};
@@ -119,9 +131,9 @@ module.exports = function(app, middleware) {
 	userRoutes(router, middleware, controllers);
 	groupRoutes(router, middleware, controllers);
 
-	app.use(relativePath, pluginRouter);
-	app.use(relativePath, router);
-	app.use(relativePath, authRouter);
+	for(var x=0;x<routers.length;x++) {
+		app.use(relativePath, routers[x]);
+	}
 
 	if (process.env.NODE_ENV === 'development') {
 		require('./debug')(app, middleware, controllers);
diff --git a/src/webserver.js b/src/webserver.js
index e667862561..28b0197dec 100644
--- a/src/webserver.js
+++ b/src/webserver.js
@@ -101,8 +101,9 @@ function initializeNodeBB(callback) {
 				middleware: middleware
 			}, next);
 		},
-		function(next) {
-			routes(app, middleware);
+		async.apply(plugins.fireHook, 'filter:hotswap.prepare', []),
+		function(hotswapIds, next) {
+			routes(app, middleware, hotswapIds);
 			next();
 		}
 	], callback);