babel-plugin-react-compiler 0.0.0-experimental-a9d296d-20250326 → 0.0.0-experimental-b6140c1-20250328

package/dist/index.d.ts

@@ -545,6 +545,7 @@ declare class Environment {
     programContext: ProgramContext;
     hasFireRewrite: boolean;
     hasInferredEffect: boolean;
+    inferredEffectLocations: Set<SourceLocation>;
     constructor(scope: Scope, fnType: ReactFunctionType, compilerMode: CompilerMode, config: EnvironmentConfig, contextIdentifiers: Set<t.Identifier>, logger: Logger | null, filename: string | null, code: string | null, programContext: ProgramContext);
     get isInferredMemoEnabled(): boolean;
     get nextIdentifierId(): IdentifierId;
@@ -1126,13 +1127,18 @@ type InstructionValue = LoadLocal | LoadContext | {
     operator: Exclude<t.UnaryExpression['operator'], 'throw' | 'delete'>;
     value: Place;
     loc: SourceLocation;
-} | {
+} | ({
     kind: 'TypeCastExpression';
     value: Place;
-    typeAnnotation: t.FlowType | t.TSType;
     type: Type;
     loc: SourceLocation;
-} | JsxExpression | {
+} & ({
+    typeAnnotation: t.FlowType;
+    typeAnnotationKind: 'cast';
+} | {
+    typeAnnotation: t.TSType;
+    typeAnnotationKind: 'as' | 'satisfies';
+})) | JsxExpression | {
     kind: 'ObjectExpression';
     properties: Array<ObjectProperty | SpreadPattern>;
     loc: SourceLocation;
@@ -1432,6 +1438,7 @@ type InstructionId = number & {
 type Options = {
     indent: number;
 };
+declare function printFunctionWithOutlined(fn: HIRFunction): string;
 declare function printHIR(ir: HIR, options?: Options | null): string;
 
 declare enum ErrorSeverity {
@@ -1513,9 +1520,11 @@ type CodegenFunction = {
         type: ReactFunctionType | null;
     }>;
     hasInferredEffect: boolean;
+    inferredEffectLocations: Set<SourceLocation>;
     hasFireRewrite: boolean;
 };
 
+declare function printReactiveFunctionWithOutlined(fn: ReactiveFunction): string;
 declare function printReactiveFunction(fn: ReactiveFunction): string;
 
 type CompilerPipelineValue = {
@@ -1640,6 +1649,7 @@ type CompileProgramResult = {
         fn: BabelFn;
         error: CompilerError;
     }>;
+    inferredEffectLocations: Set<t.SourceLocation>;
 };
 declare function compileProgram(program: NodePath$1<t.Program>, pass: CompilerPass): CompileProgramResult | null;
 
@@ -1651,4 +1661,4 @@ declare global {
     let __DEV__: boolean | null | undefined;
 }
 
-export { CompilerError, CompilerErrorDetail, type CompilerErrorDetailOptions, type CompilerPipelineValue, CompilerSuggestionOperation, Effect, type EnvironmentConfig, ErrorSeverity, type ExternalFunction, type Hook, type Logger, type LoggerEvent, OPT_IN_DIRECTIVES, OPT_OUT_DIRECTIVES, type PluginOptions, ProgramContext, type SourceLocation, ValueKind, compileFn as compile, compileProgram, BabelPluginReactCompiler as default, findDirectiveDisablingMemoization, findDirectiveEnablingMemoization, parseConfigPragmaForTests, parsePluginOptions, printHIR, printReactiveFunction, runBabelPluginReactCompiler, validateEnvironmentConfig };
+export { CompilerError, CompilerErrorDetail, type CompilerErrorDetailOptions, type CompilerPipelineValue, CompilerSuggestionOperation, Effect, type EnvironmentConfig, ErrorSeverity, type ExternalFunction, type Hook, type Logger, type LoggerEvent, OPT_IN_DIRECTIVES, OPT_OUT_DIRECTIVES, type PluginOptions, ProgramContext, type SourceLocation, ValueKind, compileFn as compile, compileProgram, BabelPluginReactCompiler as default, findDirectiveDisablingMemoization, findDirectiveEnablingMemoization, parseConfigPragmaForTests, parsePluginOptions, printFunctionWithOutlined, printHIR, printReactiveFunction, printReactiveFunctionWithOutlined, runBabelPluginReactCompiler, validateEnvironmentConfig };

package/dist/index.js

@@ -110434,8 +110434,10 @@ __export(index_exports, {
   findDirectiveEnablingMemoization: () => findDirectiveEnablingMemoization,
   parseConfigPragmaForTests: () => parseConfigPragmaForTests,
   parsePluginOptions: () => parsePluginOptions,
+  printFunctionWithOutlined: () => printFunctionWithOutlined,
   printHIR: () => printHIR,
   printReactiveFunction: () => printReactiveFunction,
+  printReactiveFunctionWithOutlined: () => printReactiveFunctionWithOutlined,
   runBabelPluginReactCompiler: () => runBabelPluginReactCompiler,
   validateEnvironmentConfig: () => validateEnvironmentConfig
 });
@@ -114959,6 +114961,14 @@ function getHookKindForType(env, type) {
 var import_generator = __toESM(require_lib26());
 
 // src/ReactiveScopes/PrintReactiveFunction.ts
+function printReactiveFunctionWithOutlined(fn) {
+  const writer = new Writer();
+  writeReactiveFunction(fn, writer);
+  for (const outlined of fn.env.getOutlinedFunctions()) {
+    writer.writeLine("\nfunction " + printFunction(outlined.fn));
+  }
+  return writer.complete();
+}
 function printReactiveFunction(fn) {
   const writer = new Writer();
   writeReactiveFunction(fn, writer);
@@ -115326,6 +115336,15 @@ _line = new WeakMap();
 _depth = new WeakMap();
 
 // src/HIR/PrintHIR.ts
+function printFunctionWithOutlined(fn) {
+  const output = [printFunction(fn)];
+  for (const outlined of fn.env.getOutlinedFunctions()) {
+    output.push(`
+function ${outlined.fn.id}:
+${printHIR(outlined.fn.body)}`);
+  }
+  return output.join("\n");
+}
 function printFunction(fn) {
   const output = [];
   let definition = "";
@@ -120998,6 +121017,19 @@ function lowerExpression(builder, exprPath) {
         kind: "TypeCastExpression",
         value: lowerExpressionToTemporary(builder, expr.get("expression")),
         typeAnnotation: typeAnnotation2.node,
+        typeAnnotationKind: "cast",
+        type: lowerType(typeAnnotation2.node),
+        loc: exprLoc
+      };
+    }
+    case "TSSatisfiesExpression": {
+      let expr = exprPath;
+      const typeAnnotation2 = expr.get("typeAnnotation");
+      return {
+        kind: "TypeCastExpression",
+        value: lowerExpressionToTemporary(builder, expr.get("expression")),
+        typeAnnotation: typeAnnotation2.node,
+        typeAnnotationKind: "satisfies",
         type: lowerType(typeAnnotation2.node),
         loc: exprLoc
       };
@@ -121009,6 +121041,7 @@ function lowerExpression(builder, exprPath) {
         kind: "TypeCastExpression",
         value: lowerExpressionToTemporary(builder, expr.get("expression")),
         typeAnnotation: typeAnnotation2.node,
+        typeAnnotationKind: "as",
         type: lowerType(typeAnnotation2.node),
         loc: exprLoc
       };
@@ -124417,6 +124450,7 @@ function parseConfigPragmaForTests(pragma, defaults) {
   const environment = parseConfigPragmaEnvironmentForTest(pragma);
   let compilationMode = defaults.compilationMode;
   let panicThreshold = "all_errors";
+  let noEmit = defaultOptions.noEmit;
   for (const token2 of pragma.split(" ")) {
     if (!token2.startsWith("@")) {
       continue;
@@ -124442,12 +124476,17 @@ function parseConfigPragmaForTests(pragma, defaults) {
         panicThreshold = "none";
         break;
       }
+      case "@noEmit": {
+        noEmit = true;
+        break;
+      }
     }
   }
   return parsePluginOptions({
     environment,
     compilationMode,
-    panicThreshold
+    panicThreshold,
+    noEmit
   });
 }
 var _globals, _shapes, _moduleTypes, _nextIdentifer, _nextBlock, _nextScope, _scope, _outlinedFunctions, _contextIdentifiers, _hoistedIdentifiers, _Environment_instances, resolveModuleType_fn, isKnownReactModule_fn, getCustomHookType_fn;
@@ -124462,6 +124501,7 @@ var Environment = class {
     __privateAdd(this, _nextScope, 0);
     __privateAdd(this, _scope);
     __privateAdd(this, _outlinedFunctions, []);
+    this.inferredEffectLocations = /* @__PURE__ */ new Set();
     __privateAdd(this, _contextIdentifiers);
     __privateAdd(this, _hoistedIdentifiers);
     __privateSet(this, _scope, scope);
@@ -129590,7 +129630,8 @@ function codegenReactiveFunction(cx, fn) {
     prunedMemoValues: countMemoBlockVisitor.prunedMemoValues,
     outlined: [],
     hasFireRewrite: fn.env.hasFireRewrite,
-    hasInferredEffect: fn.env.hasInferredEffect
+    hasInferredEffect: fn.env.hasInferredEffect,
+    inferredEffectLocations: fn.env.inferredEffectLocations
   });
 }
 var CountMemoBlockVisitor = class extends ReactiveFunctionVisitor {
@@ -131079,10 +131120,17 @@ function codegenInstructionValue(cx, instrValue) {
     }
     case "TypeCastExpression": {
       if (t3.isTSType(instrValue.typeAnnotation)) {
-        value = t3.tsAsExpression(
-          codegenPlaceToExpression(cx, instrValue.value),
-          instrValue.typeAnnotation
-        );
+        if (instrValue.typeAnnotationKind === "satisfies") {
+          value = t3.tsSatisfiesExpression(
+            codegenPlaceToExpression(cx, instrValue.value),
+            instrValue.typeAnnotation
+          );
+        } else {
+          value = t3.tsAsExpression(
+            codegenPlaceToExpression(cx, instrValue.value),
+            instrValue.typeAnnotation
+          );
+        }
       } else {
         value = t3.typeCastExpression(
           codegenPlaceToExpression(cx, instrValue.value),
@@ -136458,6 +136506,7 @@ function inferEffectDependencies(fn) {
             });
             value.args.push(__spreadProps(__spreadValues({}, depsPlace), { effect: "freeze" /* Freeze */ }));
             rewriteInstrs.set(instr.id, newInstructions);
+            fn.env.inferredEffectLocations.add(callee.loc);
           } else if (loadGlobals.has(value.args[0].identifier.id)) {
             newInstructions.push({
               id: makeInstructionId(0),
@@ -136467,6 +136516,7 @@ function inferEffectDependencies(fn) {
             });
             value.args.push(__spreadProps(__spreadValues({}, depsPlace), { effect: "freeze" /* Freeze */ }));
             rewriteInstrs.set(instr.id, newInstructions);
+            fn.env.inferredEffectLocations.add(callee.loc);
           }
         }
       }
@@ -143028,6 +143078,7 @@ function compileProgram(program, pass) {
     })
   );
   const retryErrors = [];
+  const inferredEffectLocations = /* @__PURE__ */ new Set();
   const processFn = (fn, fnType) => {
     var _a2, _b2, _c, _d, _e, _f, _g, _h, _i;
     let optInDirectives = [];
@@ -143131,6 +143182,9 @@ function compileProgram(program, pass) {
     if (!pass.opts.noEmit) {
       return compileResult.compiledFn;
     }
+    for (const loc of compileResult.compiledFn.inferredEffectLocations) {
+      if (loc !== GeneratedSource) inferredEffectLocations.add(loc);
+    }
     return null;
   };
   while (queue.length !== 0) {
@@ -143194,7 +143248,7 @@ function compileProgram(program, pass) {
   if (compiledFns.length > 0) {
     addImportsToProgram(program, programContext);
   }
-  return { retryErrors };
+  return { retryErrors, inferredEffectLocations };
 }
 function shouldSkipCompilation(program, pass) {
   if (pass.opts.sources) {
@@ -143888,7 +143942,9 @@ function assertValidEffectImportReference(numArgs, paths, context) {
     const parent = path.parentPath;
     if (parent != null && parent.isCallExpression()) {
       const args = parent.get("arguments");
-      if (args.length === numArgs) {
+      const maybeCalleeLoc = path.node.loc;
+      const hasInferredEffect = maybeCalleeLoc != null && context.inferredEffectLocations.has(maybeCalleeLoc);
+      if (args.length === numArgs && !hasInferredEffect) {
         const maybeErrorDiagnostic = matchCompilerDiagnostic(
           path,
           context.transformErrors
@@ -143922,7 +143978,7 @@ function assertValidFireImportReference(paths, context) {
     );
   }
 }
-function validateNoUntransformedReferences(path, filename, logger, env, transformErrors) {
+function validateNoUntransformedReferences(path, filename, logger, env, compileResult) {
   const moduleLoadChecks = /* @__PURE__ */ new Map();
   if (env.enableFire) {
     for (const module2 of Environment.knownReactModules) {
@@ -143943,7 +143999,7 @@ function validateNoUntransformedReferences(path, filename, logger, env, transfor
     }
   }
   if (moduleLoadChecks.size > 0) {
-    transformProgram(path, moduleLoadChecks, filename, logger, transformErrors);
+    transformProgram(path, moduleLoadChecks, filename, logger, compileResult);
   }
 }
 function validateImportSpecifier(specifier, importSpecifierChecks, state) {
@@ -144001,13 +144057,15 @@ function validateNamespacedImport(specifier, importSpecifierChecks, state) {
     checkFn(references, state);
   }
 }
-function transformProgram(path, moduleLoadChecks, filename, logger, transformErrors) {
+function transformProgram(path, moduleLoadChecks, filename, logger, compileResult) {
+  var _a, _b;
   const traversalState = {
     shouldInvalidateScopes: true,
     program: path,
     filename,
     logger,
-    transformErrors
+    transformErrors: (_a = compileResult == null ? void 0 : compileResult.retryErrors) != null ? _a : [],
+    inferredEffectLocations: (_b = compileResult == null ? void 0 : compileResult.inferredEffectLocations) != null ? _b : /* @__PURE__ */ new Set()
   };
   path.traverse({
     ImportDeclaration(path2) {
@@ -144058,7 +144116,7 @@ function BabelPluginReactCompiler(_babel) {
        */
       Program: {
         enter(prog, pass) {
-          var _a, _b, _c, _d, _e;
+          var _a, _b, _c, _d;
           const filename = (_a = pass.filename) != null ? _a : "unknown";
           if (ENABLE_REACT_COMPILER_TIMINGS === true) {
             performance.mark(`${filename}:start`, {
@@ -144088,7 +144146,7 @@ function BabelPluginReactCompiler(_babel) {
             (_d = pass.filename) != null ? _d : null,
             opts.logger,
             opts.environment,
-            (_e = result == null ? void 0 : result.retryErrors) != null ? _e : []
+            result
           );
           if (ENABLE_REACT_COMPILER_TIMINGS === true) {
             performance.mark(`${filename}:end`, {
@@ -144166,8 +144224,10 @@ var index_default = BabelPluginReactCompiler;
   findDirectiveEnablingMemoization,
   parseConfigPragmaForTests,
   parsePluginOptions,
+  printFunctionWithOutlined,
   printHIR,
   printReactiveFunction,
+  printReactiveFunctionWithOutlined,
   runBabelPluginReactCompiler,
   validateEnvironmentConfig
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "babel-plugin-react-compiler",
-  "version": "0.0.0-experimental-a9d296d-20250326",
+  "version": "0.0.0-experimental-b6140c1-20250328",
   "description": "Babel plugin for React Compiler.",
   "main": "dist/index.js",
   "license": "MIT",
@@ -12,12 +12,12 @@
     "build": "rimraf dist && tsup",
     "test": "./scripts/link-react-compiler-runtime.sh && yarn snap:ci",
     "jest": "yarn build && ts-node node_modules/.bin/jest",
-    "snap": "node ../snap/dist/main.js",
+    "snap": "yarn workspace snap run snap",
     "snap:build": "yarn workspace snap run build",
     "snap:ci": "yarn snap:build && yarn snap",
     "ts:analyze-trace": "scripts/ts-analyze-trace.sh",
     "lint": "yarn eslint src",
-    "watch": "yarn build --watch"
+    "watch": "yarn build --dts --watch"
   },
   "dependencies": {
     "@babel/types": "^7.26.0"