#include "gtest/gtest.h"
#include "tkc/buffer.h"
#include "ubjson/ubjson_writer.h"
#include "conf_io/conf_ini.h"
#include "conf_io/conf_json.h"
#include "conf_io/conf_ubjson.h"
#include "conf_io/conf_node.h"

TEST(ConfNode, basic) {
  value_t v;
  conf_doc_t* doc = conf_doc_create(100);
  conf_node_t* node = NULL;
  doc->root = conf_doc_create_node(doc, "root");

  conf_doc_append_child(doc, doc->root, conf_doc_create_node(doc, "1"));
  conf_doc_append_child(doc, doc->root, conf_doc_create_node(doc, "2"));
  conf_doc_append_child(doc, doc->root, conf_doc_create_node(doc, "3"));
  conf_doc_append_child(doc, doc->root, conf_doc_create_node(doc, "4"));

  node = conf_node_find_child(doc->root, "1");
  ASSERT_STREQ(conf_node_get_name(node), "1");
  value_set_int(&v, 1);
  ASSERT_EQ(conf_node_set_value(node, &v), RET_OK);
  value_set_int(&v, 0);
  ASSERT_EQ(conf_node_get_child_value(doc->root, "1", &v), RET_OK);
  ASSERT_EQ(value_int(&v), 1);
  ASSERT_EQ(conf_node_get_child_value_by_index(doc->root, 0, &v), RET_OK);
  ASSERT_EQ(value_int(&v), 1);

  node = conf_node_find_child(doc->root, "2");
  ASSERT_STREQ(conf_node_get_name(node), "2");
  value_set_int(&v, 2);
  ASSERT_EQ(conf_node_set_value(node, &v), RET_OK);
  value_set_int(&v, 0);
  ASSERT_EQ(conf_node_get_child_value(doc->root, "2", &v), RET_OK);
  ASSERT_EQ(value_int(&v), 2);
  ASSERT_EQ(conf_node_get_child_value_by_index(doc->root, 1, &v), RET_OK);
  ASSERT_EQ(value_int(&v), 2);

  node = conf_node_find_child(doc->root, "3");
  ASSERT_STREQ(conf_node_get_name(node), "3");
  value_set_int(&v, 3);
  ASSERT_EQ(conf_node_set_value(node, &v), RET_OK);
  value_set_int(&v, 0);
  ASSERT_EQ(conf_node_get_child_value(doc->root, "3", &v), RET_OK);
  ASSERT_EQ(value_int(&v), 3);
  ASSERT_EQ(conf_node_get_child_value_by_index(doc->root, 2, &v), RET_OK);
  ASSERT_EQ(value_int(&v), 3);

  node = conf_node_find_child(doc->root, "4");
  ASSERT_STREQ(conf_node_get_name(node), "4");
  value_set_int(&v, 4);
  ASSERT_EQ(conf_node_set_value(node, &v), RET_OK);
  value_set_int(&v, 0);
  ASSERT_EQ(conf_node_get_child_value(doc->root, "4", &v), RET_OK);
  ASSERT_EQ(value_int(&v), 4);
  ASSERT_EQ(conf_node_get_child_value_by_index(doc->root, 3, &v), RET_OK);
  ASSERT_EQ(value_int(&v), 4);
  
  ASSERT_EQ(conf_node_get_child_value(doc->root, "100", &v), RET_NOT_FOUND);

  ASSERT_EQ(conf_doc_remove_child_by_name(doc, doc->root, "2"), RET_OK);
  ASSERT_NE(conf_doc_remove_child_by_name(doc, doc->root, "2"), RET_OK);
  ASSERT_EQ(conf_doc_remove_child_by_name(doc, doc->root, "1"), RET_OK);
  ASSERT_NE(conf_doc_remove_child_by_name(doc, doc->root, "1"), RET_OK);
  ASSERT_EQ(conf_doc_remove_child_by_name(doc, doc->root, "4"), RET_OK);
  ASSERT_NE(conf_doc_remove_child_by_name(doc, doc->root, "4"), RET_OK);

  node = conf_node_find_child(doc->root, "3");
  ASSERT_EQ(conf_doc_remove_child(doc, doc->root, node), RET_OK);
  ASSERT_NE(conf_doc_remove_child_by_name(doc, doc->root, "3"), RET_OK);

  conf_doc_destroy(doc);
}

