Not to add computed property into initialisor

ShenghaiWang · Dec 20, 2023 · dc0db4c · dc0db4c
1 parent bc210ec
commit dc0db4c
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 2 deletions.
diff --git a/Sources/Macros/AddInit.swift b/Sources/Macros/AddInit.swift
@@ -33,7 +33,8 @@ public struct AddInit: MemberMacro {
             if let patternBinding = member.decl.as(VariableDeclSyntax.self)?.bindings
                 .as(PatternBindingListSyntax.self)?.first?.as(PatternBindingSyntax.self),
                let identifier = patternBinding.pattern.as(IdentifierPatternSyntax.self)?.identifier,
-               let type =  patternBinding.typeAnnotation?.as(TypeAnnotationSyntax.self)?.type {
+               let type =  patternBinding.typeAnnotation?.as(TypeAnnotationSyntax.self)?.type,
+               patternBinding.accessorBlock == nil {
                 var parameter = "\(identifier): "
                 if type.is(FunctionTypeSyntax.self) {
                     parameter += "@escaping "
@@ -57,7 +58,8 @@ public struct AddInit: MemberMacro {
             guard let patternBinding = member.decl.as(VariableDeclSyntax.self)?.bindings
                 .as(PatternBindingListSyntax.self)?.first?.as(PatternBindingSyntax.self),
                   let identifier = patternBinding.pattern.as(IdentifierPatternSyntax.self)?.identifier,
-                  let type =  patternBinding.typeAnnotation?.as(TypeAnnotationSyntax.self)?.type else { return nil }
+                  let type =  patternBinding.typeAnnotation?.as(TypeAnnotationSyntax.self)?.type,
+                  patternBinding.accessorBlock == nil else { return nil }
             let mockValue = type.mockValue(randomValue: randomValue)
             ?? type.as(OptionalTypeSyntax.self)?.mockValue(randomValue: randomValue)
             ?? "nil"

diff --git a/Tests/MacroTests/AddInitTests.swift b/Tests/MacroTests/AddInitTests.swift
@@ -34,6 +34,35 @@ final class AddInitTests: XCTestCase {
         )
     }
 
+    func testAddInitWithComputedPropertyMacro() {
+        assertMacroExpansion(
+            """
+            @AddInit
+            struct A {
+                let a: Int?
+                var b: Int {
+                    0
+                }
+            }
+            """,
+            expandedSource:
+            """
+
+            struct A {
+                let a: Int?
+                var b: Int {
+                    0
+                }
+
+                init(a: Int? = nil) {
+                    self.a = a
+                }
+            }
+            """,
+            macros: testMacros
+        )
+    }
+
     func testAddPublicInitMacro() {
         assertMacroExpansion(
             """