Fix undetectable mock annotations with attributes above. (#281)

* Fix annotation cannot detect when attributes attached above it

* Fix fragile output

* Also support class

* search more trivia

Sources/MockoloFramework/Parsers/SwiftSyntaxExtensions.swift

@@ -14,6 +14,7 @@
 //  limitations under the License.
 //
 
+import Algorithms
 import Foundation
 import SwiftSyntax
 import SwiftParser
@@ -259,7 +260,7 @@ extension IfConfigDeclSyntax {
             in: subModels,
             exclude: [:],
             fullnames: []
-        ).map({ $0 })
+        ).sorted(path: \.value.offset, fallback: \.key)
 
         let macroModel = IfMacroModel(name: name, offset: self.offset, entities: uniqueSubModels)
         return (macroModel, attrDesc, hasInit)
@@ -300,7 +301,12 @@ extension ProtocolDeclSyntax: EntityNode {
     }
 
     func annotationMetadata(with annotation: String) -> AnnotationMetadata? {
-        return leadingTrivia.annotationMetadata(with: annotation)
+        let trivias = [
+            leadingTrivia,
+            protocolKeyword.leadingTrivia,
+            modifiers.leadingTrivia,
+        ] + attributes.map(\.leadingTrivia)
+        return trivias.firstNonNil { $0.annotationMetadata(with: annotation) }
     }
 
     var hasBlankInit: Bool {
@@ -358,7 +364,12 @@ extension ClassDeclSyntax: EntityNode {
     }
 
     func annotationMetadata(with annotation: String) -> AnnotationMetadata? {
-        return leadingTrivia.annotationMetadata(with: annotation)
+        let trivias = [
+            leadingTrivia,
+            classKeyword.leadingTrivia,
+            modifiers.leadingTrivia,
+        ] + attributes.map(\.leadingTrivia)
+        return trivias.firstNonNil { $0.annotationMetadata(with: annotation) }
     }
 
     func subContainer(metadata: AnnotationMetadata?, declKind: NominalTypeDeclKind, path: String?, isProcessed: Bool) -> EntityNodeSubContainer {
@@ -863,9 +874,9 @@ extension Trivia {
     // See metadata(with:, in:) for more info on the annotation arguments.
     func annotationMetadata(with annotation: String) -> AnnotationMetadata? {
         guard !annotation.isEmpty else { return nil }
+
         var ret: AnnotationMetadata?
-        for i in 0..<count {
-            let trivia = self[i]
+        for trivia in self {
             switch trivia {
             case .docLineComment(let val):
                 ret = metadata(with: annotation, in: val)

Tests/TestActor/ActorTests.swift

@@ -13,4 +13,10 @@ final class ActorTests: MockoloTestCase {
         verify(srcContent: globalActorProtocol,
                dstContent: globalActorProtocolMock)
     }
+
+    func testAttributeAboveAnnotationComment() {
+        verify(srcContent: attributeAboveAnnotationComment,
+               dstContent: attributeAboveAnnotationCommentMock,
+               declType: .all)
+    }
 }

Tests/TestActor/FixtureGlobalActor.swift

@@ -41,3 +41,35 @@ class RootBuildableMock: RootBuildable {
     }
 }
 """
+
+
+let attributeAboveAnnotationComment = """
+@MainActor
+/// \(String.mockAnnotation)
+protocol P0 {
+}
+
+@MainActor
+/// \(String.mockAnnotation)
+@available(iOS 18.0, *) protocol P1 {
+}
+
+@MainActor
+/// \(String.mockAnnotation)
+public class C0 {
+}
+"""
+
+let attributeAboveAnnotationCommentMock = """
+class P0Mock: P0 {
+    init() { }
+}
+
+class P1Mock: P1 {
+    init() { }
+}
+
+public class C0Mock: C0 {
+    public init() { }
+}
+"""