forked from OKEAMAH/prettier
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathindex.js
More file actions
75 lines (57 loc) · 1.67 KB
/
index.js
File metadata and controls
75 lines (57 loc) · 1.67 KB
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
import path from "node:path";
import babelGenerator from "@babel/generator";
import { parse } from "@babel/parser";
import { traverseFast as traverse } from "@babel/types";
import { outdent } from "outdent";
import { PROJECT_ROOT, SOURCE_DIR } from "../../utils/index.js";
import allTransforms from "./transforms/index.js";
const generate = babelGenerator.default;
/* Doesn't work for dependencies, optional call, computed property, and spread arguments */
function transform(original, file) {
if (
!(
file.startsWith(SOURCE_DIR) ||
file.startsWith(path.join(PROJECT_ROOT, "node_modules/camelcase/")) ||
file.startsWith(
path.join(PROJECT_ROOT, "node_modules/angular-estree-parser/"),
) ||
file.startsWith(path.join(PROJECT_ROOT, "node_modules/jest-docblock/")) ||
file.startsWith(path.join(PROJECT_ROOT, "node_modules/espree/"))
)
) {
return original;
}
const transforms = allTransforms.filter(
(transform) => !transform.shouldSkip(original, file),
);
if (transforms.length === 0) {
return original;
}
let changed = false;
const injected = new Set();
const ast = parse(original, { sourceType: "module" });
traverse(ast, (node) => {
for (const transform of transforms) {
if (!transform.test(node)) {
continue;
}
transform.transform(node);
if (transform.inject) {
injected.add(transform.inject);
}
changed ||= true;
}
});
if (!changed) {
return original;
}
let { code } = generate(ast);
if (injected.size > 0) {
code = outdent`
${[...injected].join("\n")}
${code}
`;
}
return code;
}
export default transform;