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
107
108
109
110
111
112
113
114
115
116
117
118
119
|
//===--- ASTScriptParser.cpp ----------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2019 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
//
//===----------------------------------------------------------------------===//
///
/// AST script parsing.
///
//===----------------------------------------------------------------------===//
#include "swift/Basic/QuotedString.h"
#include "swift/Frontend/Frontend.h"
#include "swift/Parse/Lexer.h"
#include "swift/Parse/Token.h"
#include "llvm/Support/MemoryBuffer.h"
#include "ASTScript.h"
#include "ASTScriptConfiguration.h"
using namespace swift;
using namespace scripting;
namespace {
class ASTScriptParser {
ASTScriptConfiguration &Config;
ASTContext &Context;
DiagnosticEngine &Diags;
std::unique_ptr<llvm::MemoryBuffer> Buffer;
std::optional<Lexer> TheLexer;
Token Tok;
public:
ASTScriptParser(ASTScriptConfiguration &config)
: Config(config), Context(config.Compiler.getASTContext()),
Diags(config.Compiler.getDiags()) {
auto &compiler = config.Compiler;
// Open the file and give it to the source manager.
auto bufferOrErr = llvm::MemoryBuffer::getFile(config.ScriptFile);
if (!bufferOrErr) {
llvm::errs() << "error opening file " << QuotedString(config.ScriptFile)
<< ": " << bufferOrErr.getError().message() << "\n";
return;
}
auto &sourceManager = compiler.getSourceMgr();
auto bufferID = sourceManager.addNewSourceBuffer(std::move(*bufferOrErr));
// Build and prime the lexer.
TheLexer.emplace(Context.LangOpts, sourceManager, bufferID,
&Diags, LexerMode::Swift);
TheLexer->lex(Tok);
}
std::unique_ptr<ASTScript> parseFile() {
if (!TheLexer) return nullptr;
return parseTopLevel();
}
private:
/***************************************************************************/
/*** Parsing primitives ****************************************************/
/***************************************************************************/
void consume(tok kind) {
assert(Tok.is(kind));
TheLexer->lex(Tok);
}
bool consumeIf(tok kind) {
if (Tok.isNot(kind)) return false;
consume(kind);
return true;
}
bool consumeIfExactly(StringRef literal) {
if (Tok.isNot(tok::identifier) || Tok.getText() != literal)
return false;
consume(tok::identifier);
return true;
}
bool consumeIfIdentifier(StringRef &ident) {
if (Tok.isNot(tok::identifier)) return false;
ident = Tok.getText();
consume(tok::identifier);
return true;
}
std::optional<StringRef> consumeIfIdentifier() {
StringRef ident;
return consumeIfIdentifier(ident) ? std::optional<StringRef>(ident)
: std::nullopt;
}
/***************************************************************************/
/*** ASTScript parsing *****************************************************/
/***************************************************************************/
std::unique_ptr<ASTScript> parseTopLevel();
};
} // end anonymous namespace
/// ast-script ::= ???
std::unique_ptr<ASTScript> ASTScriptParser::parseTopLevel() {
return std::unique_ptr<ASTScript>(new ASTScript(Config));
}
std::unique_ptr<ASTScript> ASTScript::parse(ASTScriptConfiguration &config) {
return ASTScriptParser(config).parseFile();
}
|
