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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
import SwiftSyntax
import SwiftSyntaxBuilder
import SyntaxSupport
import Utils
let syntaxEnumFile = SourceFileSyntax(leadingTrivia: copyrightHeader) {
try! EnumDeclSyntax(
"""
/// Enum to exhaustively switch over all different syntax nodes.
public enum SyntaxEnum: Sendable
"""
) {
DeclSyntax("case token(TokenSyntax)")
for node in NON_BASE_SYNTAX_NODES {
DeclSyntax(
"""
\(node.apiAttributes())\
case \(node.varOrCaseName)(\(node.kind.syntaxType))
"""
)
}
}
try! ExtensionDeclSyntax(
"""
extension Syntax
"""
) {
try FunctionDeclSyntax(
"""
/// Get an enum that can be used to exhaustively switch over all syntax nodes.
public func `as`(_: SyntaxEnum.Type) -> SyntaxEnum
"""
) {
try SwitchExprSyntax("switch raw.kind") {
SwitchCaseSyntax("case .token:") {
StmtSyntax("return .token(TokenSyntax(self)!)")
}
for node in NON_BASE_SYNTAX_NODES {
SwitchCaseSyntax("case .\(node.varOrCaseName):") {
StmtSyntax("return .\(node.varOrCaseName)(\(node.kind.syntaxType)(self)!)")
}
}
}
}
}
for base in SYNTAX_NODES where base.kind.isBase {
let baseKind = base.kind
let baseName = baseKind.rawValue.withFirstCharacterUppercased;
let enumType: TypeSyntax = "\(raw: baseName)SyntaxEnum"
try! EnumDeclSyntax(
"""
/// Enum to exhaustively switch over all different \(raw: baseName) syntax nodes.
public enum \(enumType)
"""
) {
for node in NON_BASE_SYNTAX_NODES where node.base == baseKind {
DeclSyntax(
"""
\(node.apiAttributes())\
case \(node.varOrCaseName)(\(node.kind.syntaxType))
"""
)
}
}
try! ExtensionDeclSyntax(
"""
extension \(baseKind.syntaxType)
"""
) {
try FunctionDeclSyntax(
"""
/// Get an enum that can be used to exhaustively switch over all \(raw: baseName) syntax nodes.
public func `as`(_: \(enumType).Type) -> \(enumType)
"""
) {
try SwitchExprSyntax("switch raw.kind") {
for node in NON_BASE_SYNTAX_NODES where node.base == baseKind {
SwitchCaseSyntax("case .\(node.varOrCaseName):") {
StmtSyntax("return .\(node.varOrCaseName)(\(node.kind.syntaxType)(self)!)")
}
}
SwitchCaseSyntax("default:") {
ExprSyntax(#"preconditionFailure("unknown \#(raw: baseName) syntax kind")"#)
}
}
}
}
}
}
|
