fix(taroize): class 内部 wx. 开头的函数不会转换为 Taro.func()

NervJS · Dec 3, 2018 · 5d01b3c · 5d01b3c
1 parent 0c894bd
commit 5d01b3c
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/packages/taroize/src/script.ts b/packages/taroize/src/script.ts
@@ -144,6 +144,25 @@ function parsePage (
   refId?: Set<string>
 ) {
   const stateKeys: string[] = []
+  path.traverse({
+    CallExpression (path) {
+      const callee = path.get('callee')
+      if (callee.isIdentifier()) {
+        const name = callee.node.name
+        if (name === 'getApp' || name === 'getCurrentPages') {
+          callee.replaceWith(
+            t.memberExpression(t.identifier('Taro'), callee.node)
+          )
+        }
+      }
+      if (callee.isMemberExpression()) {
+        const object = callee.get('object')
+        if (object.isIdentifier({ name: 'wx' })) {
+          object.replaceWith(t.identifier('Taro'))
+        }
+      }
+    }
+  })
   if (refId) {
     refId.forEach(id => {
       if (!stateKeys.includes(id)) {