fix: remove shorthand prop if undefined

ccontrols · Mar 17, 2021 · e1234d8 · e1234d8
1 parent a30e237
commit e1234d8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/core/instrument/src/babel/extract-arguments-usage.ts b/core/instrument/src/babel/extract-arguments-usage.ts
@@ -32,7 +32,7 @@ export const addArgumentUsage = (
   story: Story,
   args: StoryArguments,
   node: any,
-  shorthand: boolean,
+  shorthand?: boolean,
 ): ArgUsageLocation | undefined => {
   const param = findArguments(args, node.name);
 
@@ -48,7 +48,10 @@ export const addArgumentUsage = (
         p.loc.end.line === loc.end.line &&
         p.loc.end.column === loc.end.column,
     );
-    const usage: ArgUsageLocation = { loc, shorthand };
+    const usage: ArgUsageLocation = { loc };
+    if (shorthand) {
+      usage.shorthand = shorthand;
+    }
     if (!existing) {
       param.usage.push(usage);
       return usage;