clang/lib/ARCMigrate/ARCMTActions.cpp

   1 //===--- ARCMTActions.cpp - ARC Migrate Tool Frontend Actions ---*- 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/ARCMigrate/ARCMTActions.h"
  10 #include "clang/ARCMigrate/ARCMT.h"
  11 #include "clang/Frontend/CompilerInstance.h"
  12
  13 using namespace clang;
  14 using namespace arcmt;
  15
  16 bool CheckAction::BeginInvocation(CompilerInstance &CI) {
  17   if (arcmt::checkForManualIssues(CI.getInvocation(), getCurrentInput(),
  18                                   CI.getPCHContainerOperations(),
  19                                   CI.getDiagnostics().getClient()))
  20     return false; // errors, stop the action.
  21
  22   // We only want to see warnings reported from arcmt::checkForManualIssues.
  23   CI.getDiagnostics().setIgnoreAllWarnings(true);
  24   return true;
  25 }
  26
  27 CheckAction::CheckAction(std::unique_ptr<FrontendAction> WrappedAction)
  28   : WrapperFrontendAction(std::move(WrappedAction)) {}
  29
  30 bool ModifyAction::BeginInvocation(CompilerInstance &CI) {
  31   return !arcmt::applyTransformations(CI.getInvocation(), getCurrentInput(),
  32                                       CI.getPCHContainerOperations(),
  33                                       CI.getDiagnostics().getClient());
  34 }
  35
  36 ModifyAction::ModifyAction(std::unique_ptr<FrontendAction> WrappedAction)
  37   : WrapperFrontendAction(std::move(WrappedAction)) {}
  38
  39 bool MigrateAction::BeginInvocation(CompilerInstance &CI) {
  40   if (arcmt::migrateWithTemporaryFiles(
  41           CI.getInvocation(), getCurrentInput(), CI.getPCHContainerOperations(),
  42           CI.getDiagnostics().getClient(), MigrateDir, EmitPremigrationARCErrors,
  43           PlistOut))
  44     return false; // errors, stop the action.
  45
  46   // We only want to see diagnostics emitted by migrateWithTemporaryFiles.
  47   CI.getDiagnostics().setIgnoreAllWarnings(true);
  48   return true;
  49 }
  50
  51 MigrateAction::MigrateAction(std::unique_ptr<FrontendAction> WrappedAction,
  52                              StringRef migrateDir,
  53                              StringRef plistOut,
  54                              bool emitPremigrationARCErrors)
  55   : WrapperFrontendAction(std::move(WrappedAction)), MigrateDir(migrateDir),
  56     PlistOut(plistOut), EmitPremigrationARCErrors(emitPremigrationARCErrors) {
  57   if (MigrateDir.empty())
  58     MigrateDir = "."; // user current directory if none is given.
  59 }