// Copyright Benoit Blanchon 2014
// MIT License
//
// Arduino JSON library
// https://github.com/bblanchon/ArduinoJson

#include "ArduinoJson/JsonObject.hpp"

#include <string.h>  // for strcmp

#include "ArduinoJson/JsonBuffer.hpp"
#include "ArduinoJson/JsonValue.hpp"
#include "ArduinoJson/Internals/JsonNode.hpp"
#include "ArduinoJson/Internals/StringBuilder.hpp"

using namespace ArduinoJson;
using namespace ArduinoJson::Internals;

JsonValue JsonObject::operator[](char const *key) {
  JsonNode *node = getOrCreateNodeAt(key);
  return JsonValue(node);
}

void JsonObject::remove(char const *key) {
  for (JsonObjectIterator it = begin(); it != end(); ++it) {
    if (!strcmp(it->key(), key)) {
      removeChild(it->_node);
    }
  }
}

JsonArray JsonObject::createNestedArray(char const *key) {
  JsonNode *node = getOrCreateNodeAt(key);

  if (node) node->setAsArray(_node->getContainerBuffer());

  return JsonArray(node);
}

JsonObject JsonObject::createNestedObject(char const *key) {
  JsonNode *node = getOrCreateNodeAt(key);

  if (node) node->setAsObject(_node->getContainerBuffer());

  return JsonObject(node);
}

JsonNode *JsonObject::getOrCreateNodeAt(const char *key) {
  for (JsonObjectIterator it = begin(); it != end(); ++it) {
    if (!strcmp(it->key(), key)) return it->value()._node;
  }

  JsonNode *newValueNode = createNode();
  if (!newValueNode) return 0;

  JsonNode *newKeyNode = createNode();
  if (!newKeyNode) return 0;

  newKeyNode->setAsObjectKeyValue(key, newValueNode);

  addChild(newKeyNode);

  return newValueNode;
}