#include "JsonContainer.h"

#include "JsonBuffer.h"
#include "Internals/StringBuilder.h"
#include "Internals/CompactJsonWriter.h"
#include "Internals/PrettyJsonWriter.h"

using namespace ArduinoJson::Internals;

size_t JsonContainer::printTo(char* buffer, size_t bufferSize) const
{
    StringBuilder sb(buffer, bufferSize);
    return printTo(sb);
}

size_t JsonContainer::printTo(Print& p) const
{
    CompactJsonWriter writer(p);
    _node->writeTo(writer);
    return writer.bytesWritten();
}

size_t JsonContainer::prettyPrintTo(char* buffer, size_t bufferSize) const
{
    StringBuilder sb(buffer, bufferSize);
    return prettyPrintTo(sb);
}

size_t JsonContainer::prettyPrintTo(IndentedPrint& p) const
{
    PrettyJsonWriter writer(p);
    _node->writeTo(writer);
    return writer.bytesWritten();
}

JsonNode* JsonContainer::createNode(JsonNodeType type)
{
    if (!_node) return 0;

    JsonBuffer* buffer = _node->getContainerBuffer();
    if (!buffer) return 0;

    return buffer->createNode(type);
}

bool JsonContainer::checkNodeType(JsonNodeType expectedType)
{
    return _node && _node->type == expectedType;
}

bool JsonContainer::operator==(const JsonContainer & other) const
{
    return _node->getContainerChild() == other._node->getContainerChild();
}

void JsonContainer::addChild(JsonNode* childToAdd)
{
    if (_node)
        _node->addChildToContainer(childToAdd);
}

void JsonContainer::removeChild(JsonNode* childToRemove)
{
    if (_node)
        _node->removeChildFromContainer(childToRemove);
}

size_t JsonContainer::size() const
{
    int size = 0;

    for (JsonNodeIterator it = beginChildren(); it != endChildren(); ++it)
    {
        size++;
    }

    return size;
}