worklets plugin: allow react-native imports in bundle mode

packages/react-native-worklets/__tests__/plugin.test.ts

@@ -540,6 +540,52 @@ describe('babel plugin', () => {
     });
   });
 
+  describe('for bundle mode runtime entry', () => {
+    const workletRuntimeEntryFilename =
+      '/tmp/react-native-worklets/src/initializers/workletRuntimeEntry.native.ts';
+
+    test('enables bundle mode in worklet runtime entry', () => {
+      const input = html`<script>
+        globalThis._WORKLETS_BUNDLE_MODE_ENABLED = false;
+      </script>`;
+
+      const { code } = runPlugin(input, undefined, {
+        bundleMode: true,
+      }, workletRuntimeEntryFilename);
+
+      expect(code).toContain('globalThis._WORKLETS_BUNDLE_MODE_ENABLED = true;');
+    });
+
+    test('allows react-native imports when explicitly configured', () => {
+      const input = html`<script>
+        globalThis._WORKLETS_REACT_NATIVE_IMPORTS_ALLOWED = false;
+      </script>`;
+
+      const { code } = runPlugin(input, undefined, {
+        bundleMode: true,
+        workletizableModules: ['react-native'],
+      }, workletRuntimeEntryFilename);
+
+      expect(code).toContain(
+        'globalThis._WORKLETS_REACT_NATIVE_IMPORTS_ALLOWED = true;'
+      );
+    });
+
+    test("doesn't allow react-native imports by default", () => {
+      const input = html`<script>
+        globalThis._WORKLETS_REACT_NATIVE_IMPORTS_ALLOWED = false;
+      </script>`;
+
+      const { code } = runPlugin(input, undefined, {
+        bundleMode: true,
+      }, workletRuntimeEntryFilename);
+
+      expect(code).toContain(
+        'globalThis._WORKLETS_REACT_NATIVE_IMPORTS_ALLOWED = false;'
+      );
+    });
+  });
+
   describe('for explicit worklets', () => {
     test('workletizes FunctionDeclaration', () => {
       const input = html`<script>

packages/react-native-worklets/plugin/src/bundleMode.ts

@@ -30,10 +30,28 @@ export function toggleBundleMode(
     return;
   }
 
+  const propertyName = getGlobalPropertyName(expressionPath);
+
+  if (propertyName === '_WORKLETS_BUNDLE_MODE_ENABLED') {
+    expressionPath.get('right').replaceWith(booleanLiteral(true));
+    return;
+  }
+
+  if (
+    propertyName === '_WORKLETS_REACT_NATIVE_IMPORTS_ALLOWED' &&
+    areReactNativeImportsAllowed(state)
+  ) {
+    expressionPath.get('right').replaceWith(booleanLiteral(true));
+  }
+}
+
+function getGlobalPropertyName(
+  expressionPath: NodePath<ExpressionStatement['expression']>
+) {
   const left = expressionPath.get('left');
 
   if (!left.isMemberExpression()) {
-    return;
+    return null;
   }
 
   const object = left.get('object');
@@ -42,13 +60,14 @@ export function toggleBundleMode(
   if (
     !object.isIdentifier() ||
     object.node.name !== 'globalThis' ||
-    !property.isIdentifier() ||
-    property.node.name !== '_WORKLETS_BUNDLE_MODE_ENABLED'
+    !property.isIdentifier()
   ) {
-    return;
+    return null;
   }
 
-  const right = expressionPath.get('right');
+  return property.node.name;
+}
 
-  right.replaceWith(booleanLiteral(true));
+function areReactNativeImportsAllowed(state: WorkletsPluginPass) {
+  return state.opts.workletizableModules?.includes('react-native') ?? false;
 }

packages/react-native-worklets/src/initializers/initializers.native.ts

@@ -152,7 +152,9 @@ function initializeWorkletRuntime() {
 
     if (__DEV__) {
       silenceHMRWarnings();
-      disallowRNImports();
+      if (!globalThis._WORKLETS_REACT_NATIVE_IMPORTS_ALLOWED) {
+        disallowRNImports();
+      }
     }
 
     if (getStaticFeatureFlag('FETCH_PREVIEW_ENABLED')) {

packages/react-native-worklets/src/initializers/workletRuntimeEntry.native.ts

@@ -20,6 +20,7 @@ export function bundleModeInit() {
   // Worklets Babel Plugin replaces `false` with `true` here
   // when Bundle Mode is enabled.
   globalThis._WORKLETS_BUNDLE_MODE_ENABLED = false;
+  globalThis._WORKLETS_REACT_NATIVE_IMPORTS_ALLOWED = false;
 
   if (!globalThis._WORKLETS_BUNDLE_MODE_ENABLED) {
     return;

packages/react-native-worklets/src/privateGlobals.d.ts

@@ -22,6 +22,7 @@ declare global {
   var _toString: (value: unknown) => string;
   var __workletsModuleProxy: WorkletsModuleProxy;
   var _WORKLETS_BUNDLE_MODE_ENABLED: boolean | undefined;
+  var _WORKLETS_REACT_NATIVE_IMPORTS_ALLOWED: boolean | undefined;
   var _WORKLETS_VERSION_CPP: string | undefined;
   var _WORKLETS_VERSION_JS: string | undefined;
   var _createSerializable: <T>(