You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							492 lines
						
					
					
						
							11 KiB
						
					
					
				
			
		
		
	
	
							492 lines
						
					
					
						
							11 KiB
						
					
					
				'use strict'; | 
						|
 | 
						|
var isObject = require('isobject'); | 
						|
var define = require('define-property'); | 
						|
var utils = require('snapdragon-util'); | 
						|
var ownNames; | 
						|
 | 
						|
/** | 
						|
 * Create a new AST `Node` with the given `val` and `type`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var node = new Node('*', 'Star'); | 
						|
 * var node = new Node({type: 'star', val: '*'}); | 
						|
 * ``` | 
						|
 * @name Node | 
						|
 * @param {String|Object} `val` Pass a matched substring, or an object to merge onto the node. | 
						|
 * @param {String} `type` The node type to use when `val` is a string. | 
						|
 * @return {Object} node instance | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
function Node(val, type, parent) { | 
						|
  if (typeof type !== 'string') { | 
						|
    parent = type; | 
						|
    type = null; | 
						|
  } | 
						|
 | 
						|
  define(this, 'parent', parent); | 
						|
  define(this, 'isNode', true); | 
						|
  define(this, 'expect', null); | 
						|
 | 
						|
  if (typeof type !== 'string' && isObject(val)) { | 
						|
    lazyKeys(); | 
						|
    var keys = Object.keys(val); | 
						|
    for (var i = 0; i < keys.length; i++) { | 
						|
      var key = keys[i]; | 
						|
      if (ownNames.indexOf(key) === -1) { | 
						|
        this[key] = val[key]; | 
						|
      } | 
						|
    } | 
						|
  } else { | 
						|
    this.type = type; | 
						|
    this.val = val; | 
						|
  } | 
						|
} | 
						|
 | 
						|
/** | 
						|
 * Returns true if the given value is a node. | 
						|
 * | 
						|
 * ```js | 
						|
 * var Node = require('snapdragon-node'); | 
						|
 * var node = new Node({type: 'foo'}); | 
						|
 * console.log(Node.isNode(node)); //=> true | 
						|
 * console.log(Node.isNode({})); //=> false | 
						|
 * ``` | 
						|
 * @param {Object} `node` | 
						|
 * @returns {Boolean} | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.isNode = function(node) { | 
						|
  return utils.isNode(node); | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Define a non-enumberable property on the node instance. | 
						|
 * Useful for adding properties that shouldn't be extended | 
						|
 * or visible during debugging. | 
						|
 * | 
						|
 * ```js | 
						|
 * var node = new Node(); | 
						|
 * node.define('foo', 'something non-enumerable'); | 
						|
 * ``` | 
						|
 * @param {String} `name` | 
						|
 * @param {any} `val` | 
						|
 * @return {Object} returns the node instance | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.define = function(name, val) { | 
						|
  define(this, name, val); | 
						|
  return this; | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Returns true if `node.val` is an empty string, or `node.nodes` does | 
						|
 * not contain any non-empty text nodes. | 
						|
 * | 
						|
 * ```js | 
						|
 * var node = new Node({type: 'text'}); | 
						|
 * node.isEmpty(); //=> true | 
						|
 * node.val = 'foo'; | 
						|
 * node.isEmpty(); //=> false | 
						|
 * ``` | 
						|
 * @param {Function} `fn` (optional) Filter function that is called on `node` and/or child nodes. `isEmpty` will return false immediately when the filter function returns false on any nodes. | 
						|
 * @return {Boolean} | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.isEmpty = function(fn) { | 
						|
  return utils.isEmpty(this, fn); | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Given node `foo` and node `bar`, push node `bar` onto `foo.nodes`, and | 
						|
 * set `foo` as `bar.parent`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * foo.push(bar); | 
						|
 * ``` | 
						|
 * @param {Object} `node` | 
						|
 * @return {Number} Returns the length of `node.nodes` | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.push = function(node) { | 
						|
  assert(Node.isNode(node), 'expected node to be an instance of Node'); | 
						|
  define(node, 'parent', this); | 
						|
 | 
						|
  this.nodes = this.nodes || []; | 
						|
  return this.nodes.push(node); | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Given node `foo` and node `bar`, unshift node `bar` onto `foo.nodes`, and | 
						|
 * set `foo` as `bar.parent`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * foo.unshift(bar); | 
						|
 * ``` | 
						|
 * @param {Object} `node` | 
						|
 * @return {Number} Returns the length of `node.nodes` | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.unshift = function(node) { | 
						|
  assert(Node.isNode(node), 'expected node to be an instance of Node'); | 
						|
  define(node, 'parent', this); | 
						|
 | 
						|
  this.nodes = this.nodes || []; | 
						|
  return this.nodes.unshift(node); | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Pop a node from `node.nodes`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var node = new Node({type: 'foo'}); | 
						|
 * node.push(new Node({type: 'a'})); | 
						|
 * node.push(new Node({type: 'b'})); | 
						|
 * node.push(new Node({type: 'c'})); | 
						|
 * node.push(new Node({type: 'd'})); | 
						|
 * console.log(node.nodes.length); | 
						|
 * //=> 4 | 
						|
 * node.pop(); | 
						|
 * console.log(node.nodes.length); | 
						|
 * //=> 3 | 
						|
 * ``` | 
						|
 * @return {Number} Returns the popped `node` | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.pop = function() { | 
						|
  return this.nodes && this.nodes.pop(); | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Shift a node from `node.nodes`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var node = new Node({type: 'foo'}); | 
						|
 * node.push(new Node({type: 'a'})); | 
						|
 * node.push(new Node({type: 'b'})); | 
						|
 * node.push(new Node({type: 'c'})); | 
						|
 * node.push(new Node({type: 'd'})); | 
						|
 * console.log(node.nodes.length); | 
						|
 * //=> 4 | 
						|
 * node.shift(); | 
						|
 * console.log(node.nodes.length); | 
						|
 * //=> 3 | 
						|
 * ``` | 
						|
 * @return {Object} Returns the shifted `node` | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.shift = function() { | 
						|
  return this.nodes && this.nodes.shift(); | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Remove `node` from `node.nodes`. | 
						|
 * | 
						|
 * ```js | 
						|
 * node.remove(childNode); | 
						|
 * ``` | 
						|
 * @param {Object} `node` | 
						|
 * @return {Object} Returns the removed node. | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.remove = function(node) { | 
						|
  assert(Node.isNode(node), 'expected node to be an instance of Node'); | 
						|
  this.nodes = this.nodes || []; | 
						|
  var idx = node.index; | 
						|
  if (idx !== -1) { | 
						|
    node.index = -1; | 
						|
    return this.nodes.splice(idx, 1); | 
						|
  } | 
						|
  return null; | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Get the first child node from `node.nodes` that matches the given `type`. | 
						|
 * If `type` is a number, the child node at that index is returned. | 
						|
 * | 
						|
 * ```js | 
						|
 * var child = node.find(1); //<= index of the node to get | 
						|
 * var child = node.find('foo'); //<= node.type of a child node | 
						|
 * var child = node.find(/^(foo|bar)$/); //<= regex to match node.type | 
						|
 * var child = node.find(['foo', 'bar']); //<= array of node.type(s) | 
						|
 * ``` | 
						|
 * @param {String} `type` | 
						|
 * @return {Object} Returns a child node or undefined. | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.find = function(type) { | 
						|
  return utils.findNode(this.nodes, type); | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Return true if the node is the given `type`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var node = new Node({type: 'bar'}); | 
						|
 * cosole.log(node.isType('foo'));          // false | 
						|
 * cosole.log(node.isType(/^(foo|bar)$/));  // true | 
						|
 * cosole.log(node.isType(['foo', 'bar'])); // true | 
						|
 * ``` | 
						|
 * @param {String} `type` | 
						|
 * @return {Boolean} | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.isType = function(type) { | 
						|
  return utils.isType(this, type); | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Return true if the `node.nodes` has the given `type`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * foo.push(bar); | 
						|
 * | 
						|
 * cosole.log(foo.hasType('qux'));          // false | 
						|
 * cosole.log(foo.hasType(/^(qux|bar)$/));  // true | 
						|
 * cosole.log(foo.hasType(['qux', 'bar'])); // true | 
						|
 * ``` | 
						|
 * @param {String} `type` | 
						|
 * @return {Boolean} | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Node.prototype.hasType = function(type) { | 
						|
  return utils.hasType(this, type); | 
						|
}; | 
						|
 | 
						|
/** | 
						|
 * Get the siblings array, or `null` if it doesn't exist. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * var baz = new Node({type: 'baz'}); | 
						|
 * foo.push(bar); | 
						|
 * foo.push(baz); | 
						|
 * | 
						|
 * console.log(bar.siblings.length) // 2 | 
						|
 * console.log(baz.siblings.length) // 2 | 
						|
 * ``` | 
						|
 * @return {Array} | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Object.defineProperty(Node.prototype, 'siblings', { | 
						|
  set: function() { | 
						|
    throw new Error('node.siblings is a getter and cannot be defined'); | 
						|
  }, | 
						|
  get: function() { | 
						|
    return this.parent ? this.parent.nodes : null; | 
						|
  } | 
						|
}); | 
						|
 | 
						|
/** | 
						|
 * Get the node's current index from `node.parent.nodes`. | 
						|
 * This should always be correct, even when the parent adds nodes. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * var baz = new Node({type: 'baz'}); | 
						|
 * var qux = new Node({type: 'qux'}); | 
						|
 * foo.push(bar); | 
						|
 * foo.push(baz); | 
						|
 * foo.unshift(qux); | 
						|
 * | 
						|
 * console.log(bar.index) // 1 | 
						|
 * console.log(baz.index) // 2 | 
						|
 * console.log(qux.index) // 0 | 
						|
 * ``` | 
						|
 * @return {Number} | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Object.defineProperty(Node.prototype, 'index', { | 
						|
  set: function(index) { | 
						|
    define(this, 'idx', index); | 
						|
  }, | 
						|
  get: function() { | 
						|
    if (!Array.isArray(this.siblings)) { | 
						|
      return -1; | 
						|
    } | 
						|
    var tok = this.idx !== -1 ? this.siblings[this.idx] : null; | 
						|
    if (tok !== this) { | 
						|
      this.idx = this.siblings.indexOf(this); | 
						|
    } | 
						|
    return this.idx; | 
						|
  } | 
						|
}); | 
						|
 | 
						|
/** | 
						|
 * Get the previous node from the siblings array or `null`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * var baz = new Node({type: 'baz'}); | 
						|
 * foo.push(bar); | 
						|
 * foo.push(baz); | 
						|
 * | 
						|
 * console.log(baz.prev.type) // 'bar' | 
						|
 * ``` | 
						|
 * @return {Object} | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Object.defineProperty(Node.prototype, 'prev', { | 
						|
  set: function() { | 
						|
    throw new Error('node.prev is a getter and cannot be defined'); | 
						|
  }, | 
						|
  get: function() { | 
						|
    if (Array.isArray(this.siblings)) { | 
						|
      return this.siblings[this.index - 1] || this.parent.prev; | 
						|
    } | 
						|
    return null; | 
						|
  } | 
						|
}); | 
						|
 | 
						|
/** | 
						|
 * Get the siblings array, or `null` if it doesn't exist. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * var baz = new Node({type: 'baz'}); | 
						|
 * foo.push(bar); | 
						|
 * foo.push(baz); | 
						|
 * | 
						|
 * console.log(bar.siblings.length) // 2 | 
						|
 * console.log(baz.siblings.length) // 2 | 
						|
 * ``` | 
						|
 * @return {Object} | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Object.defineProperty(Node.prototype, 'next', { | 
						|
  set: function() { | 
						|
    throw new Error('node.next is a getter and cannot be defined'); | 
						|
  }, | 
						|
  get: function() { | 
						|
    if (Array.isArray(this.siblings)) { | 
						|
      return this.siblings[this.index + 1] || this.parent.next; | 
						|
    } | 
						|
    return null; | 
						|
  } | 
						|
}); | 
						|
 | 
						|
/** | 
						|
 * Get the first node from `node.nodes`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * var baz = new Node({type: 'baz'}); | 
						|
 * var qux = new Node({type: 'qux'}); | 
						|
 * foo.push(bar); | 
						|
 * foo.push(baz); | 
						|
 * foo.push(qux); | 
						|
 * | 
						|
 * console.log(foo.first.type) // 'bar' | 
						|
 * ``` | 
						|
 * @return {Object} The first node, or undefiend | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Object.defineProperty(Node.prototype, 'first', { | 
						|
  get: function() { | 
						|
    return this.nodes ? this.nodes[0] : null; | 
						|
  } | 
						|
}); | 
						|
 | 
						|
/** | 
						|
 * Get the last node from `node.nodes`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * var baz = new Node({type: 'baz'}); | 
						|
 * var qux = new Node({type: 'qux'}); | 
						|
 * foo.push(bar); | 
						|
 * foo.push(baz); | 
						|
 * foo.push(qux); | 
						|
 * | 
						|
 * console.log(foo.last.type) // 'qux' | 
						|
 * ``` | 
						|
 * @return {Object} The last node, or undefiend | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Object.defineProperty(Node.prototype, 'last', { | 
						|
  get: function() { | 
						|
    return this.nodes ? utils.last(this.nodes) : null; | 
						|
  } | 
						|
}); | 
						|
 | 
						|
/** | 
						|
 * Get the last node from `node.nodes`. | 
						|
 * | 
						|
 * ```js | 
						|
 * var foo = new Node({type: 'foo'}); | 
						|
 * var bar = new Node({type: 'bar'}); | 
						|
 * var baz = new Node({type: 'baz'}); | 
						|
 * var qux = new Node({type: 'qux'}); | 
						|
 * foo.push(bar); | 
						|
 * foo.push(baz); | 
						|
 * foo.push(qux); | 
						|
 * | 
						|
 * console.log(foo.last.type) // 'qux' | 
						|
 * ``` | 
						|
 * @return {Object} The last node, or undefiend | 
						|
 * @api public | 
						|
 */ | 
						|
 | 
						|
Object.defineProperty(Node.prototype, 'scope', { | 
						|
  get: function() { | 
						|
    if (this.isScope !== true) { | 
						|
      return this.parent ? this.parent.scope : this; | 
						|
    } | 
						|
    return this; | 
						|
  } | 
						|
}); | 
						|
 | 
						|
/** | 
						|
 * Get own property names from Node prototype, but only the | 
						|
 * first time `Node` is instantiated | 
						|
 */ | 
						|
 | 
						|
function lazyKeys() { | 
						|
  if (!ownNames) { | 
						|
    ownNames = Object.getOwnPropertyNames(Node.prototype); | 
						|
  } | 
						|
} | 
						|
 | 
						|
/** | 
						|
 * Simplified assertion. Throws an error is `val` is falsey. | 
						|
 */ | 
						|
 | 
						|
function assert(val, message) { | 
						|
  if (!val) throw new Error(message); | 
						|
} | 
						|
 | 
						|
/** | 
						|
 * Expose `Node` | 
						|
 */ | 
						|
 | 
						|
exports = module.exports = Node;
 | 
						|
 |