[d565449] | 1 | var concatMap = require('concat-map');
|
---|
| 2 | var balanced = require('balanced-match');
|
---|
| 3 |
|
---|
| 4 | module.exports = expandTop;
|
---|
| 5 |
|
---|
| 6 | var escSlash = '\0SLASH'+Math.random()+'\0';
|
---|
| 7 | var escOpen = '\0OPEN'+Math.random()+'\0';
|
---|
| 8 | var escClose = '\0CLOSE'+Math.random()+'\0';
|
---|
| 9 | var escComma = '\0COMMA'+Math.random()+'\0';
|
---|
| 10 | var escPeriod = '\0PERIOD'+Math.random()+'\0';
|
---|
| 11 |
|
---|
| 12 | function numeric(str) {
|
---|
| 13 | return parseInt(str, 10) == str
|
---|
| 14 | ? parseInt(str, 10)
|
---|
| 15 | : str.charCodeAt(0);
|
---|
| 16 | }
|
---|
| 17 |
|
---|
| 18 | function escapeBraces(str) {
|
---|
| 19 | return str.split('\\\\').join(escSlash)
|
---|
| 20 | .split('\\{').join(escOpen)
|
---|
| 21 | .split('\\}').join(escClose)
|
---|
| 22 | .split('\\,').join(escComma)
|
---|
| 23 | .split('\\.').join(escPeriod);
|
---|
| 24 | }
|
---|
| 25 |
|
---|
| 26 | function unescapeBraces(str) {
|
---|
| 27 | return str.split(escSlash).join('\\')
|
---|
| 28 | .split(escOpen).join('{')
|
---|
| 29 | .split(escClose).join('}')
|
---|
| 30 | .split(escComma).join(',')
|
---|
| 31 | .split(escPeriod).join('.');
|
---|
| 32 | }
|
---|
| 33 |
|
---|
| 34 |
|
---|
| 35 | // Basically just str.split(","), but handling cases
|
---|
| 36 | // where we have nested braced sections, which should be
|
---|
| 37 | // treated as individual members, like {a,{b,c},d}
|
---|
| 38 | function parseCommaParts(str) {
|
---|
| 39 | if (!str)
|
---|
| 40 | return [''];
|
---|
| 41 |
|
---|
| 42 | var parts = [];
|
---|
| 43 | var m = balanced('{', '}', str);
|
---|
| 44 |
|
---|
| 45 | if (!m)
|
---|
| 46 | return str.split(',');
|
---|
| 47 |
|
---|
| 48 | var pre = m.pre;
|
---|
| 49 | var body = m.body;
|
---|
| 50 | var post = m.post;
|
---|
| 51 | var p = pre.split(',');
|
---|
| 52 |
|
---|
| 53 | p[p.length-1] += '{' + body + '}';
|
---|
| 54 | var postParts = parseCommaParts(post);
|
---|
| 55 | if (post.length) {
|
---|
| 56 | p[p.length-1] += postParts.shift();
|
---|
| 57 | p.push.apply(p, postParts);
|
---|
| 58 | }
|
---|
| 59 |
|
---|
| 60 | parts.push.apply(parts, p);
|
---|
| 61 |
|
---|
| 62 | return parts;
|
---|
| 63 | }
|
---|
| 64 |
|
---|
| 65 | function expandTop(str) {
|
---|
| 66 | if (!str)
|
---|
| 67 | return [];
|
---|
| 68 |
|
---|
| 69 | // I don't know why Bash 4.3 does this, but it does.
|
---|
| 70 | // Anything starting with {} will have the first two bytes preserved
|
---|
| 71 | // but *only* at the top level, so {},a}b will not expand to anything,
|
---|
| 72 | // but a{},b}c will be expanded to [a}c,abc].
|
---|
| 73 | // One could argue that this is a bug in Bash, but since the goal of
|
---|
| 74 | // this module is to match Bash's rules, we escape a leading {}
|
---|
| 75 | if (str.substr(0, 2) === '{}') {
|
---|
| 76 | str = '\\{\\}' + str.substr(2);
|
---|
| 77 | }
|
---|
| 78 |
|
---|
| 79 | return expand(escapeBraces(str), true).map(unescapeBraces);
|
---|
| 80 | }
|
---|
| 81 |
|
---|
| 82 | function identity(e) {
|
---|
| 83 | return e;
|
---|
| 84 | }
|
---|
| 85 |
|
---|
| 86 | function embrace(str) {
|
---|
| 87 | return '{' + str + '}';
|
---|
| 88 | }
|
---|
| 89 | function isPadded(el) {
|
---|
| 90 | return /^-?0\d/.test(el);
|
---|
| 91 | }
|
---|
| 92 |
|
---|
| 93 | function lte(i, y) {
|
---|
| 94 | return i <= y;
|
---|
| 95 | }
|
---|
| 96 | function gte(i, y) {
|
---|
| 97 | return i >= y;
|
---|
| 98 | }
|
---|
| 99 |
|
---|
| 100 | function expand(str, isTop) {
|
---|
| 101 | var expansions = [];
|
---|
| 102 |
|
---|
| 103 | var m = balanced('{', '}', str);
|
---|
| 104 | if (!m || /\$$/.test(m.pre)) return [str];
|
---|
| 105 |
|
---|
| 106 | var isNumericSequence = /^-?\d+\.\.-?\d+(?:\.\.-?\d+)?$/.test(m.body);
|
---|
| 107 | var isAlphaSequence = /^[a-zA-Z]\.\.[a-zA-Z](?:\.\.-?\d+)?$/.test(m.body);
|
---|
| 108 | var isSequence = isNumericSequence || isAlphaSequence;
|
---|
| 109 | var isOptions = m.body.indexOf(',') >= 0;
|
---|
| 110 | if (!isSequence && !isOptions) {
|
---|
| 111 | // {a},b}
|
---|
| 112 | if (m.post.match(/,.*\}/)) {
|
---|
| 113 | str = m.pre + '{' + m.body + escClose + m.post;
|
---|
| 114 | return expand(str);
|
---|
| 115 | }
|
---|
| 116 | return [str];
|
---|
| 117 | }
|
---|
| 118 |
|
---|
| 119 | var n;
|
---|
| 120 | if (isSequence) {
|
---|
| 121 | n = m.body.split(/\.\./);
|
---|
| 122 | } else {
|
---|
| 123 | n = parseCommaParts(m.body);
|
---|
| 124 | if (n.length === 1) {
|
---|
| 125 | // x{{a,b}}y ==> x{a}y x{b}y
|
---|
| 126 | n = expand(n[0], false).map(embrace);
|
---|
| 127 | if (n.length === 1) {
|
---|
| 128 | var post = m.post.length
|
---|
| 129 | ? expand(m.post, false)
|
---|
| 130 | : [''];
|
---|
| 131 | return post.map(function(p) {
|
---|
| 132 | return m.pre + n[0] + p;
|
---|
| 133 | });
|
---|
| 134 | }
|
---|
| 135 | }
|
---|
| 136 | }
|
---|
| 137 |
|
---|
| 138 | // at this point, n is the parts, and we know it's not a comma set
|
---|
| 139 | // with a single entry.
|
---|
| 140 |
|
---|
| 141 | // no need to expand pre, since it is guaranteed to be free of brace-sets
|
---|
| 142 | var pre = m.pre;
|
---|
| 143 | var post = m.post.length
|
---|
| 144 | ? expand(m.post, false)
|
---|
| 145 | : [''];
|
---|
| 146 |
|
---|
| 147 | var N;
|
---|
| 148 |
|
---|
| 149 | if (isSequence) {
|
---|
| 150 | var x = numeric(n[0]);
|
---|
| 151 | var y = numeric(n[1]);
|
---|
| 152 | var width = Math.max(n[0].length, n[1].length)
|
---|
| 153 | var incr = n.length == 3
|
---|
| 154 | ? Math.abs(numeric(n[2]))
|
---|
| 155 | : 1;
|
---|
| 156 | var test = lte;
|
---|
| 157 | var reverse = y < x;
|
---|
| 158 | if (reverse) {
|
---|
| 159 | incr *= -1;
|
---|
| 160 | test = gte;
|
---|
| 161 | }
|
---|
| 162 | var pad = n.some(isPadded);
|
---|
| 163 |
|
---|
| 164 | N = [];
|
---|
| 165 |
|
---|
| 166 | for (var i = x; test(i, y); i += incr) {
|
---|
| 167 | var c;
|
---|
| 168 | if (isAlphaSequence) {
|
---|
| 169 | c = String.fromCharCode(i);
|
---|
| 170 | if (c === '\\')
|
---|
| 171 | c = '';
|
---|
| 172 | } else {
|
---|
| 173 | c = String(i);
|
---|
| 174 | if (pad) {
|
---|
| 175 | var need = width - c.length;
|
---|
| 176 | if (need > 0) {
|
---|
| 177 | var z = new Array(need + 1).join('0');
|
---|
| 178 | if (i < 0)
|
---|
| 179 | c = '-' + z + c.slice(1);
|
---|
| 180 | else
|
---|
| 181 | c = z + c;
|
---|
| 182 | }
|
---|
| 183 | }
|
---|
| 184 | }
|
---|
| 185 | N.push(c);
|
---|
| 186 | }
|
---|
| 187 | } else {
|
---|
| 188 | N = concatMap(n, function(el) { return expand(el, false) });
|
---|
| 189 | }
|
---|
| 190 |
|
---|
| 191 | for (var j = 0; j < N.length; j++) {
|
---|
| 192 | for (var k = 0; k < post.length; k++) {
|
---|
| 193 | var expansion = pre + N[j] + post[k];
|
---|
| 194 | if (!isTop || isSequence || expansion)
|
---|
| 195 | expansions.push(expansion);
|
---|
| 196 | }
|
---|
| 197 | }
|
---|
| 198 |
|
---|
| 199 | return expansions;
|
---|
| 200 | }
|
---|
| 201 |
|
---|