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
|
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 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 Swift project authors
//
@testable import TestingMacros
import SwiftDiagnostics
import SwiftOperators
import SwiftParser
import SwiftSyntax
import SwiftSyntaxBuilder
import SwiftSyntaxMacros
import SwiftSyntaxMacroExpansion
fileprivate let allMacros: [String: any Macro.Type] = [
"expect": ExpectMacro.self,
"require": RequireMacro.self,
"requireAmbiguous": AmbiguousRequireMacro.self, // different name needed only for unit testing
"requireNonOptional": NonOptionalRequireMacro.self, // different name needed only for unit testing
"requireThrows": RequireThrowsMacro.self, // different name needed only for unit testing
"requireThrowsNever": RequireThrowsNeverMacro.self, // different name needed only for unit testing
"expectExitTest": ExitTestRequireMacro.self, // different name needed only for unit testing
"requireExitTest": ExitTestRequireMacro.self, // different name needed only for unit testing
"Suite": SuiteDeclarationMacro.self,
"Test": TestDeclarationMacro.self,
"Tag": TagMacro.self,
]
func parse(_ sourceCode: String, activeMacros activeMacroNames: [String] = [], removeWhitespace: Bool = false) throws -> (sourceCode: String, diagnostics: [Diagnostic]) {
let activeMacros: [String: any Macro.Type]
if activeMacroNames.isEmpty {
activeMacros = allMacros
} else {
activeMacros = allMacros.filter { activeMacroNames.contains($0.key) }
}
let operatorTable = OperatorTable.standardOperators
let originalSyntax = try operatorTable.foldAll(Parser.parse(source: sourceCode))
let context = BasicMacroExpansionContext(lexicalContext: [], expansionDiscriminator: "", sourceFiles: [:])
let syntax = try operatorTable.foldAll(
originalSyntax.expand(macros: activeMacros) { syntax in
BasicMacroExpansionContext(sharingWith: context, lexicalContext: syntax.allMacroLexicalContexts())
}
)
var sourceCode = String(describing: syntax.formatted().trimmed)
if removeWhitespace {
sourceCode = sourceCode.filter { !$0.isWhitespace }
}
return (sourceCode, context.diagnostics)
}
|
