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
|
//===--- IncludeSpeller.cpp------------------------------------------------===//
//
// 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 "clang-include-cleaner/IncludeSpeller.h"
#include "clang-include-cleaner/Types.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/ErrorHandling.h"
#include "llvm/Support/Registry.h"
#include <memory>
#include <string>
LLVM_INSTANTIATE_REGISTRY(clang::include_cleaner::IncludeSpellingStrategy)
namespace clang::include_cleaner {
namespace {
// Fallback strategy to default spelling via header search.
class DefaultIncludeSpeller : public IncludeSpeller {
public:
std::string operator()(const Input &Input) const override {
switch (Input.H.kind()) {
case Header::Standard:
return Input.H.standard().name().str();
case Header::Verbatim:
return Input.H.verbatim().str();
case Header::Physical:
bool IsAngled = false;
std::string WorkingDir;
if (auto WD = Input.HS.getFileMgr()
.getVirtualFileSystem()
.getCurrentWorkingDirectory())
WorkingDir = *WD;
std::string FinalSpelling = Input.HS.suggestPathToFileForDiagnostics(
Input.H.physical().getName(), WorkingDir,
Input.Main->tryGetRealPathName(), &IsAngled);
return IsAngled ? "<" + FinalSpelling + ">" : "\"" + FinalSpelling + "\"";
}
llvm_unreachable("Unknown clang::include_cleaner::Header::Kind enum");
}
};
} // namespace
std::string spellHeader(const IncludeSpeller::Input &Input) {
static auto *Spellers = [] {
auto *Result =
new llvm::SmallVector<std::unique_ptr<include_cleaner::IncludeSpeller>>;
for (const auto &Strategy :
include_cleaner::IncludeSpellingStrategy::entries())
Result->push_back(Strategy.instantiate());
Result->push_back(std::make_unique<DefaultIncludeSpeller>());
return Result;
}();
std::string Spelling;
for (const auto &Speller : *Spellers) {
Spelling = (*Speller)(Input);
if (!Spelling.empty())
break;
}
return Spelling;
}
} // namespace clang::include_cleaner
|
