-
Notifications
You must be signed in to change notification settings - Fork 36
/
Copy pathmacro.js
34 lines (28 loc) · 1.22 KB
/
macro.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
const { createMacro, MacroError } = require("babel-plugin-macros");
const dedent = require("./dist/dedent.js").default;
module.exports = createMacro(prevalMacros);
function prevalMacros({ references, state, babel }) {
references.default.forEach(referencePath => {
if (referencePath.parentPath.type === "TaggedTemplateExpression") {
asTag(referencePath.parentPath.get("quasi"), state, babel);
} else if (referencePath.parentPath.type === "CallExpression") {
asFunction(referencePath.parentPath.get("arguments"), state, babel);
} else {
throw new MacroError(
`dedent.macro can only be used as tagged template expression or function call. You tried ${
referencePath.parentPath.type
}.`
);
}
});
}
function asTag(quasiPath, { file: { opts: { filename } } }, babel) {
const string = quasiPath.parentPath.get("quasi").evaluate().value;
const { types: t } = babel;
quasiPath.parentPath.replaceWith(t.stringLiteral(dedent(string)));
}
function asFunction(argumentsPaths, { file: { opts: { filename } } }, babel) {
const string = argumentsPaths[0].evaluate().value;
const { types: t } = babel;
argumentsPaths[0].parentPath.replaceWith(t.stringLiteral(dedent(string)));
}