feat: allow execution of custom express middleware

shattercms · Feb 14, 2021 · 23145c5 · 23145c5
1 parent 2c647b0
commit 23145c5
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/packages/gateway/src/server/express.ts b/packages/gateway/src/server/express.ts
@@ -1,17 +1,18 @@
 import express from 'express';
 import { ConnectionOptions, createConnection } from 'typeorm';
 import { buildSchema } from 'type-graphql';
-import {
-  ApolloServer,
-  AuthenticationError,
-  SchemaDirectiveVisitor,
-} from 'apollo-server-express';
+import { ApolloServer, SchemaDirectiveVisitor } from 'apollo-server-express';
 import { BuildContext, ShatterContext } from '@shattercms/types';
 import { authMiddleware } from './middleware';
 
 export const startServer = async (context: BuildContext) => {
   const app = express();
 
+  // Execute user express middleware
+  context.config.expressMiddlewares.forEach((middleware) => {
+    app.use(middleware);
+  });
+
   // Create database connection
   const orm = await createConnection({
     ...context.config.postgres,

diff --git a/packages/gateway/src/setup/config.ts b/packages/gateway/src/setup/config.ts
@@ -5,6 +5,7 @@ const configDefault: ShatterConfig = {
   rootDir: '.',
   debug: false,
   modules: [],
+  expressMiddlewares: [],
   server: {
     host: 'localhost',
     port: 4000,

diff --git a/packages/types/src/index.ts b/packages/types/src/index.ts
@@ -1,13 +1,14 @@
 import { Connection, EntitySchema } from 'typeorm';
 import { SchemaDirectiveVisitor } from 'apollo-server-express';
-import { Request, Response } from 'express';
+import { Request, RequestHandler, Response } from 'express';
 import { CorsOptions, CorsOptionsDelegate } from 'cors';
 
 export interface ShatterConfig {
   rootDir: string;
   debug: boolean;
 
   modules: Array<ConfigModule>;
+  expressMiddlewares: Array<RequestHandler>;
 
   server: {
     host: string;