Home | History | Annotate | Download | only in libclang 
      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"
    140