1 //===- ARCMigrate.cpp - Clang-C ARC Migration Library ---------------------===//
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 // This file implements the main API hooks in the Clang-C ARC Migration library.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "clang-c/Index.h"
14 #include "CXString.h"
15 #include "clang/ARCMigrate/ARCMT.h"
16 #include "clang/Config/config.h"
17 #include "clang/Frontend/TextDiagnosticBuffer.h"
18 #include "llvm/Support/FileSystem.h"
19 
20 using namespace clang;
21 using namespace arcmt;
22 
23 namespace {
24 
25 struct Remap {
26   std::vector<std::pair<std::string, std::string> > Vec;
27 };
28 
29 } // anonymous namespace.
30 
31 //===----------------------------------------------------------------------===//
32 // libClang public APIs.
33 //===----------------------------------------------------------------------===//
34 
35 CXRemapping clang_getRemappings(const char *migrate_dir_path) {
36 #if !CLANG_ENABLE_ARCMT
37   llvm::errs() << "error: feature not enabled in this build\n";
38   return nullptr;
39 #else
40   bool Logging = ::getenv("LIBCLANG_LOGGING");
41 
42   if (!migrate_dir_path) {
43     if (Logging)
44       llvm::errs() << "clang_getRemappings was called with NULL parameter\n";
45     return nullptr;
46   }
47 
48   if (!llvm::sys::fs::exists(migrate_dir_path)) {
49     if (Logging) {
50       llvm::errs() << "Error by clang_getRemappings(\"" << migrate_dir_path
51                    << "\")\n";
52       llvm::errs() << "\"" << migrate_dir_path << "\" does not exist\n";
53     }
54     return nullptr;
55   }
56 
57   TextDiagnosticBuffer diagBuffer;
58   std::unique_ptr<Remap> remap(new Remap());
59 
60   bool err = arcmt::getFileRemappings(remap->Vec, migrate_dir_path,&diagBuffer);
61 
62   if (err) {
63     if (Logging) {
64       llvm::errs() << "Error by clang_getRemappings(\"" << migrate_dir_path
65                    << "\")\n";
66       for (TextDiagnosticBuffer::const_iterator
67              I = diagBuffer.err_begin(), E = diagBuffer.err_end(); I != E; ++I)
68         llvm::errs() << I->second << '\n';
69     }
70     return nullptr;
71   }
72 
73   return remap.release();
74 #endif
75 }
76 
77 CXRemapping clang_getRemappingsFromFileList(const char **filePaths,
78                                             unsigned numFiles) {
79 #if !CLANG_ENABLE_ARCMT
80   llvm::errs() << "error: feature not enabled in this build\n";
81   return nullptr;
82 #else
83   bool Logging = ::getenv("LIBCLANG_LOGGING");
84 
85   std::unique_ptr<Remap> remap(new Remap());
86 
87   if (numFiles == 0) {
88     if (Logging)
89       llvm::errs() << "clang_getRemappingsFromFileList was called with "
90                       "numFiles=0\n";
91     return remap.release();
92   }
93 
94   if (!filePaths) {
95     if (Logging)
96       llvm::errs() << "clang_getRemappingsFromFileList was called with "
97                       "NULL filePaths\n";
98     return nullptr;
99   }
100 
101   TextDiagnosticBuffer diagBuffer;
102   SmallVector<StringRef, 32> Files(filePaths, filePaths + numFiles);
103 
104   bool err = arcmt::getFileRemappingsFromFileList(remap->Vec, Files,
105                                                   &diagBuffer);
106 
107   if (err) {
108     if (Logging) {
109       llvm::errs() << "Error by clang_getRemappingsFromFileList\n";
110       for (TextDiagnosticBuffer::const_iterator
111              I = diagBuffer.err_begin(), E = diagBuffer.err_end(); I != E; ++I)
112         llvm::errs() << I->second << '\n';
113     }
114     return remap.release();
115   }
116 
117   return remap.release();
118 #endif
119 }
120 
121 unsigned clang_remap_getNumFiles(CXRemapping map) {
122   return static_cast<Remap *>(map)->Vec.size();
123 
124 }
125 
126 void clang_remap_getFilenames(CXRemapping map, unsigned index,
127                               CXString *original, CXString *transformed) {
128   if (original)
129     *original = cxstring::createDup(
130                     static_cast<Remap *>(map)->Vec[index].first);
131   if (transformed)
132     *transformed = cxstring::createDup(
133                     static_cast<Remap *>(map)->Vec[index].second);
134 }
135 
136 void clang_remap_dispose(CXRemapping map) {
137   delete static_cast<Remap *>(map);
138 }
139