fix(resolution): stream native bridge method scans

Author: jischeng

__tests__/react-native-bridge.test.ts

@@ -27,6 +27,9 @@ function makeContext(nodes: Node[], fileContents: Record<string, string> = {}):
     getNodesByName: (name) => byName.get(name) ?? [],
     getNodesByQualifiedName: () => { throw new Error('not used'); },
     getNodesByKind: (kind) => nodes.filter((n) => n.kind === kind),
+    iterateNodesByKind: function* (kind) {
+      yield* nodes.filter((n) => n.kind === kind);
+    },
     getNodesByLowerName: () => { throw new Error('not used'); },
     fileExists: (fp) => allFiles.has(fp),
     readFile: (fp) => fileContents[fp] ?? null,
@@ -121,6 +124,27 @@ describe('React Native bridge resolver', () => {
       expect(result?.resolvedBy).toBe('framework');
     });
 
+    it('streams method nodes when building the bridge map instead of materializing every method', () => {
+      const native = method('getCurrentPosition:', 'objc', 'RCTGeolocation.m');
+      const ctx = {
+        ...makeContext([native], {
+          'package.json': '{"dependencies":{"react-native":"^0.73"}}',
+          'RCTGeolocation.m':
+            '@implementation RCTGeolocation\n' +
+            'RCT_EXPORT_MODULE()\n' +
+            'RCT_EXPORT_METHOD(getCurrentPosition:(RCTResponseSenderBlock)cb) {}\n' +
+            '@end',
+        }),
+        getNodesByKind: () => { throw new Error('full method scan should not be used'); },
+      } satisfies ResolutionContext;
+
+      const result = reactNativeBridgeResolver.resolve(
+        ref('getCurrentPosition', 'javascript', 'App.js'),
+        ctx
+      );
+      expect(result?.targetNodeId).toBe(native.id);
+    });
+
     it('resolves via explicit module name in RCT_EXPORT_MODULE(name)', () => {
       const native = method('startScan:', 'objc', 'Bluetooth.m');
       const ctx = makeContext([native], {

__tests__/swift-objc-bridge-resolver.test.ts

@@ -21,6 +21,9 @@ function makeContext(nodes: Node[], fileContents: Record<string, string> = {}):
     getNodesByName: (name) => byName.get(name) ?? [],
     getNodesByQualifiedName: () => { throw new Error('not used'); },
     getNodesByKind: (kind) => nodes.filter((n) => n.kind === kind),
+    iterateNodesByKind: function* (kind) {
+      yield* nodes.filter((n) => n.kind === kind);
+    },
     getNodesByLowerName: () => { throw new Error('not used'); },
     fileExists: (fp) => allFiles.has(fp),
     readFile: (fp) => fileContents[fp] ?? null,
@@ -113,6 +116,20 @@ describe('swiftObjcBridgeResolver integration', () => {
       expect(result?.confidence).toBe(0.6);
     });
 
+    it('streams ObjC method nodes when building the bridge map', () => {
+      const objcTarget = method('fetchEntryForKey:', 'objc', 'Cache.m');
+      const ctx = {
+        ...makeContext([objcTarget]),
+        getNodesByKind: () => { throw new Error('full method scan should not be used'); },
+      } satisfies ResolutionContext;
+
+      const result = swiftObjcBridgeResolver.resolve(
+        ref('fetchEntry', 'swift', 'Caller.swift'),
+        ctx
+      );
+      expect(result?.targetNodeId).toBe(objcTarget.id);
+    });
+
     it('does NOT bridge generic Cocoa names like "init" or "description"', () => {
       // Bridging Swift `init()` calls to arbitrary ObjC `init*:` methods is
       // noise — every NSObject subclass has them. The regular name-matcher

src/resolution/frameworks/react-native.ts

@@ -62,6 +62,12 @@ const nativeMethodMaps: WeakMap<
   { byJsName: Map<string, NativeMethod[]> }
 > = new WeakMap();
 
+function methodNodes(context: ResolutionContext): Iterable<Node> {
+  return context.iterateNodesByKind
+    ? context.iterateNodesByKind('method')
+    : context.getNodesByKind('method');
+}
+
 // ─── Native-side extraction ─────────────────────────────────────────────────
 
 /**
@@ -270,7 +276,7 @@ function buildRNMaps(context: ResolutionContext): { byJsName: Map<string, Native
   // their bridge exports.
   const objcMethodsByFirstKw = new Map<string, Node[]>();
   const jvmMethodsByName = new Map<string, Node[]>();
-  for (const node of context.getNodesByKind('method')) {
+  for (const node of methodNodes(context)) {
     if (node.language === 'objc') {
       const firstKw = node.name.includes(':') ? node.name.split(':')[0] : node.name;
       if (firstKw) {

src/resolution/frameworks/swift-objc.ts

@@ -52,6 +52,12 @@ const objcByCandidateSwiftBase: WeakMap<
   Map<string, Node[]>
 > = new WeakMap();
 
+function methodNodes(context: ResolutionContext): Iterable<Node> {
+  return context.iterateNodesByKind
+    ? context.iterateNodesByKind('method')
+    : context.getNodesByKind('method');
+}
+
 /**
  * Build the reverse-bridge map: for every ObjC method node in the graph,
  * compute the Swift base names that would auto-bridge to its selector and
@@ -118,10 +124,8 @@ function buildObjcMap(context: ResolutionContext): Map<string, Node[]> {
   if (cached) return cached;
 
   const map = new Map<string, Node[]>();
-  const objcMethods = context
-    .getNodesByKind('method')
-    .filter((n) => n.language === 'objc');
-  for (const node of objcMethods) {
+  for (const node of methodNodes(context)) {
+    if (node.language !== 'objc') continue;
     const candidates = swiftBaseNamesForObjcSelector(node.name);
     for (const c of candidates) {
       // Skip the trivial case where the Swift base name equals the ObjC
@@ -227,9 +231,15 @@ function resolveObjcCallToSwift(
 
   const candidates = swiftBaseNamesForObjcSelector(rawSelector);
   for (const candidate of candidates) {
-    const matches = context
-      .getNodesByName(candidate)
-      .filter((n) => n.language === 'swift' && (n.kind === 'method' || n.kind === 'function'));
+    const matches = context.getNodesByNameFiltered
+      ? context.getNodesByNameFiltered(candidate, {
+          language: 'swift',
+          kinds: ['method', 'function'],
+          limit: 2000,
+        })
+      : context
+          .getNodesByName(candidate)
+          .filter((n) => n.language === 'swift' && (n.kind === 'method' || n.kind === 'function'));
     for (const match of matches) {
       const window = declarationSourceWindow(match, context);
       if (isObjcExposed(window)) {