fix: mdx components of prop

ccontrols · Jun 26, 2020 · d2d968f · d2d968f
1 parent b227844
commit d2d968f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/core/instrument/src/babel/mdx-stories.ts b/core/instrument/src/babel/mdx-stories.ts
@@ -187,6 +187,8 @@ export const extractMDXStories = (props: any) => (
             break;
           }
           default:
+            const attributes = collectAttributes(node);
+            collectComponent(attributes);
             break;
         }
         // console.log(node.name.name, attributes);

diff --git a/core/instrument/src/misc/component-attributes.ts b/core/instrument/src/misc/component-attributes.ts
@@ -1,7 +1,7 @@
 import { Story, Document, StoryParameters } from '@component-controls/core';
 
 export const componentsFromParams = (
-  element: Document | Story | StoryParameters,
+  element: (Document | Story | StoryParameters) & { of?: string },
 ): string[] => {
   const result = [];
   let { component } = element;
@@ -11,6 +11,10 @@ export const componentsFromParams = (
   if (typeof component === 'string') {
     result.push(component);
   }
+  const { of: componentShorthand } = element;
+  if (typeof componentShorthand === 'string') {
+    result.push(componentShorthand);
+  }
   let { subcomponents } = element;
   if (!subcomponents && element.parameters) {
     ({ subcomponents } = element.parameters);