Code coverage report for 6to5/transformation/transformer.js

Statements: 100% (58 / 58)      Branches: 100% (32 / 32)      Functions: 100% (9 / 9)      Lines: 100% (49 / 49)      Ignored: none     

All files » 6to5/transformation/ » transformer.js
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 851   1 1 1   1 35 35 35     1 35 2     35 75 74 74   74 74 4 14         35     1 23968 23968   23968 990       1 12042   11992 11992   11992   11992 23984 1109701 1109701   39684 39684 39684   20099       11992         11976     1 12379 12379 12379   11031 11031   11011 11011   10980    
module.exports = Transformer;
 
var traverse = require("../traverse");
var t        = require("../types");
var _        = require("lodash");
 
function Transformer(key, transformer, opts) {
  this.transformer = Transformer.normalise(transformer);
  this.opts        = opts || {};
  this.key         = key;
}
 
Transformer.normalise = function (transformer) {
  if (_.isFunction(transformer)) {
    transformer = { ast: transformer };
  }
 
  _.each(transformer, function (fns, type) {
    if (type[0] === "_") return;
    if (_.isFunction(fns)) fns = { enter: fns };
    transformer[type] = fns;
 
    var aliases = t.FLIPPED_ALIAS_KEYS[type];
    if (aliases) {
      _.each(aliases, function (alias) {
        transformer[alias] = fns;
      });
    }
  });
 
  return transformer;
};
 
Transformer.prototype.astRun = function (file, key) {
  var transformer = this.transformer;
  var ast         = file.ast;
 
  if (transformer.ast && transformer.ast[key]) {
    transformer.ast[key](ast, file);
  }
};
 
Transformer.prototype.transform = function (file) {
  if (!this.canRun(file)) return;
 
  var transformer = this.transformer;
  var ast         = file.ast;
 
  this.astRun(file, "enter");
 
  var build = function (exit) {
    return function (node, parent, scope) {
      var fns = transformer[node.type];
      if (!fns) return;
 
      var fn = fns.enter;
      if (exit) fn = fns.exit;
      if (!fn) return;
 
      return fn(node, parent, file, scope);
    };
  };
 
  traverse(ast, {
    enter: build(),
    exit:  build(true)
  });
 
  this.astRun(file, "exit");
};
 
Transformer.prototype.canRun = function (file) {
  var opts = file.opts;
  var key  = this.key;
  if (key[0] === "_") return true;
 
  var blacklist = opts.blacklist;
  if (blacklist.length && _.contains(blacklist, key)) return false;
 
  var whitelist = opts.whitelist;
  if (whitelist.length && !_.contains(whitelist, key)) return false;
 
  return true;
};