From a7a16ce272481c13c7d4f0ea33a98a98c89223b9 Mon Sep 17 00:00:00 2001
From: Dan Howard <code_dh@svexpertise.com>
Date: Thu, 14 Nov 2019 20:16:07 -0800
Subject: [PATCH] Add ability to install proxy middleware

---
 index.js                 | 5 +++++
 initializeRestHandler.js | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/index.js b/index.js
index 5773563..a27934c 100644
--- a/index.js
+++ b/index.js
@@ -26,6 +26,11 @@ exports.install = (pluginContext) => {
       description: 'Path to private SSL key (optional)'
     },
 
+    middleware: {
+      type: Object,
+      description: 'Object to install middleware'
+    },
+
     routeNotFound: {
       type: Function,
       description: 'Function to handle missing routes (optional)'
diff --git a/initializeRestHandler.js b/initializeRestHandler.js
index edd176c..b02a7d1 100644
--- a/initializeRestHandler.js
+++ b/initializeRestHandler.js
@@ -7,8 +7,13 @@ module.exports = (httpServer) => {
   const logger = httpServer.logger;
   const requestLogger = httpServer.requestLogger;
   const colorsEnabled = project.getColors();
+  const middleware = project.getMiddleware();
   const routeNotFound = project.getRouteNotFound();
 
+  if (middleware) {
+    httpServer.restHandler.middleware(middleware);
+  }
+
   httpServer.restHandler
     .on('route', (event) => {
       let desc;