summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibGUI/GMLFormatter.cpp
blob: 2391d4a4c4a4e26a64760d2f4fc8bf8c66ce9dde (plain)
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
/*
 * Copyright (c) 2021, Linus Groh <linusg@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <AK/JsonObject.h>
#include <AK/JsonValue.h>
#include <AK/StringBuilder.h>
#include <LibGUI/GMLFormatter.h>
#include <LibGUI/GMLParser.h>

namespace GUI {

static String format_gml_object(const JsonObject& node, size_t indentation = 0, bool is_inline = false)
{
    StringBuilder builder;

    auto indent = [&builder](size_t indentation) {
        for (size_t i = 0; i < indentation; ++i)
            builder.append("    ");
    };

    struct Property {
        String key;
        JsonValue value;
    };
    Vector<Property> properties;
    node.for_each_member([&](auto& key, auto& value) {
        if (key != "class" && key != "layout" && key != "children")
            properties.append({ key, value });
        return IterationDecision::Continue;
    });

    if (!is_inline)
        indent(indentation);
    builder.append('@');
    builder.append(node.get("class").as_string());
    builder.append(" {\n");

    for (auto& property : properties) {
        indent(indentation + 1);
        builder.append(property.key);
        builder.append(": ");
        if (property.value.is_array()) {
            // custom array serialization as AK's doesn't pretty-print
            // objects and arrays (we only care about arrays (for now))
            builder.append("[");
            auto first = true;
            property.value.as_array().for_each([&](auto& value) {
                if (!first)
                    builder.append(", ");
                first = false;
                value.serialize(builder);
            });
            builder.append("]");
        } else {
            property.value.serialize(builder);
        }
        builder.append("\n");
    }

    if (node.has("layout")) {
        auto layout = node.get("layout").as_object();
        if (!properties.is_empty())
            builder.append("\n");
        indent(indentation + 1);
        builder.append("layout: ");
        builder.append(format_gml_object(move(layout), indentation + 1, true));
    }

    if (node.has("children")) {
        auto children = node.get("children").as_array();
        auto first = properties.is_empty() && !node.has("layout");
        children.for_each([&](auto& value) {
            if (!first)
                builder.append("\n");
            first = false;
            builder.append(format_gml_object(value.as_object(), indentation + 1));
        });
    }

    indent(indentation);
    builder.append("}\n");

    return builder.to_string();
}

String format_gml(const StringView& string)
{
    // FIXME: Preserve comments somehow, they're not contained
    // in the JSON object returned by parse_gml()
    auto ast = parse_gml(string);
    if (ast.is_null())
        return {};
    VERIFY(ast.is_object());
    return format_gml_object(ast.as_object());
}

}