14 #include "clang/Basic/SourceLocation.h"
15 #include "clang/Basic/SourceManager.h"
16 #include "clang/Frontend/CompilerInstance.h"
17 #include "clang/Frontend/CompilerInvocation.h"
18 #include "clang/Frontend/FrontendActions.h"
19 #include "clang/Lex/HeaderSearch.h"
20 #include "llvm/ADT/StringRef.h"
21 #include "llvm/Support/Path.h"
29 RecordHeaders(
const SourceManager &SM, IncludeStructure *
Out)
34 void InclusionDirective(SourceLocation HashLoc,
const Token &IncludeTok,
37 const FileEntry *
File, llvm::StringRef ,
40 SrcMgr::CharacteristicKind FileKind)
override {
41 auto MainFID = SM.getMainFileID();
49 Out->MainFileIncludes.emplace_back();
50 auto &Inc =
Out->MainFileIncludes.back();
53 Inc.Resolved = std::string(
File ?
File->tryGetRealPathName() :
"");
54 Inc.HashOffset = SM.getFileOffset(HashLoc);
56 SM.getLineNumber(SM.getFileID(HashLoc), Inc.HashOffset) - 1;
57 Inc.FileKind = FileKind;
58 Inc.Directive = IncludeTok.getIdentifierInfo()->getPPKeywordID();
63 auto *IncludingFileEntry = SM.getFileEntryForID(SM.getFileID(HashLoc));
64 if (!IncludingFileEntry) {
65 assert(SM.getBufferName(HashLoc).startswith(
"<") &&
66 "Expected #include location to be a file or <built-in>");
68 IncludingFileEntry = SM.getFileEntryForID(MainFID);
70 Out->recordInclude(IncludingFileEntry->getName(),
File->getName(),
71 File->tryGetRealPathName());
75 void FileChanged(SourceLocation
Loc, FileChangeReason Reason,
76 SrcMgr::CharacteristicKind FileType,
77 FileID PrevFID)
override {
79 case PPCallbacks::EnterFile:
80 if (BuiltinFile.isInvalid() && SM.isWrittenInBuiltinFile(
Loc)) {
81 BuiltinFile = SM.getFileID(
Loc);
85 case PPCallbacks::ExitFile:
86 if (PrevFID == BuiltinFile)
87 InBuiltinFile =
false;
89 case PPCallbacks::RenameFile:
90 case PPCallbacks::SystemHeaderPragma:
96 const SourceManager &SM;
100 bool InBuiltinFile =
false;
102 IncludeStructure *
Out;
108 return Include.startswith(
"<") || Include.startswith(
"\"");
117 llvm::StringRef HintPath) {
122 return U.takeError();
126 return IncludePath.takeError();
127 if (!IncludePath->empty())
128 return HeaderFile{std::move(*IncludePath),
true};
132 return Resolved.takeError();
133 return HeaderFile{std::move(*Resolved),
false};
145 llvm::SmallVector<llvm::StringRef, 1> Headers;
146 for (
const auto &Include : Includes)
147 Headers.push_back(Include.IncludeHeader);
151 std::unique_ptr<PPCallbacks>
154 return std::make_unique<RecordHeaders>(SM,
Out);
158 llvm::StringRef IncludedName,
159 llvm::StringRef IncludedRealName) {
160 auto Child = fileIndex(IncludedName);
161 if (!IncludedRealName.empty() && RealPathNames[Child].empty())
162 RealPathNames[Child] = std::string(IncludedRealName);
163 auto Parent = fileIndex(IncludingName);
164 IncludeChildren[
Parent].push_back(Child);
167 unsigned IncludeStructure::fileIndex(llvm::StringRef
Name) {
168 auto R = NameToIndex.try_emplace(
Name, RealPathNames.size());
170 RealPathNames.emplace_back();
171 return R.first->getValue();
174 llvm::StringMap<unsigned>
177 llvm::StringMap<unsigned> Result;
179 std::vector<unsigned> CurrentLevel;
180 llvm::DenseSet<unsigned> Seen;
181 auto It = NameToIndex.find(Root);
182 if (It != NameToIndex.end()) {
183 CurrentLevel.push_back(It->second);
184 Seen.insert(It->second);
188 std::vector<unsigned> PreviousLevel;
189 for (
unsigned Level = 1; !CurrentLevel.empty(); ++Level) {
190 PreviousLevel.clear();
191 PreviousLevel.swap(CurrentLevel);
192 for (
const auto &
Parent : PreviousLevel) {
193 for (
const auto &Child : IncludeChildren.lookup(
Parent)) {
194 if (Seen.insert(Child).second) {
195 CurrentLevel.push_back(Child);
196 const auto &
Name = RealPathNames[Child];
199 Result[
Name] = Level;
208 IncludedHeaders.insert(Inc.
Written);
210 IncludedHeaders.insert(Inc.
Resolved);
217 assert(InsertedHeader.
valid());
218 if (!HeaderSearchInfo && !InsertedHeader.
Verbatim)
222 auto Included = [&](llvm::StringRef Header) {
223 return IncludedHeaders.find(Header) != IncludedHeaders.end();
225 return !Included(DeclaringHeader) && !Included(InsertedHeader.
File);
228 llvm::Optional<std::string>
230 llvm::StringRef IncludingFile)
const {
231 assert(InsertedHeader.
valid());
233 return InsertedHeader.
File;
234 bool IsSystem =
false;
235 std::string Suggested;
236 if (HeaderSearchInfo) {
237 Suggested = HeaderSearchInfo->suggestPathToFileForDiagnostics(
238 InsertedHeader.
File, BuildDir, IncludingFile, &IsSystem);
241 StringRef IncludingDir = llvm::sys::path::parent_path(IncludingFile);
242 SmallString<256> RelFile(InsertedHeader.
File);
244 llvm::sys::path::replace_path_prefix(RelFile, IncludingDir,
"./");
245 Suggested = llvm::sys::path::convert_to_slash(
246 llvm::sys::path::remove_leading_dotslash(RelFile));
249 if (llvm::sys::path::is_absolute(Suggested))
252 Suggested =
"<" + Suggested +
">";
254 Suggested =
"\"" + Suggested +
"\"";
258 llvm::Optional<TextEdit>
260 llvm::Optional<TextEdit>
Edit =
None;
261 if (
auto Insertion = Inserter.insert(VerbatimHeader.trim(
"\"<>"),
262 VerbatimHeader.startswith(
"<")))