tools/libclang/ARCMigrate.cpp

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