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