Merge pull request #17292 from storybookjs/fix/ui/composition_docs_mode

Core: Apply Docs mode to composed storybooks
storybookjs · Feb 4, 2022 · 9b9ea26 · 9b9ea26
2 parents 36f356d + 76ad114
commit 9b9ea26
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/lib/ui/src/components/sidebar/Sidebar.tsx b/lib/ui/src/components/sidebar/Sidebar.tsx
@@ -105,7 +105,24 @@ export const Sidebar: FunctionComponent<SidebarProps> = React.memo(
       () => (DOCS_MODE ? collapseAllStories : collapseDocsOnlyStories)(storiesHash),
       [DOCS_MODE, storiesHash]
     );
-    const dataset = useCombination(stories, storiesConfigured, storiesFailed, refs);
+    const adaptedRefs = useMemo(() => {
+      if (DOCS_MODE) {
+        return Object.keys(refs).reduce((acc: Refs, cur) => {
+          const ref = refs[cur];
+          if (ref.stories) {
+            acc[cur] = {
+              ...ref,
+              stories: collapseDocsOnlyStories(ref.stories),
+            };
+          } else {
+            acc[cur] = ref;
+          }
+          return acc;
+        }, {});
+      }
+      return refs;
+    }, [DOCS_MODE, refs]);
+    const dataset = useCombination(stories, storiesConfigured, storiesFailed, adaptedRefs);
     const isLoading = !dataset.hash[DEFAULT_REF_ID].ready;
     const lastViewedProps = useLastViewed(selected);