Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===--- JSON.h - JSON values, parsing and serialization -------*- C++ -*-===// |
2 | // |
||
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
||
4 | // See https://llvm.org/LICENSE.txt for license information. |
||
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
||
6 | // |
||
7 | //===---------------------------------------------------------------------===// |
||
8 | /// |
||
9 | /// \file |
||
10 | /// This file supports working with JSON data. |
||
11 | /// |
||
12 | /// It comprises: |
||
13 | /// |
||
14 | /// - classes which hold dynamically-typed parsed JSON structures |
||
15 | /// These are value types that can be composed, inspected, and modified. |
||
16 | /// See json::Value, and the related types json::Object and json::Array. |
||
17 | /// |
||
18 | /// - functions to parse JSON text into Values, and to serialize Values to text. |
||
19 | /// See parse(), operator<<, and format_provider. |
||
20 | /// |
||
21 | /// - a convention and helpers for mapping between json::Value and user-defined |
||
22 | /// types. See fromJSON(), ObjectMapper, and the class comment on Value. |
||
23 | /// |
||
24 | /// - an output API json::OStream which can emit JSON without materializing |
||
25 | /// all structures as json::Value. |
||
26 | /// |
||
27 | /// Typically, JSON data would be read from an external source, parsed into |
||
28 | /// a Value, and then converted into some native data structure before doing |
||
29 | /// real work on it. (And vice versa when writing). |
||
30 | /// |
||
31 | /// Other serialization mechanisms you may consider: |
||
32 | /// |
||
33 | /// - YAML is also text-based, and more human-readable than JSON. It's a more |
||
34 | /// complex format and data model, and YAML parsers aren't ubiquitous. |
||
35 | /// YAMLParser.h is a streaming parser suitable for parsing large documents |
||
36 | /// (including JSON, as YAML is a superset). It can be awkward to use |
||
37 | /// directly. YAML I/O (YAMLTraits.h) provides data mapping that is more |
||
38 | /// declarative than the toJSON/fromJSON conventions here. |
||
39 | /// |
||
40 | /// - LLVM bitstream is a space- and CPU- efficient binary format. Typically it |
||
41 | /// encodes LLVM IR ("bitcode"), but it can be a container for other data. |
||
42 | /// Low-level reader/writer libraries are in Bitstream/Bitstream*.h |
||
43 | /// |
||
44 | //===---------------------------------------------------------------------===// |
||
45 | |||
46 | #ifndef LLVM_SUPPORT_JSON_H |
||
47 | #define LLVM_SUPPORT_JSON_H |
||
48 | |||
49 | #include "llvm/ADT/DenseMap.h" |
||
50 | #include "llvm/ADT/SmallVector.h" |
||
51 | #include "llvm/ADT/StringRef.h" |
||
52 | #include "llvm/ADT/STLFunctionalExtras.h" |
||
53 | #include "llvm/Support/Error.h" |
||
54 | #include "llvm/Support/FormatVariadic.h" |
||
55 | #include "llvm/Support/raw_ostream.h" |
||
56 | #include <cmath> |
||
57 | #include <map> |
||
58 | |||
59 | namespace llvm { |
||
60 | namespace json { |
||
61 | |||
62 | // === String encodings === |
||
63 | // |
||
64 | // JSON strings are character sequences (not byte sequences like std::string). |
||
65 | // We need to know the encoding, and for simplicity only support UTF-8. |
||
66 | // |
||
67 | // - When parsing, invalid UTF-8 is a syntax error like any other |
||
68 | // |
||
69 | // - When creating Values from strings, callers must ensure they are UTF-8. |
||
70 | // with asserts on, invalid UTF-8 will crash the program |
||
71 | // with asserts off, we'll substitute the replacement character (U+FFFD) |
||
72 | // Callers can use json::isUTF8() and json::fixUTF8() for validation. |
||
73 | // |
||
74 | // - When retrieving strings from Values (e.g. asString()), the result will |
||
75 | // always be valid UTF-8. |
||
76 | |||
77 | /// Returns true if \p S is valid UTF-8, which is required for use as JSON. |
||
78 | /// If it returns false, \p Offset is set to a byte offset near the first error. |
||
79 | bool isUTF8(llvm::StringRef S, size_t *ErrOffset = nullptr); |
||
80 | /// Replaces invalid UTF-8 sequences in \p S with the replacement character |
||
81 | /// (U+FFFD). The returned string is valid UTF-8. |
||
82 | /// This is much slower than isUTF8, so test that first. |
||
83 | std::string fixUTF8(llvm::StringRef S); |
||
84 | |||
85 | class Array; |
||
86 | class ObjectKey; |
||
87 | class Value; |
||
88 | template <typename T> Value toJSON(const std::optional<T> &Opt); |
||
89 | |||
90 | /// An Object is a JSON object, which maps strings to heterogenous JSON values. |
||
91 | /// It simulates DenseMap<ObjectKey, Value>. ObjectKey is a maybe-owned string. |
||
92 | class Object { |
||
93 | using Storage = DenseMap<ObjectKey, Value, llvm::DenseMapInfo<StringRef>>; |
||
94 | Storage M; |
||
95 | |||
96 | public: |
||
97 | using key_type = ObjectKey; |
||
98 | using mapped_type = Value; |
||
99 | using value_type = Storage::value_type; |
||
100 | using iterator = Storage::iterator; |
||
101 | using const_iterator = Storage::const_iterator; |
||
102 | |||
103 | Object() = default; |
||
104 | // KV is a trivial key-value struct for list-initialization. |
||
105 | // (using std::pair forces extra copies). |
||
106 | struct KV; |
||
107 | explicit Object(std::initializer_list<KV> Properties); |
||
108 | |||
109 | iterator begin() { return M.begin(); } |
||
110 | const_iterator begin() const { return M.begin(); } |
||
111 | iterator end() { return M.end(); } |
||
112 | const_iterator end() const { return M.end(); } |
||
113 | |||
114 | bool empty() const { return M.empty(); } |
||
115 | size_t size() const { return M.size(); } |
||
116 | |||
117 | void clear() { M.clear(); } |
||
118 | std::pair<iterator, bool> insert(KV E); |
||
119 | template <typename... Ts> |
||
120 | std::pair<iterator, bool> try_emplace(const ObjectKey &K, Ts &&... Args) { |
||
121 | return M.try_emplace(K, std::forward<Ts>(Args)...); |
||
122 | } |
||
123 | template <typename... Ts> |
||
124 | std::pair<iterator, bool> try_emplace(ObjectKey &&K, Ts &&... Args) { |
||
125 | return M.try_emplace(std::move(K), std::forward<Ts>(Args)...); |
||
126 | } |
||
127 | bool erase(StringRef K); |
||
128 | void erase(iterator I) { M.erase(I); } |
||
129 | |||
130 | iterator find(StringRef K) { return M.find_as(K); } |
||
131 | const_iterator find(StringRef K) const { return M.find_as(K); } |
||
132 | // operator[] acts as if Value was default-constructible as null. |
||
133 | Value &operator[](const ObjectKey &K); |
||
134 | Value &operator[](ObjectKey &&K); |
||
135 | // Look up a property, returning nullptr if it doesn't exist. |
||
136 | Value *get(StringRef K); |
||
137 | const Value *get(StringRef K) const; |
||
138 | // Typed accessors return std::nullopt/nullptr if |
||
139 | // - the property doesn't exist |
||
140 | // - or it has the wrong type |
||
141 | std::optional<std::nullptr_t> getNull(StringRef K) const; |
||
142 | std::optional<bool> getBoolean(StringRef K) const; |
||
143 | std::optional<double> getNumber(StringRef K) const; |
||
144 | std::optional<int64_t> getInteger(StringRef K) const; |
||
145 | std::optional<llvm::StringRef> getString(StringRef K) const; |
||
146 | const json::Object *getObject(StringRef K) const; |
||
147 | json::Object *getObject(StringRef K); |
||
148 | const json::Array *getArray(StringRef K) const; |
||
149 | json::Array *getArray(StringRef K); |
||
150 | }; |
||
151 | bool operator==(const Object &LHS, const Object &RHS); |
||
152 | inline bool operator!=(const Object &LHS, const Object &RHS) { |
||
153 | return !(LHS == RHS); |
||
154 | } |
||
155 | |||
156 | /// An Array is a JSON array, which contains heterogeneous JSON values. |
||
157 | /// It simulates std::vector<Value>. |
||
158 | class Array { |
||
159 | std::vector<Value> V; |
||
160 | |||
161 | public: |
||
162 | using value_type = Value; |
||
163 | using iterator = std::vector<Value>::iterator; |
||
164 | using const_iterator = std::vector<Value>::const_iterator; |
||
165 | |||
166 | Array() = default; |
||
167 | explicit Array(std::initializer_list<Value> Elements); |
||
168 | template <typename Collection> explicit Array(const Collection &C) { |
||
169 | for (const auto &V : C) |
||
170 | emplace_back(V); |
||
171 | } |
||
172 | |||
173 | Value &operator[](size_t I); |
||
174 | const Value &operator[](size_t I) const; |
||
175 | Value &front(); |
||
176 | const Value &front() const; |
||
177 | Value &back(); |
||
178 | const Value &back() const; |
||
179 | Value *data(); |
||
180 | const Value *data() const; |
||
181 | |||
182 | iterator begin(); |
||
183 | const_iterator begin() const; |
||
184 | iterator end(); |
||
185 | const_iterator end() const; |
||
186 | |||
187 | bool empty() const; |
||
188 | size_t size() const; |
||
189 | void reserve(size_t S); |
||
190 | |||
191 | void clear(); |
||
192 | void push_back(const Value &E); |
||
193 | void push_back(Value &&E); |
||
194 | template <typename... Args> void emplace_back(Args &&...A); |
||
195 | void pop_back(); |
||
196 | iterator insert(const_iterator P, const Value &E); |
||
197 | iterator insert(const_iterator P, Value &&E); |
||
198 | template <typename It> iterator insert(const_iterator P, It A, It Z); |
||
199 | template <typename... Args> iterator emplace(const_iterator P, Args &&...A); |
||
200 | |||
201 | friend bool operator==(const Array &L, const Array &R); |
||
202 | }; |
||
203 | inline bool operator!=(const Array &L, const Array &R) { return !(L == R); } |
||
204 | |||
205 | /// A Value is an JSON value of unknown type. |
||
206 | /// They can be copied, but should generally be moved. |
||
207 | /// |
||
208 | /// === Composing values === |
||
209 | /// |
||
210 | /// You can implicitly construct Values from: |
||
211 | /// - strings: std::string, SmallString, formatv, StringRef, char* |
||
212 | /// (char*, and StringRef are references, not copies!) |
||
213 | /// - numbers |
||
214 | /// - booleans |
||
215 | /// - null: nullptr |
||
216 | /// - arrays: {"foo", 42.0, false} |
||
217 | /// - serializable things: types with toJSON(const T&)->Value, found by ADL |
||
218 | /// |
||
219 | /// They can also be constructed from object/array helpers: |
||
220 | /// - json::Object is a type like map<ObjectKey, Value> |
||
221 | /// - json::Array is a type like vector<Value> |
||
222 | /// These can be list-initialized, or used to build up collections in a loop. |
||
223 | /// json::ary(Collection) converts all items in a collection to Values. |
||
224 | /// |
||
225 | /// === Inspecting values === |
||
226 | /// |
||
227 | /// Each Value is one of the JSON kinds: |
||
228 | /// null (nullptr_t) |
||
229 | /// boolean (bool) |
||
230 | /// number (double, int64 or uint64) |
||
231 | /// string (StringRef) |
||
232 | /// array (json::Array) |
||
233 | /// object (json::Object) |
||
234 | /// |
||
235 | /// The kind can be queried directly, or implicitly via the typed accessors: |
||
236 | /// if (std::optional<StringRef> S = E.getAsString() |
||
237 | /// assert(E.kind() == Value::String); |
||
238 | /// |
||
239 | /// Array and Object also have typed indexing accessors for easy traversal: |
||
240 | /// Expected<Value> E = parse(R"( {"options": {"font": "sans-serif"}} )"); |
||
241 | /// if (Object* O = E->getAsObject()) |
||
242 | /// if (Object* Opts = O->getObject("options")) |
||
243 | /// if (std::optional<StringRef> Font = Opts->getString("font")) |
||
244 | /// assert(Opts->at("font").kind() == Value::String); |
||
245 | /// |
||
246 | /// === Converting JSON values to C++ types === |
||
247 | /// |
||
248 | /// The convention is to have a deserializer function findable via ADL: |
||
249 | /// fromJSON(const json::Value&, T&, Path) -> bool |
||
250 | /// |
||
251 | /// The return value indicates overall success, and Path is used for precise |
||
252 | /// error reporting. (The Path::Root passed in at the top level fromJSON call |
||
253 | /// captures any nested error and can render it in context). |
||
254 | /// If conversion fails, fromJSON calls Path::report() and immediately returns. |
||
255 | /// This ensures that the first fatal error survives. |
||
256 | /// |
||
257 | /// Deserializers are provided for: |
||
258 | /// - bool |
||
259 | /// - int and int64_t |
||
260 | /// - double |
||
261 | /// - std::string |
||
262 | /// - vector<T>, where T is deserializable |
||
263 | /// - map<string, T>, where T is deserializable |
||
264 | /// - std::optional<T>, where T is deserializable |
||
265 | /// ObjectMapper can help writing fromJSON() functions for object types. |
||
266 | /// |
||
267 | /// For conversion in the other direction, the serializer function is: |
||
268 | /// toJSON(const T&) -> json::Value |
||
269 | /// If this exists, then it also allows constructing Value from T, and can |
||
270 | /// be used to serialize vector<T>, map<string, T>, and std::optional<T>. |
||
271 | /// |
||
272 | /// === Serialization === |
||
273 | /// |
||
274 | /// Values can be serialized to JSON: |
||
275 | /// 1) raw_ostream << Value // Basic formatting. |
||
276 | /// 2) raw_ostream << formatv("{0}", Value) // Basic formatting. |
||
277 | /// 3) raw_ostream << formatv("{0:2}", Value) // Pretty-print with indent 2. |
||
278 | /// |
||
279 | /// And parsed: |
||
280 | /// Expected<Value> E = json::parse("[1, 2, null]"); |
||
281 | /// assert(E && E->kind() == Value::Array); |
||
282 | class Value { |
||
283 | public: |
||
284 | enum Kind { |
||
285 | Null, |
||
286 | Boolean, |
||
287 | /// Number values can store both int64s and doubles at full precision, |
||
288 | /// depending on what they were constructed/parsed from. |
||
289 | Number, |
||
290 | String, |
||
291 | Array, |
||
292 | Object, |
||
293 | }; |
||
294 | |||
295 | // It would be nice to have Value() be null. But that would make {} null too. |
||
296 | Value(const Value &M) { copyFrom(M); } |
||
297 | Value(Value &&M) { moveFrom(std::move(M)); } |
||
298 | Value(std::initializer_list<Value> Elements); |
||
299 | Value(json::Array &&Elements) : Type(T_Array) { |
||
300 | create<json::Array>(std::move(Elements)); |
||
301 | } |
||
302 | template <typename Elt> |
||
303 | Value(const std::vector<Elt> &C) : Value(json::Array(C)) {} |
||
304 | Value(json::Object &&Properties) : Type(T_Object) { |
||
305 | create<json::Object>(std::move(Properties)); |
||
306 | } |
||
307 | template <typename Elt> |
||
308 | Value(const std::map<std::string, Elt> &C) : Value(json::Object(C)) {} |
||
309 | // Strings: types with value semantics. Must be valid UTF-8. |
||
310 | Value(std::string V) : Type(T_String) { |
||
311 | if (LLVM_UNLIKELY(!isUTF8(V))) { |
||
312 | assert(false && "Invalid UTF-8 in value used as JSON"); |
||
313 | V = fixUTF8(std::move(V)); |
||
314 | } |
||
315 | create<std::string>(std::move(V)); |
||
316 | } |
||
317 | Value(const llvm::SmallVectorImpl<char> &V) |
||
318 | : Value(std::string(V.begin(), V.end())) {} |
||
319 | Value(const llvm::formatv_object_base &V) : Value(V.str()) {} |
||
320 | // Strings: types with reference semantics. Must be valid UTF-8. |
||
321 | Value(StringRef V) : Type(T_StringRef) { |
||
322 | create<llvm::StringRef>(V); |
||
323 | if (LLVM_UNLIKELY(!isUTF8(V))) { |
||
324 | assert(false && "Invalid UTF-8 in value used as JSON"); |
||
325 | *this = Value(fixUTF8(V)); |
||
326 | } |
||
327 | } |
||
328 | Value(const char *V) : Value(StringRef(V)) {} |
||
329 | Value(std::nullptr_t) : Type(T_Null) {} |
||
330 | // Boolean (disallow implicit conversions). |
||
331 | // (The last template parameter is a dummy to keep templates distinct.) |
||
332 | template <typename T, |
||
333 | typename = std::enable_if_t<std::is_same<T, bool>::value>, |
||
334 | bool = false> |
||
335 | Value(T B) : Type(T_Boolean) { |
||
336 | create<bool>(B); |
||
337 | } |
||
338 | |||
339 | // Unsigned 64-bit long integers. |
||
340 | template <typename T, |
||
341 | typename = std::enable_if_t<std::is_same<T, uint64_t>::value>, |
||
342 | bool = false, bool = false> |
||
343 | Value(T V) : Type(T_UINT64) { |
||
344 | create<uint64_t>(uint64_t{V}); |
||
345 | } |
||
346 | |||
347 | // Integers (except boolean and uint64_t). |
||
348 | // Must be non-narrowing convertible to int64_t. |
||
349 | template <typename T, typename = std::enable_if_t<std::is_integral<T>::value>, |
||
350 | typename = std::enable_if_t<!std::is_same<T, bool>::value>, |
||
351 | typename = std::enable_if_t<!std::is_same<T, uint64_t>::value>> |
||
352 | Value(T I) : Type(T_Integer) { |
||
353 | create<int64_t>(int64_t{I}); |
||
354 | } |
||
355 | // Floating point. Must be non-narrowing convertible to double. |
||
356 | template <typename T, |
||
357 | typename = std::enable_if_t<std::is_floating_point<T>::value>, |
||
358 | double * = nullptr> |
||
359 | Value(T D) : Type(T_Double) { |
||
360 | create<double>(double{D}); |
||
361 | } |
||
362 | // Serializable types: with a toJSON(const T&)->Value function, found by ADL. |
||
363 | template <typename T, |
||
364 | typename = std::enable_if_t<std::is_same< |
||
365 | Value, decltype(toJSON(*(const T *)nullptr))>::value>, |
||
366 | Value * = nullptr> |
||
367 | Value(const T &V) : Value(toJSON(V)) {} |
||
368 | |||
369 | Value &operator=(const Value &M) { |
||
370 | destroy(); |
||
371 | copyFrom(M); |
||
372 | return *this; |
||
373 | } |
||
374 | Value &operator=(Value &&M) { |
||
375 | destroy(); |
||
376 | moveFrom(std::move(M)); |
||
377 | return *this; |
||
378 | } |
||
379 | ~Value() { destroy(); } |
||
380 | |||
381 | Kind kind() const { |
||
382 | switch (Type) { |
||
383 | case T_Null: |
||
384 | return Null; |
||
385 | case T_Boolean: |
||
386 | return Boolean; |
||
387 | case T_Double: |
||
388 | case T_Integer: |
||
389 | case T_UINT64: |
||
390 | return Number; |
||
391 | case T_String: |
||
392 | case T_StringRef: |
||
393 | return String; |
||
394 | case T_Object: |
||
395 | return Object; |
||
396 | case T_Array: |
||
397 | return Array; |
||
398 | } |
||
399 | llvm_unreachable("Unknown kind"); |
||
400 | } |
||
401 | |||
402 | // Typed accessors return std::nullopt/nullptr if the Value is not of this |
||
403 | // type. |
||
404 | std::optional<std::nullptr_t> getAsNull() const { |
||
405 | if (LLVM_LIKELY(Type == T_Null)) |
||
406 | return nullptr; |
||
407 | return std::nullopt; |
||
408 | } |
||
409 | std::optional<bool> getAsBoolean() const { |
||
410 | if (LLVM_LIKELY(Type == T_Boolean)) |
||
411 | return as<bool>(); |
||
412 | return std::nullopt; |
||
413 | } |
||
414 | std::optional<double> getAsNumber() const { |
||
415 | if (LLVM_LIKELY(Type == T_Double)) |
||
416 | return as<double>(); |
||
417 | if (LLVM_LIKELY(Type == T_Integer)) |
||
418 | return as<int64_t>(); |
||
419 | if (LLVM_LIKELY(Type == T_UINT64)) |
||
420 | return as<uint64_t>(); |
||
421 | return std::nullopt; |
||
422 | } |
||
423 | // Succeeds if the Value is a Number, and exactly representable as int64_t. |
||
424 | std::optional<int64_t> getAsInteger() const { |
||
425 | if (LLVM_LIKELY(Type == T_Integer)) |
||
426 | return as<int64_t>(); |
||
427 | if (LLVM_LIKELY(Type == T_Double)) { |
||
428 | double D = as<double>(); |
||
429 | if (LLVM_LIKELY(std::modf(D, &D) == 0.0 && |
||
430 | D >= double(std::numeric_limits<int64_t>::min()) && |
||
431 | D <= double(std::numeric_limits<int64_t>::max()))) |
||
432 | return D; |
||
433 | } |
||
434 | return std::nullopt; |
||
435 | } |
||
436 | std::optional<uint64_t> getAsUINT64() const { |
||
437 | if (Type == T_UINT64) |
||
438 | return as<uint64_t>(); |
||
439 | else if (Type == T_Integer) { |
||
440 | int64_t N = as<int64_t>(); |
||
441 | if (N >= 0) |
||
442 | return as<uint64_t>(); |
||
443 | } |
||
444 | return std::nullopt; |
||
445 | } |
||
446 | std::optional<llvm::StringRef> getAsString() const { |
||
447 | if (Type == T_String) |
||
448 | return llvm::StringRef(as<std::string>()); |
||
449 | if (LLVM_LIKELY(Type == T_StringRef)) |
||
450 | return as<llvm::StringRef>(); |
||
451 | return std::nullopt; |
||
452 | } |
||
453 | const json::Object *getAsObject() const { |
||
454 | return LLVM_LIKELY(Type == T_Object) ? &as<json::Object>() : nullptr; |
||
455 | } |
||
456 | json::Object *getAsObject() { |
||
457 | return LLVM_LIKELY(Type == T_Object) ? &as<json::Object>() : nullptr; |
||
458 | } |
||
459 | const json::Array *getAsArray() const { |
||
460 | return LLVM_LIKELY(Type == T_Array) ? &as<json::Array>() : nullptr; |
||
461 | } |
||
462 | json::Array *getAsArray() { |
||
463 | return LLVM_LIKELY(Type == T_Array) ? &as<json::Array>() : nullptr; |
||
464 | } |
||
465 | |||
466 | private: |
||
467 | void destroy(); |
||
468 | void copyFrom(const Value &M); |
||
469 | // We allow moving from *const* Values, by marking all members as mutable! |
||
470 | // This hack is needed to support initializer-list syntax efficiently. |
||
471 | // (std::initializer_list<T> is a container of const T). |
||
472 | void moveFrom(const Value &&M); |
||
473 | friend class Array; |
||
474 | friend class Object; |
||
475 | |||
476 | template <typename T, typename... U> void create(U &&... V) { |
||
477 | new (reinterpret_cast<T *>(&Union)) T(std::forward<U>(V)...); |
||
478 | } |
||
479 | template <typename T> T &as() const { |
||
480 | // Using this two-step static_cast via void * instead of reinterpret_cast |
||
481 | // silences a -Wstrict-aliasing false positive from GCC6 and earlier. |
||
482 | void *Storage = static_cast<void *>(&Union); |
||
483 | return *static_cast<T *>(Storage); |
||
484 | } |
||
485 | |||
486 | friend class OStream; |
||
487 | |||
488 | enum ValueType : char16_t { |
||
489 | T_Null, |
||
490 | T_Boolean, |
||
491 | T_Double, |
||
492 | T_Integer, |
||
493 | T_UINT64, |
||
494 | T_StringRef, |
||
495 | T_String, |
||
496 | T_Object, |
||
497 | T_Array, |
||
498 | }; |
||
499 | // All members mutable, see moveFrom(). |
||
500 | mutable ValueType Type; |
||
501 | mutable llvm::AlignedCharArrayUnion<bool, double, int64_t, uint64_t, |
||
502 | llvm::StringRef, std::string, json::Array, |
||
503 | json::Object> |
||
504 | Union; |
||
505 | friend bool operator==(const Value &, const Value &); |
||
506 | }; |
||
507 | |||
508 | bool operator==(const Value &, const Value &); |
||
509 | inline bool operator!=(const Value &L, const Value &R) { return !(L == R); } |
||
510 | |||
511 | // Array Methods |
||
512 | inline Value &Array::operator[](size_t I) { return V[I]; } |
||
513 | inline const Value &Array::operator[](size_t I) const { return V[I]; } |
||
514 | inline Value &Array::front() { return V.front(); } |
||
515 | inline const Value &Array::front() const { return V.front(); } |
||
516 | inline Value &Array::back() { return V.back(); } |
||
517 | inline const Value &Array::back() const { return V.back(); } |
||
518 | inline Value *Array::data() { return V.data(); } |
||
519 | inline const Value *Array::data() const { return V.data(); } |
||
520 | |||
521 | inline typename Array::iterator Array::begin() { return V.begin(); } |
||
522 | inline typename Array::const_iterator Array::begin() const { return V.begin(); } |
||
523 | inline typename Array::iterator Array::end() { return V.end(); } |
||
524 | inline typename Array::const_iterator Array::end() const { return V.end(); } |
||
525 | |||
526 | inline bool Array::empty() const { return V.empty(); } |
||
527 | inline size_t Array::size() const { return V.size(); } |
||
528 | inline void Array::reserve(size_t S) { V.reserve(S); } |
||
529 | |||
530 | inline void Array::clear() { V.clear(); } |
||
531 | inline void Array::push_back(const Value &E) { V.push_back(E); } |
||
532 | inline void Array::push_back(Value &&E) { V.push_back(std::move(E)); } |
||
533 | template <typename... Args> inline void Array::emplace_back(Args &&...A) { |
||
534 | V.emplace_back(std::forward<Args>(A)...); |
||
535 | } |
||
536 | inline void Array::pop_back() { V.pop_back(); } |
||
537 | inline typename Array::iterator Array::insert(const_iterator P, const Value &E) { |
||
538 | return V.insert(P, E); |
||
539 | } |
||
540 | inline typename Array::iterator Array::insert(const_iterator P, Value &&E) { |
||
541 | return V.insert(P, std::move(E)); |
||
542 | } |
||
543 | template <typename It> |
||
544 | inline typename Array::iterator Array::insert(const_iterator P, It A, It Z) { |
||
545 | return V.insert(P, A, Z); |
||
546 | } |
||
547 | template <typename... Args> |
||
548 | inline typename Array::iterator Array::emplace(const_iterator P, Args &&...A) { |
||
549 | return V.emplace(P, std::forward<Args>(A)...); |
||
550 | } |
||
551 | inline bool operator==(const Array &L, const Array &R) { return L.V == R.V; } |
||
552 | |||
553 | /// ObjectKey is a used to capture keys in Object. Like Value but: |
||
554 | /// - only strings are allowed |
||
555 | /// - it's optimized for the string literal case (Owned == nullptr) |
||
556 | /// Like Value, strings must be UTF-8. See isUTF8 documentation for details. |
||
557 | class ObjectKey { |
||
558 | public: |
||
559 | ObjectKey(const char *S) : ObjectKey(StringRef(S)) {} |
||
560 | ObjectKey(std::string S) : Owned(new std::string(std::move(S))) { |
||
561 | if (LLVM_UNLIKELY(!isUTF8(*Owned))) { |
||
562 | assert(false && "Invalid UTF-8 in value used as JSON"); |
||
563 | *Owned = fixUTF8(std::move(*Owned)); |
||
564 | } |
||
565 | Data = *Owned; |
||
566 | } |
||
567 | ObjectKey(llvm::StringRef S) : Data(S) { |
||
568 | if (LLVM_UNLIKELY(!isUTF8(Data))) { |
||
569 | assert(false && "Invalid UTF-8 in value used as JSON"); |
||
570 | *this = ObjectKey(fixUTF8(S)); |
||
571 | } |
||
572 | } |
||
573 | ObjectKey(const llvm::SmallVectorImpl<char> &V) |
||
574 | : ObjectKey(std::string(V.begin(), V.end())) {} |
||
575 | ObjectKey(const llvm::formatv_object_base &V) : ObjectKey(V.str()) {} |
||
576 | |||
577 | ObjectKey(const ObjectKey &C) { *this = C; } |
||
578 | ObjectKey(ObjectKey &&C) : ObjectKey(static_cast<const ObjectKey &&>(C)) {} |
||
579 | ObjectKey &operator=(const ObjectKey &C) { |
||
580 | if (C.Owned) { |
||
581 | Owned.reset(new std::string(*C.Owned)); |
||
582 | Data = *Owned; |
||
583 | } else { |
||
584 | Data = C.Data; |
||
585 | } |
||
586 | return *this; |
||
587 | } |
||
588 | ObjectKey &operator=(ObjectKey &&) = default; |
||
589 | |||
590 | operator llvm::StringRef() const { return Data; } |
||
591 | std::string str() const { return Data.str(); } |
||
592 | |||
593 | private: |
||
594 | // FIXME: this is unneccesarily large (3 pointers). Pointer + length + owned |
||
595 | // could be 2 pointers at most. |
||
596 | std::unique_ptr<std::string> Owned; |
||
597 | llvm::StringRef Data; |
||
598 | }; |
||
599 | |||
600 | inline bool operator==(const ObjectKey &L, const ObjectKey &R) { |
||
601 | return llvm::StringRef(L) == llvm::StringRef(R); |
||
602 | } |
||
603 | inline bool operator!=(const ObjectKey &L, const ObjectKey &R) { |
||
604 | return !(L == R); |
||
605 | } |
||
606 | inline bool operator<(const ObjectKey &L, const ObjectKey &R) { |
||
607 | return StringRef(L) < StringRef(R); |
||
608 | } |
||
609 | |||
610 | struct Object::KV { |
||
611 | ObjectKey K; |
||
612 | Value V; |
||
613 | }; |
||
614 | |||
615 | inline Object::Object(std::initializer_list<KV> Properties) { |
||
616 | for (const auto &P : Properties) { |
||
617 | auto R = try_emplace(P.K, nullptr); |
||
618 | if (R.second) |
||
619 | R.first->getSecond().moveFrom(std::move(P.V)); |
||
620 | } |
||
621 | } |
||
622 | inline std::pair<Object::iterator, bool> Object::insert(KV E) { |
||
623 | return try_emplace(std::move(E.K), std::move(E.V)); |
||
624 | } |
||
625 | inline bool Object::erase(StringRef K) { |
||
626 | return M.erase(ObjectKey(K)); |
||
627 | } |
||
628 | |||
629 | /// A "cursor" marking a position within a Value. |
||
630 | /// The Value is a tree, and this is the path from the root to the current node. |
||
631 | /// This is used to associate errors with particular subobjects. |
||
632 | class Path { |
||
633 | public: |
||
634 | class Root; |
||
635 | |||
636 | /// Records that the value at the current path is invalid. |
||
637 | /// Message is e.g. "expected number" and becomes part of the final error. |
||
638 | /// This overwrites any previously written error message in the root. |
||
639 | void report(llvm::StringLiteral Message); |
||
640 | |||
641 | /// The root may be treated as a Path. |
||
642 | Path(Root &R) : Parent(nullptr), Seg(&R) {} |
||
643 | /// Derives a path for an array element: this[Index] |
||
644 | Path index(unsigned Index) const { return Path(this, Segment(Index)); } |
||
645 | /// Derives a path for an object field: this.Field |
||
646 | Path field(StringRef Field) const { return Path(this, Segment(Field)); } |
||
647 | |||
648 | private: |
||
649 | /// One element in a JSON path: an object field (.foo) or array index [27]. |
||
650 | /// Exception: the root Path encodes a pointer to the Path::Root. |
||
651 | class Segment { |
||
652 | uintptr_t Pointer; |
||
653 | unsigned Offset; |
||
654 | |||
655 | public: |
||
656 | Segment() = default; |
||
657 | Segment(Root *R) : Pointer(reinterpret_cast<uintptr_t>(R)) {} |
||
658 | Segment(llvm::StringRef Field) |
||
659 | : Pointer(reinterpret_cast<uintptr_t>(Field.data())), |
||
660 | Offset(static_cast<unsigned>(Field.size())) {} |
||
661 | Segment(unsigned Index) : Pointer(0), Offset(Index) {} |
||
662 | |||
663 | bool isField() const { return Pointer != 0; } |
||
664 | StringRef field() const { |
||
665 | return StringRef(reinterpret_cast<const char *>(Pointer), Offset); |
||
666 | } |
||
667 | unsigned index() const { return Offset; } |
||
668 | Root *root() const { return reinterpret_cast<Root *>(Pointer); } |
||
669 | }; |
||
670 | |||
671 | const Path *Parent; |
||
672 | Segment Seg; |
||
673 | |||
674 | Path(const Path *Parent, Segment S) : Parent(Parent), Seg(S) {} |
||
675 | }; |
||
676 | |||
677 | /// The root is the trivial Path to the root value. |
||
678 | /// It also stores the latest reported error and the path where it occurred. |
||
679 | class Path::Root { |
||
680 | llvm::StringRef Name; |
||
681 | llvm::StringLiteral ErrorMessage; |
||
682 | std::vector<Path::Segment> ErrorPath; // Only valid in error state. Reversed. |
||
683 | |||
684 | friend void Path::report(llvm::StringLiteral Message); |
||
685 | |||
686 | public: |
||
687 | Root(llvm::StringRef Name = "") : Name(Name), ErrorMessage("") {} |
||
688 | // No copy/move allowed as there are incoming pointers. |
||
689 | Root(Root &&) = delete; |
||
690 | Root &operator=(Root &&) = delete; |
||
691 | Root(const Root &) = delete; |
||
692 | Root &operator=(const Root &) = delete; |
||
693 | |||
694 | /// Returns the last error reported, or else a generic error. |
||
695 | Error getError() const; |
||
696 | /// Print the root value with the error shown inline as a comment. |
||
697 | /// Unrelated parts of the value are elided for brevity, e.g. |
||
698 | /// { |
||
699 | /// "id": 42, |
||
700 | /// "name": /* expected string */ null, |
||
701 | /// "properties": { ... } |
||
702 | /// } |
||
703 | void printErrorContext(const Value &, llvm::raw_ostream &) const; |
||
704 | }; |
||
705 | |||
706 | // Standard deserializers are provided for primitive types. |
||
707 | // See comments on Value. |
||
708 | inline bool fromJSON(const Value &E, std::string &Out, Path P) { |
||
709 | if (auto S = E.getAsString()) { |
||
710 | Out = std::string(*S); |
||
711 | return true; |
||
712 | } |
||
713 | P.report("expected string"); |
||
714 | return false; |
||
715 | } |
||
716 | inline bool fromJSON(const Value &E, int &Out, Path P) { |
||
717 | if (auto S = E.getAsInteger()) { |
||
718 | Out = *S; |
||
719 | return true; |
||
720 | } |
||
721 | P.report("expected integer"); |
||
722 | return false; |
||
723 | } |
||
724 | inline bool fromJSON(const Value &E, int64_t &Out, Path P) { |
||
725 | if (auto S = E.getAsInteger()) { |
||
726 | Out = *S; |
||
727 | return true; |
||
728 | } |
||
729 | P.report("expected integer"); |
||
730 | return false; |
||
731 | } |
||
732 | inline bool fromJSON(const Value &E, double &Out, Path P) { |
||
733 | if (auto S = E.getAsNumber()) { |
||
734 | Out = *S; |
||
735 | return true; |
||
736 | } |
||
737 | P.report("expected number"); |
||
738 | return false; |
||
739 | } |
||
740 | inline bool fromJSON(const Value &E, bool &Out, Path P) { |
||
741 | if (auto S = E.getAsBoolean()) { |
||
742 | Out = *S; |
||
743 | return true; |
||
744 | } |
||
745 | P.report("expected boolean"); |
||
746 | return false; |
||
747 | } |
||
748 | inline bool fromJSON(const Value &E, uint64_t &Out, Path P) { |
||
749 | if (auto S = E.getAsUINT64()) { |
||
750 | Out = *S; |
||
751 | return true; |
||
752 | } |
||
753 | P.report("expected uint64_t"); |
||
754 | return false; |
||
755 | } |
||
756 | inline bool fromJSON(const Value &E, std::nullptr_t &Out, Path P) { |
||
757 | if (auto S = E.getAsNull()) { |
||
758 | Out = *S; |
||
759 | return true; |
||
760 | } |
||
761 | P.report("expected null"); |
||
762 | return false; |
||
763 | } |
||
764 | template <typename T> |
||
765 | bool fromJSON(const Value &E, std::optional<T> &Out, Path P) { |
||
766 | if (E.getAsNull()) { |
||
767 | Out = std::nullopt; |
||
768 | return true; |
||
769 | } |
||
770 | T Result; |
||
771 | if (!fromJSON(E, Result, P)) |
||
772 | return false; |
||
773 | Out = std::move(Result); |
||
774 | return true; |
||
775 | } |
||
776 | template <typename T> |
||
777 | bool fromJSON(const Value &E, std::vector<T> &Out, Path P) { |
||
778 | if (auto *A = E.getAsArray()) { |
||
779 | Out.clear(); |
||
780 | Out.resize(A->size()); |
||
781 | for (size_t I = 0; I < A->size(); ++I) |
||
782 | if (!fromJSON((*A)[I], Out[I], P.index(I))) |
||
783 | return false; |
||
784 | return true; |
||
785 | } |
||
786 | P.report("expected array"); |
||
787 | return false; |
||
788 | } |
||
789 | template <typename T> |
||
790 | bool fromJSON(const Value &E, std::map<std::string, T> &Out, Path P) { |
||
791 | if (auto *O = E.getAsObject()) { |
||
792 | Out.clear(); |
||
793 | for (const auto &KV : *O) |
||
794 | if (!fromJSON(KV.second, Out[std::string(llvm::StringRef(KV.first))], |
||
795 | P.field(KV.first))) |
||
796 | return false; |
||
797 | return true; |
||
798 | } |
||
799 | P.report("expected object"); |
||
800 | return false; |
||
801 | } |
||
802 | |||
803 | // Allow serialization of std::optional<T> for supported T. |
||
804 | template <typename T> Value toJSON(const std::optional<T> &Opt) { |
||
805 | return Opt ? Value(*Opt) : Value(nullptr); |
||
806 | } |
||
807 | |||
808 | /// Helper for mapping JSON objects onto protocol structs. |
||
809 | /// |
||
810 | /// Example: |
||
811 | /// \code |
||
812 | /// bool fromJSON(const Value &E, MyStruct &R, Path P) { |
||
813 | /// ObjectMapper O(E, P); |
||
814 | /// // When returning false, error details were already reported. |
||
815 | /// return O && O.map("mandatory_field", R.MandatoryField) && |
||
816 | /// O.mapOptional("optional_field", R.OptionalField); |
||
817 | /// } |
||
818 | /// \endcode |
||
819 | class ObjectMapper { |
||
820 | public: |
||
821 | /// If O is not an object, this mapper is invalid and an error is reported. |
||
822 | ObjectMapper(const Value &E, Path P) : O(E.getAsObject()), P(P) { |
||
823 | if (!O) |
||
824 | P.report("expected object"); |
||
825 | } |
||
826 | |||
827 | /// True if the expression is an object. |
||
828 | /// Must be checked before calling map(). |
||
829 | operator bool() const { return O; } |
||
830 | |||
831 | /// Maps a property to a field. |
||
832 | /// If the property is missing or invalid, reports an error. |
||
833 | template <typename T> bool map(StringLiteral Prop, T &Out) { |
||
834 | assert(*this && "Must check this is an object before calling map()"); |
||
835 | if (const Value *E = O->get(Prop)) |
||
836 | return fromJSON(*E, Out, P.field(Prop)); |
||
837 | P.field(Prop).report("missing value"); |
||
838 | return false; |
||
839 | } |
||
840 | |||
841 | /// Maps a property to a field, if it exists. |
||
842 | /// If the property exists and is invalid, reports an error. |
||
843 | /// (Optional requires special handling, because missing keys are OK). |
||
844 | template <typename T> bool map(StringLiteral Prop, std::optional<T> &Out) { |
||
845 | assert(*this && "Must check this is an object before calling map()"); |
||
846 | if (const Value *E = O->get(Prop)) |
||
847 | return fromJSON(*E, Out, P.field(Prop)); |
||
848 | Out = std::nullopt; |
||
849 | return true; |
||
850 | } |
||
851 | |||
852 | /// Maps a property to a field, if it exists. |
||
853 | /// If the property exists and is invalid, reports an error. |
||
854 | /// If the property does not exist, Out is unchanged. |
||
855 | template <typename T> bool mapOptional(StringLiteral Prop, T &Out) { |
||
856 | assert(*this && "Must check this is an object before calling map()"); |
||
857 | if (const Value *E = O->get(Prop)) |
||
858 | return fromJSON(*E, Out, P.field(Prop)); |
||
859 | return true; |
||
860 | } |
||
861 | |||
862 | private: |
||
863 | const Object *O; |
||
864 | Path P; |
||
865 | }; |
||
866 | |||
867 | /// Parses the provided JSON source, or returns a ParseError. |
||
868 | /// The returned Value is self-contained and owns its strings (they do not refer |
||
869 | /// to the original source). |
||
870 | llvm::Expected<Value> parse(llvm::StringRef JSON); |
||
871 | |||
872 | class ParseError : public llvm::ErrorInfo<ParseError> { |
||
873 | const char *Msg; |
||
874 | unsigned Line, Column, Offset; |
||
875 | |||
876 | public: |
||
877 | static char ID; |
||
878 | ParseError(const char *Msg, unsigned Line, unsigned Column, unsigned Offset) |
||
879 | : Msg(Msg), Line(Line), Column(Column), Offset(Offset) {} |
||
880 | void log(llvm::raw_ostream &OS) const override { |
||
881 | OS << llvm::formatv("[{0}:{1}, byte={2}]: {3}", Line, Column, Offset, Msg); |
||
882 | } |
||
883 | std::error_code convertToErrorCode() const override { |
||
884 | return llvm::inconvertibleErrorCode(); |
||
885 | } |
||
886 | }; |
||
887 | |||
888 | /// Version of parse() that converts the parsed value to the type T. |
||
889 | /// RootName describes the root object and is used in error messages. |
||
890 | template <typename T> |
||
891 | Expected<T> parse(const llvm::StringRef &JSON, const char *RootName = "") { |
||
892 | auto V = parse(JSON); |
||
893 | if (!V) |
||
894 | return V.takeError(); |
||
895 | Path::Root R(RootName); |
||
896 | T Result; |
||
897 | if (fromJSON(*V, Result, R)) |
||
898 | return std::move(Result); |
||
899 | return R.getError(); |
||
900 | } |
||
901 | |||
902 | /// json::OStream allows writing well-formed JSON without materializing |
||
903 | /// all structures as json::Value ahead of time. |
||
904 | /// It's faster, lower-level, and less safe than OS << json::Value. |
||
905 | /// It also allows emitting more constructs, such as comments. |
||
906 | /// |
||
907 | /// Only one "top-level" object can be written to a stream. |
||
908 | /// Simplest usage involves passing lambdas (Blocks) to fill in containers: |
||
909 | /// |
||
910 | /// json::OStream J(OS); |
||
911 | /// J.array([&]{ |
||
912 | /// for (const Event &E : Events) |
||
913 | /// J.object([&] { |
||
914 | /// J.attribute("timestamp", int64_t(E.Time)); |
||
915 | /// J.attributeArray("participants", [&] { |
||
916 | /// for (const Participant &P : E.Participants) |
||
917 | /// J.value(P.toString()); |
||
918 | /// }); |
||
919 | /// }); |
||
920 | /// }); |
||
921 | /// |
||
922 | /// This would produce JSON like: |
||
923 | /// |
||
924 | /// [ |
||
925 | /// { |
||
926 | /// "timestamp": 19287398741, |
||
927 | /// "participants": [ |
||
928 | /// "King Kong", |
||
929 | /// "Miley Cyrus", |
||
930 | /// "Cleopatra" |
||
931 | /// ] |
||
932 | /// }, |
||
933 | /// ... |
||
934 | /// ] |
||
935 | /// |
||
936 | /// The lower level begin/end methods (arrayBegin()) are more flexible but |
||
937 | /// care must be taken to pair them correctly: |
||
938 | /// |
||
939 | /// json::OStream J(OS); |
||
940 | // J.arrayBegin(); |
||
941 | /// for (const Event &E : Events) { |
||
942 | /// J.objectBegin(); |
||
943 | /// J.attribute("timestamp", int64_t(E.Time)); |
||
944 | /// J.attributeBegin("participants"); |
||
945 | /// for (const Participant &P : E.Participants) |
||
946 | /// J.value(P.toString()); |
||
947 | /// J.attributeEnd(); |
||
948 | /// J.objectEnd(); |
||
949 | /// } |
||
950 | /// J.arrayEnd(); |
||
951 | /// |
||
952 | /// If the call sequence isn't valid JSON, asserts will fire in debug mode. |
||
953 | /// This can be mismatched begin()/end() pairs, trying to emit attributes inside |
||
954 | /// an array, and so on. |
||
955 | /// With asserts disabled, this is undefined behavior. |
||
956 | class OStream { |
||
957 | public: |
||
958 | using Block = llvm::function_ref<void()>; |
||
959 | // If IndentSize is nonzero, output is pretty-printed. |
||
960 | explicit OStream(llvm::raw_ostream &OS, unsigned IndentSize = 0) |
||
961 | : OS(OS), IndentSize(IndentSize) { |
||
962 | Stack.emplace_back(); |
||
963 | } |
||
964 | ~OStream() { |
||
965 | assert(Stack.size() == 1 && "Unmatched begin()/end()"); |
||
966 | assert(Stack.back().Ctx == Singleton); |
||
967 | assert(Stack.back().HasValue && "Did not write top-level value"); |
||
968 | } |
||
969 | |||
970 | /// Flushes the underlying ostream. OStream does not buffer internally. |
||
971 | void flush() { OS.flush(); } |
||
972 | |||
973 | // High level functions to output a value. |
||
974 | // Valid at top-level (exactly once), in an attribute value (exactly once), |
||
975 | // or in an array (any number of times). |
||
976 | |||
977 | /// Emit a self-contained value (number, string, vector<string> etc). |
||
978 | void value(const Value &V); |
||
979 | /// Emit an array whose elements are emitted in the provided Block. |
||
980 | void array(Block Contents) { |
||
981 | arrayBegin(); |
||
982 | Contents(); |
||
983 | arrayEnd(); |
||
984 | } |
||
985 | /// Emit an object whose elements are emitted in the provided Block. |
||
986 | void object(Block Contents) { |
||
987 | objectBegin(); |
||
988 | Contents(); |
||
989 | objectEnd(); |
||
990 | } |
||
991 | /// Emit an externally-serialized value. |
||
992 | /// The caller must write exactly one valid JSON value to the provided stream. |
||
993 | /// No validation or formatting of this value occurs. |
||
994 | void rawValue(llvm::function_ref<void(raw_ostream &)> Contents) { |
||
995 | rawValueBegin(); |
||
996 | Contents(OS); |
||
997 | rawValueEnd(); |
||
998 | } |
||
999 | void rawValue(llvm::StringRef Contents) { |
||
1000 | rawValue([&](raw_ostream &OS) { OS << Contents; }); |
||
1001 | } |
||
1002 | /// Emit a JavaScript comment associated with the next printed value. |
||
1003 | /// The string must be valid until the next attribute or value is emitted. |
||
1004 | /// Comments are not part of standard JSON, and many parsers reject them! |
||
1005 | void comment(llvm::StringRef); |
||
1006 | |||
1007 | // High level functions to output object attributes. |
||
1008 | // Valid only within an object (any number of times). |
||
1009 | |||
1010 | /// Emit an attribute whose value is self-contained (number, vector<int> etc). |
||
1011 | void attribute(llvm::StringRef Key, const Value& Contents) { |
||
1012 | attributeImpl(Key, [&] { value(Contents); }); |
||
1013 | } |
||
1014 | /// Emit an attribute whose value is an array with elements from the Block. |
||
1015 | void attributeArray(llvm::StringRef Key, Block Contents) { |
||
1016 | attributeImpl(Key, [&] { array(Contents); }); |
||
1017 | } |
||
1018 | /// Emit an attribute whose value is an object with attributes from the Block. |
||
1019 | void attributeObject(llvm::StringRef Key, Block Contents) { |
||
1020 | attributeImpl(Key, [&] { object(Contents); }); |
||
1021 | } |
||
1022 | |||
1023 | // Low-level begin/end functions to output arrays, objects, and attributes. |
||
1024 | // Must be correctly paired. Allowed contexts are as above. |
||
1025 | |||
1026 | void arrayBegin(); |
||
1027 | void arrayEnd(); |
||
1028 | void objectBegin(); |
||
1029 | void objectEnd(); |
||
1030 | void attributeBegin(llvm::StringRef Key); |
||
1031 | void attributeEnd(); |
||
1032 | raw_ostream &rawValueBegin(); |
||
1033 | void rawValueEnd(); |
||
1034 | |||
1035 | private: |
||
1036 | void attributeImpl(llvm::StringRef Key, Block Contents) { |
||
1037 | attributeBegin(Key); |
||
1038 | Contents(); |
||
1039 | attributeEnd(); |
||
1040 | } |
||
1041 | |||
1042 | void valueBegin(); |
||
1043 | void flushComment(); |
||
1044 | void newline(); |
||
1045 | |||
1046 | enum Context { |
||
1047 | Singleton, // Top level, or object attribute. |
||
1048 | Array, |
||
1049 | Object, |
||
1050 | RawValue, // External code writing a value to OS directly. |
||
1051 | }; |
||
1052 | struct State { |
||
1053 | Context Ctx = Singleton; |
||
1054 | bool HasValue = false; |
||
1055 | }; |
||
1056 | llvm::SmallVector<State, 16> Stack; // Never empty. |
||
1057 | llvm::StringRef PendingComment; |
||
1058 | llvm::raw_ostream &OS; |
||
1059 | unsigned IndentSize; |
||
1060 | unsigned Indent = 0; |
||
1061 | }; |
||
1062 | |||
1063 | /// Serializes this Value to JSON, writing it to the provided stream. |
||
1064 | /// The formatting is compact (no extra whitespace) and deterministic. |
||
1065 | /// For pretty-printing, use the formatv() format_provider below. |
||
1066 | inline llvm::raw_ostream &operator<<(llvm::raw_ostream &OS, const Value &V) { |
||
1067 | OStream(OS).value(V); |
||
1068 | return OS; |
||
1069 | } |
||
1070 | } // namespace json |
||
1071 | |||
1072 | /// Allow printing json::Value with formatv(). |
||
1073 | /// The default style is basic/compact formatting, like operator<<. |
||
1074 | /// A format string like formatv("{0:2}", Value) pretty-prints with indent 2. |
||
1075 | template <> struct format_provider<llvm::json::Value> { |
||
1076 | static void format(const llvm::json::Value &, raw_ostream &, StringRef); |
||
1077 | }; |
||
1078 | } // namespace llvm |
||
1079 | |||
1080 | #endif |