package com.ishchuk.antlr.json.model;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

public class JsonObject extends JsonElement {
    private Map<String, JsonElement> map;

    public Map<String, JsonElement> getMap() {
        return map;
    }

    public void setMap(Map<String, JsonElement> map) {
        this.map = map;
    }

    public JsonObject() {
        this.map = new HashMap<>();
    }

    public String toString() {
        StringBuilder builder = new StringBuilder();
        builder.append("{");
        this.map.keySet().forEach(key -> {
            builder.append(key)
                    .append(":")
                    .append(map.get(key).toString())
                    .append(",");
        });
        int lastInd = builder.lastIndexOf(",");
        if (lastInd > 0) {
            builder.deleteCharAt(lastInd);
        }
        builder.append("}");

        return builder.toString();
    }

    @Override
    public String toYml(String delay) {
        StringBuilder builder = new StringBuilder();
        this.map.keySet().forEach(key -> {
            builder.append(delay).append(key).append(": ");
            if (this.map.get(key) instanceof JsonPrimitive) {
                builder.append(this.map.get(key).toString()).append("\n");
            } else {
                builder.append("\n").append(this.map.get(key).toYml(delay + "  "));
            }
        });
        return builder.toString();
    }
}