mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 22:02:23 +01:00
Jon Meow
6fa42a7f3c
This replaces GetAstMatcher with AddMatcher because cxxForRangeStmt is a StatementMatcher. addMatcher has multiple definitions (https://clang.llvm.org/doxygen/classclang_1_1ast__matchers_1_1MatchFinder.html) and so this approach allows using the right addMatcher without writing per-call overloads. To handle the `var`, I'm considering something like moving VarDecl logic into a VarMatcherBase so that I can just use CXXForRangeStmt's getLoopVariable. The problem is a for-range statement has multiple VarDecls, and getLoopVariable may be the easiest way to identify the real one.
50 lines
1.8 KiB
C++
50 lines
1.8 KiB
C++
// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
|
|
// Exceptions. See /LICENSE for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
#include "migrate_cpp/cpp_refactoring/fn_inserter.h"
|
|
|
|
#include "clang/ASTMatchers/ASTMatchers.h"
|
|
|
|
namespace cam = ::clang::ast_matchers;
|
|
|
|
namespace Carbon {
|
|
|
|
static constexpr char Label[] = "FnInserter";
|
|
|
|
void FnInserter::Run() {
|
|
const auto& decl = GetNodeAsOrDie<clang::FunctionDecl>(Label);
|
|
|
|
// For names like "Class::Method", replace up to "Class" not "Method".
|
|
clang::NestedNameSpecifierLoc qual_loc = decl.getQualifierLoc();
|
|
clang::SourceLocation name_begin_loc =
|
|
qual_loc.hasQualifier() ? qual_loc.getBeginLoc() : decl.getLocation();
|
|
auto range =
|
|
clang::CharSourceRange::getCharRange(decl.getBeginLoc(), name_begin_loc);
|
|
|
|
// In order to handle keywords like "virtual" in "virtual auto Foo() -> ...",
|
|
// scan the replaced text and only drop auto/void entries.
|
|
llvm::SmallVector<llvm::StringRef> split;
|
|
GetSourceText(range).split(split, ' ', /*MaxSplit=*/-1, /*KeepEmpty=*/false);
|
|
std::string new_text = "fn ";
|
|
for (llvm::StringRef t : split) {
|
|
if (t != "auto" && t != "void") {
|
|
new_text += t.str() + " ";
|
|
}
|
|
}
|
|
AddReplacement(range, new_text);
|
|
}
|
|
|
|
void FnInserterFactory::AddMatcher(cam::MatchFinder* finder,
|
|
cam::MatchFinder::MatchCallback* callback) {
|
|
finder->addMatcher(
|
|
cam::functionDecl(cam::anyOf(cam::hasTrailingReturn(),
|
|
cam::returns(cam::asString("void"))),
|
|
cam::unless(cam::anyOf(cam::cxxConstructorDecl(),
|
|
cam::cxxDestructorDecl())))
|
|
.bind(Label),
|
|
callback);
|
|
}
|
|
|
|
} // namespace Carbon
|