294 lines
6.3 KiB
JavaScript
294 lines
6.3 KiB
JavaScript
/*!
|
|
* to-regex-range <https://github.com/jonschlinkert/to-regex-range>
|
|
*
|
|
* Copyright (c) 2015, 2017, Jon Schlinkert.
|
|
* Released under the MIT License.
|
|
*/
|
|
|
|
'use strict';
|
|
|
|
var repeat = require('repeat-string');
|
|
var isNumber = require('is-number');
|
|
var cache = {};
|
|
|
|
function toRegexRange(min, max, options) {
|
|
if (isNumber(min) === false) {
|
|
throw new RangeError('toRegexRange: first argument is invalid.');
|
|
}
|
|
|
|
if (typeof max === 'undefined' || min === max) {
|
|
return String(min);
|
|
}
|
|
|
|
if (isNumber(max) === false) {
|
|
throw new RangeError('toRegexRange: second argument is invalid.');
|
|
}
|
|
|
|
options = options || {};
|
|
var relax = String(options.relaxZeros);
|
|
var shorthand = String(options.shorthand);
|
|
var capture = String(options.capture);
|
|
var key = min + ':' + max + '=' + relax + shorthand + capture;
|
|
if (cache.hasOwnProperty(key)) {
|
|
return cache[key].result;
|
|
}
|
|
|
|
var a = Math.min(min, max);
|
|
var b = Math.max(min, max);
|
|
|
|
if (Math.abs(a - b) === 1) {
|
|
var result = min + '|' + max;
|
|
if (options.capture) {
|
|
return '(' + result + ')';
|
|
}
|
|
return result;
|
|
}
|
|
|
|
var isPadded = padding(min) || padding(max);
|
|
var positives = [];
|
|
var negatives = [];
|
|
|
|
var tok = {min: min, max: max, a: a, b: b};
|
|
if (isPadded) {
|
|
tok.isPadded = isPadded;
|
|
tok.maxLen = String(tok.max).length;
|
|
}
|
|
|
|
if (a < 0) {
|
|
var newMin = b < 0 ? Math.abs(b) : 1;
|
|
var newMax = Math.abs(a);
|
|
negatives = splitToPatterns(newMin, newMax, tok, options);
|
|
a = tok.a = 0;
|
|
}
|
|
|
|
if (b >= 0) {
|
|
positives = splitToPatterns(a, b, tok, options);
|
|
}
|
|
|
|
tok.negatives = negatives;
|
|
tok.positives = positives;
|
|
tok.result = siftPatterns(negatives, positives, options);
|
|
|
|
if (options.capture && (positives.length + negatives.length) > 1) {
|
|
tok.result = '(' + tok.result + ')';
|
|
}
|
|
|
|
cache[key] = tok;
|
|
return tok.result;
|
|
}
|
|
|
|
function siftPatterns(neg, pos, options) {
|
|
var onlyNegative = filterPatterns(neg, pos, '-', false, options) || [];
|
|
var onlyPositive = filterPatterns(pos, neg, '', false, options) || [];
|
|
var intersected = filterPatterns(neg, pos, '-?', true, options) || [];
|
|
var subpatterns = onlyNegative.concat(intersected).concat(onlyPositive);
|
|
return subpatterns.join('|');
|
|
}
|
|
|
|
function splitToRanges(min, max) {
|
|
min = Number(min);
|
|
max = Number(max);
|
|
|
|
var nines = 1;
|
|
var stops = [max];
|
|
var stop = +countNines(min, nines);
|
|
|
|
while (min <= stop && stop <= max) {
|
|
stops = push(stops, stop);
|
|
nines += 1;
|
|
stop = +countNines(min, nines);
|
|
}
|
|
|
|
var zeros = 1;
|
|
stop = countZeros(max + 1, zeros) - 1;
|
|
|
|
while (min < stop && stop <= max) {
|
|
stops = push(stops, stop);
|
|
zeros += 1;
|
|
stop = countZeros(max + 1, zeros) - 1;
|
|
}
|
|
|
|
stops.sort(compare);
|
|
return stops;
|
|
}
|
|
|
|
/**
|
|
* Convert a range to a regex pattern
|
|
* @param {Number} `start`
|
|
* @param {Number} `stop`
|
|
* @return {String}
|
|
*/
|
|
|
|
function rangeToPattern(start, stop, options) {
|
|
if (start === stop) {
|
|
return {pattern: String(start), digits: []};
|
|
}
|
|
|
|
var zipped = zip(String(start), String(stop));
|
|
var len = zipped.length, i = -1;
|
|
|
|
var pattern = '';
|
|
var digits = 0;
|
|
|
|
while (++i < len) {
|
|
var numbers = zipped[i];
|
|
var startDigit = numbers[0];
|
|
var stopDigit = numbers[1];
|
|
|
|
if (startDigit === stopDigit) {
|
|
pattern += startDigit;
|
|
|
|
} else if (startDigit !== '0' || stopDigit !== '9') {
|
|
pattern += toCharacterClass(startDigit, stopDigit);
|
|
|
|
} else {
|
|
digits += 1;
|
|
}
|
|
}
|
|
|
|
if (digits) {
|
|
pattern += options.shorthand ? '\\d' : '[0-9]';
|
|
}
|
|
|
|
return { pattern: pattern, digits: [digits] };
|
|
}
|
|
|
|
function splitToPatterns(min, max, tok, options) {
|
|
var ranges = splitToRanges(min, max);
|
|
var len = ranges.length;
|
|
var idx = -1;
|
|
|
|
var tokens = [];
|
|
var start = min;
|
|
var prev;
|
|
|
|
while (++idx < len) {
|
|
var range = ranges[idx];
|
|
var obj = rangeToPattern(start, range, options);
|
|
var zeros = '';
|
|
|
|
if (!tok.isPadded && prev && prev.pattern === obj.pattern) {
|
|
if (prev.digits.length > 1) {
|
|
prev.digits.pop();
|
|
}
|
|
prev.digits.push(obj.digits[0]);
|
|
prev.string = prev.pattern + toQuantifier(prev.digits);
|
|
start = range + 1;
|
|
continue;
|
|
}
|
|
|
|
if (tok.isPadded) {
|
|
zeros = padZeros(range, tok);
|
|
}
|
|
|
|
obj.string = zeros + obj.pattern + toQuantifier(obj.digits);
|
|
tokens.push(obj);
|
|
start = range + 1;
|
|
prev = obj;
|
|
}
|
|
|
|
return tokens;
|
|
}
|
|
|
|
function filterPatterns(arr, comparison, prefix, intersection, options) {
|
|
var res = [];
|
|
|
|
for (var i = 0; i < arr.length; i++) {
|
|
var tok = arr[i];
|
|
var ele = tok.string;
|
|
|
|
if (options.relaxZeros !== false) {
|
|
if (prefix === '-' && ele.charAt(0) === '0') {
|
|
if (ele.charAt(1) === '{') {
|
|
ele = '0*' + ele.replace(/^0\{\d+\}/, '');
|
|
} else {
|
|
ele = '0*' + ele.slice(1);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!intersection && !contains(comparison, 'string', ele)) {
|
|
res.push(prefix + ele);
|
|
}
|
|
|
|
if (intersection && contains(comparison, 'string', ele)) {
|
|
res.push(prefix + ele);
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
/**
|
|
* Zip strings (`for in` can be used on string characters)
|
|
*/
|
|
|
|
function zip(a, b) {
|
|
var arr = [];
|
|
for (var ch in a) arr.push([a[ch], b[ch]]);
|
|
return arr;
|
|
}
|
|
|
|
function compare(a, b) {
|
|
return a > b ? 1 : b > a ? -1 : 0;
|
|
}
|
|
|
|
function push(arr, ele) {
|
|
if (arr.indexOf(ele) === -1) arr.push(ele);
|
|
return arr;
|
|
}
|
|
|
|
function contains(arr, key, val) {
|
|
for (var i = 0; i < arr.length; i++) {
|
|
if (arr[i][key] === val) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
function countNines(min, len) {
|
|
return String(min).slice(0, -len) + repeat('9', len);
|
|
}
|
|
|
|
function countZeros(integer, zeros) {
|
|
return integer - (integer % Math.pow(10, zeros));
|
|
}
|
|
|
|
function toQuantifier(digits) {
|
|
var start = digits[0];
|
|
var stop = digits[1] ? (',' + digits[1]) : '';
|
|
if (!stop && (!start || start === 1)) {
|
|
return '';
|
|
}
|
|
return '{' + start + stop + '}';
|
|
}
|
|
|
|
function toCharacterClass(a, b) {
|
|
return '[' + a + ((b - a === 1) ? '' : '-') + b + ']';
|
|
}
|
|
|
|
function padding(str) {
|
|
return /^-?(0+)\d/.exec(str);
|
|
}
|
|
|
|
function padZeros(val, tok) {
|
|
if (tok.isPadded) {
|
|
var diff = Math.abs(tok.maxLen - String(val).length);
|
|
switch (diff) {
|
|
case 0:
|
|
return '';
|
|
case 1:
|
|
return '0';
|
|
default: {
|
|
return '0{' + diff + '}';
|
|
}
|
|
}
|
|
}
|
|
return val;
|
|
}
|
|
|
|
/**
|
|
* Expose `toRegexRange`
|
|
*/
|
|
|
|
module.exports = toRegexRange;
|