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 16CXRewriter 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 27void 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 34void 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 42void 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 48int clang_CXRewriter_overwriteChangedFiles(CXRewriter Rew) { 49 assert(Rew); 50 clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew); 51 return R.overwriteChangedFiles(); 52} 53 54void 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 60void clang_CXRewriter_dispose(CXRewriter Rew) { 61 if (Rew) 62 delete reinterpret_cast<clang::Rewriter *>(Rew); 63} 64