blob: 8d96eabf1a63a198c82d2cdaf0b88dafb2f5338b (
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
|
export default function ({ types: t }) {
return {
visitor: {
Program(path, state) {
let shouldAdd = false
const MyVisitor = {
Function: {
enter: function enter(path) {
if (path.node.async) {
shouldAdd = true
}
}
},
Identifier(path) {
if (path.node.name === "Promise") {
shouldAdd = true
}
}
};
path.traverse(MyVisitor)
if (shouldAdd) {
let depth = state.filename.replace(state.cwd, "").split(/[\\/]/g).length - 2
const identifier = t.identifier('Promise');
const importDefaultSpecifier = t.importDefaultSpecifier(identifier);
const importDeclaration = t.importDeclaration([importDefaultSpecifier], t.stringLiteral("../".repeat(depth) + 'PromiseV2'));
path.unshiftContainer('body', importDeclaration);
}
let shouldAdd2 = false
const MyVisitor2 = {
Identifier(path) {
if (path.node.name === "fetch") {
shouldAdd2 = true
}
}
};
path.traverse(MyVisitor2)
if (shouldAdd2 && !state.filename.includes("networkUtils")) {
let depth = state.filename.replace(state.cwd, "").split(/[\\/]/g).length - 2
const identifier = t.identifier('fetch');
const importDefaultSpecifier = t.importDefaultSpecifier(identifier);
const importDeclaration = t.importDeclaration([importDefaultSpecifier], t.stringLiteral("../".repeat(depth) + 'SoopyV2/utils/networkUtils'));
path.unshiftContainer('body', importDeclaration);
}
}
}
};
};
|
