feature file processor to support csf3 format (#37)

Author: fa93hws

__tests__/fixtures/feature-storybook-ref/b.stories.tsx

@@ -3,4 +3,11 @@
 import { storiesOf } from "@storybook/react";
 import * as React from "react";
 
-storiesOf("b", module).add("Foo", () => null);
+export default {
+  title: 'b',
+  component: null,
+};
+
+export const Foo = {
+  render: () => null,
+}

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@canva/dependency-tree",
-  "version": "3.2.1",
+  "version": "3.3.0",
   "description": "Calculates a dependency tree for set of files",
   "main": "dist/index.js",
   "author": "Canva Pty Ltd",
@@ -36,7 +36,7 @@
     "enhanced-resolve": "^4.1.0",
     "esquery": "^1.3.1",
     "fast-glob": "^3.0.4",
-    "gherkin": "^7.0.3",
+    "gherkin": "^8.2.1",
     "lodash.escaperegexp": "^4.1.2",
     "lodash.memoize": "^4.1.2",
     "memoize-fs": "^2.2.0",

package.json

@@ -16,6 +16,7 @@ const info = logger.extend('info');
 const warn = logger.extend('warn');
 
 const STORIES_IMPORT = 'storiesOf';
+const CSF3_EXPORT_TITLE_FIELD = 'title';
 const STORIES_PACKAGE = '@storybook/react';
 const STORIES_FILE_RE = new RegExp(
   `([^${escapeRegExp(path.sep)}]+)\\.stories\\.tsx?$`,
@@ -107,6 +108,28 @@ export class FeatureFileProcessor extends TypeScriptFileProcessor {
     return node.kind === ts.SyntaxKind.CallExpression;
   }
 
+  private static isExportAssignment(
+    node: ts.Node,
+  ): node is ts.ExportAssignment {
+    return node.kind === ts.SyntaxKind.ExportAssignment;
+  }
+
+  private static isObjectLiteralExpression(
+    node: ts.Node,
+  ): node is ts.ObjectLiteralExpression {
+    return node.kind === ts.SyntaxKind.ObjectLiteralExpression;
+  }
+
+  private static isPropertyAssignment(
+    node: ts.Node,
+  ): node is ts.PropertyAssignment {
+    return node.kind === ts.SyntaxKind.PropertyAssignment;
+  }
+
+  private static isIdentifier(node: ts.Node): node is ts.Identifier {
+    return node.kind === ts.SyntaxKind.Identifier;
+  }
+
   private static walkStories(
     sourceFile: ts.SourceFile,
     callback: (storybook: Storybook) => void,
@@ -129,6 +152,27 @@ export class FeatureFileProcessor extends TypeScriptFileProcessor {
         return;
       }
 
+      if (FeatureFileProcessor.isExportAssignment(node)) {
+        const { expression } = node;
+        if (
+          FeatureFileProcessor.isObjectLiteralExpression(expression) &&
+          expression.properties.length > 0
+        ) {
+          const property = expression.properties[0];
+          if (FeatureFileProcessor.isPropertyAssignment(property)) {
+            const { name, initializer } = property;
+            if (
+              FeatureFileProcessor.isIdentifier(name) &&
+              FeatureFileProcessor.isStringLiteral(initializer) &&
+              name.escapedText === CSF3_EXPORT_TITLE_FIELD
+            ) {
+              callback(initializer.text);
+              return;
+            }
+          }
+        }
+      }
+
       ts.forEachChild(node, walkTree);
     };