sammccall created this revision.
sammccall added a reviewer: usaxena95.
Herald added a subscriber: kadircet.
sammccall requested review of this revision.
Herald added subscribers: cfe-commits, ilya-biryukov.
Herald added a project: clang.
Previously, loading one from a file meant allowing the library to do the IO.
Clangd would prefer to do such IO itself (e.g. to allow caching).
Repository:
rG LLVM Github Monorepo
https://reviews.llvm.org/D92640
Files:
clang/include/clang/Tooling/CompilationDatabase.h
clang/lib/Tooling/CompilationDatabase.cpp
clang/unittests/Tooling/CompilationDatabaseTest.cpp
Index: clang/unittests/Tooling/CompilationDatabaseTest.cpp
===================================================================
--- clang/unittests/Tooling/CompilationDatabaseTest.cpp
+++ clang/unittests/Tooling/CompilationDatabaseTest.cpp
@@ -541,6 +541,27 @@
EXPECT_EQ(0ul, Database.getAllCompileCommands().size());
}
+TEST(FixedCompilationDatabase, FromBuffer) {
+ const char *Data = R"(
+
+ -DFOO=BAR
+
+--baz
+
+ )";
+ std::string ErrorMsg;
+ auto CDB =
FixedCompilationDatabase::loadFromBuffer("/path/compile_flags.txt",
+ Data, ErrorMsg);
+
+ std::vector<CompileCommand> Result = CDB->getCompileCommands("/foo/bar.cc");
+ ASSERT_EQ(1ul, Result.size());
+ EXPECT_EQ("/path", Result.front().Directory);
+ EXPECT_EQ("/foo/bar.cc", Result.front().Filename);
+ EXPECT_THAT(
+ Result.front().CommandLine,
+ ElementsAre(EndsWith("clang-tool"), "-DFOO=BAR", "--baz",
"/foo/bar.cc"));
+}
+
TEST(ParseFixedCompilationDatabase, ReturnsNullOnEmptyArgumentList) {
int Argc = 0;
std::string ErrorMsg;
Index: clang/lib/Tooling/CompilationDatabase.cpp
===================================================================
--- clang/lib/Tooling/CompilationDatabase.cpp
+++ clang/lib/Tooling/CompilationDatabase.cpp
@@ -348,9 +348,17 @@
ErrorMsg = "Error while opening fixed database: " + Result.message();
return nullptr;
}
+ return loadFromBuffer(Path, (*File)->getBuffer(), ErrorMsg);
+}
+
+std::unique_ptr<FixedCompilationDatabase>
+FixedCompilationDatabase::loadFromBuffer(StringRef Path, StringRef Data,
+ std::string &ErrorMsg) {
+ ErrorMsg.clear();
std::vector<std::string> Args;
- for (llvm::StringRef Line :
- llvm::make_range(llvm::line_iterator(**File), llvm::line_iterator())) {
+ StringRef Line;
+ while (!Data.empty()) {
+ std::tie(Line, Data) = Data.split('\n');
// Stray whitespace is almost certainly unintended.
Line = Line.trim();
if (!Line.empty())
Index: clang/include/clang/Tooling/CompilationDatabase.h
===================================================================
--- clang/include/clang/Tooling/CompilationDatabase.h
+++ clang/include/clang/Tooling/CompilationDatabase.h
@@ -184,11 +184,15 @@
int &Argc, const char *const *Argv, std::string &ErrorMsg,
Twine Directory = ".");
- /// Reads flags from the given file, one-per line.
+ /// Reads flags from the given file, one-per-line.
/// Returns nullptr and sets ErrorMessage if we can't read the file.
static std::unique_ptr<FixedCompilationDatabase>
loadFromFile(StringRef Path, std::string &ErrorMsg);
+ /// Reads flags from the given buffer, one-per-line.
+ static std::unique_ptr<FixedCompilationDatabase>
+ loadFromBuffer(StringRef Path, StringRef Data, std::string &ErrorMsg);
+
/// Constructs a compilation data base from a specified directory
/// and command line.
FixedCompilationDatabase(Twine Directory, ArrayRef<std::string> CommandLine);
Index: clang/unittests/Tooling/CompilationDatabaseTest.cpp
===================================================================
--- clang/unittests/Tooling/CompilationDatabaseTest.cpp
+++ clang/unittests/Tooling/CompilationDatabaseTest.cpp
@@ -541,6 +541,27 @@
EXPECT_EQ(0ul, Database.getAllCompileCommands().size());
}
+TEST(FixedCompilationDatabase, FromBuffer) {
+ const char *Data = R"(
+
+ -DFOO=BAR
+
+--baz
+
+ )";
+ std::string ErrorMsg;
+ auto CDB = FixedCompilationDatabase::loadFromBuffer("/path/compile_flags.txt",
+ Data, ErrorMsg);
+
+ std::vector<CompileCommand> Result = CDB->getCompileCommands("/foo/bar.cc");
+ ASSERT_EQ(1ul, Result.size());
+ EXPECT_EQ("/path", Result.front().Directory);
+ EXPECT_EQ("/foo/bar.cc", Result.front().Filename);
+ EXPECT_THAT(
+ Result.front().CommandLine,
+ ElementsAre(EndsWith("clang-tool"), "-DFOO=BAR", "--baz", "/foo/bar.cc"));
+}
+
TEST(ParseFixedCompilationDatabase, ReturnsNullOnEmptyArgumentList) {
int Argc = 0;
std::string ErrorMsg;
Index: clang/lib/Tooling/CompilationDatabase.cpp
===================================================================
--- clang/lib/Tooling/CompilationDatabase.cpp
+++ clang/lib/Tooling/CompilationDatabase.cpp
@@ -348,9 +348,17 @@
ErrorMsg = "Error while opening fixed database: " + Result.message();
return nullptr;
}
+ return loadFromBuffer(Path, (*File)->getBuffer(), ErrorMsg);
+}
+
+std::unique_ptr<FixedCompilationDatabase>
+FixedCompilationDatabase::loadFromBuffer(StringRef Path, StringRef Data,
+ std::string &ErrorMsg) {
+ ErrorMsg.clear();
std::vector<std::string> Args;
- for (llvm::StringRef Line :
- llvm::make_range(llvm::line_iterator(**File), llvm::line_iterator())) {
+ StringRef Line;
+ while (!Data.empty()) {
+ std::tie(Line, Data) = Data.split('\n');
// Stray whitespace is almost certainly unintended.
Line = Line.trim();
if (!Line.empty())
Index: clang/include/clang/Tooling/CompilationDatabase.h
===================================================================
--- clang/include/clang/Tooling/CompilationDatabase.h
+++ clang/include/clang/Tooling/CompilationDatabase.h
@@ -184,11 +184,15 @@
int &Argc, const char *const *Argv, std::string &ErrorMsg,
Twine Directory = ".");
- /// Reads flags from the given file, one-per line.
+ /// Reads flags from the given file, one-per-line.
/// Returns nullptr and sets ErrorMessage if we can't read the file.
static std::unique_ptr<FixedCompilationDatabase>
loadFromFile(StringRef Path, std::string &ErrorMsg);
+ /// Reads flags from the given buffer, one-per-line.
+ static std::unique_ptr<FixedCompilationDatabase>
+ loadFromBuffer(StringRef Path, StringRef Data, std::string &ErrorMsg);
+
/// Constructs a compilation data base from a specified directory
/// and command line.
FixedCompilationDatabase(Twine Directory, ArrayRef<std::string> CommandLine);
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
