离梦之殇 如梦之境

/**
 * 通过 `size` 和 `init` 函数来创建一个数组
 * 
 * @param size 数组的大小
 * @param init 初始化函数：(index) => element
 * @returns {*[]}
 */
const buildArray = (size, init) =>
  Array(size).fill().map((_, i) => init(i));

/**
 * 为一个函数动态地设置参数的默认值。
 * 支持复杂对象，支持链式调用，支持绑定
 *
 * 其中的 `getParameterNames` 函数
 * 来自于 https://www.npmjs.com/package/get-parameter-names
 *
 * @param fn 需要设置默认值的函数
 * @param defaultArgs 格式：{ 参数名 或者 参数位置的下标: 默认值，... } 或者使用数组
 * @returns { Function }
 */
function setDefaultArgs (fn, defaultArgs) {
  // See: https://www.npmjs.com/package/get-parameter-names
  function getParameterNames (fn) {
    const COMMENTS = /((\/\/.*$)|(\/\*[\s\S]*?\*\/))/mg;
    const FAT_ARROWS = /=>.*$/mg;
    const DEFAULT_PARAMS = /=[^,]+/mg;

    const code = fn.toString()
      .replace(COMMENTS, '')
      .replace(FAT_ARROWS, '')
      .replace(DEFAULT_PARAMS, '');

    const result = code.slice(code.indexOf('(') + 1, code.indexOf(')'))
      .match(/([^\s,]+)/g);

    return result || [];
  }

  const names = getParameterNames(fn);

  const functionTemplate = (fn, names, defaultArgs) =>
    function ($PARAMS$) {
      const fixedArgs = Array(names.length).fill('_').map((_, i) =>
        arguments[i] || defaultArgs[names[i]] || defaultArgs[i]);
      return fn.apply(this, fixedArgs);
    };

  const template = `"use strict"; return ${functionTemplate}`
    .replace('$PARAMS$', names.join(', '))
    .replace(/^ {4}/gm, '');

  return Function(template)()(fn, names, defaultArgs);
}

/**
 * 基于原型链进行调用的版本、
 *
 * @see setDefaultArgs
 * @param defaultArgs
 * @returns {*}
 */
Function.prototype.setDefaultArgs = function (defaultArgs) {
  return setDefaultArgs(this, defaultArgs);
};