oxc/crates/oxc_traverse/scripts/build.mjs
overlookmotel eba5033a58 refactor(traverse): codegen ChildScopeCollector (#5119)
Codegen `ChildScopeCollector`, so that if more types have scopes added, we don't forget to add them (like the problem #5118 fixed).

The methods are in different order in the generated version, but otherwise identical to before this PR.

`visit_finally_clause` has to be added manually, as `oxc_traverse` codegen does not read or understand `#[visit(as)]` attrs.
2024-08-23 12:49:33 +00:00

44 lines
1.7 KiB
JavaScript

/*
* Codegen for `traverse`.
*
* Parses Rust AST type definitions from files in `crates/oxc_ast/src/ast`, and generates:
* - `src/traverse.rs`
* - `src/ancestor.rs`
* - `src/walk.rs`
*
* This is a quick-and-dirty version written in JS for speed of implementation.
* We should do this properly with a Rust build script using `syn` etc.
*/
import {writeFile} from 'fs/promises';
import {exec} from 'child_process';
import {join as pathJoin} from 'path';
import {fileURLToPath} from 'url';
import {promisify} from 'util';
import getTypesFromCode from './lib/parse.mjs';
import generateTraverseTraitCode from './lib/traverse.mjs';
import generateAncestorsCode from './lib/ancestor.mjs';
import generateWalkFunctionsCode from './lib/walk.mjs';
import generateScopesCollectorCode from './lib/scopes_collector.mjs';
const execAsync = promisify(exec);
const PREAMBLE = '// Auto-generated code, DO NOT EDIT DIRECTLY!\n'
+ '// Generated by `oxc_traverse/scripts/build.mjs`.\n'
+ '// To alter this generated file you have to edit the codegen.\n\n';
const types = await getTypesFromCode();
const outputDirPath = pathJoin(fileURLToPath(import.meta.url), '../../src/generated');
await writeToFile('traverse.rs', generateTraverseTraitCode(types));
await writeToFile('ancestor.rs', generateAncestorsCode(types));
await writeToFile('walk.rs', generateWalkFunctionsCode(types));
await writeToFile('scopes_collector.rs', generateScopesCollectorCode(types));
async function writeToFile(filename, code) {
code = `${PREAMBLE}${code}`;
const path = pathJoin(outputDirPath, filename);
console.log('Writing:', path);
await writeFile(path, code);
await execAsync(`rustfmt ${JSON.stringify(path)}`);
}