@vyr/pack-var-rename-plugin 0.0.4

package/index.js

@@ -0,0 +1,74 @@
+const types = require('@babel/types');
+const { parse } = require('@babel/parser')
+const { default: traverse } = require('@babel/traverse')
+const { default: generate } = require('@babel/generator')
+
+module.exports = (packages, outputFile) => {
+    const varRename = (ast, depName) => {
+        // 记录已改过的作用域
+        const scopeRecord = {};
+        // 替换作用域内变量
+        function replaceScopeVarsName(path, name, newName) {
+            // 防止重复修改
+            if (scopeRecord[path.scope.uid]) return;
+            const item = path.scope.bindings[name]
+            // // 变量名应该唯一， 这里用随机字符凑合一下
+            // const newName = randomString()
+            // 替换变量名
+            item.identifier.name = newName
+            // 替换引用
+            item.referencePaths.forEach(function (refItem) {
+                refItem.node.name = newName;
+            })
+            scopeRecord[path.scope.uid] = true
+        }
+        const rebinds = []
+        traverse(ast, {
+            VariableDeclaration(path) {
+                path.node.declarations.forEach(declarator => {
+                    // 检查该声明是否是通过 require 引入 {depName}
+                    if (
+                        declarator.init &&
+                        declarator.init.type === 'CallExpression' &&
+                        declarator.init.callee.name === 'require' &&
+                        declarator.init.arguments.length === 1 &&
+                        declarator.init.arguments[0].value === depName
+                    ) {
+                        const replaceName = `__vyr_${declarator.id.name}`
+                        replaceScopeVarsName(path, declarator.id.name, replaceName)
+                        const newName = `'__${depName.toUpperCase().replace(/-/g, "_")}'`
+                        rebinds.push({ replaceName, member: types.memberExpression(types.identifier('global'), types.identifier(newName), true) })
+                        path.remove()
+                    }
+                });
+            }
+        })
+
+        traverse(ast, {
+            Identifier(path) {
+                for (const rebind of rebinds) {
+                    if (rebind.replaceName !== path.node.name) continue
+                    path.replaceWith(rebind.member)
+                    return
+                }
+            }
+        })
+    }
+
+    const generateBundle = (options, bundle, isWrite) => {
+        const file = bundle[outputFile]
+
+        const ast = parse(file.code)
+
+        for (const _package of packages) varRename(ast, _package)
+
+        const { code } = generate(ast, { jsescOption: { minimal: true } })
+
+        file.code = code
+    }
+
+    return {
+        name: 'varRename',
+        generateBundle,
+    }
+}

package/package.json

@@ -0,0 +1 @@
+{"name":"@vyr/pack-var-rename-plugin","version":"0.0.4","description":"","main":"./index.js","author":"","license":"ISC"}