// Copyright Benoit Blanchon 2014-2017
// MIT License
//
// Arduino JSON library
// https://github.com/bblanchon/ArduinoJson
// If you like this project, please add a star!

#pragma once

#include "Data/JsonBufferAllocated.hpp"
#include "Data/List.hpp"
#include "Data/ReferenceType.hpp"
#include "Data/ValueSetter.hpp"
#include "JsonPair.hpp"
#include "Serialization/JsonPrintable.hpp"
#include "StringTraits/StringTraits.hpp"
#include "TypeTraits/ConstRefOrConstPtr.hpp"
#include "TypeTraits/EnableIf.hpp"
#include "TypeTraits/IsFloatingPoint.hpp"
#include "TypeTraits/IsSame.hpp"

// Returns the size (in bytes) of an object with n elements.
// Can be very handy to determine the size of a StaticJsonBuffer.
#define JSON_OBJECT_SIZE(NUMBER_OF_ELEMENTS) \
  (sizeof(JsonObject) + (NUMBER_OF_ELEMENTS) * sizeof(JsonObject::node_type))

namespace ArduinoJson {

// Forward declarations
class JsonArray;
class JsonBuffer;

// A dictionary of JsonVariant indexed by string (char*)
//
// The constructor is private, instances must be created via
// JsonBuffer::createObject() or JsonBuffer::parseObject().
// A JsonObject can be serialized to a JSON string via JsonObject::printTo().
// It can also be deserialized from a JSON string via JsonBuffer::parseObject().
class JsonObject : public Internals::JsonPrintable<JsonObject>,
                   public Internals::ReferenceType,
                   public Internals::List<JsonPair>,
                   public Internals::JsonBufferAllocated {
 public:
  // Create an empty JsonArray attached to the specified JsonBuffer.
  // You should not use this constructor directly.
  // Instead, use JsonBuffer::createObject() or JsonBuffer.parseObject().
  explicit JsonObject(JsonBuffer* buffer) : Internals::List<JsonPair>(buffer) {}

  // Gets or sets the value associated with the specified key.
  template <typename TString>
  JsonObjectSubscript<TString> operator[](const TString& key);

  // Gets the value associated with the specified key.
  template <typename TString>
  JsonVariant operator[](const TString& key) const {
    return get<JsonVariant>(key);
  }

  // Sets the specified key with the specified value.
  // bool set(TKey key, bool value);
  // bool set(TKey key, char value);
  // bool set(TKey key, long value);
  // bool set(TKey key, int value);
  // bool set(TKey key, short value);
  // bool set(TKey key, float value);
  // bool set(TKey key, double value);
  // bool set(TKey key, const char* value);
  // bool set(TKey key, RawJson value);
  // bool set(Key, String&);
  // bool set(Key, JsonArray&);
  // bool set(Key, JsonObject&);
  // bool set(Key, JsonVariant&);
  template <typename TValue, typename TString>
  bool set(const TString& key, const TValue& value) {
    // reduce the number of template function instanciation to reduce code size
    return setNodeAt<typename TypeTraits::ConstRefOrConstPtr<TString>::type,
                     typename TypeTraits::ConstRefOrConstPtr<TValue>::type>(
        key, value);
  }
  // bool set(Key, float value, uint8_t decimals);
  // bool set(Key, double value, uint8_t decimals);
  template <typename TValue, typename TString>
  typename TypeTraits::EnableIf<TypeTraits::IsFloatingPoint<TValue>::value,
                                bool>::type
  set(const TString& key, TValue value, uint8_t decimals) {
    return set(key, JsonVariant(value, decimals));
  }

  // Gets the value associated with the specified key.
  template <typename TValue, typename TString>
  typename Internals::JsonVariantAs<TValue>::type get(
      const TString& key) const {
    node_type* node = getNodeAt(key);
    return node ? node->content.value.as<TValue>()
                : Internals::JsonVariantDefault<TValue>::get();
  }

  // Checks the type of the value associated with the specified key.
  template <typename TValue, typename TString>
  bool is(const TString& key) const {
    node_type* node = getNodeAt(key);
    return node ? node->content.value.is<TValue>() : false;
  }

  // Creates and adds a JsonArray.
  // This is a shortcut for JsonBuffer::createArray() and JsonObject::add().
  template <typename TString>
  JsonArray& createNestedArray(const TString& key);

  // Creates and adds a JsonObject.
  // This is a shortcut for JsonBuffer::createObject() and JsonObject::add().
  template <typename TString>
  JsonObject& createNestedObject(const TString& key);

  // Tells weither the specified key is present and associated with a value.
  template <typename TString>
  bool containsKey(const TString& key) const {
    return getNodeAt(key) != NULL;
  }

  // Removes the specified key and the associated value.
  template <typename TString>
  void remove(const TString& key) {
    removeNode(getNodeAt(key));
  }

  // Returns a reference an invalid JsonObject.
  // This object is meant to replace a NULL pointer.
  // This is used when memory allocation or JSON parsing fail.
  static JsonObject& invalid() {
    static JsonObject instance(NULL);
    return instance;
  }

 private:
  // Returns the list node that matches the specified key.
  template <typename TString>
  node_type* getNodeAt(const TString& key) const {
    // reduce the number of template function instanciation to reduce code size
    return getNodeAtImpl<
        typename TypeTraits::ConstRefOrConstPtr<TString>::type>(key);
  }

  template <typename TStringRef>
  node_type* getNodeAtImpl(TStringRef key) const {
    for (node_type* node = _firstNode; node; node = node->next) {
      if (Internals::StringFuncs<TStringRef>::equals(key, node->content.key))
        return node;
    }
    return NULL;
  }

  template <typename TStringRef, typename TValueRef>
  bool setNodeAt(TStringRef key, TValueRef value) {
    node_type* node = getNodeAtImpl<TStringRef>(key);
    if (!node) {
      node = addNewNode();
      if (!node) return false;

      bool key_ok = Internals::ValueSetter<TStringRef>::set(
          _buffer, node->content.key, key);
      if (!key_ok) return false;
    }
    return Internals::ValueSetter<TValueRef>::set(_buffer, node->content.value,
                                                  value);
  }
};

namespace Internals {
template <>
struct JsonVariantDefault<JsonObject> {
  static JsonObject& get() {
    return JsonObject::invalid();
  }
};
}
}