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