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