1 //===--- RefactoringCallbacks.cpp - Structural query framework ------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // 11 //===----------------------------------------------------------------------===// 12 #include "clang/Lex/Lexer.h" 13 #include "clang/Tooling/RefactoringCallbacks.h" 14 15 namespace clang { 16 namespace tooling { 17 RefactoringCallback()18 RefactoringCallback::RefactoringCallback() {} getReplacements()19 tooling::Replacements &RefactoringCallback::getReplacements() { 20 return Replace; 21 } 22 replaceStmtWithText(SourceManager & Sources,const Stmt & From,StringRef Text)23 static Replacement replaceStmtWithText(SourceManager &Sources, 24 const Stmt &From, 25 StringRef Text) { 26 return tooling::Replacement(Sources, CharSourceRange::getTokenRange( 27 From.getSourceRange()), Text); 28 } replaceStmtWithStmt(SourceManager & Sources,const Stmt & From,const Stmt & To)29 static Replacement replaceStmtWithStmt(SourceManager &Sources, 30 const Stmt &From, 31 const Stmt &To) { 32 return replaceStmtWithText(Sources, From, Lexer::getSourceText( 33 CharSourceRange::getTokenRange(To.getSourceRange()), 34 Sources, LangOptions())); 35 } 36 ReplaceStmtWithText(StringRef FromId,StringRef ToText)37 ReplaceStmtWithText::ReplaceStmtWithText(StringRef FromId, StringRef ToText) 38 : FromId(FromId), ToText(ToText) {} 39 run(const ast_matchers::MatchFinder::MatchResult & Result)40 void ReplaceStmtWithText::run( 41 const ast_matchers::MatchFinder::MatchResult &Result) { 42 if (const Stmt *FromMatch = Result.Nodes.getStmtAs<Stmt>(FromId)) { 43 Replace.insert(tooling::Replacement( 44 *Result.SourceManager, 45 CharSourceRange::getTokenRange(FromMatch->getSourceRange()), 46 ToText)); 47 } 48 } 49 ReplaceStmtWithStmt(StringRef FromId,StringRef ToId)50 ReplaceStmtWithStmt::ReplaceStmtWithStmt(StringRef FromId, StringRef ToId) 51 : FromId(FromId), ToId(ToId) {} 52 run(const ast_matchers::MatchFinder::MatchResult & Result)53 void ReplaceStmtWithStmt::run( 54 const ast_matchers::MatchFinder::MatchResult &Result) { 55 const Stmt *FromMatch = Result.Nodes.getStmtAs<Stmt>(FromId); 56 const Stmt *ToMatch = Result.Nodes.getStmtAs<Stmt>(ToId); 57 if (FromMatch && ToMatch) 58 Replace.insert(replaceStmtWithStmt( 59 *Result.SourceManager, *FromMatch, *ToMatch)); 60 } 61 ReplaceIfStmtWithItsBody(StringRef Id,bool PickTrueBranch)62 ReplaceIfStmtWithItsBody::ReplaceIfStmtWithItsBody(StringRef Id, 63 bool PickTrueBranch) 64 : Id(Id), PickTrueBranch(PickTrueBranch) {} 65 run(const ast_matchers::MatchFinder::MatchResult & Result)66 void ReplaceIfStmtWithItsBody::run( 67 const ast_matchers::MatchFinder::MatchResult &Result) { 68 if (const IfStmt *Node = Result.Nodes.getStmtAs<IfStmt>(Id)) { 69 const Stmt *Body = PickTrueBranch ? Node->getThen() : Node->getElse(); 70 if (Body) { 71 Replace.insert(replaceStmtWithStmt(*Result.SourceManager, *Node, *Body)); 72 } else if (!PickTrueBranch) { 73 // If we want to use the 'else'-branch, but it doesn't exist, delete 74 // the whole 'if'. 75 Replace.insert(replaceStmtWithText(*Result.SourceManager, *Node, "")); 76 } 77 } 78 } 79 80 } // end namespace tooling 81 } // end namespace clang 82