From 59203acdfa3603d9dc6af612c05d59329a63d643 Mon Sep 17 00:00:00 2001
From: Alicia Sykes <alicia@omg.lol>
Date: Sun, 3 Sep 2023 15:32:58 +0100
Subject: [PATCH] Update the env check, to determine what handler format to use

---
 api/_common/middleware.js | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/api/_common/middleware.js b/api/_common/middleware.js
index 2ac0629..a80e16e 100644
--- a/api/_common/middleware.js
+++ b/api/_common/middleware.js
@@ -70,14 +70,13 @@ const commonMiddleware = (handler) => {
     }
   };
 
-  let platform;
-  if (process.env.PLATFORM === 'vercel' || process.env.VERCEL) {
-    platform = 'vercel';
-  } else {
-    platform = 'netlify';
-  }
-  return platform === 'vercel' ? vercelHandler : netlifyHandler;
-  
+  // The format of the handlers varies between platforms
+  // E.g. Netlify + AWS expect Lambda functions, but Vercel or Node needs standard handler
+  const platformEnv = (process.env.PLATFORM || '').toUpperCase(); // Has user set platform manually?
+
+  const nativeMode = (['VERCEL', 'NODE'].includes(platformEnv) || process.env.VERCEL || process.env.WC_SERVER);
+
+  return nativeMode ? vercelHandler : netlifyHandler;
 };
 
 module.exports = commonMiddleware;