validator-picker/regex_picker.js

validator and picker

const lodash = require('lodash');
const SPECIAL_KEY_NAME = '~~SPECIAL_KEY_NAME~~';

function prepend(v) {
  return lodash.isArray(v) ? [ SPECIAL_KEY_NAME, v.map(prepend) ] : v;
}

/**
 * 正则挑选
 * @param regex {RegExp}
 * @param fields {string[]}
 * @return {function(*, *=): Array}
 */
function regexPicker(regex, fields) {
  const array = lodash.flattenDeep(prepend(fields));

  return function (str, flag = '') {
    const results = [];

    str.replace(
      new RegExp(regex, flag),
      (...args) => {
        let obj = lodash.zipObject(array, args);
        obj[ SPECIAL_KEY_NAME ] = undefined;
        results.push(lodash.omitBy(obj, lodash.isUndefined));
      },
    );

    return results;
  };
}

module.exports = regexPicker;