1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
//===- CompilationDatabase.cpp - LSP Compilation Database -----------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "mlir/Tools/lsp-server-support/CompilationDatabase.h"
#include "mlir/Support/FileUtilities.h"
#include "mlir/Tools/lsp-server-support/Logging.h"
#include "mlir/Tools/lsp-server-support/Protocol.h"
#include "llvm/ADT/SetVector.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/YAMLTraits.h"
using namespace mlir;
using namespace mlir::lsp;
//===----------------------------------------------------------------------===//
// YamlFileInfo
//===----------------------------------------------------------------------===//
namespace {
struct YamlFileInfo {
/// The absolute path to the file.
std::string filename;
/// The include directories available for the file.
std::vector<std::string> includeDirs;
};
} // namespace
//===----------------------------------------------------------------------===//
// CompilationDatabase
//===----------------------------------------------------------------------===//
LLVM_YAML_IS_DOCUMENT_LIST_VECTOR(YamlFileInfo)
namespace llvm {
namespace yaml {
template <>
struct MappingTraits<YamlFileInfo> {
static void mapping(IO &io, YamlFileInfo &info) {
// Parse the filename and normalize it to the form we will expect from
// incoming URIs.
io.mapRequired("filepath", info.filename);
// Normalize the filename to avoid incompatability with incoming URIs.
if (Expected<lsp::URIForFile> uri =
lsp::URIForFile::fromFile(info.filename))
info.filename = uri->file().str();
// Parse the includes from the yaml stream. These are in the form of a
// semi-colon delimited list.
std::string combinedIncludes;
io.mapRequired("includes", combinedIncludes);
for (StringRef include : llvm::split(combinedIncludes, ";")) {
if (!include.empty())
info.includeDirs.push_back(include.str());
}
}
};
} // end namespace yaml
} // end namespace llvm
CompilationDatabase::CompilationDatabase(ArrayRef<std::string> databases) {
for (StringRef filename : databases)
loadDatabase(filename);
}
const CompilationDatabase::FileInfo &
CompilationDatabase::getFileInfo(StringRef filename) const {
auto it = files.find(filename);
return it == files.end() ? defaultFileInfo : it->second;
}
void CompilationDatabase::loadDatabase(StringRef filename) {
if (filename.empty())
return;
// Set up the input file.
std::string errorMessage;
std::unique_ptr<llvm::MemoryBuffer> inputFile =
openInputFile(filename, &errorMessage);
if (!inputFile) {
Logger::error("Failed to open compilation database: {0}", errorMessage);
return;
}
llvm::yaml::Input yaml(inputFile->getBuffer());
// Parse the yaml description and add any new files to the database.
std::vector<YamlFileInfo> parsedFiles;
yaml >> parsedFiles;
SetVector<StringRef> knownIncludes;
for (auto &file : parsedFiles) {
auto it = files.try_emplace(file.filename, std::move(file.includeDirs));
// If we encounter a duplicate file, log a warning and ignore it.
if (!it.second) {
Logger::info("Duplicate file in compilation database: {0}",
file.filename);
continue;
}
// Track the includes for the file.
for (StringRef include : it.first->second.includeDirs)
knownIncludes.insert(include);
}
// Add all of the known includes to the default file info. We don't know any
// information about how to treat these files, but these may be project files
// that we just don't yet have information for. In these cases, providing some
// heuristic information provides a better user experience, and generally
// shouldn't lead to any negative side effects.
for (StringRef include : knownIncludes)
defaultFileInfo.includeDirs.push_back(include.str());
}
|
