Added macro

Author: khvorost-dish

Sources/SwiftDocCoverage/DeclProtocol.swift

@@ -48,12 +48,21 @@ public enum AccessLevel: Int {
 }
 
 protocol DeclProtocol: SyntaxProtocol {
-  var id: String { get }
+  
+  // `attributes`: ``AttributeListSyntax``
   var modifiers: DeclModifierListSyntax { get }
+  // `classKeyword`: `'class'`
+  //var name: TokenSyntax { get }
+  // `genericParameterClause`: ``GenericParameterClauseSyntax``?
+  // `inheritanceClause`: ``InheritanceClauseSyntax``?
+  // `genericWhereClause`: ``GenericWhereClauseSyntax``?
+  // `memberBlock`: ``MemberBlockSyntax``
+
+  var id: String { get }
 }
 
 extension DeclProtocol {
-  
+
   var comments: [Comment] {
     return leadingTrivia.compactMap {
       switch $0 {
@@ -136,6 +145,10 @@ extension ExtensionDeclSyntax: DeclProtocol {
   var id: String {
     extendedType.trimmed.description
   }
+  
+  var name: TokenSyntax {
+    TokenSyntax(.identifier(extendedType.trimmedDescription), presence: .present)
+  }
 }
 
 extension FunctionDeclSyntax: DeclProtocol {
@@ -188,3 +201,10 @@ extension PrecedenceGroupDeclSyntax: DeclProtocol {
     "\(precedencegroupKeyword.trimmed) \(name.trimmed)"
   }
 }
+
+extension MacroDeclSyntax: DeclProtocol {
+  var id: String {
+    let generic = genericParameterClause?.trimmed.description ?? ""
+    return "\(name.trimmed)\(generic)\(signature.trimmed)"
+  }
+}

Sources/SwiftDocCoverage/Declaration.swift

@@ -26,11 +26,11 @@ import SwiftSyntax
 
 fileprivate extension String {
 
-  static let regexNewLine = try! NSRegularExpression(pattern: "\\n\\s+", options: [])
+  static let regexNewLine = try! NSRegularExpression(pattern: #"\n\s*"#, options: [])
 
   func refine() -> String {
     let range = NSRange(startIndex..., in: self)
-    return Self.regexNewLine.stringByReplacingMatches(in: self, options: [], range: range, withTemplate: " ")
+    return Self.regexNewLine.stringByReplacingMatches(in: self, options: [], range: range, withTemplate: "")
   }
 }
 

Sources/SwiftDocCoverage/Source.swift

@@ -45,10 +45,7 @@ fileprivate class Visitor: SyntaxVisitor {
     }
 
     let startLocation = decl.startLocation(converter: converter, afterLeadingTrivia: true)
-    let declaration = Declaration(decl: decl,
-                                  context: context,
-                                  line: startLocation.line,
-                                  column: startLocation.column)
+    let declaration = Declaration(decl: decl, context: context, line: startLocation.line, column: startLocation.column)
     declarations.append(declaration)
   }
 
@@ -143,6 +140,11 @@ fileprivate class Visitor: SyntaxVisitor {
     return .skipChildren
   }
 
+  override func visit(_ node: MacroDeclSyntax) -> SyntaxVisitorContinueKind {
+    append(decl: node)
+    return .skipChildren
+  }
+  
   // MARK: -
 
   override func visitPost(_ node: ClassDeclSyntax) {