chore: fix storybook beta 7 bug with viewMode

ccontrols · May 17, 2020 · 81cb550 · 81cb550
1 parent 7262b7f
commit 81cb550
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/misc/storybook-custom-docs/src/preview-loader.ts b/misc/storybook-custom-docs/src/preview-loader.ts
@@ -72,13 +72,15 @@ ${pages
   .join('\n')}
 
   window.addEventListener('load', () => {
+    
     const selection =
     window &&
     //@ts-ignore
     window.__STORYBOOK_CLIENT_API__ &&
     //@ts-ignore
     window.__STORYBOOK_CLIENT_API__.store().getSelection();
-    const viewMode = selection ? selection.viewMode : undefined;
+    const urlParams = new URLSearchParams(window.location.search);
+    const viewMode = urlParams.get('viewMode') || (selection ? selection.viewMode : undefined);
     attachPages(pageConfigs, viewMode);
   });
 `;