/
index.js
91 lines (83 loc) · 2.5 KB
/
index.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
85
86
87
88
89
90
91
import {realpathSync} from 'fs'
import {dirname} from 'path'
import {programVisitor} from 'istanbul-lib-instrument'
const testExclude = require('test-exclude')
const findUp = require('find-up')
function getRealpath (n) {
try {
return realpathSync(n) || n
} catch (e) {
return n
}
}
function makeShouldSkip () {
let exclude
return function shouldSkip (file, opts) {
if (!exclude) {
const cwd = getRealpath(process.env.NYC_CWD || process.cwd())
const nycConfig = process.env.NYC_CONFIG ? JSON.parse(process.env.NYC_CONFIG) : {}
let config = {}
if (Object.keys(opts).length > 0) {
// explicitly configuring options in babel
// takes precedence.
config = opts
} else if (nycConfig.include || nycConfig.exclude) {
// nyc was configured in a parent process (keep these settings).
config = {
include: nycConfig.include,
exclude: nycConfig.exclude,
// Make sure this is true unless explicitly set to `false`. `undefined` is still `true`.
excludeNodeModules: nycConfig.excludeNodeModules !== false
}
} else {
// fallback to loading config from key in package.json.
config = {
configKey: 'nyc',
configPath: dirname(findUp.sync('package.json', { cwd }))
}
}
exclude = testExclude(Object.assign(
{ cwd },
config
))
}
return !exclude.shouldInstrument(file)
}
}
function makeVisitor ({types: t}) {
const shouldSkip = makeShouldSkip()
return {
visitor: {
Program: {
enter (path) {
this.__dv__ = null
const realPath = getRealpath(this.file.opts.filename)
if (shouldSkip(realPath, this.opts)) {
return
}
let { inputSourceMap } = this.opts
if (this.opts.useInlineSourceMaps !== false) {
if (!inputSourceMap && this.file.inputMap) {
inputSourceMap = this.file.inputMap.sourcemap
}
}
this.__dv__ = programVisitor(t, realPath, {
coverageVariable: '__coverage__',
inputSourceMap
})
this.__dv__.enter(path)
},
exit (path) {
if (!this.__dv__) {
return
}
const result = this.__dv__.exit(path)
if (this.opts.onCover) {
this.opts.onCover(getRealpath(this.file.opts.filename), result.fileCoverage)
}
}
}
}
}
}
export default makeVisitor