TEST(ConfNode, set_get) {
  value_t v;
  conf_doc_t* doc = conf_doc_create(100);

  ASSERT_EQ(conf_doc_set(doc, "person.[0].name", value_set_str(&v, "jim")), RET_OK);
  ASSERT_EQ(conf_doc_set(doc, "person.[1].name", value_set_str(&v, "tom")), RET_OK);
  ASSERT_EQ(conf_doc_set(doc, "person.[2].name", value_set_str(&v, "mike")), RET_OK);

  ASSERT_EQ(conf_doc_get(doc, "person.[0].name", &v), RET_OK);
  ASSERT_STREQ(value_str(&v), "jim");

  ASSERT_EQ(conf_doc_get(doc, "person.[1].name", &v), RET_OK);
  ASSERT_STREQ(value_str(&v), "tom");

  ASSERT_EQ(conf_doc_get(doc, "person.[2].name", &v), RET_OK);
  ASSERT_STREQ(value_str(&v), "mike");

  ASSERT_NE(conf_doc_set(doc, "person.[20].name", value_set_str(&v, "anny")), RET_OK);

  conf_doc_destroy(doc);
}

TEST(ConfNode, set_get_str) {
  value_t v;
  str_t str;
  conf_doc_t* doc = conf_doc_create(100);

  str_init(&str, 0);
  ASSERT_EQ(conf_doc_set(doc, "names.[0]", value_set_str(&v, "jim")), RET_OK);
  ASSERT_EQ(conf_doc_set(doc, "names.[1]", value_set_str(&v, "tom")), RET_OK);
  ASSERT_EQ(conf_doc_set(doc, "names.[2]", value_set_str(&v, "anny")), RET_OK);

  ASSERT_EQ(conf_doc_save_json(doc, &str), RET_OK);
  ASSERT_STREQ(
      str.str,
      "{\n    \"names\" : [\n        \"jim\",\n        \"tom\",\n        \"anny\"\n    ]\n}");

  conf_doc_destroy(doc);
  str_reset(&str);
}

TEST(ConfNode, set_get_wstr) {
  value_t v;
  conf_doc_t* doc = conf_doc_create(100);

  ASSERT_EQ(conf_doc_set(doc, "names.[0]", value_set_wstr(&v, L"jim")), RET_OK);
  ASSERT_EQ(conf_doc_set(doc, "names.[1]", value_set_wstr(&v, L"tom")), RET_OK);
  ASSERT_EQ(conf_doc_set(doc, "names.[2]", value_set_wstr(&v, L"anny")), RET_OK);
  
  ASSERT_EQ(conf_doc_get(doc, "names.[0]", &v), RET_OK);
  ASSERT_EQ(wcscmp(value_wstr(&v), L"jim"), 0);
  ASSERT_EQ(conf_doc_get(doc, "names.[1]", &v), RET_OK);
  ASSERT_EQ(wcscmp(value_wstr(&v), L"tom"), 0);
  ASSERT_EQ(conf_doc_get(doc, "names.[2]", &v), RET_OK);
  ASSERT_EQ(wcscmp(value_wstr(&v), L"anny"), 0);
  
  str_t str;
  str_init(&str, 0);
  ASSERT_EQ(conf_doc_save_json(doc, &str), RET_OK);
  ASSERT_STREQ(
      str.str,
      "{\n    \"names\" : [\n        \"jim\",\n        \"tom\",\n        \"anny\"\n    ]\n}");

  str_reset(&str);
  
  str_init(&str, 0);
  ASSERT_EQ(conf_doc_save_ini(doc, &str), RET_OK);
  ASSERT_STREQ(
      str.str,
      "[names]\n  [0] = jim\n  [1] = tom\n  [2] = anny\n");
  str_reset(&str);
 
  wbuffer_t wb; 
  ubjson_writer_t ub; 
  wbuffer_init_extendable(&wb);
  ubjson_writer_init(&ub, (ubjson_write_callback_t)wbuffer_write_binary, &wb);
  conf_doc_save_ubjson(doc, &ub);

  conf_doc_destroy(doc);
  doc = conf_doc_load_ubjson(wb.data, wb.cursor);
  
  ASSERT_EQ(conf_doc_get(doc, "names.[0]", &v), RET_OK);
  ASSERT_EQ(wcscmp(value_wstr(&v), L"jim"), 0);
  ASSERT_EQ(conf_doc_get(doc, "names.[1]", &v), RET_OK);
  ASSERT_EQ(wcscmp(value_wstr(&v), L"tom"), 0);
  ASSERT_EQ(conf_doc_get(doc, "names.[2]", &v), RET_OK);
  ASSERT_EQ(wcscmp(value_wstr(&v), L"anny"), 0);

  conf_doc_destroy(doc);
  wbuffer_deinit(&wb);
}