objcjs-types 0.2.1 → 0.4.0

package/generator/discover.ts

@@ -0,0 +1,111 @@
+/**
+ * Auto-discovers ObjC classes and protocols by scanning framework header files.
+ * Eliminates the need for manually maintaining class/protocol lists and header mappings.
+ */
+
+import { readdir } from "fs/promises";
+import { join } from "path";
+
+export interface DiscoveryResult {
+  /** Maps class name → header file name (without .h extension) */
+  classes: Map<string, string>;
+  /** Maps protocol name → header file name (without .h extension) */
+  protocols: Map<string, string>;
+  /** Maps integer enum name → header file name (without .h extension) */
+  integerEnums: Map<string, string>;
+  /** Maps string enum name → header file name (without .h extension) */
+  stringEnums: Map<string, string>;
+}
+
+/** Matches `@interface ClassName` — captures class name */
+const INTERFACE_RE = /@interface\s+(\w+)/;
+
+/** Matches category or extension: `@interface ClassName (` or `@interface ClassName<Type> (` */
+const CATEGORY_RE = /@interface\s+\w+(?:<[^>]*>)?\s*\(/;
+
+/** Matches `@protocol ProtocolName` — captures protocol name */
+const PROTOCOL_RE = /@protocol\s+(\w+)/;
+
+/**
+ * Matches NS_ENUM / NS_OPTIONS integer enum declarations.
+ * e.g., `typedef NS_ENUM(NSInteger, NSWindowStyleMask) {`
+ * e.g., `typedef NS_OPTIONS(NSUInteger, NSWindowStyleMask) {`
+ */
+const NS_ENUM_RE = /typedef\s+NS_(?:ENUM|OPTIONS)\s*\(\s*\w+\s*,\s*(\w+)\s*\)/;
+
+/**
+ * Matches NS_TYPED_EXTENSIBLE_ENUM / NS_STRING_ENUM / NS_TYPED_ENUM string enum declarations.
+ * These are NSString * typedefs whose values are extern constants.
+ * e.g., `typedef NSString * ASAuthorizationPublicKeyCredentialUserVerificationPreference NS_TYPED_EXTENSIBLE_ENUM;`
+ * Also matches the multi-line form where the macro is on the typedef.
+ */
+const NS_STRING_ENUM_RE = /typedef\s+NSString\s*\*\s*(\w+)\s+NS_(?:TYPED_EXTENSIBLE_ENUM|STRING_ENUM|TYPED_ENUM)/;
+
+/**
+ * Scan all .h files in a framework's Headers directory and discover
+ * every ObjC class and protocol declaration, mapping each to its header file.
+ *
+ * Class declarations: `@interface Foo : Bar` or `@interface Foo <Protocol>`
+ * Protocol declarations: `@protocol Foo <Bar>` (not forward decls like `@protocol Foo;`)
+ * Skipped: categories (`@interface Foo (Cat)`), extensions (`@interface Foo ()`)
+ */
+export async function discoverFramework(headersPath: string): Promise<DiscoveryResult> {
+  const classes = new Map<string, string>();
+  const protocols = new Map<string, string>();
+  const integerEnums = new Map<string, string>();
+  const stringEnums = new Map<string, string>();
+
+  const entries = await readdir(headersPath);
+  const headerFiles = entries.filter((f) => f.endsWith(".h")).sort();
+
+  for (const file of headerFiles) {
+    const headerName = file.slice(0, -2); // strip .h
+    const content = await Bun.file(join(headersPath, file)).text();
+    const lines = content.split("\n");
+
+    for (const line of lines) {
+      // --- Class declarations ---
+      // Match @interface Foo : Bar, @interface Foo <Protocol>, etc.
+      // Skip categories: @interface Foo (Category) or extensions: @interface Foo ()
+      const ifaceMatch = INTERFACE_RE.exec(line);
+      if (ifaceMatch && !CATEGORY_RE.test(line)) {
+        const name = ifaceMatch[1]!;
+        // Only record first occurrence — handles redeclarations in other headers
+        if (!classes.has(name)) {
+          classes.set(name, headerName);
+        }
+      }
+
+      // --- Protocol declarations ---
+      // Match @protocol Foo <Bar> or @protocol Foo (followed by newline/methods)
+      // Skip forward declarations: @protocol Foo; or @protocol Foo, Bar;
+      const protoMatch = PROTOCOL_RE.exec(line);
+      if (protoMatch && !line.includes(";")) {
+        const name = protoMatch[1]!;
+        if (!protocols.has(name)) {
+          protocols.set(name, headerName);
+        }
+      }
+
+      // --- Integer enum declarations (NS_ENUM / NS_OPTIONS) ---
+      const enumMatch = NS_ENUM_RE.exec(line);
+      if (enumMatch) {
+        const name = enumMatch[1]!;
+        if (!integerEnums.has(name)) {
+          integerEnums.set(name, headerName);
+        }
+      }
+
+      // --- String enum declarations (NS_TYPED_EXTENSIBLE_ENUM etc.) ---
+      const stringEnumMatch = NS_STRING_ENUM_RE.exec(line);
+      if (stringEnumMatch) {
+        const name = stringEnumMatch[1]!;
+        if (!stringEnums.has(name)) {
+          stringEnums.set(name, headerName);
+        }
+      }
+    }
+  }
+
+  return { classes, protocols, integerEnums, stringEnums };
+}