aboutsummaryrefslogtreecommitdiffstats
path: root/yql/essentials/sql/v1/highlight/sql_highlight_json.cpp
blob: e4af680cdb9b07ad0818964c28cb162775231e5b (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