Revert Prism macro back to member macro instead of ext

Author: luizmb

Sources/SwiftRexMacroImplementation/Macros/Prism/Prism.swift

@@ -28,14 +28,12 @@ public struct PrismCase: PeerMacro {
  }
 }
 
-public struct Prism: ExtensionMacro {
+public struct Prism: MemberMacro {
  public static func expansion(
  of node: SwiftSyntax.AttributeSyntax,
- attachedTo declaration: some SwiftSyntax.DeclGroupSyntax,
- providingExtensionsOf type: some SwiftSyntax.TypeSyntaxProtocol,
- conformingTo protocols: [SwiftSyntax.TypeSyntax],
+ providingMembersOf declaration: some SwiftSyntax.DeclGroupSyntax,
  in context: some SwiftSyntaxMacros.MacroExpansionContext
- ) throws -> [SwiftSyntax.ExtensionDeclSyntax] {
+ ) throws -> [SwiftSyntax.DeclSyntax] {
  guard let enumDeclaration = declaration.as(EnumDeclSyntax.self) else {
  throw MacroError.notAnEnum
  }
@@ -49,26 +47,13 @@ public struct Prism: ExtensionMacro {
  $0.elements.compactMap { $0.name.text.isEmpty ? nil : $0 }
  }
 
- let (prisms, predicates) = (
+ return cases
+ .map(\.prism)
+ .map { $0.set(visibility: node.visibility) }
+ .compactMap { $0.as(DeclSyntax.self) } +
  cases
- .map(\.prism).map { $0.set(visibility: node.visibility) }
- .map { MemberBlockItemSyntax(decl: $0) },
- cases
- .map(\.predicate).map { $0.set(visibility: node.visibility) }
- .map { MemberBlockItemSyntax(decl: $0) }
- )
-
- return [
- prisms.isEmpty ? nil : ExtensionDeclSyntax(
- attributes: AttributeListSyntax(""),
- extendedType: type,
- memberBlock: .init(members: MemberBlockItemListSyntax(prisms))
- ),
- predicates.isEmpty ? nil : ExtensionDeclSyntax(
- attributes: AttributeListSyntax(""),
- extendedType: type,
- memberBlock: .init(members: MemberBlockItemListSyntax(predicates))
- )
- ].compactMap { $0 }
+ .map(\.predicate)
+ .map { $0.set(visibility: node.visibility) }
+ .compactMap { $0.as(DeclSyntax.self) }
  }
 }

Sources/SwiftRexMacros/SwiftRexMacros.swift

@@ -101,7 +101,7 @@ public enum DeclarationVisibility: String {
 /// - Use `PrismCase` in a case if you want a different visibility only for that case generated code.
 /// - Use only `PrismCase` without `Prism` in the enum if you want code generated only for that case.
 /// - Use `NoPrism` in a case if you don't want code generated for that case.
-@attached(extension, names: arbitrary)
+@attached(member, names: arbitrary)
 public macro Prism(visibility: DeclarationVisibility = .internal) = #externalMacro(module: "SwiftRexMacroImplementation", type: "Prism")
 
 /// A macro that produces predicates and prisms for a single case of an Enum.