contrib/llvm/tools/clang/lib/ARCMigrate/ARCMTActions.cpp

   1 //===--- ARCMTActions.cpp - ARC Migrate Tool Frontend Actions ---*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "clang/ARCMigrate/ARCMTActions.h"
  11 #include "clang/ARCMigrate/ARCMT.h"
  12 #include "clang/Frontend/CompilerInstance.h"
  13
  14 using namespace clang;
  15 using namespace arcmt;
  16
  17 bool CheckAction::BeginInvocation(CompilerInstance &CI) {
  18   if (arcmt::checkForManualIssues(CI.getInvocation(), getCurrentInput(),
  19                                   CI.getPCHContainerOperations(),
  20                                   CI.getDiagnostics().getClient()))
  21     return false; // errors, stop the action.
  22
  23   // We only want to see warnings reported from arcmt::checkForManualIssues.
  24   CI.getDiagnostics().setIgnoreAllWarnings(true);
  25   return true;
  26 }
  27
  28 CheckAction::CheckAction(std::unique_ptr<FrontendAction> WrappedAction)
  29   : WrapperFrontendAction(std::move(WrappedAction)) {}
  30
  31 bool ModifyAction::BeginInvocation(CompilerInstance &CI) {
  32   return !arcmt::applyTransformations(CI.getInvocation(), getCurrentInput(),
  33                                       CI.getPCHContainerOperations(),
  34                                       CI.getDiagnostics().getClient());
  35 }
  36
  37 ModifyAction::ModifyAction(std::unique_ptr<FrontendAction> WrappedAction)
  38   : WrapperFrontendAction(std::move(WrappedAction)) {}
  39
  40 bool MigrateAction::BeginInvocation(CompilerInstance &CI) {
  41   if (arcmt::migrateWithTemporaryFiles(
  42           CI.getInvocation(), getCurrentInput(), CI.getPCHContainerOperations(),
  43           CI.getDiagnostics().getClient(), MigrateDir, EmitPremigrationARCErros,
  44           PlistOut))
  45     return false; // errors, stop the action.
  46
  47   // We only want to see diagnostics emitted by migrateWithTemporaryFiles.
  48   CI.getDiagnostics().setIgnoreAllWarnings(true);
  49   return true;
  50 }
  51
  52 MigrateAction::MigrateAction(std::unique_ptr<FrontendAction> WrappedAction,
  53                              StringRef migrateDir,
  54                              StringRef plistOut,
  55                              bool emitPremigrationARCErrors)
  56   : WrapperFrontendAction(std::move(WrappedAction)), MigrateDir(migrateDir),
  57     PlistOut(plistOut), EmitPremigrationARCErros(emitPremigrationARCErrors) {
  58   if (MigrateDir.empty())
  59     MigrateDir = "."; // user current directory if none is given.
  60 }