Update middleware to support changes in Storybook v8

Fixes torchbox#14
lb- · Nov 5, 2024 · be087f6 · be087f6
1 parent 93a870e
commit be087f6
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/src/middleware.js b/src/middleware.js
@@ -20,15 +20,18 @@ const restream = (proxyReq, req) => {
 const createDjangoAPIMiddleware = (options) => {
   const { proxy, apiPath, origin } = options;
   const createProxy = proxy || createProxyMiddleware;
+  const paths = Array.isArray(apiPath) ? apiPath : [apiPath];
   const middleware = (router) => {
-    router.use(
-      apiPath,
-      createProxy({
-        target: origin,
-        changeOrigin: true,
-        onProxyReq: restream,
-      }),
-    );
+    paths.forEach((path) => {
+      router.use(
+        path,
+        createProxy({
+          target: origin,
+          changeOrigin: true,
+          onProxyReq: restream,
+        }),
+      );
+    });
   };
 
   return middleware;