rnxsim 0.0.0 → 0.1.312

package/src/worklets-babel.ts

@@ -0,0 +1,134 @@
+import { createRequire } from 'node:module'
+import path from 'node:path'
+import { fileURLToPath } from 'node:url'
+import type { TransformOptions } from '@babel/core'
+import type { TransformResult } from 'vite'
+
+const sootsimPluginRequire = createRequire(import.meta.url)
+const sootsimRoot = path.resolve(path.dirname(fileURLToPath(import.meta.url)), '..')
+const workspaceRoot = path.resolve(sootsimRoot, '..', '..')
+
+// handle files containing upstream reanimated/worklets worklet directives and
+// inline updaters to auto-workletizable hooks. the babel transform lifts those
+// into functions carrying `__closure`, `__workletHash`, and `__initData`.
+export const REANIMATED_AUTOWORKLETIZATION_KEYWORDS = [
+  'worklet',
+  'useAnimatedGestureHandler',
+  'useAnimatedScrollHandler',
+  'useFrameCallback',
+  'useAnimatedStyle',
+  'useAnimatedProps',
+  'createAnimatedPropAdapter',
+  'useDerivedValue',
+  'useAnimatedReaction',
+  'useWorkletCallback',
+  'withTiming',
+  'withSpring',
+  'withDecay',
+  'withRepeat',
+  'runOnUI',
+  'executeOnUIRuntimeSync',
+]
+
+export const REANIMATED_WORKLETIZATION_REGEX = new RegExp(
+  REANIMATED_AUTOWORKLETIZATION_KEYWORDS.join('|'),
+)
+// false-positive packages: they mention the keywords in strings/comments but
+// don't actually use worklets, or transforming them causes circular deps /
+// breaks the build. keep this minimal; the regex above is already
+// content-gated so most of node_modules skips for free.
+const REANIMATED_IGNORED_PATHS_REGEX =
+  /node_modules\/(react|react-dom|react-native|react-native-web)\//
+const SOOTSIM_ENGINE_SRC_PATH = '/packages/sootsim-engine/src/'
+const SOOTSIM_ENGINE_TEST_FIXTURE_PATH = '/packages/sootsim-engine/src/test-fixtures/'
+const REANIMATED_IMPORT_REGEX =
+  /(?:from\s+['"]react-native-reanimated(?:\/[^'"]*)?['"]|require\(\s*['"]react-native-reanimated(?:\/[^'"]*)?['"]\s*\))/
+
+let workletsPluginPath: string | null = null
+let babelCorePromise: Promise<typeof import('@babel/core')> | null = null
+
+export function shouldApplyWorkletsPlugin(id: string, code: string): boolean {
+  if (!/\.(tsx?|jsx?|mjs|cjs|mts|cts)$/.test(id)) return false
+  if (REANIMATED_IGNORED_PATHS_REGEX.test(id)) return false
+  // sootsim engine source has local helpers named like upstream reanimated
+  // hooks. only transform engine files that actually import upstream
+  // react-native-reanimated; otherwise the plugin rewrites callbacks for our
+  // own engine hooks and expects a `this.__closure` binding they never provide.
+  if (
+    id.includes(SOOTSIM_ENGINE_SRC_PATH) &&
+    !id.includes(SOOTSIM_ENGINE_TEST_FIXTURE_PATH) &&
+    !REANIMATED_IMPORT_REGEX.test(code)
+  ) {
+    return false
+  }
+  return REANIMATED_WORKLETIZATION_REGEX.test(code)
+}
+
+function getWorkletsPluginPath(): string {
+  if (!workletsPluginPath) {
+    workletsPluginPath = path.resolve(
+      workspaceRoot,
+      'node_modules',
+      'react-native-worklets',
+      'plugin',
+    )
+  }
+  return workletsPluginPath
+}
+
+function isTypescriptFile(id: string): boolean {
+  return /\.(tsx?|mts|cts)$/.test(id)
+}
+
+function createWorkletsBabelOptions(
+  id: string,
+  sourceMaps: TransformOptions['sourceMaps'],
+): TransformOptions {
+  const isTSX = id.endsWith('.tsx')
+  const isTS = isTypescriptFile(id)
+  return {
+    filename: id,
+    babelrc: false,
+    configFile: false,
+    sourceMaps,
+    presets: isTS
+      ? [
+          [
+            sootsimPluginRequire.resolve('@babel/preset-typescript'),
+            { isTSX, allExtensions: isTSX, allowDeclareFields: true },
+          ],
+        ]
+      : [],
+    plugins: [
+      [sootsimPluginRequire.resolve('@babel/plugin-syntax-jsx')],
+      [getWorkletsPluginPath(), { processNestedWorklets: true }],
+    ],
+  }
+}
+
+export async function transformWorkletsCode(
+  code: string,
+  id: string,
+  sourceMaps: TransformOptions['sourceMaps'] = true,
+): Promise<Pick<TransformResult, 'code' | 'map'> | null> {
+  if (!babelCorePromise) {
+    babelCorePromise = import('@babel/core') as Promise<typeof import('@babel/core')>
+  }
+  const { transformAsync } = await babelCorePromise
+  const result = await transformAsync(code, createWorkletsBabelOptions(id, sourceMaps))
+  if (!result?.code) return null
+  return { code: result.code, map: (result.map ?? null) as TransformResult['map'] }
+}
+
+export function transformWorkletsCodeSync(
+  code: string,
+  id: string,
+  sourceMaps: TransformOptions['sourceMaps'] = false,
+): Pick<TransformResult, 'code' | 'map'> | null {
+  const { transformSync } = sootsimPluginRequire(
+    '@babel/core',
+  ) as typeof import('@babel/core')
+  const result = transformSync(code, createWorkletsBabelOptions(id, sourceMaps))
+  if (!result?.code) return null
+  return { code: result.code, map: (result.map ?? null) as TransformResult['map'] }
+}