Facebook/react
1
0
Fork 0
mirror of https://github.com/facebook/react.git synced 2025-11-01 09:12:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Babel] Refactor visitor to separate function

Browse Source 
This lets us extend to ArrowFunctionExpression
This commit is contained in:
Sathya Gunasekaran
2023-03-20 17:42:56 +00:00
parent 3605199646
commit 93b79e4aa0
1 changed files with 41 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/forget/src/Babel/BabelPlugin.ts
+41 -34
View File
@@ -39,45 +39,52 @@ export default function ReactForgetBabelPlugin(
): BabelCore.PluginObj {
let hasForgetCompiledCode: boolean = false;
function visitFn(
fn: BabelCore.NodePath<t.FunctionDeclaration>,
pass: BabelPluginPass
): void {
if (pass.opts.enableOnlyOnUseForgetDirective) {
if (!hasUseForgetDirective(fn.node.body.directives)) {
return;
}
}
if (fn.scope.getProgramParent() !== fn.scope.parent) {
return;
}
hasForgetCompiledCode = true;
const compiled = compile(fn, pass.opts.environment);
if (pass.opts.gatingModule) {
// Rename existing function
invariant(fn.node.id, "FunctionDeclaration must have a name");
const original = fn.node.id;
fn.node.id = addSuffix(fn.node.id, "_uncompiled");
// Rename and append compiled function
invariant(compiled.id, "FunctionDeclaration must produce a name");
compiled.id = addSuffix(compiled.id, "_forget");
const compiledFn = fn.insertAfter(compiled)[0];
compiledFn.skip();
// Build and append gating test
compiledFn.insertAfter(buildGatingTest(fn, compiled.id, original));
} else {
fn.replaceWith(compiled);
}
// We are generating a new FunctionDeclaration node, so we must skip over it or this
// traversal will loop infinitely.
fn.skip();
}
const visitor = {
FunctionDeclaration(
fn: BabelCore.NodePath<t.FunctionDeclaration>,
pass: BabelPluginPass
): void {
if (pass.opts.enableOnlyOnUseForgetDirective) {
if (!hasUseForgetDirective(fn.node.body.directives)) {
return;
}
}
if (fn.scope.getProgramParent() !== fn.scope.parent) {
return;
}
hasForgetCompiledCode = true;
const compiled = compile(fn, pass.opts.environment);
if (pass.opts.gatingModule) {
// Rename existing function
invariant(fn.node.id, "FunctionDeclaration must have a name");
const original = fn.node.id;
fn.node.id = addSuffix(fn.node.id, "_uncompiled");
// Rename and append compiled function
invariant(compiled.id, "FunctionDeclaration must produce a name");
compiled.id = addSuffix(compiled.id, "_forget");
const compiledFn = fn.insertAfter(compiled)[0];
compiledFn.skip();
// Build and append gating test
compiledFn.insertAfter(buildGatingTest(fn, compiled.id, original));
} else {
fn.replaceWith(compiled);
}
// We are generating a new FunctionDeclaration node, so we must skip over it or this
// traversal will loop infinitely.
fn.skip();
visitFn(fn, pass);
},
};