Home | History | Annotate | Line # | Download | only in Transformer 
      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 
     24 void Transformer::registerMatchers(MatchFinder *MatchFinder) {
     25   for (auto &Matcher : transformer::detail::buildMatchers(Rule))
     26     MatchFinder->addDynamicMatcher(Matcher, this);
     27 }
     28 
     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