18 #include "llvm/ADT/SmallString.h"
19 #include "llvm/Support/Path.h"
20 #include <system_error>
31 class CommandLineArgumentParser {
36 std::vector<std::string> parse() {
37 bool HasMoreInput =
true;
38 while (HasMoreInput && nextNonWhitespace()) {
40 HasMoreInput = parseStringInto(Argument);
49 bool parseStringInto(std::string &String) {
52 if (!parseDoubleQuotedStringInto(String))
return false;
54 if (!parseSingleQuotedStringInto(String))
return false;
56 if (!parseFreeStringInto(String))
return false;
62 bool parseDoubleQuotedStringInto(std::string &String) {
63 if (!next())
return false;
65 if (!skipEscapeCharacter())
return false;
67 if (!next())
return false;
72 bool parseSingleQuotedStringInto(std::string &String) {
73 if (!next())
return false;
76 if (!next())
return false;
81 bool parseFreeStringInto(std::string &String) {
83 if (!skipEscapeCharacter())
return false;
85 if (!next())
return false;
90 bool skipEscapeCharacter() {
97 bool nextNonWhitespace() {
99 if (!next())
return false;
114 std::vector<std::string> unescapeCommandLine(
115 StringRef EscapedCommandLine) {
116 CommandLineArgumentParser parser(EscapedCommandLine);
117 return parser.parse();
120 class JSONCompilationDatabasePlugin :
public CompilationDatabasePlugin {
121 std::unique_ptr<CompilationDatabase>
122 loadFromDirectory(StringRef Directory, std::string &ErrorMessage)
override {
123 SmallString<1024> JSONDatabasePath(Directory);
124 llvm::sys::path::append(JSONDatabasePath,
"compile_commands.json");
125 std::unique_ptr<CompilationDatabase> Database(
137 static CompilationDatabasePluginRegistry::Add<JSONCompilationDatabasePlugin>
138 X(
"json-compilation-database",
"Reads JSON formatted compilation databases");
144 std::unique_ptr<JSONCompilationDatabase>
146 std::string &ErrorMessage) {
147 llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> DatabaseBuffer =
148 llvm::MemoryBuffer::getFile(FilePath);
149 if (std::error_code
Result = DatabaseBuffer.getError()) {
150 ErrorMessage =
"Error while opening JSON database: " +
Result.message();
153 std::unique_ptr<JSONCompilationDatabase> Database(
155 if (!Database->parse(ErrorMessage))
160 std::unique_ptr<JSONCompilationDatabase>
162 std::string &ErrorMessage) {
163 std::unique_ptr<llvm::MemoryBuffer> DatabaseBuffer(
164 llvm::MemoryBuffer::getMemBuffer(DatabaseString));
165 std::unique_ptr<JSONCompilationDatabase> Database(
167 if (!Database->parse(ErrorMessage))
172 std::vector<CompileCommand>
175 llvm::sys::path::native(FilePath, NativeFilePath);
178 llvm::raw_string_ostream ES(Error);
181 return std::vector<CompileCommand>();
182 llvm::StringMap< std::vector<CompileCommandRef> >::const_iterator
183 CommandsRefI = IndexByFile.find(Match);
184 if (CommandsRefI == IndexByFile.end())
185 return std::vector<CompileCommand>();
186 std::vector<CompileCommand> Commands;
187 getCommands(CommandsRefI->getValue(), Commands);
191 std::vector<std::string>
193 std::vector<std::string>
Result;
195 llvm::StringMap< std::vector<CompileCommandRef> >::const_iterator
196 CommandsRefI = IndexByFile.begin();
197 const llvm::StringMap< std::vector<CompileCommandRef> >::const_iterator
198 CommandsRefEnd = IndexByFile.end();
199 for (; CommandsRefI != CommandsRefEnd; ++CommandsRefI) {
200 Result.push_back(CommandsRefI->first().str());
206 std::vector<CompileCommand>
208 std::vector<CompileCommand> Commands;
209 for (llvm::StringMap< std::vector<CompileCommandRef> >::const_iterator
210 CommandsRefI = IndexByFile.begin(), CommandsRefEnd = IndexByFile.end();
211 CommandsRefI != CommandsRefEnd; ++CommandsRefI) {
212 getCommands(CommandsRefI->getValue(), Commands);
217 void JSONCompilationDatabase::getCommands(
219 std::vector<CompileCommand> &Commands)
const {
220 for (
int I = 0, E = CommandsRef.size(); I != E; ++I) {
223 Commands.emplace_back(
225 CommandsRef[I].first->getValue(DirectoryStorage),
226 unescapeCommandLine(CommandsRef[I].second->getValue(CommandStorage)));
230 bool JSONCompilationDatabase::parse(std::string &ErrorMessage) {
231 llvm::yaml::document_iterator I = YAMLStream.begin();
232 if (I == YAMLStream.end()) {
233 ErrorMessage =
"Error while parsing YAML.";
238 ErrorMessage =
"Error while parsing YAML.";
241 llvm::yaml::SequenceNode *Array = dyn_cast<llvm::yaml::SequenceNode>(Root);
243 ErrorMessage =
"Expected array.";
246 for (llvm::yaml::SequenceNode::iterator AI = Array->begin(),
249 llvm::yaml::MappingNode *Object = dyn_cast<llvm::yaml::MappingNode>(&*AI);
251 ErrorMessage =
"Expected object.";
254 llvm::yaml::ScalarNode *Directory =
nullptr;
255 llvm::yaml::ScalarNode *Command =
nullptr;
256 llvm::yaml::ScalarNode *File =
nullptr;
257 for (llvm::yaml::MappingNode::iterator KVI = Object->begin(),
262 ErrorMessage =
"Expected value.";
265 llvm::yaml::ScalarNode *ValueString =
266 dyn_cast<llvm::yaml::ScalarNode>(
Value);
268 ErrorMessage =
"Expected string as value.";
271 llvm::yaml::ScalarNode *KeyString =
272 dyn_cast<llvm::yaml::ScalarNode>((*KVI).getKey());
274 ErrorMessage =
"Expected strings as key.";
277 SmallString<8> KeyStorage;
278 if (KeyString->getValue(KeyStorage) ==
"directory") {
279 Directory = ValueString;
280 }
else if (KeyString->getValue(KeyStorage) ==
"command") {
281 Command = ValueString;
282 }
else if (KeyString->getValue(KeyStorage) ==
"file") {
285 ErrorMessage = (
"Unknown key: \"" +
286 KeyString->getRawValue() +
"\"").str();
291 ErrorMessage =
"Missing key: \"file\".";
295 ErrorMessage =
"Missing key: \"command\".";
299 ErrorMessage =
"Missing key: \"directory\".";
302 SmallString<8> FileStorage;
303 StringRef FileName = File->getValue(FileStorage);
304 SmallString<128> NativeFilePath;
305 if (llvm::sys::path::is_relative(FileName)) {
306 SmallString<8> DirectoryStorage;
307 SmallString<128> AbsolutePath(
308 Directory->getValue(DirectoryStorage));
309 llvm::sys::path::append(AbsolutePath, FileName);
310 llvm::sys::path::native(AbsolutePath, NativeFilePath);
312 llvm::sys::path::native(FileName, NativeFilePath);
314 IndexByFile[NativeFilePath].push_back(
315 CompileCommandRef(Directory, Command));
316 MatchTrie.
insert(NativeFilePath);
The result type of a method or function.
ast_type_traits::DynTypedNode Node
std::vector< std::string > CommandLine
StringRef::iterator Position