| const { MacroError, createMacro } = require("babel-plugin-macros"); |
| const dedent = require("./dist/dedent.js").default; |
|
|
| module.exports = createMacro(prevalMacros); |
|
|
| function prevalMacros({ babel, references, state }) { |
| 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, _, babel) { |
| const string = quasiPath.parentPath.get("quasi").evaluate().value; |
| const { types: t } = babel; |
|
|
| quasiPath.parentPath.replaceWith(t.stringLiteral(dedent(string))); |
| } |
|
|
| function asFunction(argumentsPaths, _, babel) { |
| const string = argumentsPaths[0].evaluate().value; |
| const { types: t } = babel; |
|
|
| argumentsPaths[0].parentPath.replaceWith(t.stringLiteral(dedent(string))); |
| } |
|
|