利用闭包与WeakMap实现可链式调用的JavaScript函数默认参数设置器

function add(a, b) {
    return a + b;
}

// 第一次调用：为add函数设置b的默认值为9
var add_ = defaultMethod(add, { b: 9 });
console.log(add_(10)); // 预期输出 19 (10 + 9)

// 第二次调用：为add_函数（它本身就是defaultMethod的返回结果）设置a=2, b=3的默认值
add_ = defaultMethod(add_, { b: 3, a: 2 });
console.log(add_(10)); // 预期输出 13 (10 (传入a) + 3 (新的b默认值))
console.log(add_());    // 预期输出 5 (2 (a默认值) + 3 (b默认值))

function defaultMethod(func, params) {
    // 尝试从函数字符串中解析参数名
    const funcStr = func.toString();
    const requiredArgs = funcStr
        .slice(funcStr.indexOf('(') + 1, funcStr.indexOf(')'))
        .match(/([^\s,]+)/g) || [];

    return function (...args) {
        let calledArgs = [...args]; // 复制一份，避免直接修改args

        // 填充缺失的默认参数
        for (let i = calledArgs.length; i < requiredArgs.length; i++) {
            if (calledArgs[i] === undefined) {
                calledArgs[i] = params[requiredArgs[i]];
            }
        }
        return func(...calledArgs);
    };
}

function add(a, b) { return a + b; }

// 使用IIFE创建一个闭包，用于封装WeakMap注册表
const defaultMethod = (function () {
    const registry = new WeakMap(); // 声明一个私有的WeakMap注册表

    return function (func, params) {
        let requiredArgs = registry.get(func); // 尝试从注册表中获取参数名

        // 如果func是第一次被defaultMethod处理的原始函数，或者是一个未注册的函数
        if (!requiredArgs) {
            const funcStr = func.toString();
            requiredArgs = funcStr
                .slice(funcStr.indexOf('(') + 1, funcStr.indexOf(')'))
                .match(/([^\s,]+)/g) || [];
        }

        // 返回一个新的修饰函数
        const decoratedFunc = function (...args) {
            let calledArgs = [...args]; // 复制一份，避免直接修改args

            // 填充缺失的默认参数
            for (let i = calledArgs.length; i < requiredArgs.length; i++) {
                if (calledArgs[i] === undefined) {
                    calledArgs[i] = params[requiredArgs[i]];
                }
            }
            return func(...calledArgs);
        };

        // 将新创建的修饰函数及其对应的参数名注册到WeakMap中
        registry.set(decoratedFunc, requiredArgs);
        return decoratedFunc;
    };
})();

console.log("--- 初始设置与调用 ---");
console.log("为add函数设置b的默认值为9");
let add_ = defaultMethod(add, { b: 9 });
console.log("调用 add_(10): 预期 19");
console.log("结果:", add_(10)); // 19 (10 + 9)
console.log("调用 add_(10, 7): 预期 17");
console.log("结果:", add_(10, 7)); // 17 (10 + 7)
console.log("调用 add_(): 预期 NaN");
console.log("结果:", add_()); // NaN (undefined + 9)

console.log("\n--- 链式调用与更新默认值 ---");
console.log("为add_函数（已修饰）设置a=2, b=3的默认值");
add_ = defaultMethod(add_, { b: 3, a: 2 });
console.log("调用 add_(10): 预期 13");
console.log("结果:", add_(10)); // 13 (10 (传入a) + 3 (新的b默认值))
console.log("调用 add_(): 预期 5");
console.log("结果:", add_()); // 5 (2 (a默认值) + 3 (b默认值))

console.log("\n--- 进一步链式调用（无关参数） ---");
console.log("为add_函数设置c=3（c不是add的参数）");
add_ = defaultMethod(add_, { c: 3 }); // 这次调用不会改变a,b的默认行为
console.log("调用 add_(10): 预期 13");
console.log("结果:", add_(10)); // 13 (10 (传入a) + 3 (b默认值))