Skip to content

Commit 290d309

Browse files
authored
Merge pull request #70258 from rintaro/astgen-internal
[ASTGen] Small cleanups
2 parents 9ce6e12 + 813f1d8 commit 290d309

File tree

6 files changed

+51
-59
lines changed

6 files changed

+51
-59
lines changed

lib/ASTGen/Sources/ASTGen/ASTGen.swift

Lines changed: 6 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,6 @@ import ParseBridging
1818

1919
import struct SwiftDiagnostics.Diagnostic
2020

21-
extension UnsafePointer {
22-
public var raw: UnsafeMutableRawPointer {
23-
UnsafeMutableRawPointer(mutating: self)
24-
}
25-
}
26-
2721
enum ASTNode {
2822
case decl(BridgedDecl)
2923
case stmt(BridgedStmt)
@@ -102,7 +96,7 @@ struct ASTGenVisitor {
10296
self.legacyParse = legacyParser
10397
}
10498

105-
public func generate(sourceFile node: SourceFileSyntax) -> [BridgedDecl] {
99+
func generate(sourceFile node: SourceFileSyntax) -> [BridgedDecl] {
106100
var out = [BridgedDecl]()
107101

108102
for element in node.statements {
@@ -129,8 +123,6 @@ struct ASTGenVisitor {
129123
endLoc: loc
130124
)
131125
out.append(topLevelDecl.asDecl)
132-
default:
133-
fatalError("Top level nodes must be decls, stmts, or exprs.")
134126
}
135127
}
136128

@@ -241,15 +233,15 @@ extension ASTGenVisitor {
241233
// Misc visits.
242234
// TODO: Some of these are called within a single file/method; we may want to move them to the respective files.
243235
extension ASTGenVisitor {
244-
public func generate(memberBlockItem node: MemberBlockItemSyntax) -> BridgedDecl {
236+
func generate(memberBlockItem node: MemberBlockItemSyntax) -> BridgedDecl {
245237
generate(decl: node.decl)
246238
}
247239

248-
public func generate(initializerClause node: InitializerClauseSyntax) -> BridgedExpr {
240+
func generate(initializerClause node: InitializerClauseSyntax) -> BridgedExpr {
249241
generate(expr: node.value)
250242
}
251243

252-
public func generate(conditionElement node: ConditionElementSyntax) -> ASTNode {
244+
func generate(conditionElement node: ConditionElementSyntax) -> ASTNode {
253245
// FIXME: returning ASTNode is wrong, non-expression conditions are not ASTNode.
254246
switch node.condition {
255247
case .availability(_):
@@ -264,7 +256,7 @@ extension ASTGenVisitor {
264256
fatalError("unimplemented")
265257
}
266258

267-
public func generate(codeBlockItem node: CodeBlockItemSyntax) -> ASTNode {
259+
func generate(codeBlockItem node: CodeBlockItemSyntax) -> ASTNode {
268260
switch node.item {
269261
case .decl(let node):
270262
return .decl(self.generate(decl: node))
@@ -275,7 +267,7 @@ extension ASTGenVisitor {
275267
}
276268
}
277269

278-
public func generate(arrayElement node: ArrayElementSyntax) -> BridgedExpr {
270+
func generate(arrayElement node: ArrayElementSyntax) -> BridgedExpr {
279271
generate(expr: node.expression)
280272
}
281273

lib/ASTGen/Sources/ASTGen/Decls.swift

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ extension ASTGenVisitor {
7272
return self.generateWithLegacy(node)
7373
}
7474

75-
public func generate(typeAliasDecl node: TypeAliasDeclSyntax) -> BridgedTypeAliasDecl {
75+
func generate(typeAliasDecl node: TypeAliasDeclSyntax) -> BridgedTypeAliasDecl {
7676
let (name, nameLoc) = self.generateIdentifierAndSourceLoc(node.name)
7777

7878
return .createParsed(
@@ -88,7 +88,7 @@ extension ASTGenVisitor {
8888
)
8989
}
9090

91-
public func generate(enumDecl node: EnumDeclSyntax) -> BridgedNominalTypeDecl {
91+
func generate(enumDecl node: EnumDeclSyntax) -> BridgedNominalTypeDecl {
9292
let (name, nameLoc) = self.generateIdentifierAndSourceLoc(node.name)
9393

9494
let decl = BridgedEnumDecl.createParsed(
@@ -113,7 +113,7 @@ extension ASTGenVisitor {
113113
return decl
114114
}
115115

116-
public func generate(structDecl node: StructDeclSyntax) -> BridgedNominalTypeDecl {
116+
func generate(structDecl node: StructDeclSyntax) -> BridgedNominalTypeDecl {
117117
let (name, nameLoc) = self.generateIdentifierAndSourceLoc(node.name)
118118

119119
let decl = BridgedStructDecl.createParsed(
@@ -138,7 +138,7 @@ extension ASTGenVisitor {
138138
return decl
139139
}
140140

141-
public func generate(classDecl node: ClassDeclSyntax) -> BridgedNominalTypeDecl {
141+
func generate(classDecl node: ClassDeclSyntax) -> BridgedNominalTypeDecl {
142142
let (name, nameLoc) = self.generateIdentifierAndSourceLoc(node.name)
143143

144144
let decl = BridgedClassDecl.createParsed(
@@ -164,7 +164,7 @@ extension ASTGenVisitor {
164164
return decl
165165
}
166166

167-
public func generate(actorDecl node: ActorDeclSyntax) -> BridgedNominalTypeDecl {
167+
func generate(actorDecl node: ActorDeclSyntax) -> BridgedNominalTypeDecl {
168168
let (name, nameLoc) = self.generateIdentifierAndSourceLoc(node.name)
169169

170170
let decl = BridgedClassDecl.createParsed(
@@ -288,7 +288,7 @@ extension ASTGenVisitor {
288288
// MARK: - AbstractStorageDecl
289289

290290
extension ASTGenVisitor {
291-
public func generate(variableDecl node: VariableDeclSyntax) -> BridgedPatternBindingDecl {
291+
func generate(variableDecl node: VariableDeclSyntax) -> BridgedPatternBindingDecl {
292292
let pattern = generate(pattern: node.bindings.first!.pattern)
293293
let initializer = generate(initializerClause: node.bindings.first!.initializer!)
294294

@@ -310,7 +310,7 @@ extension ASTGenVisitor {
310310
// MARK: - AbstractFunctionDecl
311311

312312
extension ASTGenVisitor {
313-
public func generate(functionDecl node: FunctionDeclSyntax) -> BridgedFuncDecl {
313+
func generate(functionDecl node: FunctionDeclSyntax) -> BridgedFuncDecl {
314314
// FIXME: Compute this location
315315
let staticLoc: BridgedSourceLoc = nil
316316

lib/ASTGen/Sources/ASTGen/Exprs.swift

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,7 @@ extension ASTGenVisitor {
199199
preconditionFailure("isExprMigrated() mismatch")
200200
}
201201

202-
public func generate(arrowExpr node: ArrowExprSyntax) -> BridgedArrowExpr {
202+
func generate(arrowExpr node: ArrowExprSyntax) -> BridgedArrowExpr {
203203
let asyncLoc: BridgedSourceLoc
204204
let throwsLoc: BridgedSourceLoc
205205
let thrownTypeExpr: BridgedNullableExpr
@@ -231,15 +231,15 @@ extension ASTGenVisitor {
231231
)
232232
}
233233

234-
public func generate(assignmentExpr node: AssignmentExprSyntax) -> BridgedAssignExpr {
234+
func generate(assignmentExpr node: AssignmentExprSyntax) -> BridgedAssignExpr {
235235
return .createParsed(self.ctx, equalsLoc: self.generateSourceLoc(node.equal))
236236
}
237237

238-
public func generate(binaryOperatorExpr node: BinaryOperatorExprSyntax) -> BridgedUnresolvedDeclRefExpr {
238+
func generate(binaryOperatorExpr node: BinaryOperatorExprSyntax) -> BridgedUnresolvedDeclRefExpr {
239239
return createOperatorRefExpr(token: node.operator, kind: .binaryOperator)
240240
}
241241

242-
public func generate(closureExpr node: ClosureExprSyntax) -> BridgedClosureExpr {
242+
func generate(closureExpr node: ClosureExprSyntax) -> BridgedClosureExpr {
243243
let body = BridgedBraceStmt.createParsed(
244244
self.ctx,
245245
lBraceLoc: self.generateSourceLoc(node.leftBrace),
@@ -251,7 +251,7 @@ extension ASTGenVisitor {
251251
return .createParsed(self.ctx, declContext: self.declContext, body: body)
252252
}
253253

254-
public func generate(functionCallExpr node: FunctionCallExprSyntax) -> BridgedCallExpr {
254+
func generate(functionCallExpr node: FunctionCallExprSyntax) -> BridgedCallExpr {
255255
if !node.arguments.isEmpty || node.trailingClosure == nil {
256256
if node.leftParen == nil {
257257
self.diagnose(
@@ -333,7 +333,7 @@ extension ASTGenVisitor {
333333
}
334334
}
335335

336-
public func generate(declReferenceExpr node: DeclReferenceExprSyntax) -> BridgedUnresolvedDeclRefExpr {
336+
func generate(declReferenceExpr node: DeclReferenceExprSyntax) -> BridgedUnresolvedDeclRefExpr {
337337
let nameAndLoc = createDeclNameRef(declReferenceExpr: node)
338338
return .createParsed(
339339
self.ctx,
@@ -343,11 +343,11 @@ extension ASTGenVisitor {
343343
)
344344
}
345345

346-
public func generate(discardAssignmentExpr node: DiscardAssignmentExprSyntax) -> BridgedDiscardAssignmentExpr {
346+
func generate(discardAssignmentExpr node: DiscardAssignmentExprSyntax) -> BridgedDiscardAssignmentExpr {
347347
return .createParsed(self.ctx, loc: self.generateSourceLoc(node.wildcard))
348348
}
349349

350-
public func generate(memberAccessExpr node: MemberAccessExprSyntax) -> BridgedExpr {
350+
func generate(memberAccessExpr node: MemberAccessExprSyntax) -> BridgedExpr {
351351
let nameAndLoc = createDeclNameRef(declReferenceExpr: node.declName)
352352

353353
if let base = node.base {
@@ -368,7 +368,7 @@ extension ASTGenVisitor {
368368
}
369369
}
370370

371-
public func generate(ifExpr node: IfExprSyntax) -> BridgedSingleValueStmtExpr {
371+
func generate(ifExpr node: IfExprSyntax) -> BridgedSingleValueStmtExpr {
372372
let stmt = makeIfStmt(node).asStmt
373373

374374
// Wrap in a SingleValueStmtExpr to embed as an expression.
@@ -380,7 +380,7 @@ extension ASTGenVisitor {
380380
)
381381
}
382382

383-
public func generate(postfixOperatorExpr node: PostfixOperatorExprSyntax) -> BridgedPostfixUnaryExpr {
383+
func generate(postfixOperatorExpr node: PostfixOperatorExprSyntax) -> BridgedPostfixUnaryExpr {
384384
return .createParsed(
385385
self.ctx,
386386
operator: self.createOperatorRefExpr(
@@ -391,7 +391,7 @@ extension ASTGenVisitor {
391391
)
392392
}
393393

394-
public func generate(prefixOperatorExpr node: PrefixOperatorExprSyntax) -> BridgedPrefixUnaryExpr {
394+
func generate(prefixOperatorExpr node: PrefixOperatorExprSyntax) -> BridgedPrefixUnaryExpr {
395395
return .createParsed(
396396
self.ctx,
397397
operator: self.createOperatorRefExpr(
@@ -402,7 +402,7 @@ extension ASTGenVisitor {
402402
)
403403
}
404404

405-
public func generate(sequenceExpr node: SequenceExprSyntax) -> BridgedExpr {
405+
func generate(sequenceExpr node: SequenceExprSyntax) -> BridgedExpr {
406406
assert(
407407
!node.elements.count.isMultiple(of: 2),
408408
"SequenceExpr must have odd number of elements"
@@ -458,18 +458,18 @@ extension ASTGenVisitor {
458458
).asExpr
459459
}
460460

461-
public func generate(tupleExpr node: TupleExprSyntax) -> BridgedTupleExpr {
461+
func generate(tupleExpr node: TupleExprSyntax) -> BridgedTupleExpr {
462462
return self.generate(labeledExprList: node.elements, leftParen: node.leftParen, rightParen: node.rightParen)
463463
}
464464

465-
public func generate(typeExpr node: TypeExprSyntax) -> BridgedTypeExpr {
465+
func generate(typeExpr node: TypeExprSyntax) -> BridgedTypeExpr {
466466
return .createParsed(
467467
self.ctx,
468468
type: self.generate(type: node.type)
469469
)
470470
}
471471

472-
public func generate(unresolvedAsExpr node: UnresolvedAsExprSyntax, typeExpr typeNode: TypeExprSyntax) -> BridgedExpr {
472+
func generate(unresolvedAsExpr node: UnresolvedAsExprSyntax, typeExpr typeNode: TypeExprSyntax) -> BridgedExpr {
473473
let type = self.generate(type: typeNode.type)
474474
let asLoc = self.generateSourceLoc(node.asKeyword)
475475

@@ -499,15 +499,15 @@ extension ASTGenVisitor {
499499
}
500500
}
501501

502-
public func generate(unresolvedIsExpr node: UnresolvedIsExprSyntax, typeExpr typeNode: TypeExprSyntax) -> BridgedIsExpr {
502+
func generate(unresolvedIsExpr node: UnresolvedIsExprSyntax, typeExpr typeNode: TypeExprSyntax) -> BridgedIsExpr {
503503
return .createParsed(
504504
self.ctx,
505505
isLoc: self.generateSourceLoc(node.isKeyword),
506506
type: self.generate(type: typeNode.type)
507507
)
508508
}
509509

510-
public func generate(unresolvedTernaryExpr node: UnresolvedTernaryExprSyntax) -> BridgedTernaryExpr {
510+
func generate(unresolvedTernaryExpr node: UnresolvedTernaryExprSyntax) -> BridgedTernaryExpr {
511511
return .createParsed(
512512
self.ctx,
513513
questionLoc: self.generateSourceLoc(node.questionMark),

lib/ASTGen/Sources/ASTGen/Literals.swift

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ import ASTBridging
1414
import SwiftSyntax
1515

1616
extension ASTGenVisitor {
17-
public func generate(stringLiteralExpr node: StringLiteralExprSyntax) -> BridgedStringLiteralExpr {
17+
func generate(stringLiteralExpr node: StringLiteralExprSyntax) -> BridgedStringLiteralExpr {
1818
let openDelimiterOrQuoteLoc = self.generateSourceLoc(node.openingPounds ?? node.openingQuote)
1919

2020
// FIXME: Handle interpolated strings.
@@ -25,7 +25,7 @@ extension ASTGenVisitor {
2525
}
2626
}
2727

28-
public func generate(integerLiteralExpr node: IntegerLiteralExprSyntax) -> BridgedIntegerLiteralExpr {
28+
func generate(integerLiteralExpr node: IntegerLiteralExprSyntax) -> BridgedIntegerLiteralExpr {
2929
// FIXME: Avoid 'String' instantiation
3030
// FIXME: Strip '_'.
3131
var segment = node.literal.text
@@ -38,7 +38,7 @@ extension ASTGenVisitor {
3838
}
3939
}
4040

41-
public func generate(booleanLiteralExpr node: BooleanLiteralExprSyntax) -> BridgedBooleanLiteralExpr {
41+
func generate(booleanLiteralExpr node: BooleanLiteralExprSyntax) -> BridgedBooleanLiteralExpr {
4242
let value = node.literal.tokenKind == .keyword(.true)
4343
return .createParsed(
4444
ctx,
@@ -47,7 +47,7 @@ extension ASTGenVisitor {
4747
)
4848
}
4949

50-
public func generate(arrayExpr node: ArrayExprSyntax) -> BridgedArrayExpr {
50+
func generate(arrayExpr node: ArrayExprSyntax) -> BridgedArrayExpr {
5151
let expressions = node.elements.lazy.map(self.generate)
5252

5353
let commaLocations = node.elements.compactMap(in: self) {

lib/ASTGen/Sources/ASTGen/Stmts.swift

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ extension ASTGenVisitor {
5454
return self.generateWithLegacy(node)
5555
}
5656

57-
public func generate(codeBlock node: CodeBlockSyntax) -> BridgedBraceStmt {
57+
func generate(codeBlock node: CodeBlockSyntax) -> BridgedBraceStmt {
5858
BridgedBraceStmt.createParsed(
5959
self.ctx,
6060
lBraceLoc: self.generateSourceLoc(node.leftBrace),
@@ -86,7 +86,7 @@ extension ASTGenVisitor {
8686
)
8787
}
8888

89-
public func generate(expressionStmt node: ExpressionStmtSyntax) -> BridgedStmt {
89+
func generate(expressionStmt node: ExpressionStmtSyntax) -> BridgedStmt {
9090
switch Syntax(node.expression).as(SyntaxEnum.self) {
9191
case .ifExpr(let e):
9292
return makeIfStmt(e).asStmt
@@ -95,7 +95,7 @@ extension ASTGenVisitor {
9595
}
9696
}
9797

98-
public func generate(returnStmt node: ReturnStmtSyntax) -> BridgedReturnStmt {
98+
func generate(returnStmt node: ReturnStmtSyntax) -> BridgedReturnStmt {
9999
.createParsed(
100100
self.ctx,
101101
returnKeywordLoc: self.generateSourceLoc(node.returnKeyword),

0 commit comments

Comments
 (0)