summaryrefslogtreecommitdiffstats
path: root/yql/essentials/sql/v1/highlight/sql_highlight_json.cpp
blob: eb41141358f593c8f6ddf72f9a5b4a8eb5fce7b8 (plain) (blame)
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
#include "sql_highlight_json.h"

#include <util/string/cast.h>

namespace NSQLHighlight {

struct {
    const char* Units = "units";
    struct {
        const char* Kind = "kind";
        const char* Patterns = "patterns";
        const char* PatternsANSI = "patterns-ansi";
    } Unit;
    struct {
        const char* Body = "body";
        const char* After = "after";
        const char* IsCaseInsensitive = "is-case-insensitive";
    } Pattern;
} JsonKey;

NJson::TJsonValue ToJson(const NSQLTranslationV1::TRegexPattern& pattern) {
    NJson::TJsonMap map;
    map[JsonKey.Pattern.Body] = pattern.Body;
    if (!pattern.After.empty()) {
        map[JsonKey.Pattern.After] = pattern.After;
    }
    if (pattern.IsCaseInsensitive) {
        map[JsonKey.Pattern.IsCaseInsensitive] = pattern.IsCaseInsensitive;
    }
    return map;
}

NJson::TJsonValue ToJson(const TVector<NSQLTranslationV1::TRegexPattern>& patterns) {
    NJson::TJsonArray array;
    for (const auto& pattern : patterns) {
        array.AppendValue(ToJson(pattern));
    }
    return array;
}

NJson::TJsonValue ToJson(const TUnit& unit) {
    NJson::TJsonMap map;
    map[JsonKey.Unit.Kind] = ToString(unit.Kind);
    if (!unit.Patterns.empty()) {
        map[JsonKey.Unit.Patterns] = ToJson(unit.Patterns);
    }
    if (!unit.PatternsANSI.Empty()) {
        map[JsonKey.Unit.PatternsANSI] = ToJson(*unit.PatternsANSI);
    }
    return map;
}

NJson::TJsonValue ToJson(const TVector<TUnit>& units) {
    NJson::TJsonArray array;
    for (const auto& unit : units) {
        array.AppendValue(ToJson(unit));
    }
    return array;
}

NJson::TJsonValue ToJson(const THighlighting& highlighting) {
    NJson::TJsonMap map;
    map[JsonKey.Units] = ToJson(highlighting.Units);
    return map;
}

} // namespace NSQLHighlight