blob: 61a7e37bee0f423f05ad1a03f28940b725f4b1a1 (
plain)
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
|
'use strict';
var CallMatcher = require('call-matcher');
var babylon = require('babylon');
var assign = require('core-js/library/fn/object/assign');
var find = require('core-js/library/fn/array/find');
var BabelAssertionVisitor = require('./babel-assertion-visitor');
function BabelEspowerVisitor (babel, opts) {
this.babel = babel;
this.matchers = opts.patterns.map(function (pattern) {
var signatureAst = babylon.parse(pattern);
var expression = signatureAst.program.body[0].expression;
return new CallMatcher(expression, opts);
});
this.options = opts;
}
BabelEspowerVisitor.prototype.enter = function (nodePath) {
var currentNode = nodePath.node;
var file = nodePath.hub.file;
var assertionVisitor = file.get('espowerAssertionVisitor');
if (assertionVisitor) {
if (assertionVisitor.isGeneratedNode(nodePath) || assertionVisitor.toBeSkipped(nodePath)) {
// skipping this Node
// MEMO: exit() will not be called when skip() is called
nodePath.skip();
return;
}
if (!assertionVisitor.isCapturingArgument() && !this.isCalleeOfParentCallExpression(nodePath)) {
// entering argument
assertionVisitor.enterArgument(nodePath);
}
} else if (nodePath.isCallExpression()) {
var matcher = find(this.matchers, function (m) { return m.test(currentNode); });
if (matcher) {
// entering assertion
var espowerOptions = assign({
path: file.opts.filename, // or opts.sourceFileName?
sourceMap: file.opts.inputSourceMap
}, this.options);
assertionVisitor = new BabelAssertionVisitor(this.babel, matcher, espowerOptions);
assertionVisitor.enter(nodePath);
file.set('espowerAssertionVisitor', assertionVisitor);
}
}
};
BabelEspowerVisitor.prototype.exit = function (nodePath) {
var currentNode = nodePath.node;
var resultTree = currentNode;
var file = nodePath.hub.file;
var assertionVisitor = file.get('espowerAssertionVisitor');
if (!assertionVisitor) {
return;
}
if (assertionVisitor.isLeavingAssertion(nodePath)) {
// leaving assertion
assertionVisitor.leave(nodePath);
file.delete('espowerAssertionVisitor');
return;
}
if (!assertionVisitor.isCapturingArgument()) {
return;
}
if (assertionVisitor.toBeCaptured(nodePath)) {
// capturing Node
resultTree = assertionVisitor.captureNode(nodePath);
}
if (assertionVisitor.isLeavingArgument(nodePath)) {
// capturing whole argument on leaving argument
resultTree = assertionVisitor.leaveArgument(resultTree);
}
if (resultTree !== currentNode) {
nodePath.replaceWith(resultTree);
}
};
BabelEspowerVisitor.prototype.isCalleeOfParentCallExpression = function (nodePath) {
var currentKey = nodePath.key;
var parentNode = nodePath.parent;
var types = this.babel.types;
return types.isCallExpression(parentNode) && currentKey === 'callee';
};
module.exports = BabelEspowerVisitor;
|
