feat(typescript): support ObjectPattern (#43)

tyankatsu0105 · Sep 13, 2020 · 3fdc74e · 3fdc74e
1 parent 594bc13
commit 3fdc74e
Show file tree

Hide file tree

Showing 3 changed files with 59 additions and 7 deletions.
diff --git a/src/shared/helpers/parse/__test__/fixtures/function.ts b/src/shared/helpers/parse/__test__/fixtures/function.ts
@@ -4,4 +4,21 @@ export const fn2 = (x, y) => x + y, fn3 = () => 1;
 
 export function fn4() { return 1 };
 
-function fn5() { return 1 };
+function fn5() { return 1 };
+
+const includeSomeFn = {
+  fn6() {
+    return 1
+  },
+  fn7() {
+    return 1
+  },
+
+  prop: {
+    insideFn(){
+      return 1
+    }
+  }
+}
+
+export const { fn6, fn7, prop: { insideFn } } = includeSomeFn
diff --git a/src/shared/helpers/parse/__test__/parse-typescript.spec.ts b/src/shared/helpers/parse/__test__/parse-typescript.spec.ts
@@ -36,7 +36,15 @@ describe('parse-typescrript', () => {
       const { ast } = getAST('function.ts');
 
       const { exportItems } = getExportItems(ast);
-      expect(exportItems).toEqual(['fn', 'fn2', 'fn3', 'fn4']);
+      expect(exportItems).toEqual([
+        'fn',
+        'fn2',
+        'fn3',
+        'fn4',
+        'fn6',
+        'fn7',
+        'insideFn',
+      ]);
     });
 
     it('Class', () => {

diff --git a/src/shared/helpers/parse/parse-typescript.ts b/src/shared/helpers/parse/parse-typescript.ts
@@ -8,6 +8,27 @@ import {
 
 import { Generate } from '../../../types';
 
+const traverseObjectPattern = (
+  properties: (
+    | TSESTree.PropertyComputedName
+    | TSESTree.PropertyNonComputedName
+    | TSESTree.RestElement
+  )[],
+  callback: (
+    property: TSESTree.PropertyComputedName | TSESTree.PropertyNonComputedName
+  ) => void
+) => {
+  for (const property of properties) {
+    if (property.type === AST_NODE_TYPES['Property']) {
+      if (property.value.type === AST_NODE_TYPES['Identifier']) {
+        callback(property);
+      } else if (property.value.type === AST_NODE_TYPES['ObjectPattern']) {
+        traverseObjectPattern(property.value.properties, callback);
+      }
+    }
+  }
+};
+
 /**
  * Get names at "Identifier"
  * Only support ES modules
@@ -31,11 +52,17 @@ export const getExportItems = (ast: TSESTree.Program) => {
 
   simpleTraverse(filterdAst, {
     enter(node) {
-      if (
-        node.type === AST_NODE_TYPES['VariableDeclarator'] &&
-        node.id.type === AST_NODE_TYPES['Identifier']
-      ) {
-        exportItems.push(node.id.name);
+      if (node.type === AST_NODE_TYPES['VariableDeclarator']) {
+        if (node.id.type === AST_NODE_TYPES['Identifier']) {
+          exportItems.push(node.id.name);
+        }
+
+        if (node.id.type === AST_NODE_TYPES['ObjectPattern']) {
+          traverseObjectPattern(node.id.properties, (property) => {
+            property.value.type === AST_NODE_TYPES['Identifier'] &&
+              exportItems.push(property.value.name);
+          });
+        }
       }
 
       if (node.type === AST_NODE_TYPES['FunctionDeclaration'] && node.id) {