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
|
var util = require('util'),
os = require('os'),
path = require('path'),
mkdirp = require('mkdirp'),
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);
}
};
|
