@expo/expo-modules-macros-plugin 0.0.9 → 0.2.0

package/apple/Sources/ExpoModulesMacros/MacroHelpers.swift

@@ -39,6 +39,122 @@ internal func classListArgument(of attribute: AttributeSyntax, label: String) ->
   return []
 }
 
+/**
+ Returns true if the class's inheritance clause names any of the given identifiers.
+ Matches by base identifier only, so `Module`, `ExpoModulesCore.Module`, and
+ `Module<Foo>` all match an entry of "Module".
+ */
+internal func inheritsFromAny(_ classDecl: ClassDeclSyntax, names: Set<String>) -> Bool {
+  guard let inherited = classDecl.inheritanceClause?.inheritedTypes else {
+    return false
+  }
+  for entry in inherited {
+    if let name = baseIdentifier(of: entry.type), names.contains(name) {
+      return true
+    }
+  }
+  return false
+}
+
+/**
+ Returns the rightmost identifier of a type, e.g. `Foo` for `Foo`, `Foo` for
+ `Module.Foo`, and nil for composed or generic shapes the macro doesn't need to handle.
+ */
+internal func baseIdentifier(of type: TypeSyntax) -> String? {
+  if let identifier = type.as(IdentifierTypeSyntax.self) {
+    return identifier.name.text
+  }
+  if let member = type.as(MemberTypeSyntax.self) {
+    return member.name.text
+  }
+  return nil
+}
+
+/**
+ True if the declaration carries a `@JS` attribute. Works for functions, properties, and inits;
+ returns false for any other decl kind.
+ */
+internal func memberHasJSAttribute(_ decl: DeclSyntaxProtocol) -> Bool {
+  if let funcDecl = decl.as(FunctionDeclSyntax.self) {
+    return funcDecl.attributes.firstAttribute(named: "JS") != nil
+  }
+  if let varDecl = decl.as(VariableDeclSyntax.self) {
+    return varDecl.attributes.firstAttribute(named: "JS") != nil
+  }
+  if let initDecl = decl.as(InitializerDeclSyntax.self) {
+    return initDecl.attributes.firstAttribute(named: "JS") != nil
+  }
+  return false
+}
+
+/**
+ Decides whether the macro should stamp `@JavaScriptActor` on a `@JS`-marked member.
+ The macro defers to the user when they've already chosen an isolation:
+ - the `nonisolated` modifier is present on the member
+ - any attribute whose name matches a known global actor (`@MainActor`, `@JavaScriptActor`)
+   or follows the `*Actor` naming convention is present on the member or its enclosing type
+ Async members never get the stamp because `AsyncFunction` controls their dispatch separately.
+ */
+internal func shouldStampJavaScriptActor(
+  on member: DeclSyntaxProtocol,
+  enclosedBy enclosing: some DeclGroupSyntax
+) -> Bool {
+  let modifiers = memberModifiers(of: member)
+  if modifiers.contains(where: { $0.name.text == "nonisolated" }) {
+    return false
+  }
+
+  if let funcDecl = member.as(FunctionDeclSyntax.self),
+    funcDecl.signature.effectSpecifiers?.asyncSpecifier != nil {
+    return false
+  }
+
+  let memberAttributes = memberAttributes(of: member)
+  if memberAttributes.contains(where: hasGlobalActorShape) {
+    return false
+  }
+
+  if enclosing.attributes.contains(where: hasGlobalActorShape) {
+    return false
+  }
+
+  return true
+}
+
+private func memberModifiers(of decl: DeclSyntaxProtocol) -> DeclModifierListSyntax {
+  if let funcDecl = decl.as(FunctionDeclSyntax.self) {
+    return funcDecl.modifiers
+  }
+  if let varDecl = decl.as(VariableDeclSyntax.self) {
+    return varDecl.modifiers
+  }
+  if let initDecl = decl.as(InitializerDeclSyntax.self) {
+    return initDecl.modifiers
+  }
+  return DeclModifierListSyntax()
+}
+
+internal func memberAttributes(of decl: DeclSyntaxProtocol) -> AttributeListSyntax {
+  if let funcDecl = decl.as(FunctionDeclSyntax.self) {
+    return funcDecl.attributes
+  }
+  if let varDecl = decl.as(VariableDeclSyntax.self) {
+    return varDecl.attributes
+  }
+  if let initDecl = decl.as(InitializerDeclSyntax.self) {
+    return initDecl.attributes
+  }
+  return AttributeListSyntax()
+}
+
+private func hasGlobalActorShape(_ element: AttributeListSyntax.Element) -> Bool {
+  guard let attribute = element.as(AttributeSyntax.self) else {
+    return false
+  }
+  let name = attribute.attributeName.trimmedDescription
+  return name.hasSuffix("Actor")
+}
+
 extension AttributeListSyntax {
   internal func firstAttribute(named name: String) -> AttributeSyntax? {
     for element in self {