fix(cli): 修正对是否是 Taro 组件的判断，close #3751

NervJS · Jul 10, 2019 · 0db4742 · 0db4742
1 parent 81d90ac
commit 0db4742
Showing 1 changed file with 2 additions and 26 deletions.
diff --git a/packages/taro-cli/src/mini/component.ts b/packages/taro-cli/src/mini/component.ts
@@ -79,32 +79,8 @@ export function isFileToBeTaroComponent (
   let isTaroComponent = false
 
   traverse(ast, {
-    ClassDeclaration (astPath) {
-      astPath.traverse({
-        ClassMethod (astPath) {
-          if (astPath.get('key').isIdentifier({ name: 'render' })) {
-            astPath.traverse({
-              JSXElement () {
-                isTaroComponent = true
-              }
-            })
-          }
-        }
-      })
-    },
-
-    ClassExpression (astPath) {
-      astPath.traverse({
-        ClassMethod (astPath) {
-          if (astPath.get('key').isIdentifier({ name: 'render' })) {
-            astPath.traverse({
-              JSXElement () {
-                isTaroComponent = true
-              }
-            })
-          }
-        }
-      })
+    JSXElement (astPath) {
+      isTaroComponent = !!astPath.findParent(p => (p.isReturnStatement() || p.isArrowFunctionExpression()))
     }
   })