#pragma once

#include "JsonWriter.h"
#include "IndentedPrint.h"

using namespace ArduinoJson::Generator;

class PrettyJsonWriter : public JsonWriter
{
public:
    explicit PrettyJsonWriter(IndentedPrint* sink)
        : JsonWriter(sink), _indenter(sink)
    {
    }

    virtual void beginArray()
    {
        _length += _sink->write('[');
        indent();
    }

    virtual void endArray()
    {
        unindent();
        _length += _sink->write(']');
    }

    virtual void writeColon()
    {
        _length += _sink->print(": ");
    }

    virtual void writeComma()
    {
        _length += _sink->write(',');
        _length += _indenter->println();
    }

    virtual void beginObject()
    {
        _length += _sink->write('{');
        indent();
    }

    virtual void endObject()
    {
        unindent();
        _length += _sink->write('}');
    }

private:
    IndentedPrint* _indenter;

    void indent()
    {
        _indenter->indent();
        _length += _indenter->println();
    }

    void unindent()
    {
        _length += _indenter->println();
        _indenter->unindent();
    }
};