Add ability to install proxy middleware

marko-js-archive · Jul 8, 2020 · a7a16ce · a7a16ce
1 parent b9862cb
commit a7a16ce
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git 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
@@ -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;