1
|
'use strict';
|
2
|
|
3
|
var extend = require('extend-shallow');
|
4
|
var Snapdragon = require('snapdragon');
|
5
|
var compilers = require('./compilers');
|
6
|
var parsers = require('./parsers');
|
7
|
var utils = require('./utils');
|
8
|
|
9
|
/**
|
10
|
* Customize Snapdragon parser and renderer
|
11
|
*/
|
12
|
|
13
|
function Braces(options) {
|
14
|
this.options = extend({}, options);
|
15
|
}
|
16
|
|
17
|
/**
|
18
|
* Initialize braces
|
19
|
*/
|
20
|
|
21
|
Braces.prototype.init = function(options) {
|
22
|
if (this.isInitialized) return;
|
23
|
this.isInitialized = true;
|
24
|
var opts = utils.createOptions({}, this.options, options);
|
25
|
this.snapdragon = this.options.snapdragon || new Snapdragon(opts);
|
26
|
this.compiler = this.snapdragon.compiler;
|
27
|
this.parser = this.snapdragon.parser;
|
28
|
|
29
|
compilers(this.snapdragon, opts);
|
30
|
parsers(this.snapdragon, opts);
|
31
|
|
32
|
/**
|
33
|
* Call Snapdragon `.parse` method. When AST is returned, we check to
|
34
|
* see if any unclosed braces are left on the stack and, if so, we iterate
|
35
|
* over the stack and correct the AST so that compilers are called in the correct
|
36
|
* order and unbalance braces are properly escaped.
|
37
|
*/
|
38
|
|
39
|
utils.define(this.snapdragon, 'parse', function(pattern, options) {
|
40
|
var parsed = Snapdragon.prototype.parse.apply(this, arguments);
|
41
|
this.parser.ast.input = pattern;
|
42
|
|
43
|
var stack = this.parser.stack;
|
44
|
while (stack.length) {
|
45
|
addParent({type: 'brace.close', val: ''}, stack.pop());
|
46
|
}
|
47
|
|
48
|
function addParent(node, parent) {
|
49
|
utils.define(node, 'parent', parent);
|
50
|
parent.nodes.push(node);
|
51
|
}
|
52
|
|
53
|
// add non-enumerable parser reference
|
54
|
utils.define(parsed, 'parser', this.parser);
|
55
|
return parsed;
|
56
|
});
|
57
|
};
|
58
|
|
59
|
/**
|
60
|
* Decorate `.parse` method
|
61
|
*/
|
62
|
|
63
|
Braces.prototype.parse = function(ast, options) {
|
64
|
if (ast && typeof ast === 'object' && ast.nodes) return ast;
|
65
|
this.init(options);
|
66
|
return this.snapdragon.parse(ast, options);
|
67
|
};
|
68
|
|
69
|
/**
|
70
|
* Decorate `.compile` method
|
71
|
*/
|
72
|
|
73
|
Braces.prototype.compile = function(ast, options) {
|
74
|
if (typeof ast === 'string') {
|
75
|
ast = this.parse(ast, options);
|
76
|
} else {
|
77
|
this.init(options);
|
78
|
}
|
79
|
return this.snapdragon.compile(ast, options);
|
80
|
};
|
81
|
|
82
|
/**
|
83
|
* Expand
|
84
|
*/
|
85
|
|
86
|
Braces.prototype.expand = function(pattern) {
|
87
|
var ast = this.parse(pattern, {expand: true});
|
88
|
return this.compile(ast, {expand: true});
|
89
|
};
|
90
|
|
91
|
/**
|
92
|
* Optimize
|
93
|
*/
|
94
|
|
95
|
Braces.prototype.optimize = function(pattern) {
|
96
|
var ast = this.parse(pattern, {optimize: true});
|
97
|
return this.compile(ast, {optimize: true});
|
98
|
};
|
99
|
|
100
|
/**
|
101
|
* Expose `Braces`
|
102
|
*/
|
103
|
|
104
|
module.exports = Braces;
|