fix: add isMDXComponent check

ccontrols · Apr 4, 2021 · 324d190 · 324d190
1 parent a7e4896
commit 324d190
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/core/instrument/src/babel/mdx-stories.ts b/core/instrument/src/babel/mdx-stories.ts
@@ -67,7 +67,9 @@ export const extractMDXStories: (
     return attrComponents.length > 0 ? attrComponents[0] : undefined;
   };
   const { title, name, ...rest } = props;
-  const doc = props ? { ...rest, title: title || name } : undefined;
+  const doc = props
+    ? { isMDXComponent: true, ...rest, title: title || name }
+    : undefined;
   const store: PartialStore = {
     stories: {},
     doc,

diff --git a/plugins/cc-cli/src/document-template.ts b/plugins/cc-cli/src/document-template.ts
@@ -64,7 +64,7 @@ export const createDocumentTemplate: TemplateFunction<StoryTemplateOptions> = as
       }));
     }
   }
-  if (stories.length <= 0 || doc.MDXPage) {
+  if (stories.length === 0 || doc.isMDXComponent) {
     return '';
   }
   const store = bundle ? 'bundle' : 'imports';

diff --git a/plugins/cc-cli/src/stories-template.ts b/plugins/cc-cli/src/stories-template.ts
@@ -64,7 +64,7 @@ export const createStoriesTemplate: TemplateFunction<StoryTemplateOptions> = asy
       }));
     }
   }
-  if (stories.length <= 0 || doc.MDXPage) {
+  if (stories.length === 0 || doc.isMDXComponent) {
     return '';
   }
   const store = bundle ? 'bundle' : 'imports';