clang/tools/libclang/Rewrite.cpp

   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 }