stylus/lib/visitor/index.js

Robust, expressive, and feature-rich CSS superset

/*!
 * Stylus - Visitor
 * Copyright (c) Automattic <developer.wordpress.com>
 * MIT Licensed
 */

module.exports = class Visitor {
  /**
   * Initialize a new `Visitor` with the given `root` Node.
   *
   * @param {Node} root
   * @api private
   */

  constructor(root) {
    this.root = root;
  }

  /**
   * Visit the given `node`.
   *
   * @param {Node|Array} node
   * @api public
   */

  visit(node, fn) {
    var method = 'visit' + node.constructor.name;
    if (this[method]) return this[method](node);
    return node;
  };
};