147 lines
4.6 KiB
JavaScript
147 lines
4.6 KiB
JavaScript
exports.init = function(grunt) {
|
|
'use strict';
|
|
|
|
var chalk = require('chalk');
|
|
|
|
var fs = require('fs');
|
|
var path = require('path');
|
|
|
|
var flow = require('nue').flow;
|
|
var as = require('nue').as;
|
|
|
|
var istanbul = require('istanbul');
|
|
|
|
function flowEnd(err, done) {
|
|
if (err) {
|
|
grunt.fail.fatal(err);
|
|
}
|
|
done();
|
|
}
|
|
|
|
function makeReporters(options) {
|
|
var result = [];
|
|
var reporters = options.reporters &&
|
|
typeof options.reporters === 'object' ? options.reporters : {};
|
|
Object.keys(reporters).forEach(function(n) {
|
|
if(reporters[n]) {
|
|
result.push({ type : n, options : reporters[n] });
|
|
}
|
|
});
|
|
|
|
var append = function(t) {
|
|
if(t && !reporters[t]) {
|
|
result.push({ type : t, options : options});
|
|
reporters[t] = true;
|
|
}
|
|
};
|
|
|
|
if (Array.isArray(options.type)) {
|
|
options.type.forEach(append);
|
|
} else {
|
|
append(options.type);
|
|
}
|
|
|
|
var mapping = {
|
|
'none' : [],
|
|
'detail': ['text'],
|
|
'both' : ['text', 'text-summary']
|
|
};
|
|
var a = mapping[options.print];
|
|
if(a) {
|
|
a.forEach(append);
|
|
} else {
|
|
append('text-summary');
|
|
}
|
|
return result;
|
|
}
|
|
|
|
return {
|
|
instrument : function(files, options, done) {
|
|
var outFile = function(file) {
|
|
return path.join(options.basePath, options.flatten === true ? path.basename(file) : file);
|
|
};
|
|
|
|
var tally = { instrumented : 0, skipped : 0 };
|
|
|
|
var instFlow = flow(
|
|
function instrumentFile(f) {
|
|
var code = grunt.file.read(f.name);
|
|
var instrumenter = new istanbul.Instrumenter(options);
|
|
instrumenter.instrument(code, f.name, this.async({
|
|
name : f.name,
|
|
code : as(1)
|
|
}));
|
|
}, function write(result) {
|
|
var out = outFile(result.name);
|
|
grunt.file.write(out, result.code);
|
|
tally.instrumented++;
|
|
this.next();
|
|
}, function end() {
|
|
flowEnd(this.err, this.next.bind(this));
|
|
});
|
|
|
|
var dateCheckFlow = flow(
|
|
function readStat(f) {
|
|
if (grunt.file.exists(f.name)) {
|
|
grunt.log.debug('reading stat for ' + f.name);
|
|
fs.stat(f.name, this.async({ name : f.name, stat : as(1) }));
|
|
fs.stat(outFile(f.name), this.async({ name : f.name, stat : as(1) }));
|
|
} else {
|
|
grunt.verbose.writeln('instrumented file does not exist ' + f.name);
|
|
this.end({ name : f.name, instrument : true });
|
|
}
|
|
}, function decision(i, o) {
|
|
var reinstrument = i.stat.mtime.getTime() > o.stat.mtime.getTime();
|
|
grunt.log.debug('make a decision about instrumenting ' + i.name + ': ' + reinstrument);
|
|
this.end({ name: i.name, instrument: reinstrument });
|
|
}, function end(f) {
|
|
if (f.instrument) {
|
|
this.exec(instFlow, { name : f.name }, this.async());
|
|
} else {
|
|
tally.skipped++;
|
|
flowEnd(this.err, this.next.bind(this));
|
|
}
|
|
});
|
|
|
|
flow(function(filelist) {
|
|
this.asyncEach(filelist, function(file, group) {
|
|
this.exec((options.lazy ? dateCheckFlow : instFlow), { name : file }, group.async(as(1)));
|
|
});
|
|
}, function outputSummary() {
|
|
grunt.log.write('Instrumented ' + chalk.cyan(tally.instrumented) + ' ' +
|
|
grunt.util.pluralize(tally.instrumented, 'file/files'));
|
|
if (options.lazy) {
|
|
grunt.log.write(' (skipped ' + chalk.cyan(tally.skipped) + ' ' +
|
|
grunt.util.pluralize(tally.skipped, 'file/files') + ')');
|
|
}
|
|
grunt.log.writeln();
|
|
this.next();
|
|
}, done)(files);
|
|
},
|
|
storeCoverage : function(coverage, options, done) {
|
|
flow(function write_json(cov) {
|
|
var json = path.resolve(options.dir, options.json);
|
|
grunt.file.write(json, JSON.stringify(cov));
|
|
this.next();
|
|
}, function() {
|
|
flowEnd(this.err, done);
|
|
})(coverage);
|
|
},
|
|
makeReport : function(files, options, done) {
|
|
flow(function(filelist) {
|
|
var collector = new istanbul.Collector();
|
|
filelist.forEach(function(file) {
|
|
collector.add(grunt.file.readJSON(file));
|
|
});
|
|
makeReporters(options).forEach(function(repoDef) {
|
|
var reporter = istanbul.Report.create(repoDef.type, repoDef.options);
|
|
reporter.writeReport(collector, true);
|
|
});
|
|
this.next();
|
|
}, function() {
|
|
flowEnd(this.err, done);
|
|
})(files);
|
|
}
|
|
};
|
|
};
|