Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -540,7 +540,7 @@ bool SwiftASTManipulator::RewriteResult() {
// First step, walk the function body converting returns to assignments to
// temp variables + return:

for (swift::Decl *decl : m_source_file.Decls) {
for (swift::Decl *decl : m_source_file.getTopLevelDecls()) {
if (auto top_level_code_decl =
llvm::dyn_cast<swift::TopLevelCodeDecl>(decl)) {
return_finder.SetDeclContext(top_level_code_decl);
Expand All @@ -551,8 +551,8 @@ bool SwiftASTManipulator::RewriteResult() {
// Second step, fetch the last expression, and if it is non-null, set it to
// a temp result as well:

if (!m_source_file.Decls.empty()) {
swift::Decl *last_decl = *(m_source_file.Decls.end() - 1);
if (!m_source_file.getTopLevelDecls().empty()) {
swift::Decl *last_decl = *(m_source_file.getTopLevelDecls().end() - 1);

if (auto last_top_level_code_decl =
llvm::dyn_cast<swift::TopLevelCodeDecl>(last_decl)) {
Expand Down Expand Up @@ -754,7 +754,7 @@ void SwiftASTManipulator::FindVariableDeclarations(
};

if (m_repl) {
for (swift::Decl *decl : m_source_file.Decls) {
for (swift::Decl *decl : m_source_file.getTopLevelDecls()) {
if (swift::VarDecl *var_decl = llvm::dyn_cast<swift::VarDecl>(decl)) {
if (!var_decl->getName().str().startswith("$")) {
register_one_var(var_decl);
Expand Down Expand Up @@ -797,7 +797,7 @@ void SwiftASTManipulator::FindNonVariableDeclarations(
if (!m_repl)
return; // we don't do this for non-REPL expressions... yet

for (swift::Decl *decl : m_source_file.Decls) {
for (swift::Decl *decl : m_source_file.getTopLevelDecls()) {
if (swift::ValueDecl *value_decl = llvm::dyn_cast<swift::ValueDecl>(decl)) {
if (!llvm::isa<swift::VarDecl>(value_decl) && value_decl->hasName()) {
non_variables.push_back(value_decl);
Expand Down Expand Up @@ -1125,9 +1125,8 @@ bool SwiftASTManipulator::AddExternalVariables(

redirected_var_decl->setImplicit(true);

m_source_file.Decls.insert(m_source_file.Decls.begin(), top_level_code);
m_source_file.Decls.insert(m_source_file.Decls.begin(),
redirected_var_decl);
m_source_file.addTopLevelDecl(top_level_code);
m_source_file.addTopLevelDecl(redirected_var_decl);

variable.m_decl = redirected_var_decl;

Expand Down Expand Up @@ -1402,7 +1401,7 @@ swift::ValueDecl *SwiftASTManipulator::MakeGlobalTypealias(
if (make_private) {
type_alias_decl->overwriteAccess(swift::AccessLevel::Private);
}
m_source_file.Decls.push_back(type_alias_decl);
m_source_file.addTopLevelDecl(type_alias_decl);
}

return type_alias_decl;
Expand Down