Docs: Add autodocs filter function parameter

storybookjs · Jan 2, 2024 · 685b825 · 685b825
1 parent e2c05c6
commit 685b825
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/code/addons/docs/src/preview.ts b/code/addons/docs/src/preview.ts
@@ -1,8 +1,11 @@
+import type { PreparedStory } from '@storybook/types';
+
 export const parameters: any = {
   docs: {
     renderer: async () => {
       const { DocsRenderer } = (await import('./DocsRenderer')) as any;
       return new DocsRenderer();
     },
+    autodocsFilter: (story: PreparedStory) => !story.parameters?.docs?.disable,
   },
 };
diff --git a/code/ui/blocks/src/blocks/Stories.tsx b/code/ui/blocks/src/blocks/Stories.tsx
@@ -27,9 +27,13 @@ const StyledHeading: typeof Heading = styled(Heading)(({ theme }) => ({
 }));
 
 export const Stories: FC<StoriesProps> = ({ title = 'Stories', includePrimary = true }) => {
-  const { componentStories } = useContext(DocsContext);
+  const { componentStories, projectAnnotations, getStoryContext } = useContext(DocsContext);
 
-  let stories = componentStories().filter((story) => !story.parameters?.docs?.disable);
+  let stories = componentStories();
+  const { autodocsFilter } = projectAnnotations.parameters?.docs || {};
+  if (autodocsFilter) {
+    stories = stories.filter((story) => autodocsFilter(story, getStoryContext(story)));
+  }
 
   if (!includePrimary) stories = stories.slice(1);