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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2019 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
//
//===----------------------------------------------------------------------===//
import SwiftSyntax
extension Trivia {
var hasAnyComments: Bool {
return contains {
switch $0 {
case .lineComment, .docLineComment, .blockComment, .docBlockComment:
return true
default:
return false
}
}
}
/// Returns whether the trivia contains at least 1 `lineComment`.
var hasLineComment: Bool {
return self.contains {
if case .lineComment = $0 { return true }
return false
}
}
/// Returns this set of trivia, without any leading spaces.
func withoutLeadingSpaces() -> Trivia {
return Trivia(pieces: self.pieces.drop(while: \.isSpaceOrTab))
}
func withoutTrailingSpaces() -> Trivia {
guard let lastNonSpaceIndex = self.pieces.lastIndex(where: \.isSpaceOrTab) else {
return self
}
return Trivia(pieces: self[..<lastNonSpaceIndex])
}
/// Returns this trivia, excluding the last newline and anything following it.
///
/// If there is no newline in the trivia, it is returned unmodified.
func withoutLastLine() -> Trivia {
var maybeLastNewlineOffset: Int? = nil
for (offset, piece) in self.enumerated() {
switch piece {
case .newlines, .carriageReturns, .carriageReturnLineFeeds:
maybeLastNewlineOffset = offset
default:
break
}
}
guard let lastNewlineOffset = maybeLastNewlineOffset else { return self }
return Trivia(pieces: self.dropLast(self.count - lastNewlineOffset))
}
/// Returns `true` if this trivia contains any newlines.
var containsNewlines: Bool {
return contains(
where: {
if case .newlines = $0 { return true }
return false
})
}
/// Returns `true` if this trivia contains any spaces.
var containsSpaces: Bool {
return contains(
where: {
if case .spaces = $0 { return true }
if case .tabs = $0 { return true }
return false
})
}
/// Returns `true` if this trivia contains any backslashes (used for multiline string newline
/// suppression).
var containsBackslashes: Bool {
return contains(
where: {
if case .backslashes = $0 { return true }
return false
})
}
}
|
