1 //===--- Transformer.cpp - Transformer library implementation ---*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "clang/Tooling/Transformer/Transformer.h"
10 #include "clang/ASTMatchers/ASTMatchFinder.h"
11 #include "clang/ASTMatchers/ASTMatchersInternal.h"
12 #include "clang/Basic/SourceLocation.h"
13 #include "clang/Tooling/Refactoring/AtomicChange.h"
14 #include "llvm/Support/Error.h"
15 #include <map>
16 #include <utility>
17 #include <vector>
18
19 using namespace clang;
20 using namespace tooling;
21
22 using ast_matchers::MatchFinder;
23
registerMatchers(MatchFinder * MatchFinder)24 void Transformer::registerMatchers(MatchFinder *MatchFinder) {
25 for (auto &Matcher : transformer::detail::buildMatchers(Rule))
26 MatchFinder->addDynamicMatcher(Matcher, this);
27 }
28
run(const MatchFinder::MatchResult & Result)29 void Transformer::run(const MatchFinder::MatchResult &Result) {
30 if (Result.Context->getDiagnostics().hasErrorOccurred())
31 return;
32
33 transformer::RewriteRule::Case Case =
34 transformer::detail::findSelectedCase(Result, Rule);
35 auto Transformations = Case.Edits(Result);
36 if (!Transformations) {
37 Consumer(Transformations.takeError());
38 return;
39 }
40
41 if (Transformations->empty())
42 return;
43
44 // Group the transformations, by file, into AtomicChanges, each anchored by
45 // the location of the first change in that file.
46 std::map<FileID, AtomicChange> ChangesByFileID;
47 for (const auto &T : *Transformations) {
48 auto ID = Result.SourceManager->getFileID(T.Range.getBegin());
49 auto Iter = ChangesByFileID
50 .emplace(ID, AtomicChange(*Result.SourceManager,
51 T.Range.getBegin(), T.Metadata))
52 .first;
53 auto &AC = Iter->second;
54 switch (T.Kind) {
55 case transformer::EditKind::Range:
56 if (auto Err =
57 AC.replace(*Result.SourceManager, T.Range, T.Replacement)) {
58 Consumer(std::move(Err));
59 return;
60 }
61 break;
62 case transformer::EditKind::AddInclude:
63 AC.addHeader(T.Replacement);
64 break;
65 }
66 }
67
68 for (auto &IDChangePair : ChangesByFileID)
69 Consumer(std::move(IDChangePair.second));
70 }
71