From 3d6689f68f17df25973589afe7e5df4ad1b2924e Mon Sep 17 00:00:00 2001
From: Ian VanSchooten <ian.vanschooten@gmail.com>
Date: Thu, 21 Apr 2022 22:41:27 -0400
Subject: [PATCH] Avoid hiding all non-storybook body children

---
 packages/builder-vite/plugins/no-fouc.ts | 25 ++++++++++++++----------
 1 file changed, 15 insertions(+), 10 deletions(-)

diff --git a/packages/builder-vite/plugins/no-fouc.ts b/packages/builder-vite/plugins/no-fouc.ts
index 6d1e0608..f4e81160 100644
--- a/packages/builder-vite/plugins/no-fouc.ts
+++ b/packages/builder-vite/plugins/no-fouc.ts
@@ -21,22 +21,27 @@ export function noFouc(): Plugin {
   };
 }
 
+/**
+ * Insert default styles to hide storybook elements as the page loads until JS can
+ * add the official storybook default head styles and scripts.  These lines are mostly
+ * taken from https://github.com/storybookjs/storybook/blob/next/lib/core-common/templates/base-preview-head.html#L6-L20
+ */
 function insertHeadStyles(html: string) {
   return html.replace(
     '</head>',
     `
   <style>
-    /* Fix for elements flashing */
-    body > * {
-      display: none !important;
+    .sb-show-preparing-story:not(.sb-show-main) > :not(.sb-preparing-story) {
+      display: none;
+    }
+    .sb-show-preparing-docs:not(.sb-show-main) > :not(.sb-preparing-docs) {
+      display: none;
     }
-    .sb-show-preparing-story > .sb-preparing-story,
-    .sb-show-preparing-docs > .sb-preparing-docs,
-    .sb-show-nopreview > .sb-nopreview,
-    .sb-show-errordisplay > .sb-errordisplay,
-    .sb-show-main > #root,
-    .sb-show-main > #docs-root {
-      display: block !important;
+    :not(.sb-show-preparing-story) > .sb-preparing-story,
+    :not(.sb-show-preparing-docs) > .sb-preparing-docs,
+    :not(.sb-show-nopreview) > .sb-nopreview,
+    :not(.sb-show-errordisplay) > .sb-errordisplay {
+      display: none;
     }
     #root[hidden],
     #docs-root[hidden] {