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