除非，我們今天將 [a] 改寫成 [a = a]

除了前面介紹的 visitor 以外

我們可以透過實作 SyntaxRewriter

去更改特定語法

ClosureCaptureItemSyntax

我們可以透過 AST Explorer 得知

[weak a = a] 是 ClosureCaptureItemSyntax 語法

以及得知其語法架構

/// ClosureCaptureItemSyntax
/// specifier name assignToken expression
/// weak      a    =           a          [weak a = a]
/// weak                       a          [weak     a]

CaptureListRewriter

於是我們就可以開始偷偷改 code

• 先檢查確實沒有 = a
• 在檢查 a 是 ID
  • node.expression.as(IdentifierExprSyntax.self)
• 最後用 withXXX 將特定語法替換掉
  • 新版會是 with(KeyPath)

import Foundation
import SwiftSyntax

final public class CaptureListRewriter: SyntaxRewriter {
    public override func visit(_ node: ClosureCaptureItemSyntax) -> ClosureCaptureItemSyntax {
        guard node.name == nil, node.assignToken == nil else {
            return node
        }

        guard let name = node.expression.as(IdentifierExprSyntax.self)?.identifier else {
            return node
        }

        return node
            .withName(name.withoutTrivia())
            .withAssignToken(TokenSyntax.equalToken(leadingTrivia: .space, trailingTrivia: .space))
    }
}

最終

let newSource = rewriter.visit(source)