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
|
//===- ClangCASTest.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/Basic/Diagnostic.h"
#include "clang/CAS/IncludeTree.h"
#include "clang/Frontend/CompileJobCacheKey.h"
#include "clang/Frontend/CompilerInstance.h"
#include "llvm/CAS/ObjectStore.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/Error.h"
#include "llvm/Support/InitLLVM.h"
using namespace clang;
using namespace clang::cas;
using namespace llvm;
static ExitOnError ExitOnErr("clang-cas-test: ");
static void printCompileJobCacheKey(llvm::cas::ObjectStore &CAS,
StringRef Key) {
auto KeyID = ExitOnErr(CAS.parseID(Key));
ExitOnErr(printCompileJobCacheKey(CAS, KeyID, outs()));
}
static void printIncludeTree(llvm::cas::ObjectStore &CAS, StringRef Key) {
auto ID = ExitOnErr(CAS.parseID(Key));
llvm::cas::ObjectRef Ref = *CAS.getReference(ID);
auto IncludeTree = ExitOnErr(IncludeTreeRoot::get(CAS, Ref));
ExitOnErr(IncludeTree.print(outs()));
}
int main(int Argc, const char **Argv) {
llvm::InitLLVM X(Argc, Argv);
enum ActionType {
None,
PrintCompileJobCacheKey,
PrintIncludeTree,
};
cl::opt<ActionType> Action(
cl::desc("Action:"), cl::init(ActionType::None),
cl::values(clEnumValN(PrintCompileJobCacheKey,
"print-compile-job-cache-key",
"Print a compile-job result cache key's structure"),
clEnumValN(PrintIncludeTree, "print-include-tree",
"Print include tree structure")));
cl::opt<std::string> CASPath("cas", cl::desc("On-disk CAS path"),
cl::value_desc("path"));
cl::opt<std::string> CASPluginPath("fcas-plugin-path",
cl::desc("Path to plugin CAS library"),
cl::value_desc("path"));
cl::list<std::string> CASPluginOpts("fcas-plugin-option",
cl::desc("Plugin CAS Options"));
cl::list<std::string> Inputs(cl::Positional, cl::desc("<input>..."));
cl::ParseCommandLineOptions(Argc, Argv, "clang-cas-test");
if (Action == ActionType::None) {
errs() << "error: action required; pass '-help' for options\n";
return 1;
}
IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
CompilerInstance::createDiagnostics(new DiagnosticOptions));
CASOptions Opts;
// Printing a cache key only makes sense in an existing CAS, so default to
// on-disk instead of in-memory if no --cas path is specified.
Opts.CASPath = CASPath.empty() ? std::string("auto") : CASPath;
if (!CASPluginPath.empty()) {
Opts.PluginPath = CASPluginPath;
for (const auto &PluginOpt : CASPluginOpts) {
auto [Name, Val] = StringRef(PluginOpt).split('=');
Opts.PluginOptions.push_back({std::string(Name), std::string(Val)});
}
}
auto CAS =
Opts.getOrCreateDatabases(*Diags, /*CreateEmptyCASOnFailure=*/false)
.first;
if (!CAS)
return 1;
if (Action == ActionType::PrintCompileJobCacheKey) {
if (Inputs.empty()) {
errs() << "error: missing compile-job cache key in inputs'\n";
return 1;
}
for (StringRef CacheKey : Inputs)
printCompileJobCacheKey(*CAS, CacheKey);
} else if (Action == ActionType::PrintIncludeTree) {
if (Inputs.empty()) {
errs() << "error: missing include tree ID in inputs\n";
return 1;
}
for (StringRef CacheKey : Inputs)
printIncludeTree(*CAS, CacheKey);
}
return 0;
}
|
