[Bug] Annotations aren't being extracted from initializers (#1311)

krzysztofzablocki · Mar 27, 2024 · 2f9cf2c · 2f9cf2c
1 parent b2d484c
commit 2f9cf2c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/SourceryFramework/Sources/Parsing/Utils/AnnotationsParser.swift b/SourceryFramework/Sources/Parsing/Utils/AnnotationsParser.swift
@@ -199,8 +199,10 @@ public struct AnnotationsParser {
         if prefix.isEmpty {
             shouldUsePositionBeforeTrailing = true
             (prefix, sourceLine) = findPrefix(positionBeforeTrailingTrivia, shouldUsePositionBeforeTrailing)
-            if shouldUsePositionBeforeTrailing {
+            if shouldUsePositionBeforeTrailing && !prefix.isEmpty {
                 position = positionBeforeTrailingTrivia
+            } else {
+                shouldUsePositionBeforeTrailing = false
             }
         }
         guard !prefix.isEmpty else { return ([:], shouldUsePositionBeforeTrailing) }

diff --git a/SourceryTests/Parsing/FileParser_MethodsSpec.swift b/SourceryTests/Parsing/FileParser_MethodsSpec.swift
@@ -456,6 +456,20 @@ class FileParserMethodsSpec: QuickSpec {
                         ]))
                 }
 
+                it("extracts method annotations from initializers") {
+                    expect(parse("""
+                    class Foo {
+                        // sourcery: annotation
+                        init() {
+                        }
+                    }
+                    """)).to(equal([
+                        Class(name: "Foo", methods: [
+                                Method(name: "init()", selectorName: "init", returnTypeName: TypeName(name: "Foo"), isStatic: true, annotations: ["annotation": NSNumber(value: true)], definedInTypeName: TypeName(name: "Foo"))
+                            ])
+                        ]))
+                }
+
                 it("extracts method inline annotations") {
                     expect(parse("class Foo {\n /* sourcery: annotation */func foo() {} }")).to(equal([
                         Class(name: "Foo", methods: [