@@ -21,57 +21,57 @@ import SwiftSyntax
21
21
/// An entity node is either a Protocol or Class syntax node
22
22
protocol EntityNode : SyntaxNodeWithModifiers {
23
23
var typeName : String { get }
24
- var inheritanceClause : TypeInheritanceClauseSyntax ? { get }
24
+ var inheritanceClause : InheritanceClauseSyntax ? { get }
25
25
}
26
26
27
27
extension EntityNode {
28
28
/// Checks whether the entity inherits from a certain type with `typeName`
29
29
func inherits( from typeName: String ) -> Bool {
30
30
31
- let inheritedTypeSyntax = inheritanceClause? . inheritedTypeCollection . first? . typeName
31
+ let inheritedTypeSyntax = inheritanceClause? . inheritedTypes . first? . type
32
32
// Usually, first token is the inherited type name. But sometimes it could also be the module prefix.
33
33
// In that case, we need to look for the actual type name by checking for `MemberTypeIdentifierSyntax`
34
- if inheritedTypeSyntax? . firstToken? . nextToken? . tokenKind != TokenKind . period {
35
- return inheritedTypeSyntax? . firstToken? . text == typeName
34
+ if inheritedTypeSyntax? . firstToken ( viewMode : . sourceAccurate ) ? . nextToken ( viewMode : . sourceAccurate ) ? . tokenKind != TokenKind . period {
35
+ return inheritedTypeSyntax? . firstToken ( viewMode : . sourceAccurate ) ? . text == typeName
36
36
} else {
37
- return inheritedTypeSyntax? . as ( MemberTypeIdentifierSyntax . self) ? . name. text == typeName
37
+ return inheritedTypeSyntax? . as ( MemberTypeSyntax . self) ? . name. text == typeName
38
38
}
39
39
}
40
40
41
41
var inheritanceHasGenericArgument : Bool {
42
- let inheritanceTypeToken = inheritanceClause? . inheritedTypeCollection . first? . typeName
43
- return inheritanceTypeToken? . as ( SimpleTypeIdentifierSyntax . self) ? . genericArgumentClause != nil ||
44
- inheritanceTypeToken? . as ( MemberTypeIdentifierSyntax . self) ? . genericArgumentClause != nil
42
+ let inheritanceTypeToken = inheritanceClause? . inheritedTypes . first? . type
43
+ return inheritanceTypeToken? . as ( IdentifierTypeSyntax . self) ? . genericArgumentClause != nil ||
44
+ inheritanceTypeToken? . as ( MemberTypeSyntax . self) ? . genericArgumentClause != nil
45
45
}
46
46
}
47
47
48
48
protocol SyntaxNodeWithModifiers {
49
- var modifiers : ModifierListSyntax ? { get }
49
+ var modifiers : DeclModifierListSyntax { get }
50
50
}
51
51
52
52
extension SyntaxNodeWithModifiers {
53
53
var isInternal : Bool {
54
- modifiers? . first? . name. text == nil || modifiers? . first? . name. text == " internal "
54
+ modifiers. first? . name. text == nil || modifiers. first? . name. text == " internal "
55
55
}
56
56
57
57
var isPublic : Bool {
58
- modifiers? . first? . name. text == " public "
58
+ modifiers. first? . name. text == " public "
59
59
}
60
60
61
61
var isPrivate : Bool {
62
- modifiers? . first? . name. text == " private "
62
+ modifiers. first? . name. text == " private "
63
63
}
64
64
65
65
var isFileprivate : Bool {
66
- modifiers? . first? . name. text == " fileprivate "
66
+ modifiers. first? . name. text == " fileprivate "
67
67
}
68
68
}
69
69
70
70
// MARK: - SwiftSyntax Protocol Extensions
71
71
72
- extension IdentifiedDeclSyntax {
72
+ extension NamedDeclSyntax {
73
73
var typeName : String {
74
- return identifier . description. trimmed
74
+ return name . description. trimmed
75
75
}
76
76
}
77
77
0 commit comments