Support CaptureListRewriter Origin syntax location

Author: yume190

DesignDetail.md

@@ -11,12 +11,12 @@
 ```swift
 func xxx() {
     let xxx = XXX()
-//      ^ <-------------¬
-    // closure          |
+//      ^ <─────────────┐
+    // closure          │
     { [weak self, yyy = xxx] _ in
 //          ^     ^
-//          |     └---- yyy
-//          └---- self
+//          │     └──── yyy
+//          └──── self
     }
 }
 ```
@@ -27,20 +27,20 @@ func xxx() {
 class XXX {
     func xxx() {
 //       ^
-//       └----¬
+//       └────┐
         print(self)
     }
 
     lazy var yyy: Int = {
 //           ^
-//           └¬
+//           └┐
         print(self)
         return 1
     }()
 
     var zzz: XXX {
 //               ^
-//            ⌜--⌟
+//            ┌──┘
         print(self)
     }
 }

Sources/LeakDetectKit/Capture/CaptureListRewriter.swift

@@ -10,8 +10,14 @@ import SwiftSyntax
 
 final public class CaptureListRewriter: SyntaxRewriter {
     struct Info {
-        let originSyntax: SyntaxProtocol
-        let expresion: Int
+        let originSyntax: TokenSyntax
+        let expression: Int
+    }
+  
+    subscript(offset: Int) -> TokenSyntax? {
+        infos.first { info in
+            info.expression == offset
+        }?.originSyntax
     }
 
     private(set) var infos: [Info] = []
@@ -35,7 +41,7 @@ final public class CaptureListRewriter: SyntaxRewriter {
 
         infos.append(.init(
             originSyntax: name,
-            expresion: origin + offset + newOffset))
+            expression: origin + offset + newOffset))
 
         offset += newOffset
 

Sources/LeakDetectKit/Capture/DeclsVisitor.swift

@@ -15,23 +15,23 @@ import SKClient
 /// Source Code use ``customWalk`` to walk `static func` and `sub DeclsVisitor`
 public final class DeclsVisitor: SyntaxVisitor {
     private lazy var _subVisitors: [DeclsVisitor] = []
-    public let client: SKClient
     private let leak: LeakVisitor
-    public init(client: SKClient) {
-        self.client = client
+
+    
+    public init(client: SKClient, _ rewriter: CaptureListRewriter) {
         self.leak = LeakVisitor(
             context: .global(.file),
             client: client,
-            parentVisitor: nil)
+            rewriter: rewriter)
+        
         super.init(viewMode: .sourceAccurate)
     }
 
-    init(client: SKClient, _ global: LeakVisitor.Context.Global) {
-        self.client = client
+    init(client: SKClient, _ rewriter: CaptureListRewriter, _ global: LeakVisitor.Context.Global) {
         self.leak = LeakVisitor(
             context: .global(global),
             client: client,
-            parentVisitor: nil)
+            rewriter: rewriter)
         super.init(viewMode: .sourceAccurate)
     }
 
@@ -81,7 +81,7 @@ extension DeclsVisitor {
     }
 
     private final func append<Syntax: SyntaxProtocol>(_ syntax: Syntax, _ global: LeakVisitor.Context.Global) {
-        let visitor = DeclsVisitor(client: client, global)
+        let visitor = DeclsVisitor(client: leak.client, leak.rewriter, global)
         self._subVisitors.append(visitor)
         visitor.customWalk(syntax)
     }