1 //===--- RewriteTest.cpp - Rewriter playground ----------------------------===//
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 // This is a testbed.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "clang/Lex/Preprocessor.h"
14 #include "clang/Rewrite/Core/TokenRewriter.h"
15 #include "clang/Rewrite/Frontend/Rewriters.h"
16 #include "llvm/Support/raw_ostream.h"
17 
18 void clang::DoRewriteTest(Preprocessor &PP, raw_ostream *OS) {
19   SourceManager &SM = PP.getSourceManager();
20   const LangOptions &LangOpts = PP.getLangOpts();
21 
22   TokenRewriter Rewriter(SM.getMainFileID(), SM, LangOpts);
23 
24   // Throw <i> </i> tags around comments.
25   for (TokenRewriter::token_iterator I = Rewriter.token_begin(),
26        E = Rewriter.token_end(); I != E; ++I) {
27     if (I->isNot(tok::comment)) continue;
28 
29     Rewriter.AddTokenBefore(I, "<i>");
30     Rewriter.AddTokenAfter(I, "</i>");
31   }
32 
33 
34   // Print out the output.
35   for (TokenRewriter::token_iterator I = Rewriter.token_begin(),
36        E = Rewriter.token_end(); I != E; ++I)
37     *OS << PP.getSpelling(*I);
38 }
39