Index: lib/Tooling/Refactoring.cpp =================================================================== --- lib/Tooling/Refactoring.cpp +++ lib/Tooling/Refactoring.cpp @@ -105,16 +105,21 @@ const std::pair DecomposedLocation = Sources.getDecomposedLoc(Start); const FileEntry *Entry = Sources.getFileEntryForID(DecomposedLocation.first); - if (Entry != NULL) { // Make FilePath absolute so replacements can be applied correctly when - // relative paths for files are used. - llvm::SmallString<256> FilePath(Entry->getName()); - llvm::error_code EC = llvm::sys::fs::make_absolute(FilePath); - // Don't change the FilePath if the file is a virtual file. - this->FilePath = EC ? FilePath.c_str() : Entry->getName(); - } else + // relative paths for file are used. But we don't want to change virtual + // files. + if (llvm::sys::fs::exists(Entry->getName())) { + llvm::SmallString<256> FilePath(Entry->getName()); + llvm::sys::fs::make_absolute(FilePath); + this->FilePath = FilePath.c_str(); + } + else { + this->FilePath = Entry->getName(); + } + } else { this->FilePath = InvalidLocation; + } this->ReplacementRange = Range(DecomposedLocation.second, Length); this->ReplacementText = ReplacementText; } Index: unittests/Tooling/RefactoringTest.cpp =================================================================== --- unittests/Tooling/RefactoringTest.cpp +++ unittests/Tooling/RefactoringTest.cpp @@ -244,6 +244,10 @@ return Context.Sources.createFileID(File, SourceLocation(), SrcMgr::C_User); } + StringRef getFilePath(llvm::StringRef Name) { + return TemporaryFiles.lookup(Name); + } + std::string getFileContentFromDisk(llvm::StringRef Name) { std::string Path = TemporaryFiles.lookup(Name); assert(!Path.empty()); @@ -270,6 +274,55 @@ getFileContentFromDisk("input.cpp")); } +TEST_F(FlushRewrittenFilesTest, GetFilePath) { + // Create a temporary file. + createFile("input.cpp", "line1\nline2\nline3\nline4"); + + StringRef FilePath = getFilePath("input.cpp"); + StringRef TempPath = llvm::sys::path::parent_path(FilePath); + StringRef TempFile = llvm::sys::path::filename(FilePath); + + // Save current path. + SmallString<512> CurrentPath; + llvm::sys::fs::current_path(CurrentPath); + + // Change directory to the temporary directory. + EXPECT_FALSE(chdir(TempPath.str().c_str())); + llvm::sys::fs::current_path(CurrentPath); + + // Get a FileEntry from the current directory. + FileManager Files((FileSystemOptions())); + const FileEntry *Entry = Files.getFile(TempFile); + ASSERT_TRUE(Entry != NULL); + + // We expect to get just the filename and "." in the FileEntry + // since paths are relative to the current directory. + EXPECT_EQ(TempFile, Entry->getName()); + EXPECT_STREQ(".", Entry->getDir()->getName()); + + FileID ID = Context.Sources.createFileID(Entry, SourceLocation(), + SrcMgr::C_User); + + Replacement R = Replacement(Context.Sources, Context.getLocation(ID, 2, 1), + 5, "replaced"); + + // Change back to the original path so we can verify that replacements + // are being applied independently of the location. + EXPECT_FALSE(chdir(CurrentPath.c_str())); + + // We expect that the file path of the replacement is using the absolute + // path. + EXPECT_EQ(R.getFilePath(), getFilePath("input.cpp")); + + // Apply replacements. + Replacements Replaces; + Replaces.insert(R); + EXPECT_TRUE(applyAllReplacements(Replaces, Context.Rewrite)); + EXPECT_FALSE(Context.Rewrite.overwriteChangedFiles()); + EXPECT_EQ("line1\nreplaced\nline3\nline4", + getFileContentFromDisk("input.cpp")); +} + namespace { template class TestVisitor : public clang::RecursiveASTVisitor {