From f8c7aa42efa7b51e7dceb765a1c657952388fd96 Mon Sep 17 00:00:00 2001
From: Tom Sherman <the.tomsherman@gmail.com>
Date: Sat, 27 Jan 2024 12:24:24 +0000
Subject: [PATCH] Allow passing cors options

---
 src/express.ts | 3 ++-
 src/types.ts   | 3 +++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/express.ts b/src/express.ts
index 0ceb4a2..64105f5 100644
--- a/src/express.ts
+++ b/src/express.ts
@@ -37,6 +37,7 @@ function createExpressApp({
     resetScenariosPath = '/reset-scenarios',
     scenariosPath = '/scenarios',
     cookieMode = false,
+    cors: corsOptions,
   } = options;
 
   const scenarioIds = Object.keys(scenarioMap);
@@ -60,7 +61,7 @@ function createExpressApp({
   let serverContext = getContextFromScenarios([defaultScenario]);
 
   const app = express();
-  app.use(cors({ credentials: true }));
+  app.use(cors({ ...corsOptions, credentials: true }));
   app.use(cookieParser());
   app.use(uiPath, express.static(path.join(__dirname, 'assets')));
   app.use(express.urlencoded({ extended: false }));
diff --git a/src/types.ts b/src/types.ts
index 387b8b9..95b6f9d 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -1,3 +1,5 @@
+import { CorsOptions } from 'cors';
+
 export type Result = {
   status: number;
   headers?: Record<string, string>;
@@ -92,6 +94,7 @@ export type Options = {
   resetScenariosPath?: string;
   scenariosPath?: string;
   cookieMode?: boolean;
+  cors?: CorsOptions;
 };
 
 export type Context = Record<string, any>;