blob: d2d5ed17df189b0cbdeb84d3762781a79289fda3 [file] [log] [blame]
var util = require('util'),
os = require('os'),
path = require('path'),
mkdirp = require('make-dir'),
rimraf = require('rimraf'),
fs = require('fs');
function SourceStore(/*opts*/) {}
SourceStore.prototype.registerSource = function(/* filePath, sourceText */) {
throw new Error('registerSource must be overridden');
};
SourceStore.prototype.getSource = function(/* filePath */) {
throw new Error('getSource must be overridden');
};
SourceStore.prototype.dispose = function() {};
function MemoryStore() {
this.data = {};
}
util.inherits(MemoryStore, SourceStore);
MemoryStore.prototype.registerSource = function(filePath, sourceText) {
this.data[filePath] = sourceText;
};
MemoryStore.prototype.getSource = function(filePath) {
return this.data[filePath] || null;
};
function FileStore(opts) {
opts = opts || {};
var tmpDir = opts.tmpdir || os.tmpdir();
this.counter = 0;
this.mappings = [];
this.basePath = path.resolve(tmpDir, '.istanbul', 'cache_');
mkdirp.sync(path.dirname(this.basePath));
}
util.inherits(FileStore, SourceStore);
FileStore.prototype.registerSource = function(filePath, sourceText) {
if (this.mappings[filePath]) {
return;
}
this.counter += 1;
var mapFile = this.basePath + this.counter;
this.mappings[filePath] = mapFile;
fs.writeFileSync(mapFile, sourceText, 'utf8');
};
FileStore.prototype.getSource = function(filePath) {
var mapFile = this.mappings[filePath];
if (!mapFile) {
return null;
}
return fs.readFileSync(mapFile, 'utf8');
};
FileStore.prototype.dispose = function() {
this.mappings = [];
rimraf.sync(path.dirname(this.basePath));
};
module.exports = {
create: function(type, opts) {
opts = opts || {};
type = (type || 'memory').toLowerCase();
if (type === 'file') {
return new FileStore(opts);
}
return new MemoryStore(opts);
}
};