1*e038c9c4Sjoerg //===- Rewrite.cpp --------------------------------------------------------===//
2*e038c9c4Sjoerg //
3*e038c9c4Sjoerg // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*e038c9c4Sjoerg // See https://llvm.org/LICENSE.txt for license information.
5*e038c9c4Sjoerg // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*e038c9c4Sjoerg //
7*e038c9c4Sjoerg //===----------------------------------------------------------------------===//
8*e038c9c4Sjoerg
9*e038c9c4Sjoerg #include "clang-c/Rewrite.h"
10*e038c9c4Sjoerg #include "CXSourceLocation.h"
11*e038c9c4Sjoerg #include "CXTranslationUnit.h"
12*e038c9c4Sjoerg #include "clang/Basic/SourceManager.h"
13*e038c9c4Sjoerg #include "clang/Frontend/ASTUnit.h"
14*e038c9c4Sjoerg #include "clang/Rewrite/Core/Rewriter.h"
15*e038c9c4Sjoerg
clang_CXRewriter_create(CXTranslationUnit TU)16*e038c9c4Sjoerg CXRewriter clang_CXRewriter_create(CXTranslationUnit TU) {
17*e038c9c4Sjoerg if (clang::cxtu::isNotUsableTU(TU)) {
18*e038c9c4Sjoerg LOG_BAD_TU(TU);
19*e038c9c4Sjoerg return {};
20*e038c9c4Sjoerg }
21*e038c9c4Sjoerg clang::ASTUnit *AU = clang::cxtu::getASTUnit(TU);
22*e038c9c4Sjoerg assert(AU);
23*e038c9c4Sjoerg return reinterpret_cast<CXRewriter>(
24*e038c9c4Sjoerg new clang::Rewriter(AU->getSourceManager(), AU->getLangOpts()));
25*e038c9c4Sjoerg }
26*e038c9c4Sjoerg
clang_CXRewriter_insertTextBefore(CXRewriter Rew,CXSourceLocation Loc,const char * Insert)27*e038c9c4Sjoerg void clang_CXRewriter_insertTextBefore(CXRewriter Rew, CXSourceLocation Loc,
28*e038c9c4Sjoerg const char *Insert) {
29*e038c9c4Sjoerg assert(Rew);
30*e038c9c4Sjoerg clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
31*e038c9c4Sjoerg R.InsertTextBefore(clang::cxloc::translateSourceLocation(Loc), Insert);
32*e038c9c4Sjoerg }
33*e038c9c4Sjoerg
clang_CXRewriter_replaceText(CXRewriter Rew,CXSourceRange ToBeReplaced,const char * Replacement)34*e038c9c4Sjoerg void clang_CXRewriter_replaceText(CXRewriter Rew, CXSourceRange ToBeReplaced,
35*e038c9c4Sjoerg const char *Replacement) {
36*e038c9c4Sjoerg assert(Rew);
37*e038c9c4Sjoerg clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
38*e038c9c4Sjoerg R.ReplaceText(clang::cxloc::translateCXRangeToCharRange(ToBeReplaced),
39*e038c9c4Sjoerg Replacement);
40*e038c9c4Sjoerg }
41*e038c9c4Sjoerg
clang_CXRewriter_removeText(CXRewriter Rew,CXSourceRange ToBeRemoved)42*e038c9c4Sjoerg void clang_CXRewriter_removeText(CXRewriter Rew, CXSourceRange ToBeRemoved) {
43*e038c9c4Sjoerg assert(Rew);
44*e038c9c4Sjoerg clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
45*e038c9c4Sjoerg R.RemoveText(clang::cxloc::translateCXRangeToCharRange(ToBeRemoved));
46*e038c9c4Sjoerg }
47*e038c9c4Sjoerg
clang_CXRewriter_overwriteChangedFiles(CXRewriter Rew)48*e038c9c4Sjoerg int clang_CXRewriter_overwriteChangedFiles(CXRewriter Rew) {
49*e038c9c4Sjoerg assert(Rew);
50*e038c9c4Sjoerg clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
51*e038c9c4Sjoerg return R.overwriteChangedFiles();
52*e038c9c4Sjoerg }
53*e038c9c4Sjoerg
clang_CXRewriter_writeMainFileToStdOut(CXRewriter Rew)54*e038c9c4Sjoerg void clang_CXRewriter_writeMainFileToStdOut(CXRewriter Rew) {
55*e038c9c4Sjoerg assert(Rew);
56*e038c9c4Sjoerg clang::Rewriter &R = *reinterpret_cast<clang::Rewriter *>(Rew);
57*e038c9c4Sjoerg R.getEditBuffer(R.getSourceMgr().getMainFileID()).write(llvm::outs());
58*e038c9c4Sjoerg }
59*e038c9c4Sjoerg
clang_CXRewriter_dispose(CXRewriter Rew)60*e038c9c4Sjoerg void clang_CXRewriter_dispose(CXRewriter Rew) {
61*e038c9c4Sjoerg if (Rew)
62*e038c9c4Sjoerg delete reinterpret_cast<clang::Rewriter *>(Rew);
63*e038c9c4Sjoerg }
64