Index: include/clang/Tooling/Core/Replacement.h =================================================================== --- include/clang/Tooling/Core/Replacement.h +++ include/clang/Tooling/Core/Replacement.h @@ -19,6 +19,7 @@ #ifndef LLVM_CLANG_TOOLING_CORE_REPLACEMENT_H #define LLVM_CLANG_TOOLING_CORE_REPLACEMENT_H +#include "clang/Basic/FileManager.h" #include "clang/Basic/LangOptions.h" #include "clang/Basic/SourceLocation.h" #include "llvm/ADT/StringRef.h" @@ -293,9 +294,9 @@ const std::vector &Ranges); /// \brief If there are multiple pairs with the same file -/// path after removing dots, we only keep one pair (with path after dots being -/// removed) and discard the rest. +/// entry, we only keep one pair and discard the rest. std::map groupReplacementsByFile( + FileManager &FileMgr, const std::map &FileToReplaces); template Index: lib/Tooling/Core/Replacement.cpp =================================================================== --- lib/Tooling/Core/Replacement.cpp +++ lib/Tooling/Core/Replacement.cpp @@ -20,6 +20,7 @@ #include "clang/Basic/SourceManager.h" #include "clang/Lex/Lexer.h" #include "clang/Rewrite/Core/Rewriter.h" +#include "llvm/ADT/SmallPtrSet.h" #include "llvm/Support/FileSystem.h" #include "llvm/Support/Path.h" #include "llvm/Support/raw_os_ostream.h" @@ -566,13 +567,13 @@ } std::map groupReplacementsByFile( + FileManager &FileMgr, const std::map &FileToReplaces) { std::map Result; - for (const auto &Entry : FileToReplaces) { - llvm::SmallString<256> CleanPath(Entry.first.data()); - llvm::sys::path::remove_dots(CleanPath, /*remove_dot_dot=*/true); - Result[CleanPath.str()] = std::move(Entry.second); - } + llvm::SmallPtrSet ProcessedFileEntries; + for (const auto &Entry : FileToReplaces) + if (ProcessedFileEntries.insert(FileMgr.getFile(Entry.first)).second) + Result[Entry.first] = std::move(Entry.second); return Result; } Index: lib/Tooling/Refactoring.cpp =================================================================== --- lib/Tooling/Refactoring.cpp +++ lib/Tooling/Refactoring.cpp @@ -57,7 +57,8 @@ bool RefactoringTool::applyAllReplacements(Rewriter &Rewrite) { bool Result = true; - for (const auto &Entry : groupReplacementsByFile(FileToReplaces)) + for (const auto &Entry : groupReplacementsByFile( + Rewrite.getSourceMgr().getFileManager(), FileToReplaces)) Result = tooling::applyAllReplacements(Entry.second, Rewrite) && Result; return Result; } @@ -73,7 +74,8 @@ FileManager &Files = SM.getFileManager(); bool Result = true; - for (const auto &FileAndReplaces : groupReplacementsByFile(FileToReplaces)) { + for (const auto &FileAndReplaces : groupReplacementsByFile( + Rewrite.getSourceMgr().getFileManager(), FileToReplaces)) { const std::string &FilePath = FileAndReplaces.first; auto &CurReplaces = FileAndReplaces.second; Index: unittests/Tooling/RefactoringTest.cpp =================================================================== --- unittests/Tooling/RefactoringTest.cpp +++ unittests/Tooling/RefactoringTest.cpp @@ -972,8 +972,9 @@ toReplacements({{"", 0, 3, "cc"}, {"", 3, 3, "dd"}})); } -TEST(DeduplicateByFileTest, LeaveLeadingDotDot) { +TEST(DeduplicateByFileTest, PathsWithDots) { std::map FileToReplaces; + FileManager *FileMgr = new FileManager(FileSystemOptions()); #if !defined(LLVM_ON_WIN32) FileToReplaces["../../a/b/.././c.h"] = Replacements(); FileToReplaces["../../a/c.h"] = Replacements(); @@ -981,30 +982,31 @@ FileToReplaces["..\\..\\a\\b\\..\\.\\c.h"] = Replacements(); FileToReplaces["..\\..\\a\\c.h"] = Replacements(); #endif - FileToReplaces = groupReplacementsByFile(FileToReplaces); + FileToReplaces = groupReplacementsByFile(*FileMgr, FileToReplaces); EXPECT_EQ(1u, FileToReplaces.size()); #if !defined(LLVM_ON_WIN32) - EXPECT_EQ("../../a/c.h", FileToReplaces.begin()->first); + EXPECT_EQ("../../a/b/.././c.h", FileToReplaces.begin()->first); #else - EXPECT_EQ("..\\..\\a\\c.h", FileToReplaces.begin()->first); + EXPECT_EQ("..\\..\\a\\b\\..\\.\\c.h", FileToReplaces.begin()->first); #endif } -TEST(DeduplicateByFileTest, RemoveDotSlash) { +TEST(DeduplicateByFileTest, PathWithDotSlash) { std::map FileToReplaces; + FileManager *FileMgr = new FileManager(FileSystemOptions()); #if !defined(LLVM_ON_WIN32) - FileToReplaces["./a/b/.././c.h"] = Replacements(); - FileToReplaces["a/c.h"] = Replacements(); + FileToReplaces["./a/b/c.h"] = Replacements(); + FileToReplaces["a/b/c.h"] = Replacements(); #else - FileToReplaces[".\\a\\b\\..\\.\\c.h"] = Replacements(); - FileToReplaces["a\\c.h"] = Replacements(); + FileToReplaces[".\\a\\b\\c.h"] = Replacements(); + FileToReplaces["a\\b\\c.h"] = Replacements(); #endif - FileToReplaces = groupReplacementsByFile(FileToReplaces); + FileToReplaces = groupReplacementsByFile(*FileMgr, FileToReplaces); EXPECT_EQ(1u, FileToReplaces.size()); #if !defined(LLVM_ON_WIN32) - EXPECT_EQ("a/c.h", FileToReplaces.begin()->first); + EXPECT_EQ("./a/b/c.h", FileToReplaces.begin()->first); #else - EXPECT_EQ("a\\c.h", FileToReplaces.begin()->first); + EXPECT_EQ(".\\a\\b\\c.h", FileToReplaces.begin()->first); #endif }