Home | History | Annotate | Line # | Download | only in libclang 
      1 //===- Rewrite.cpp --------------------------------------------------------===//
      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 #include "clang-c/Rewrite.h"
     10 #include "CXSourceLocation.h"
     11 #include "CXTranslationUnit.h"
     12 #include "clang/Basic/SourceManager.h"
     13 #include "clang/Frontend/ASTUnit.h"
     14 #include "clang/Rewrite/Core/Rewriter.h"
     15 
     16 CXRewriter clang_CXRewriter_create(CXTranslationUnit TU) {
     17   if (clang::cxtu::isNotUsableTU(TU)) {
     18     LOG_BAD_TU(TU);
     19     return {};
     20   }
     21   clang::ASTUnit *AU = clang::cxtu::getASTUnit(TU);
     22   assert(AU);
     23   return reinterpret_cast<CXRewriter>(
     24       new clang::Rewriter(AU->getSourceManager(), AU->getLangOpts()));
     25 }
     26 
     27 void clang_CXRewriter_insertTextBefore(CXRewriter Rew, CXSourceLocation Loc,
     28                             const char *Insert) {
     29   assert(Rew);
     30   clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
     31   R.InsertTextBefore(clang::cxloc::translateSourceLocation(Loc), Insert);
     32 }
     33 
     34 void clang_CXRewriter_replaceText(CXRewriter Rew, CXSourceRange ToBeReplaced,
     35                        const char *Replacement) {
     36   assert(Rew);
     37   clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
     38   R.ReplaceText(clang::cxloc::translateCXRangeToCharRange(ToBeReplaced),
     39                 Replacement);
     40 }
     41 
     42 void clang_CXRewriter_removeText(CXRewriter Rew, CXSourceRange ToBeRemoved) {
     43   assert(Rew);
     44   clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
     45   R.RemoveText(clang::cxloc::translateCXRangeToCharRange(ToBeRemoved));
     46 }
     47 
     48 int clang_CXRewriter_overwriteChangedFiles(CXRewriter Rew) {
     49   assert(Rew);
     50   clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
     51   return R.overwriteChangedFiles();
     52 }
     53 
     54 void clang_CXRewriter_writeMainFileToStdOut(CXRewriter Rew) {
     55   assert(Rew);
     56   clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
     57   R.getEditBuffer(R.getSourceMgr().getMainFileID()).write(llvm::outs());
     58 }
     59 
     60 void clang_CXRewriter_dispose(CXRewriter Rew) {
     61   if (Rew)
     62     delete reinterpret_cast<clang::Rewriter *>(Rew);
     63 }
     64