stylus/lib/visitor/deps-resolver.js

Robust, expressive, and feature-rich CSS superset

/**
 * Module dependencies.
 */

var Visitor = require('./')
  , Parser = require('../parser')
  , nodes = require('../nodes')
  , utils = require('../utils')
  , dirname = require('path').dirname
  , fs = require('fs');

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

  constructor(root, options) {
    super(root)
    this.filename = options.filename;
    this.paths = options.paths || [];
    this.paths.push(dirname(options.filename || '.'));
    this.options = options;
    this.functions = {};
    this.deps = [];
  }


  visit(node) {
    switch (node.nodeName) {
      case 'root':
      case 'block':
      case 'expression':
        this.visitRoot(node);
        break;
      case 'group':
      case 'media':
      case 'atblock':
      case 'atrule':
      case 'keyframes':
      case 'each':
      case 'supports':
        this.visit(node.block);
        break;
      default:
        super.visit(node);
    }
  };

  /**
   * Visit Root.
   */

  visitRoot(block) {
    for (var i = 0, len = block.nodes.length; i < len; ++i) {
      this.visit(block.nodes[i]);
    }
  };

  /**
   * Visit Ident.
   */

  visitIdent(ident) {
    this.visit(ident.val);
  };

  /**
   * Visit If.
   */

  visitIf(node) {
    this.visit(node.block);
    this.visit(node.cond);
    for (var i = 0, len = node.elses.length; i < len; ++i) {
      this.visit(node.elses[i]);
    }
  };

  /**
   * Visit Function.
   */

  visitFunction(fn) {
    this.functions[fn.name] = fn.block;
  };

  /**
   * Visit Call.
   */

  visitCall(call) {
    if (call.name in this.functions) this.visit(this.functions[call.name]);
    if (call.block) this.visit(call.block);
  };

  /**
   * Visit Import.
   */

  visitImport(node) {
    // If it's a url() call, skip
    if (node.path.first.name === 'url') return;

    var path = !node.path.first.val.isNull && node.path.first.val || node.path.first.name
      , literal, found, oldPath;

    if (!path) return;

    literal = /\.css(?:"|$)/.test(path);

    // support optional .styl
    if (!literal && !/\.styl$/i.test(path)) {
      oldPath = path;
      path += '.styl';
    }

    // Lookup
    found = utils.find(path, this.paths, this.filename);

    // support optional index
    if (!found && oldPath) found = utils.lookupIndex(oldPath, this.paths, this.filename);

    if (!found) return;

    this.deps = this.deps.concat(found);

    if (literal) return;

    // nested imports
    for (var i = 0, len = found.length; i < len; ++i) {
      var file = found[i]
        , dir = dirname(file)
        , str = fs.readFileSync(file, 'utf-8')
        , block = new nodes.Block
        , parser = new Parser(str, utils.merge({ root: block }, this.options));

      if (!~this.paths.indexOf(dir)) this.paths.push(dir);

      try {
        block = parser.parse();
      } catch (err) {
        err.filename = file;
        err.lineno = parser.lexer.lineno;
        err.column = parser.lexer.column;
        err.input = str;
        throw err;
      }

      this.visit(block);
    }
  };

  /**
   * Get dependencies.
   */

  resolve() {
    this.visit(this.root);
    return utils.uniq(this.deps);
  };
};