Home | History | Annotate | Line # | Download | only in libclang 
      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