1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2023 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#if swift(>=6)
public import SwiftSyntax
#else
import SwiftSyntax
#endif
extension Trivia {
func indented(indentation: Trivia) -> Trivia {
let mappedPieces = self.flatMap { (piece) -> [TriviaPiece] in
if piece.isNewline {
return [piece] + indentation.pieces
} else {
return [piece]
}
}
return Trivia(pieces: mappedPieces)
}
}
/// Adds a given amount of indentation after every newline in a syntax tree.
public class Indenter: SyntaxRewriter {
let indentation: Trivia
init(indentation: Trivia) {
self.indentation = indentation
super.init(viewMode: .sourceAccurate)
}
/// Add `indentation` after all newlines in the syntax tree.
public static func indent<SyntaxType: SyntaxProtocol>(
_ node: SyntaxType,
indentation: Trivia
) -> SyntaxType {
return Indenter(indentation: indentation).rewrite(node).as(SyntaxType.self)!
}
public override func visit(_ token: TokenSyntax) -> TokenSyntax {
return TokenSyntax(
token.tokenKind,
leadingTrivia: token.leadingTrivia.indented(indentation: indentation),
trailingTrivia: token.trailingTrivia.indented(indentation: indentation),
presence: token.presence
)
}
}
|
