64 lines · 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-exception6//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