Merge branch 'master' of https://github.com/didi/mpx

Author: hiyuki

packages/webpack-plugin/lib/template-compiler/compiler.js

@@ -176,11 +176,11 @@ const i18nWxsPath = normalize.lib('runtime/i18n.wxs')
 const i18nWxsLoaderPath = normalize.lib('wxs/i18n-loader.js')
 // 添加~前缀避免wxs绝对路径在存在projectRoot时被拼接为错误路径
 const i18nWxsRequest = '~' + i18nWxsLoaderPath + '!' + i18nWxsPath
-const i18nModuleName = '_i'
+const i18nModuleName = '_i_'
 const stringifyWxsPath = '~' + normalize.lib('runtime/stringify.wxs')
-const stringifyModuleName = '_s'
+const stringifyModuleName = '_s_'
 const optionalChainWxsPath = '~' + normalize.lib('runtime/oc.wxs')
-const optionalChainWxsName = '_oc' // 改成_oc解决web下_o重名问题
+const optionalChainWxsName = '_oc_' // 改成_oc解决web下_o重名问题
 
 const tagRES = /(\{\{(?:.|\n|\r)+?\}\})(?!})/
 const tagRE = /\{\{((?:.|\n|\r)+?)\}\}(?!})/

packages/webpack-plugin/lib/utils/chain-assign.js

@@ -0,0 +1,47 @@
+/**
+ * 链式合并方法的工具函数
+ *
+ * 在多条件分支下使用 Object.assign 会导致同名方法被覆盖，
+ * 这个函数通过创建组合函数来确保所有方法都能按顺序执行。
+ *
+ * @param {Object} target - 目标 visitor 对象
+ * @param {Object} source - 要链式分配的 visitor 方法对象
+ *
+ * @example
+ * const visitor = {}
+ *
+ * // 第一次合并
+ * chainAssign(visitor, {
+ *   CallExpression(path) {
+ *     console.log('第一个处理器')
+ *   }
+ * })
+ *
+ * // 第二次合并 - 不会覆盖，而是组合执行
+ * chainAssign(visitor, {
+ *   CallExpression(path) {
+ *     console.log('第二个处理器')
+ *   }
+ * })
+ *
+ * // 执行时会依次输出:
+ * // 第一个处理器
+ * // 第二个处理器
+ */
+module.exports = function chainAssign (target, source) {
+  for (const [key, value] of Object.entries(source)) {
+    if (target[key]) {
+      // 如果已存在同名方法，创建组合函数依次执行
+      const originalMethod = target[key]
+      target[key] = function (path) {
+        originalMethod.call(this, path)
+        // 只有当节点没有停止遍历或被移除时才继续执行
+        if (!path.removed && !path.shouldStop) {
+          value.call(this, path)
+        }
+      }
+    } else {
+      target[key] = value
+    }
+  }
+}

packages/webpack-plugin/lib/wxs/pre-loader.js

@@ -4,6 +4,7 @@ const t = require('@babel/types')
 const generate = require('@babel/generator').default
 const parseRequest = require('../utils/parse-request')
 const isEmptyObject = require('../utils/is-empty-object')
+const chainAssign = require('../utils/chain-assign')
 const parseQuery = require('loader-utils').parseQuery
 
 module.exports = function (content) {
@@ -30,8 +31,7 @@ module.exports = function (content) {
         '  __mpx_args__[i] = arguments[i];\n' +
         '}'
       ).program.body
-      // todo Object.assign可能会覆盖，未来存在非预期的覆盖case时需要改进处理
-      Object.assign(visitor, {
+      chainAssign(visitor, {
         Identifier (path) {
           if (path.node.name === 'arguments') {
             path.node.name = '__mpx_args__'
@@ -66,7 +66,7 @@ module.exports = function (content) {
     }
 
     if (mode !== 'wx') {
-      Object.assign(visitor, {
+      chainAssign(visitor, {
         CallExpression (path) {
           const callee = path.node.callee
           if (t.isIdentifier(callee) && callee.name === 'getRegExp') {
@@ -81,7 +81,7 @@ module.exports = function (content) {
   }
 
   if (mode === 'dd') {
-    Object.assign(visitor, {
+    chainAssign(visitor, {
       MemberExpression (path) {
         const property = path.node.property
         if (
@@ -96,7 +96,7 @@ module.exports = function (content) {
   }
 
   if (!module.wxs) {
-    Object.assign(visitor, {
+    chainAssign(visitor, {
       MemberExpression (path) {
         const property = path.node.property
         if (