/*
* Arduino JSON library
* Benoit Blanchon 2014 - MIT License
*/

#include "PrettyPrintDecorator.h"

size_t PrettyPrintDecorator::write(uint8_t c)
{
    size_t n = inString ? handleStringChar(c) : handleMarkupChar(c);    
    previousChar = c;
    return n;
}

size_t PrettyPrintDecorator::handleStringChar(uint8_t c)
{
    bool isQuote = c == '"' && previousChar != '\\';

    return isQuote ? writeQuote() : writeNormalChar(c);
}

size_t PrettyPrintDecorator::handleMarkupChar(uint8_t c)
{
    switch (c)
    {
    case '{':
    case '[':
        return writeOpening(c);

    case '}':
    case ']':
        return writeClosing(c);

    case ',':
        return writeComma();

    case ':':
        return writeColumn();

    case '"':
        return writeQuote();

    default:
        return writeNormalChar(c); // <- should not happen anyway
    }
}

size_t PrettyPrintDecorator::writeNormalChar(uint8_t c)
{
    bool inEmptyBlock = !inString && (previousChar == '{' || previousChar == '[');

    return inEmptyBlock ? writeln() + sink.write(c) : sink.write(c);
}

size_t PrettyPrintDecorator::writeColumn()
{
    return sink.write(':') + sink.write(' ');
}

size_t PrettyPrintDecorator::writeComma()
{
    return sink.write(',') + writeln();
}

size_t PrettyPrintDecorator::writeOpening(uint8_t c)
{
    bool inEmptyBlock = previousChar == '{' || previousChar == '[';

    size_t n = inEmptyBlock ? writeln() + sink.write(c) : sink.write(c);

    indent++;

    return n;
}

size_t PrettyPrintDecorator::writeQuote()
{
    size_t n = writeNormalChar('"');
    inString = !inString;
    return n;
}

size_t PrettyPrintDecorator::writeClosing(uint8_t c)
{
    bool inEmptyBlock = previousChar == '{' || previousChar == '[';

    indent--;

    return inEmptyBlock ? sink.write(c) : writeln() + sink.write(c);
}

size_t PrettyPrintDecorator::writeln()
{
    size_t n = sink.write('\n');

    for (int i = 0; i < indent; i++)
        n += sink.write(' ');

    return n;
}