#include <string.h>

#include <event2/buffer.h>

#include "transmission.h"
#include "rpcimpl.h"
#include "utils.h"
#include "variant.h"

#include "libtransmission-test.h"

static int
test_list (void)
{
  size_t len;
  int64_t i;
  const char * str;
  tr_variant top;

  tr_rpc_parse_list_str (&top, "12", -1);
  check (tr_variantIsInt (&top));
  check (tr_variantGetInt (&top, &i));
  check_int_eq (12, i);
  tr_variantFree (&top);

  tr_rpc_parse_list_str (&top, "12", 1);
  check (tr_variantIsInt (&top));
  check (tr_variantGetInt (&top, &i));
  check_int_eq (1, i);
  tr_variantFree (&top);

  tr_rpc_parse_list_str (&top, "6,7", -1);
  check (tr_variantIsList (&top));
  check (tr_variantListSize (&top) == 2);
  check (tr_variantGetInt (tr_variantListChild (&top, 0), &i));
  check_int_eq (6, i);
  check (tr_variantGetInt (tr_variantListChild (&top, 1), &i));
  check_int_eq (7, i);
  tr_variantFree (&top);

  tr_rpc_parse_list_str (&top, "asdf", -1);
  check (tr_variantIsString (&top));
  check (tr_variantGetStr (&top, &str, &len));
  check_int_eq (4, len);
  check_streq ("asdf", str);
  tr_variantFree (&top);

  tr_rpc_parse_list_str (&top, "1,3-5", -1);
  check (tr_variantIsList (&top));
  check (tr_variantListSize (&top) == 4);
  check (tr_variantGetInt (tr_variantListChild (&top, 0), &i));
  check_int_eq (1, i);
  check (tr_variantGetInt (tr_variantListChild (&top, 1), &i));
  check_int_eq (3, i);
  check (tr_variantGetInt (tr_variantListChild (&top, 2), &i));
  check_int_eq (4, i);
  check (tr_variantGetInt (tr_variantListChild (&top, 3), &i));
  check_int_eq (5, i);
  tr_variantFree (&top);

  return 0;
}

/***
****
***/

static void
rpc_response_func (tr_session      * session    UNUSED,
                   struct evbuffer * response,
                   void            * setme)
{
  tr_variantFromBuf (setme, TR_VARIANT_FMT_JSON, evbuffer_pullup(response,-1), evbuffer_get_length(response), NULL, NULL);
}

static int
test_session_get_and_set (void)
{
  const char * json;
  tr_variant response;
  tr_variant * args;
  tr_torrent * tor = libtransmission_test_zero_torrent_init ();
  check (tor != NULL);

  json = "{\"method\":\"session-get\"}";
  tr_rpc_request_exec_json (session, json, strlen(json), rpc_response_func, &response);

  check (tr_variantIsDict(&response));
  check (tr_variantIsDict(&response));
  check (tr_variantDictFindDict (&response, TR_KEY_arguments, &args));
  check (tr_variantDictFind (args, TR_KEY_alt_speed_down) != NULL);
  check (tr_variantDictFind (args, TR_KEY_alt_speed_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_alt_speed_time_begin) != NULL);
  check (tr_variantDictFind (args, TR_KEY_alt_speed_time_day) != NULL);
  check (tr_variantDictFind (args, TR_KEY_alt_speed_time_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_alt_speed_time_end) != NULL);
  check (tr_variantDictFind (args, TR_KEY_alt_speed_up) != NULL);
  check (tr_variantDictFind (args, TR_KEY_blocklist_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_blocklist_size) != NULL);
  check (tr_variantDictFind (args, TR_KEY_blocklist_url) != NULL);
  check (tr_variantDictFind (args, TR_KEY_cache_size_mb) != NULL);
  check (tr_variantDictFind (args, TR_KEY_config_dir) != NULL);
  check (tr_variantDictFind (args, TR_KEY_dht_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_download_dir) != NULL);
  check (tr_variantDictFind (args, TR_KEY_download_dir_free_space) != NULL);
  check (tr_variantDictFind (args, TR_KEY_download_queue_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_download_queue_size) != NULL);
  check (tr_variantDictFind (args, TR_KEY_encryption) != NULL);
  check (tr_variantDictFind (args, TR_KEY_idle_seeding_limit) != NULL);
  check (tr_variantDictFind (args, TR_KEY_idle_seeding_limit_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_incomplete_dir) != NULL);
  check (tr_variantDictFind (args, TR_KEY_incomplete_dir_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_lpd_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_peer_limit_global) != NULL);
  check (tr_variantDictFind (args, TR_KEY_peer_limit_per_torrent) != NULL);
  check (tr_variantDictFind (args, TR_KEY_peer_port) != NULL);
  check (tr_variantDictFind (args, TR_KEY_peer_port_random_on_start) != NULL);
  check (tr_variantDictFind (args, TR_KEY_pex_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_port_forwarding_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_queue_stalled_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_queue_stalled_minutes) != NULL);
  check (tr_variantDictFind (args, TR_KEY_rename_partial_files) != NULL);
  check (tr_variantDictFind (args, TR_KEY_rpc_version) != NULL);
  check (tr_variantDictFind (args, TR_KEY_rpc_version_minimum) != NULL);
  check (tr_variantDictFind (args, TR_KEY_script_torrent_done_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_script_torrent_done_filename) != NULL);
  check (tr_variantDictFind (args, TR_KEY_seed_queue_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_seed_queue_size) != NULL);
  check (tr_variantDictFind (args, TR_KEY_seedRatioLimit) != NULL);
  check (tr_variantDictFind (args, TR_KEY_seedRatioLimited) != NULL);
  check (tr_variantDictFind (args, TR_KEY_speed_limit_down) != NULL);
  check (tr_variantDictFind (args, TR_KEY_speed_limit_down_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_speed_limit_up) != NULL);
  check (tr_variantDictFind (args, TR_KEY_speed_limit_up_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_start_added_torrents) != NULL);
  check (tr_variantDictFind (args, TR_KEY_trash_original_torrent_files) != NULL);
  check (tr_variantDictFind (args, TR_KEY_units) != NULL);
  check (tr_variantDictFind (args, TR_KEY_utp_enabled) != NULL);
  check (tr_variantDictFind (args, TR_KEY_version) != NULL);
  tr_variantFree (&response);

  /* cleanup */
  tr_torrentRemove (tor, false, NULL);
  return 0;
}

/***
****
***/

int
main (void)
{
  int ret;
  const testFunc tests[] = { test_list,
                             test_session_get_and_set };

  libtransmission_test_session_init ();
  ret = runTests (tests, NUM_TESTS (tests));
  libtransmission_test_session_close ();

  return ret;
}