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
|
//===----------------------------------------------------------------------===//
//
// 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 SyntaxSupport
/// Extension to the ``Node`` type to provide functionality specific to
/// SwiftSyntaxBuilder.
extension Node {
/// The node's syntax kind as ``SyntaxBuildableType``.
public var type: SyntaxBuildableType {
SyntaxBuildableType(kind: .node(kind: kind))
}
/// The node's syntax kind as ``SyntaxBuildableType``.
public var baseType: SyntaxBuildableType {
if base == .syntaxCollection {
return SyntaxBuildableType(kind: .node(kind: .syntax))
} else {
return SyntaxBuildableType(kind: .node(kind: base))
}
}
public static func from(type: SyntaxBuildableType) -> Node {
guard case .node(kind: let kind) = type.kind, let node = SYNTAX_NODE_MAP[kind] else {
fatalError("Base name \(type) does not have a syntax node")
}
return node
}
}
extension LayoutNode {
/// Assuming this node has a single child without a default value, that child.
public var singleNonDefaultedChild: Child {
let nonDefaultedParams = children.filter { $0.defaultInitialization == nil }
precondition(nonDefaultedParams.count == 1)
return nonDefaultedParams[0]
}
}
extension CollectionNode {
/// Assuming this node is a syntax collection, the type of its elements.
public var collectionElementType: SyntaxBuildableType {
if elementChoices.count > 1 {
return SyntaxBuildableType(kind: .node(kind: .syntax))
} else {
return SyntaxBuildableType(kind: .node(kind: elementChoices.first!))
}
}
}
|
