ARCMT.h revision 341825
1224135Sdim//===-- ARCMT.h - ARC Migration Rewriter ------------------------*- C++ -*-===// 2224135Sdim// 3224135Sdim// The LLVM Compiler Infrastructure 4224135Sdim// 5224135Sdim// This file is distributed under the University of Illinois Open Source 6224135Sdim// License. See LICENSE.TXT for details. 7224135Sdim// 8224135Sdim//===----------------------------------------------------------------------===// 9224135Sdim 10224135Sdim#ifndef LLVM_CLANG_ARCMIGRATE_ARCMT_H 11224135Sdim#define LLVM_CLANG_ARCMIGRATE_ARCMT_H 12224135Sdim 13224135Sdim#include "clang/ARCMigrate/FileRemapper.h" 14249423Sdim#include "clang/Basic/SourceLocation.h" 15224135Sdim#include "clang/Frontend/CompilerInvocation.h" 16224135Sdim 17224135Sdimnamespace clang { 18224135Sdim class ASTContext; 19226633Sdim class DiagnosticConsumer; 20288943Sdim class PCHContainerOperations; 21224135Sdim 22224135Sdimnamespace arcmt { 23224135Sdim class MigrationPass; 24224135Sdim 25341825Sdim/// Creates an AST with the provided CompilerInvocation but with these 26224135Sdim/// changes: 27224135Sdim/// -if a PCH/PTH is set, the original header is used instead 28224135Sdim/// -Automatic Reference Counting mode is enabled 29224135Sdim/// 30224135Sdim/// It then checks the AST and produces errors/warning for ARC migration issues 31224135Sdim/// that the user needs to handle manually. 32224135Sdim/// 33226633Sdim/// \param emitPremigrationARCErrors if true all ARC errors will get emitted 34226633Sdim/// even if the migrator can fix them, but the function will still return false 35226633Sdim/// if all ARC errors can be fixed. 36226633Sdim/// 37226633Sdim/// \param plistOut if non-empty, it is the file path to store the plist with 38226633Sdim/// the pre-migration ARC diagnostics. 39226633Sdim/// 40224135Sdim/// \returns false if no error is produced, true otherwise. 41288943Sdimbool 42288943SdimcheckForManualIssues(CompilerInvocation &CI, const FrontendInputFile &Input, 43288943Sdim std::shared_ptr<PCHContainerOperations> PCHContainerOps, 44288943Sdim DiagnosticConsumer *DiagClient, 45288943Sdim bool emitPremigrationARCErrors = false, 46288943Sdim StringRef plistOut = StringRef()); 47224135Sdim 48341825Sdim/// Works similar to checkForManualIssues but instead of checking, it 49224135Sdim/// applies automatic modifications to source files to conform to ARC. 50224135Sdim/// 51224135Sdim/// \returns false if no error is produced, true otherwise. 52288943Sdimbool 53288943SdimapplyTransformations(CompilerInvocation &origCI, 54288943Sdim const FrontendInputFile &Input, 55288943Sdim std::shared_ptr<PCHContainerOperations> PCHContainerOps, 56288943Sdim DiagnosticConsumer *DiagClient); 57224135Sdim 58341825Sdim/// Applies automatic modifications and produces temporary files 59243830Sdim/// and metadata into the \p outputDir path. 60224135Sdim/// 61226633Sdim/// \param emitPremigrationARCErrors if true all ARC errors will get emitted 62226633Sdim/// even if the migrator can fix them, but the function will still return false 63226633Sdim/// if all ARC errors can be fixed. 64226633Sdim/// 65226633Sdim/// \param plistOut if non-empty, it is the file path to store the plist with 66226633Sdim/// the pre-migration ARC diagnostics. 67226633Sdim/// 68224135Sdim/// \returns false if no error is produced, true otherwise. 69288943Sdimbool migrateWithTemporaryFiles( 70288943Sdim CompilerInvocation &origCI, const FrontendInputFile &Input, 71288943Sdim std::shared_ptr<PCHContainerOperations> PCHContainerOps, 72288943Sdim DiagnosticConsumer *DiagClient, StringRef outputDir, 73288943Sdim bool emitPremigrationARCErrors, StringRef plistOut); 74224135Sdim 75341825Sdim/// Get the set of file remappings from the \p outputDir path that 76224135Sdim/// migrateWithTemporaryFiles produced. 77224135Sdim/// 78224135Sdim/// \returns false if no error is produced, true otherwise. 79224135Sdimbool getFileRemappings(std::vector<std::pair<std::string,std::string> > &remap, 80226633Sdim StringRef outputDir, 81226633Sdim DiagnosticConsumer *DiagClient); 82224135Sdim 83341825Sdim/// Get the set of file remappings from a list of files with remapping 84234353Sdim/// info. 85234353Sdim/// 86234353Sdim/// \returns false if no error is produced, true otherwise. 87234353Sdimbool getFileRemappingsFromFileList( 88234353Sdim std::vector<std::pair<std::string,std::string> > &remap, 89234353Sdim ArrayRef<StringRef> remapFiles, 90234353Sdim DiagnosticConsumer *DiagClient); 91234353Sdim 92224135Sdimtypedef void (*TransformFn)(MigrationPass &pass); 93224135Sdim 94234353Sdimstd::vector<TransformFn> getAllTransformations(LangOptions::GCMode OrigGCMode, 95234353Sdim bool NoFinalizeRemoval); 96224135Sdim 97224135Sdimclass MigrationProcess { 98224135Sdim CompilerInvocation OrigCI; 99288943Sdim std::shared_ptr<PCHContainerOperations> PCHContainerOps; 100226633Sdim DiagnosticConsumer *DiagClient; 101224135Sdim FileRemapper Remapper; 102224135Sdim 103224135Sdimpublic: 104261991Sdim bool HadARCErrors; 105261991Sdim 106288943Sdim MigrationProcess(const CompilerInvocation &CI, 107288943Sdim std::shared_ptr<PCHContainerOperations> PCHContainerOps, 108288943Sdim DiagnosticConsumer *diagClient, 109226633Sdim StringRef outputDir = StringRef()); 110224135Sdim 111224135Sdim class RewriteListener { 112224135Sdim public: 113224135Sdim virtual ~RewriteListener(); 114224135Sdim 115224135Sdim virtual void start(ASTContext &Ctx) { } 116224135Sdim virtual void finish() { } 117224135Sdim 118226633Sdim virtual void insert(SourceLocation loc, StringRef text) { } 119224135Sdim virtual void remove(CharSourceRange range) { } 120224135Sdim }; 121224135Sdim 122276479Sdim bool applyTransform(TransformFn trans, RewriteListener *listener = nullptr); 123224135Sdim 124224135Sdim FileRemapper &getRemapper() { return Remapper; } 125224135Sdim}; 126224135Sdim 127224135Sdim} // end namespace arcmt 128224135Sdim 129224135Sdim} // end namespace clang 130224135Sdim 131224135Sdim#endif 132