forked from projectmoon/tenebrous-dicebot
534 lines
11 KiB
JavaScript
534 lines
11 KiB
JavaScript
|
'use strict';
|
||
|
|
||
|
var use = require('use');
|
||
|
var util = require('util');
|
||
|
var Cache = require('map-cache');
|
||
|
var define = require('define-property');
|
||
|
var debug = require('debug')('snapdragon:parser');
|
||
|
var Position = require('./position');
|
||
|
var utils = require('./utils');
|
||
|
|
||
|
/**
|
||
|
* Create a new `Parser` with the given `input` and `options`.
|
||
|
* @param {String} `input`
|
||
|
* @param {Object} `options`
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
function Parser(options) {
|
||
|
debug('initializing', __filename);
|
||
|
this.options = utils.extend({source: 'string'}, options);
|
||
|
this.init(this.options);
|
||
|
use(this);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Prototype methods
|
||
|
*/
|
||
|
|
||
|
Parser.prototype = {
|
||
|
constructor: Parser,
|
||
|
|
||
|
init: function(options) {
|
||
|
this.orig = '';
|
||
|
this.input = '';
|
||
|
this.parsed = '';
|
||
|
|
||
|
this.column = 1;
|
||
|
this.line = 1;
|
||
|
|
||
|
this.regex = new Cache();
|
||
|
this.errors = this.errors || [];
|
||
|
this.parsers = this.parsers || {};
|
||
|
this.types = this.types || [];
|
||
|
this.sets = this.sets || {};
|
||
|
this.fns = this.fns || [];
|
||
|
this.currentType = 'root';
|
||
|
|
||
|
var pos = this.position();
|
||
|
this.bos = pos({type: 'bos', val: ''});
|
||
|
|
||
|
this.ast = {
|
||
|
type: 'root',
|
||
|
errors: this.errors,
|
||
|
nodes: [this.bos]
|
||
|
};
|
||
|
|
||
|
define(this.bos, 'parent', this.ast);
|
||
|
this.nodes = [this.ast];
|
||
|
|
||
|
this.count = 0;
|
||
|
this.setCount = 0;
|
||
|
this.stack = [];
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Throw a formatted error with the cursor column and `msg`.
|
||
|
* @param {String} `msg` Message to use in the Error.
|
||
|
*/
|
||
|
|
||
|
error: function(msg, node) {
|
||
|
var pos = node.position || {start: {column: 0, line: 0}};
|
||
|
var line = pos.start.line;
|
||
|
var column = pos.start.column;
|
||
|
var source = this.options.source;
|
||
|
|
||
|
var message = source + ' <line:' + line + ' column:' + column + '>: ' + msg;
|
||
|
var err = new Error(message);
|
||
|
err.source = source;
|
||
|
err.reason = msg;
|
||
|
err.pos = pos;
|
||
|
|
||
|
if (this.options.silent) {
|
||
|
this.errors.push(err);
|
||
|
} else {
|
||
|
throw err;
|
||
|
}
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Define a non-enumberable property on the `Parser` instance.
|
||
|
*
|
||
|
* ```js
|
||
|
* parser.define('foo', 'bar');
|
||
|
* ```
|
||
|
* @name .define
|
||
|
* @param {String} `key` propery name
|
||
|
* @param {any} `val` property value
|
||
|
* @return {Object} Returns the Parser instance for chaining.
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
define: function(key, val) {
|
||
|
define(this, key, val);
|
||
|
return this;
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Mark position and patch `node.position`.
|
||
|
*/
|
||
|
|
||
|
position: function() {
|
||
|
var start = { line: this.line, column: this.column };
|
||
|
var self = this;
|
||
|
|
||
|
return function(node) {
|
||
|
define(node, 'position', new Position(start, self));
|
||
|
return node;
|
||
|
};
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Set parser `name` with the given `fn`
|
||
|
* @param {String} `name`
|
||
|
* @param {Function} `fn`
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
set: function(type, fn) {
|
||
|
if (this.types.indexOf(type) === -1) {
|
||
|
this.types.push(type);
|
||
|
}
|
||
|
this.parsers[type] = fn.bind(this);
|
||
|
return this;
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Get parser `name`
|
||
|
* @param {String} `name`
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
get: function(name) {
|
||
|
return this.parsers[name];
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Push a `token` onto the `type` stack.
|
||
|
*
|
||
|
* @param {String} `type`
|
||
|
* @return {Object} `token`
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
push: function(type, token) {
|
||
|
this.sets[type] = this.sets[type] || [];
|
||
|
this.count++;
|
||
|
this.stack.push(token);
|
||
|
return this.sets[type].push(token);
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Pop a token off of the `type` stack
|
||
|
* @param {String} `type`
|
||
|
* @returns {Object} Returns a token
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
pop: function(type) {
|
||
|
this.sets[type] = this.sets[type] || [];
|
||
|
this.count--;
|
||
|
this.stack.pop();
|
||
|
return this.sets[type].pop();
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Return true if inside a `stack` node. Types are `braces`, `parens` or `brackets`.
|
||
|
*
|
||
|
* @param {String} `type`
|
||
|
* @return {Boolean}
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
isInside: function(type) {
|
||
|
this.sets[type] = this.sets[type] || [];
|
||
|
return this.sets[type].length > 0;
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Return true if `node` is the given `type`.
|
||
|
*
|
||
|
* ```js
|
||
|
* parser.isType(node, 'brace');
|
||
|
* ```
|
||
|
* @param {Object} `node`
|
||
|
* @param {String} `type`
|
||
|
* @return {Boolean}
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
isType: function(node, type) {
|
||
|
return node && node.type === type;
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Get the previous AST node
|
||
|
* @return {Object}
|
||
|
*/
|
||
|
|
||
|
prev: function(n) {
|
||
|
return this.stack.length > 0
|
||
|
? utils.last(this.stack, n)
|
||
|
: utils.last(this.nodes, n);
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Update line and column based on `str`.
|
||
|
*/
|
||
|
|
||
|
consume: function(len) {
|
||
|
this.input = this.input.substr(len);
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Update column based on `str`.
|
||
|
*/
|
||
|
|
||
|
updatePosition: function(str, len) {
|
||
|
var lines = str.match(/\n/g);
|
||
|
if (lines) this.line += lines.length;
|
||
|
var i = str.lastIndexOf('\n');
|
||
|
this.column = ~i ? len - i : this.column + len;
|
||
|
this.parsed += str;
|
||
|
this.consume(len);
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Match `regex`, return captures, and update the cursor position by `match[0]` length.
|
||
|
* @param {RegExp} `regex`
|
||
|
* @return {Object}
|
||
|
*/
|
||
|
|
||
|
match: function(regex) {
|
||
|
var m = regex.exec(this.input);
|
||
|
if (m) {
|
||
|
this.updatePosition(m[0], m[0].length);
|
||
|
return m;
|
||
|
}
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Capture `type` with the given regex.
|
||
|
* @param {String} `type`
|
||
|
* @param {RegExp} `regex`
|
||
|
* @return {Function}
|
||
|
*/
|
||
|
|
||
|
capture: function(type, regex) {
|
||
|
if (typeof regex === 'function') {
|
||
|
return this.set.apply(this, arguments);
|
||
|
}
|
||
|
|
||
|
this.regex.set(type, regex);
|
||
|
this.set(type, function() {
|
||
|
var parsed = this.parsed;
|
||
|
var pos = this.position();
|
||
|
var m = this.match(regex);
|
||
|
if (!m || !m[0]) return;
|
||
|
|
||
|
var prev = this.prev();
|
||
|
var node = pos({
|
||
|
type: type,
|
||
|
val: m[0],
|
||
|
parsed: parsed,
|
||
|
rest: this.input
|
||
|
});
|
||
|
|
||
|
if (m[1]) {
|
||
|
node.inner = m[1];
|
||
|
}
|
||
|
|
||
|
define(node, 'inside', this.stack.length > 0);
|
||
|
define(node, 'parent', prev);
|
||
|
prev.nodes.push(node);
|
||
|
}.bind(this));
|
||
|
return this;
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Create a parser with open and close for parens,
|
||
|
* brackets or braces
|
||
|
*/
|
||
|
|
||
|
capturePair: function(type, openRegex, closeRegex, fn) {
|
||
|
this.sets[type] = this.sets[type] || [];
|
||
|
|
||
|
/**
|
||
|
* Open
|
||
|
*/
|
||
|
|
||
|
this.set(type + '.open', function() {
|
||
|
var parsed = this.parsed;
|
||
|
var pos = this.position();
|
||
|
var m = this.match(openRegex);
|
||
|
if (!m || !m[0]) return;
|
||
|
|
||
|
var val = m[0];
|
||
|
this.setCount++;
|
||
|
this.specialChars = true;
|
||
|
var open = pos({
|
||
|
type: type + '.open',
|
||
|
val: val,
|
||
|
rest: this.input
|
||
|
});
|
||
|
|
||
|
if (typeof m[1] !== 'undefined') {
|
||
|
open.inner = m[1];
|
||
|
}
|
||
|
|
||
|
var prev = this.prev();
|
||
|
var node = pos({
|
||
|
type: type,
|
||
|
nodes: [open]
|
||
|
});
|
||
|
|
||
|
define(node, 'rest', this.input);
|
||
|
define(node, 'parsed', parsed);
|
||
|
define(node, 'prefix', m[1]);
|
||
|
define(node, 'parent', prev);
|
||
|
define(open, 'parent', node);
|
||
|
|
||
|
if (typeof fn === 'function') {
|
||
|
fn.call(this, open, node);
|
||
|
}
|
||
|
|
||
|
this.push(type, node);
|
||
|
prev.nodes.push(node);
|
||
|
});
|
||
|
|
||
|
/**
|
||
|
* Close
|
||
|
*/
|
||
|
|
||
|
this.set(type + '.close', function() {
|
||
|
var pos = this.position();
|
||
|
var m = this.match(closeRegex);
|
||
|
if (!m || !m[0]) return;
|
||
|
|
||
|
var parent = this.pop(type);
|
||
|
var node = pos({
|
||
|
type: type + '.close',
|
||
|
rest: this.input,
|
||
|
suffix: m[1],
|
||
|
val: m[0]
|
||
|
});
|
||
|
|
||
|
if (!this.isType(parent, type)) {
|
||
|
if (this.options.strict) {
|
||
|
throw new Error('missing opening "' + type + '"');
|
||
|
}
|
||
|
|
||
|
this.setCount--;
|
||
|
node.escaped = true;
|
||
|
return node;
|
||
|
}
|
||
|
|
||
|
if (node.suffix === '\\') {
|
||
|
parent.escaped = true;
|
||
|
node.escaped = true;
|
||
|
}
|
||
|
|
||
|
parent.nodes.push(node);
|
||
|
define(node, 'parent', parent);
|
||
|
});
|
||
|
|
||
|
return this;
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Capture end-of-string
|
||
|
*/
|
||
|
|
||
|
eos: function() {
|
||
|
var pos = this.position();
|
||
|
if (this.input) return;
|
||
|
var prev = this.prev();
|
||
|
|
||
|
while (prev.type !== 'root' && !prev.visited) {
|
||
|
if (this.options.strict === true) {
|
||
|
throw new SyntaxError('invalid syntax:' + util.inspect(prev, null, 2));
|
||
|
}
|
||
|
|
||
|
if (!hasDelims(prev)) {
|
||
|
prev.parent.escaped = true;
|
||
|
prev.escaped = true;
|
||
|
}
|
||
|
|
||
|
visit(prev, function(node) {
|
||
|
if (!hasDelims(node.parent)) {
|
||
|
node.parent.escaped = true;
|
||
|
node.escaped = true;
|
||
|
}
|
||
|
});
|
||
|
|
||
|
prev = prev.parent;
|
||
|
}
|
||
|
|
||
|
var tok = pos({
|
||
|
type: 'eos',
|
||
|
val: this.append || ''
|
||
|
});
|
||
|
|
||
|
define(tok, 'parent', this.ast);
|
||
|
return tok;
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Run parsers to advance the cursor position
|
||
|
*/
|
||
|
|
||
|
next: function() {
|
||
|
var parsed = this.parsed;
|
||
|
var len = this.types.length;
|
||
|
var idx = -1;
|
||
|
var tok;
|
||
|
|
||
|
while (++idx < len) {
|
||
|
if ((tok = this.parsers[this.types[idx]].call(this))) {
|
||
|
define(tok, 'rest', this.input);
|
||
|
define(tok, 'parsed', parsed);
|
||
|
this.last = tok;
|
||
|
return tok;
|
||
|
}
|
||
|
}
|
||
|
},
|
||
|
|
||
|
/**
|
||
|
* Parse the given string.
|
||
|
* @return {Array}
|
||
|
*/
|
||
|
|
||
|
parse: function(input) {
|
||
|
if (typeof input !== 'string') {
|
||
|
throw new TypeError('expected a string');
|
||
|
}
|
||
|
|
||
|
this.init(this.options);
|
||
|
this.orig = input;
|
||
|
this.input = input;
|
||
|
var self = this;
|
||
|
|
||
|
function parse() {
|
||
|
// check input before calling `.next()`
|
||
|
input = self.input;
|
||
|
|
||
|
// get the next AST ndoe
|
||
|
var node = self.next();
|
||
|
if (node) {
|
||
|
var prev = self.prev();
|
||
|
if (prev) {
|
||
|
define(node, 'parent', prev);
|
||
|
if (prev.nodes) {
|
||
|
prev.nodes.push(node);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (self.sets.hasOwnProperty(prev.type)) {
|
||
|
self.currentType = prev.type;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// if we got here but input is not changed, throw an error
|
||
|
if (self.input && input === self.input) {
|
||
|
throw new Error('no parsers registered for: "' + self.input.slice(0, 5) + '"');
|
||
|
}
|
||
|
}
|
||
|
|
||
|
while (this.input) parse();
|
||
|
if (this.stack.length && this.options.strict) {
|
||
|
var node = this.stack.pop();
|
||
|
throw this.error('missing opening ' + node.type + ': "' + this.orig + '"');
|
||
|
}
|
||
|
|
||
|
var eos = this.eos();
|
||
|
var tok = this.prev();
|
||
|
if (tok.type !== 'eos') {
|
||
|
this.ast.nodes.push(eos);
|
||
|
}
|
||
|
|
||
|
return this.ast;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
/**
|
||
|
* Visit `node` with the given `fn`
|
||
|
*/
|
||
|
|
||
|
function visit(node, fn) {
|
||
|
if (!node.visited) {
|
||
|
define(node, 'visited', true);
|
||
|
return node.nodes ? mapVisit(node.nodes, fn) : fn(node);
|
||
|
}
|
||
|
return node;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Map visit over array of `nodes`.
|
||
|
*/
|
||
|
|
||
|
function mapVisit(nodes, fn) {
|
||
|
var len = nodes.length;
|
||
|
var idx = -1;
|
||
|
while (++idx < len) {
|
||
|
visit(nodes[idx], fn);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
function hasOpen(node) {
|
||
|
return node.nodes && node.nodes[0].type === (node.type + '.open');
|
||
|
}
|
||
|
|
||
|
function hasClose(node) {
|
||
|
return node.nodes && utils.last(node.nodes).type === (node.type + '.close');
|
||
|
}
|
||
|
|
||
|
function hasDelims(node) {
|
||
|
return hasOpen(node) && hasClose(node);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Expose `Parser`
|
||
|
*/
|
||
|
|
||
|
module.exports = Parser;
|