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
clang_getRemappings(const char * migrate_dir_path)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
clang_getRemappingsFromFileList(const char ** filePaths,unsigned numFiles)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
clang_remap_getNumFiles(CXRemapping map)121 unsigned clang_remap_getNumFiles(CXRemapping map) {
122 return static_cast<Remap *>(map)->Vec.size();
123
124 }
125
clang_remap_getFilenames(CXRemapping map,unsigned index,CXString * original,CXString * transformed)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
clang_remap_dispose(CXRemapping map)136 void clang_remap_dispose(CXRemapping map) {
137 delete static_cast<Remap *>(map);
138 }
139