fix(mini-runner): 修复页面 hooks config 失效问题

NervJS · Jan 7, 2020 · 545f327 · 545f327
1 parent 376138b
commit 545f327
Show file tree

Hide file tree

Showing 2 changed files with 71 additions and 1 deletion.
diff --git a/packages/taro-mini-runner/src/loaders/fileParseLoader.ts b/packages/taro-mini-runner/src/loaders/fileParseLoader.ts
@@ -4,7 +4,7 @@ import { getOptions } from 'loader-utils'
 import { transform, transformFromAst } from 'babel-core'
 import * as t from 'babel-types'
 import generate from 'better-babel-generator'
-import traverse from 'babel-traverse'
+import traverse, { NodePath } from 'babel-traverse'
 import * as _ from 'lodash'
 
 import {
@@ -49,6 +49,8 @@ function processAst (
   let hasComponentDidHide
   let hasComponentDidShow
   let hasComponentWillMount
+  let needSetConfigFromHooks = false
+  let configFromHooks
   if (isQuickApp) {
     cannotRemoves.push(taroJsComponents)
   }
@@ -354,6 +356,19 @@ function processAst (
       }
     },
 
+    AssignmentExpression (astPath) {
+      const node = astPath.node
+      const left = node.left
+      if (t.isMemberExpression(left) && t.isIdentifier(left.object)) {
+        if (left.object.name === componentClassName
+            && t.isIdentifier(left.property)
+            && left.property.name === 'config') {
+          needSetConfigFromHooks = true
+          configFromHooks = node.right
+        }
+      }
+    },
+
     Program: {
       exit (astPath) {
         astPath.traverse({
@@ -398,6 +413,15 @@ function processAst (
                   `Taro.eventCenter.off('TaroEvent:setNavigationBar')`
               )]), false, false))
             }
+            if (needSetConfigFromHooks) {
+              const classPath = astPath.findParent((p: NodePath<t.Node>) => p.isClassExpression() || p.isClassDeclaration()) as NodePath<t.ClassDeclaration>
+              classPath.node.body.body.unshift(
+                t.classProperty(
+                  t.identifier('config'),
+                  configFromHooks as t.ObjectExpression
+                )
+              )
+            }
           },
           ClassMethod (astPath) {
             if (isQuickApp) {

diff --git a/packages/taro-mini-runner/src/utils/parseAst.ts b/packages/taro-mini-runner/src/utils/parseAst.ts
@@ -19,6 +19,7 @@ export default function parseAst (
   let hasEnablePageScroll
   const taroSelfComponents = new Set<string>()
   const isQuickApp = buildAdapter === BUILD_TYPES.QUICKAPP
+  let componentClassName: string = ''
 
   const newAst = transformFromAst(ast, '', {
     plugins: [
@@ -62,6 +63,40 @@ export default function parseAst (
               }
             }
           })
+          if (node.id === null) {
+            componentClassName = '_TaroComponentClass'
+          } else if (node.id.name === 'App') {
+            componentClassName = '_App'
+          } else {
+            componentClassName = node.id.name
+          }
+        }
+      }
+    },
+    ClassExpression (astPath) {
+      const node = astPath.node
+      if (node.superClass) {
+        let hasCreateData = false
+        astPath.traverse({
+          ClassMethod (astPath) {
+            if (astPath.get('key').isIdentifier({ name: '_createData' })) {
+              hasCreateData = true
+            }
+          }
+        })
+        if (hasCreateData) {
+          if (node.id === null) {
+            const parentNode = astPath.parentPath.node as any
+            if (t.isVariableDeclarator(astPath.parentPath)) {
+              componentClassName = parentNode.id.name
+            } else {
+              componentClassName = '_TaroComponentClass'
+            }
+          } else if (node.id.name === 'App') {
+            componentClassName = '_App'
+          } else {
+            componentClassName = node.id.name
+          }
         }
       }
     },
@@ -115,6 +150,17 @@ export default function parseAst (
           astPath.remove()
         }
       }
+    },
+    AssignmentExpression (astPath) {
+      const node = astPath.node
+      const left = node.left
+      if (t.isMemberExpression(left) && t.isIdentifier(left.object)) {
+        if (left.object.name === componentClassName
+            && t.isIdentifier(left.property)
+            && left.property.name === 'config') {
+          configObj = traverseObjectNode(node.right, buildAdapter)
+        }
+      }
     }
   })