将所有变量声明包装在功能中

Question

我有一个JavaScript函数声明为字符串（从function.tostring），我想用函数包装所有变量声明（也在JavaScript中），例如 const value = 42 to const value = wrapper（42）。

首先，我想到使用正直值来获取原始值和位置，然后用包装值替换它们，但是由于需要考虑多行字符串和对象之类的事物，因此正则是非常复杂的。使用Regex也会影响其他人为该项目做出贡献的便利性。

之后，我研究了为此使用一个模块，我找到了橡子（由Babel，Svelte。将JavaScript解析为estree，javascript抽象语法树的规格）： https://github.com/acornjs/acorn ，但是我找不到将Estree解析回Javascript函数声明的方法，在进行修改后声明。

是否有一种方法可以将estree解析回功能或其他更好的解决方案？

Answer 1

您实际上不需要将树将树重新串起来的函数。相反，请注意应该发生更改的位置 offsets ，然后不要将更改应用于原始字符串中。

这是带有橡子API的演示：

function test () { // The function we want to tamper with
    const value = 42, prefix = "prefix";
    
    let x = 3;
    for (let i = 0; i < 10; i++) {
        x = (x * 997 + value) % 1000;
    }
    return prefix + " " + x;
}

function addInitWrappers(str) { // Returns an updated string
    let ast = acorn.parse(str, {ecmaVersion: 2020});
    
    function* iter(node) {
        if (Object(node) !== node) return; // Primitive
        if (node.type == "VariableDeclaration" && node.kind == "const") {
            for (let {init} of node.declarations) {
                yield init; // yield the offset where this initialisation occurs
            }
        }
        for (let value of Object.values(node)) {
            yield* iter(value);
        }
    }
    
    // Inject the wrapper -- starting at the back
    for (let {start, end} of [...iter(ast)].reverse()) {
        str = str.slice(0, start) + "wrapper(" + str.slice(start, end) + ")" + str.slice(end);
    }
    return str;
}

function wrapper(value) { // A wrapper function to demo with
    return value + 1;
}

console.log("before wrapping test() returns:", test());
let str = test.toString();
str = addInitWrappers(str);
eval(str); // Override the test function with its new definition
console.log("after wrapping test() returns:", test());

<script src="https://cdnjs.cloudflare.com/ajax/libs/acorn/8.7.1/acorn.min.js"></script>

Answer 2

const {code} = putout('const a = 42', {
    plugins: [
        ['wrap', {
            report: () => `Add wrap()`,
            replace: () => ({
                'const __a = __b': 'const __a = wrap(__b)',
            })
        }]
    ]
});

console.log(code);

<script src="https://cdn.jsdelivr.net/npm/@putout/[email protected]/bundle/putout-iife.js"></script>

是：

import {parse, print} from "@putout/recast";
console.log(print(parse(source)).code);

您可以使用 recast 要打印结果，它将 支持babel，acorn，甚至typescript解析器。

但是最简单的方法是使用