full site update
This commit is contained in:
229
node_modules/@swc/helpers/scripts/ast_grep.js
generated
vendored
Normal file
229
node_modules/@swc/helpers/scripts/ast_grep.js
generated
vendored
Normal file
@@ -0,0 +1,229 @@
|
||||
import { parseFiles } from "@ast-grep/napi";
|
||||
import MagicString from "magic-string";
|
||||
import { chalk, fs, path } from "zx";
|
||||
import { errors } from "./errors.js";
|
||||
import { root } from "./utils.js";
|
||||
|
||||
/**
|
||||
* @typedef {import("@ast-grep/napi").SgNode} SgNode
|
||||
*/
|
||||
|
||||
export function ast_grep() {
|
||||
const task_queue = [];
|
||||
|
||||
const task = parseFiles([root("esm")], (err, tree) => {
|
||||
const filename = path.basename(tree.filename(), ".js");
|
||||
if (filename === "index") {
|
||||
return;
|
||||
}
|
||||
|
||||
const source = new MagicString(tree.root().text());
|
||||
source.prepend(`"use strict";\n\n`);
|
||||
|
||||
if (filename.startsWith("_ts")) {
|
||||
const match = tree.root().find(`export { $NAME as _ } from "tslib"`);
|
||||
if (match) {
|
||||
const name = match.getMatch("NAME").text();
|
||||
|
||||
const range = match.range();
|
||||
|
||||
source.update(
|
||||
range.start.index,
|
||||
range.end.index,
|
||||
`exports._ = require("tslib").${name};`,
|
||||
);
|
||||
task_queue.push(
|
||||
fs.writeFile(root("cjs", `${filename}.cjs`), source.toString(), {
|
||||
encoding: "utf-8",
|
||||
}),
|
||||
);
|
||||
} else {
|
||||
report_noexport(tree.filename());
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
// rewrite export named function
|
||||
const match = tree.root().find({
|
||||
rule: {
|
||||
kind: "export_statement",
|
||||
pattern: "export { $FUNC as _ }",
|
||||
},
|
||||
});
|
||||
|
||||
if (match) {
|
||||
const func = match.getMatch("FUNC");
|
||||
const func_name = func.text();
|
||||
if (func_name !== filename) {
|
||||
report_export_mismatch(tree.filename(), match);
|
||||
}
|
||||
|
||||
const range = match.range();
|
||||
source.update(
|
||||
range.start.index,
|
||||
range.end.index,
|
||||
`exports._ = ${func_name};`,
|
||||
);
|
||||
|
||||
// since we match the { export x as _ } pattern,
|
||||
// we need to find the assignment expression from the root
|
||||
tree
|
||||
.root()
|
||||
.findAll({
|
||||
rule: {
|
||||
pattern: func_name,
|
||||
kind: "identifier",
|
||||
inside: { kind: "assignment_expression", field: "left" },
|
||||
},
|
||||
})
|
||||
.forEach((match) => {
|
||||
const range = match.range();
|
||||
|
||||
source.prependLeft(range.start.index, `exports._ = `);
|
||||
});
|
||||
} else {
|
||||
report_noexport(tree.filename(tree.filename()));
|
||||
}
|
||||
|
||||
// rewrite import
|
||||
tree
|
||||
.root()
|
||||
.findAll({ rule: { pattern: `import { _ as $BINDING } from "$SOURCE"` } })
|
||||
.forEach((match) => {
|
||||
const import_binding = match.getMatch("BINDING").text();
|
||||
const import_source = match.getMatch("SOURCE").text();
|
||||
|
||||
const import_basename = path.basename(import_source, ".js");
|
||||
|
||||
if (import_binding !== import_basename) {
|
||||
report_import_mismatch(tree.filename(), match);
|
||||
}
|
||||
|
||||
const range = match.range();
|
||||
|
||||
source.update(
|
||||
range.start.index,
|
||||
range.end.index,
|
||||
`var ${import_binding} = require("./${import_binding}.cjs");`,
|
||||
);
|
||||
|
||||
tree
|
||||
.root()
|
||||
.findAll({
|
||||
rule: {
|
||||
pattern: import_binding,
|
||||
kind: "identifier",
|
||||
inside: {
|
||||
not: {
|
||||
kind: "import_specifier",
|
||||
},
|
||||
},
|
||||
},
|
||||
})
|
||||
.forEach((match) => {
|
||||
const range = match.range();
|
||||
const ref_name = match.text();
|
||||
|
||||
source.update(
|
||||
range.start.index,
|
||||
range.end.index,
|
||||
`${ref_name}._`,
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
task_queue.push(
|
||||
fs.writeFile(root("cjs", `${filename}.cjs`), source.toString(), {
|
||||
encoding: "utf-8",
|
||||
}),
|
||||
);
|
||||
});
|
||||
|
||||
task_queue.push(task);
|
||||
|
||||
return task_queue;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param {string} filename
|
||||
* @param {SgNode} match
|
||||
*/
|
||||
function report_export_mismatch(filename, match) {
|
||||
const func = match.getMatch("FUNC");
|
||||
const func_range = func.range();
|
||||
|
||||
const text = match.text().split("\n");
|
||||
const offset = func_range.start.line - match.range().start.line;
|
||||
|
||||
text.splice(
|
||||
offset + 1,
|
||||
text.length,
|
||||
chalk.red(
|
||||
[
|
||||
" ".repeat(func_range.start.column),
|
||||
"^".repeat(func_range.end.column - func_range.start.column),
|
||||
]
|
||||
.join(""),
|
||||
),
|
||||
);
|
||||
|
||||
errors.push(
|
||||
[
|
||||
`${chalk.bold.red("error")}: mismatch exported function name.`,
|
||||
"",
|
||||
`${chalk.blue("-->")} ${filename}:${func_range.start.line + 1}:${func_range.start.column + 1}`,
|
||||
"",
|
||||
...text,
|
||||
"",
|
||||
`${
|
||||
chalk.bold(
|
||||
"note:",
|
||||
)
|
||||
} The exported name should be the same as the filename.`,
|
||||
"",
|
||||
]
|
||||
.join("\n"),
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param {string} filename
|
||||
* @param {SgNode} match
|
||||
*/
|
||||
function report_import_mismatch(filename, match) {
|
||||
const binding_range = match.getMatch("BINDING").range();
|
||||
const source_range = match.getMatch("SOURCE").range();
|
||||
|
||||
errors.push(
|
||||
[
|
||||
`${chalk.bold.red("error")}: mismatch imported binding name.`,
|
||||
"",
|
||||
`${chalk.blue("-->")} ${filename}:${match.range().start.line + 1}`,
|
||||
"",
|
||||
match.text(),
|
||||
[
|
||||
" ".repeat(binding_range.start.column),
|
||||
chalk.red("^".repeat(binding_range.end.column - binding_range.start.column)),
|
||||
" ".repeat(source_range.start.column - binding_range.end.column),
|
||||
chalk.blue("-".repeat(source_range.end.column - source_range.start.column)),
|
||||
]
|
||||
.join(""),
|
||||
`${
|
||||
chalk.bold(
|
||||
"note:",
|
||||
)
|
||||
} The imported binding name should be the same as the import source basename.`,
|
||||
"",
|
||||
]
|
||||
.join("\n"),
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param {string} filename
|
||||
*/
|
||||
function report_noexport(filename) {
|
||||
errors.push(
|
||||
[`${chalk.bold.red("error")}: exported name not found`, `${chalk.blue("-->")} ${filename}`].join("\n"),
|
||||
);
|
||||
}
|
Reference in New Issue
Block a user