Add middleware options

Author: pwfisher

README.md

@@ -191,7 +191,7 @@ following interface:
 
 #### `subscribe(notify)`
 
-The `subscribe()` method ony our notifier is passed a `notify` function.
+The `subscribe()` method on your notifier is passed a `notify` function.
 If you detect that a new version of your app has been deployed (whether
 via polling or a push notification), call this function to trigger a
 reload.

src/express-http-server.js

@@ -13,18 +13,23 @@ class ExpressHTTPServer {
     this.password = options.password;
     this.cache = options.cache;
     this.gzip = options.gzip || false;
+    this.preFastbootMiddlewares = options.preFastbootMiddlewares || [];
+    this.postFastbootMiddlewares = options.postFastbootMiddlewares || [];
 
     this.app = express();
-    if (options.gzip) {
-      this.app.use(require('compression')());
-    }
   }
 
-  serve(middleware) {
+  serve(fastbootMiddleware) {
     let app = this.app;
     let username = this.username;
     let password = this.password;
 
+    this.preFastbootMiddlewares.forEach(args => app.use(...args));
+
+    if (this.gzip) {
+      this.app.use(require('compression')());
+    }
+
     if (username !== undefined || password !== undefined) {
       this.ui.writeLine(`adding basic auth; username=${username}; password=${password}`);
       app.use(basicAuth(username, password));
@@ -35,14 +40,16 @@ class ExpressHTTPServer {
     }
 
     if (this.distPath) {
-      app.get('/', middleware);
+      app.get('/', fastbootMiddleware);
       app.use(express.static(this.distPath));
       app.get('/assets/*', function(req, res) {
         res.sendStatus(404);
       });
     }
 
-    app.get('/*', middleware);
+    app.get('/*', fastbootMiddleware);
+
+    this.postFastbootMiddlewares.forEach(args => app.use(...args));
 
     return new Promise(resolve => {
       let listener = app.listen(process.env.PORT || 3000, () => {

src/fastboot-app-server.js

@@ -15,8 +15,10 @@ class FastBootAppServer {
     this.notifier = options.notifier;
     this.cache = options.cache;
     this.ui = options.ui;
-    this.gzip = options.gzip || false;
+    this.gzip = options.gzip;
     this.httpServer = options.httpServer;
+    this.preFastbootMiddlewares = options.preFastbootMiddlewares;
+    this.postFastbootMiddlewares = options.postFastbootMiddlewares;
 
     if (!this.ui) {
       let UI = require('./ui');

src/worker.js

@@ -10,7 +10,9 @@ class Worker {
     this.httpServer = options.httpServer;
     this.ui = options.ui;
     this.cache = options.cache;
-    this.gzip = options.gzip || false;
+    this.gzip = options.gzip;
+    this.preFastbootMiddlewares = options.preFastbootMiddlewares;
+    this.postFastbootMiddlewares = options.postFastbootMiddlewares;
 
     if (!this.httpServer) {
       this.httpServer = new ExpressHTTPServer({