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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
#pragma once
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#endif
//===- JsonSupport.h - JSON Output Utilities --------------------*- C++ -*-===//
//
// 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
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_CLANG_BASIC_JSONSUPPORT_H
#define LLVM_CLANG_BASIC_JSONSUPPORT_H
#include "clang/Basic/LLVM.h"
#include "clang/Basic/SourceManager.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/Path.h"
#include "llvm/Support/raw_ostream.h"
#include <iterator>
namespace clang {
inline raw_ostream &Indent(raw_ostream &Out, const unsigned int Space,
bool IsDot) {
for (unsigned int I = 0; I < Space * 2; ++I)
Out << (IsDot ? " " : " ");
return Out;
}
inline std::string JsonFormat(StringRef RawSR, bool AddQuotes) {
if (RawSR.empty())
return "null";
// Trim special characters.
std::string Str = RawSR.trim().str();
size_t Pos = 0;
// Escape backslashes.
while (true) {
Pos = Str.find('\\', Pos);
if (Pos == std::string::npos)
break;
// Prevent bad conversions.
size_t TempPos = (Pos != 0) ? Pos - 1 : 0;
// See whether the current backslash is not escaped.
if (TempPos != Str.find("\\\\", Pos)) {
Str.insert(Pos, "\\");
++Pos; // As we insert the backslash move plus one.
}
++Pos;
}
// Escape double quotes.
Pos = 0;
while (true) {
Pos = Str.find('\"', Pos);
if (Pos == std::string::npos)
break;
// Prevent bad conversions.
size_t TempPos = (Pos != 0) ? Pos - 1 : 0;
// See whether the current double quote is not escaped.
if (TempPos != Str.find("\\\"", Pos)) {
Str.insert(Pos, "\\");
++Pos; // As we insert the escape-character move plus one.
}
++Pos;
}
// Remove new-lines.
llvm::erase_value(Str, '\n');
if (!AddQuotes)
return Str;
return '\"' + Str + '\"';
}
inline void printSourceLocationAsJson(raw_ostream &Out, SourceLocation Loc,
const SourceManager &SM,
bool AddBraces = true) {
// Mostly copy-pasted from SourceLocation::print.
if (!Loc.isValid()) {
Out << "null";
return;
}
if (Loc.isFileID()) {
PresumedLoc PLoc = SM.getPresumedLoc(Loc);
if (PLoc.isInvalid()) {
Out << "null";
return;
}
// The macro expansion and spelling pos is identical for file locs.
if (AddBraces)
Out << "{ ";
std::string filename(PLoc.getFilename());
if (is_style_windows(llvm::sys::path::Style::native)) {
// Remove forbidden Windows path characters
auto RemoveIt =
std::remove_if(filename.begin(), filename.end(), [](auto Char) {
static const char ForbiddenChars[] = "<>*?\"|";
return std::find(std::begin(ForbiddenChars),
std::end(ForbiddenChars),
Char) != std::end(ForbiddenChars);
});
filename.erase(RemoveIt, filename.end());
// Handle windows-specific path delimiters.
std::replace(filename.begin(), filename.end(), '\\', '/');
}
Out << "\"line\": " << PLoc.getLine()
<< ", \"column\": " << PLoc.getColumn()
<< ", \"file\": \"" << filename << "\"";
if (AddBraces)
Out << " }";
return;
}
// We want 'location: { ..., spelling: { ... }}' but not
// 'location: { ... }, spelling: { ... }', hence the dance
// with braces.
Out << "{ ";
printSourceLocationAsJson(Out, SM.getExpansionLoc(Loc), SM, false);
Out << ", \"spelling\": ";
printSourceLocationAsJson(Out, SM.getSpellingLoc(Loc), SM, true);
Out << " }";
}
} // namespace clang
#endif // LLVM_CLANG_BASIC_JSONSUPPORT_H
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
|
