123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282 |
- 'use strict';
-
- var utils = require('./utils');
-
- module.exports = function(braces, options) {
- braces.compiler
-
- /**
- * bos
- */
-
- .set('bos', function() {
- if (this.output) return;
- this.ast.queue = isEscaped(this.ast) ? [this.ast.val] : [];
- this.ast.count = 1;
- })
-
- /**
- * Square brackets
- */
-
- .set('bracket', function(node) {
- var close = node.close;
- var open = !node.escaped ? '[' : '\\[';
- var negated = node.negated;
- var inner = node.inner;
-
- inner = inner.replace(/\\(?=[\\\w]|$)/g, '\\\\');
- if (inner === ']-') {
- inner = '\\]\\-';
- }
-
- if (negated && inner.indexOf('.') === -1) {
- inner += '.';
- }
- if (negated && inner.indexOf('/') === -1) {
- inner += '/';
- }
-
- var val = open + negated + inner + close;
- var queue = node.parent.queue;
- var last = utils.arrayify(queue.pop());
-
- queue.push(utils.join(last, val));
- queue.push.apply(queue, []);
- })
-
- /**
- * Brace
- */
-
- .set('brace', function(node) {
- node.queue = isEscaped(node) ? [node.val] : [];
- node.count = 1;
- return this.mapVisit(node.nodes);
- })
-
- /**
- * Open
- */
-
- .set('brace.open', function(node) {
- node.parent.open = node.val;
- })
-
- /**
- * Inner
- */
-
- .set('text', function(node) {
- var queue = node.parent.queue;
- var escaped = node.escaped;
- var segs = [node.val];
-
- if (node.optimize === false) {
- options = utils.extend({}, options, {optimize: false});
- }
-
- if (node.multiplier > 1) {
- node.parent.count *= node.multiplier;
- }
-
- if (options.quantifiers === true && utils.isQuantifier(node.val)) {
- escaped = true;
-
- } else if (node.val.length > 1) {
- if (isType(node.parent, 'brace') && !isEscaped(node)) {
- var expanded = utils.expand(node.val, options);
- segs = expanded.segs;
-
- if (expanded.isOptimized) {
- node.parent.isOptimized = true;
- }
-
- // if nothing was expanded, we probably have a literal brace
- if (!segs.length) {
- var val = (expanded.val || node.val);
- if (options.unescape !== false) {
- // unescape unexpanded brace sequence/set separators
- val = val.replace(/\\([,.])/g, '$1');
- // strip quotes
- val = val.replace(/["'`]/g, '');
- }
-
- segs = [val];
- escaped = true;
- }
- }
-
- } else if (node.val === ',') {
- if (options.expand) {
- node.parent.queue.push(['']);
- segs = [''];
- } else {
- segs = ['|'];
- }
- } else {
- escaped = true;
- }
-
- if (escaped && isType(node.parent, 'brace')) {
- if (node.parent.nodes.length <= 4 && node.parent.count === 1) {
- node.parent.escaped = true;
- } else if (node.parent.length <= 3) {
- node.parent.escaped = true;
- }
- }
-
- if (!hasQueue(node.parent)) {
- node.parent.queue = segs;
- return;
- }
-
- var last = utils.arrayify(queue.pop());
- if (node.parent.count > 1 && options.expand) {
- last = multiply(last, node.parent.count);
- node.parent.count = 1;
- }
-
- queue.push(utils.join(utils.flatten(last), segs.shift()));
- queue.push.apply(queue, segs);
- })
-
- /**
- * Close
- */
-
- .set('brace.close', function(node) {
- var queue = node.parent.queue;
- var prev = node.parent.parent;
- var last = prev.queue.pop();
- var open = node.parent.open;
- var close = node.val;
-
- if (open && close && isOptimized(node, options)) {
- open = '(';
- close = ')';
- }
-
- // if a close brace exists, and the previous segment is one character
- // don't wrap the result in braces or parens
- var ele = utils.last(queue);
- if (node.parent.count > 1 && options.expand) {
- ele = multiply(queue.pop(), node.parent.count);
- node.parent.count = 1;
- queue.push(ele);
- }
-
- if (close && typeof ele === 'string' && ele.length === 1) {
- open = '';
- close = '';
- }
-
- if ((isLiteralBrace(node, options) || noInner(node)) && !node.parent.hasEmpty) {
- queue.push(utils.join(open, queue.pop() || ''));
- queue = utils.flatten(utils.join(queue, close));
- }
-
- if (typeof last === 'undefined') {
- prev.queue = [queue];
- } else {
- prev.queue.push(utils.flatten(utils.join(last, queue)));
- }
- })
-
- /**
- * eos
- */
-
- .set('eos', function(node) {
- if (this.input) return;
-
- if (options.optimize !== false) {
- this.output = utils.last(utils.flatten(this.ast.queue));
- } else if (Array.isArray(utils.last(this.ast.queue))) {
- this.output = utils.flatten(this.ast.queue.pop());
- } else {
- this.output = utils.flatten(this.ast.queue);
- }
-
- if (node.parent.count > 1 && options.expand) {
- this.output = multiply(this.output, node.parent.count);
- }
-
- this.output = utils.arrayify(this.output);
- this.ast.queue = [];
- });
-
- };
-
- /**
- * Multiply the segments in the current brace level
- */
-
- function multiply(queue, n, options) {
- return utils.flatten(utils.repeat(utils.arrayify(queue), n));
- }
-
- /**
- * Return true if `node` is escaped
- */
-
- function isEscaped(node) {
- return node.escaped === true;
- }
-
- /**
- * Returns true if regex parens should be used for sets. If the parent `type`
- * is not `brace`, then we're on a root node, which means we should never
- * expand segments and open/close braces should be `{}` (since this indicates
- * a brace is missing from the set)
- */
-
- function isOptimized(node, options) {
- if (node.parent.isOptimized) return true;
- return isType(node.parent, 'brace')
- && !isEscaped(node.parent)
- && options.expand !== true;
- }
-
- /**
- * Returns true if the value in `node` should be wrapped in a literal brace.
- * @return {Boolean}
- */
-
- function isLiteralBrace(node, options) {
- return isEscaped(node.parent) || options.optimize !== false;
- }
-
- /**
- * Returns true if the given `node` does not have an inner value.
- * @return {Boolean}
- */
-
- function noInner(node, type) {
- if (node.parent.queue.length === 1) {
- return true;
- }
- var nodes = node.parent.nodes;
- return nodes.length === 3
- && isType(nodes[0], 'brace.open')
- && !isType(nodes[1], 'text')
- && isType(nodes[2], 'brace.close');
- }
-
- /**
- * Returns true if the given `node` is the given `type`
- * @return {Boolean}
- */
-
- function isType(node, type) {
- return typeof node !== 'undefined' && node.type === type;
- }
-
- /**
- * Returns true if the given `node` has a non-empty queue.
- * @return {Boolean}
- */
-
- function hasQueue(node) {
- return Array.isArray(node.queue) && node.queue.length;
- }
|