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