/* Auto Tests: Save and load friends.

 */

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include "../testing/misc_tools.h"

#include "../toxcore/ccompat.h"

#include "../toxcore/crypto_core.h"

#include "../toxcore/tox.h"

#include "auto_test_support.h"

#include "check_compat.h"

struct test_data {

    uint8_t *name;

    uint8_t *status_message;

    bool received_name;

    bool received_status_message;

};

static void set_random(Tox *m, const Random *rng, bool (*setter)(Tox *, const uint8_t *, size_t, Tox_Err_Set_Info *), size_t length)

{

    VLA(uint8_t, text, length);

    for (uint32_t i = 0; i < length; ++i) {

        text[i] = random_u08(rng);

    }

    setter(m, text, length, nullptr);

}

static void alloc_string(uint8_t **to, size_t length)

{

    free(*to);

    *to = (uint8_t *)malloc(length);

    ck_assert(*to != nullptr);

}

static void set_string(uint8_t **to, const uint8_t *from, size_t length)

{

    alloc_string(to, length);

    memcpy(*to, from, length);

}

static void namechange_callback(Tox *tox, const Tox_Event_Friend_Name *event, void *user_data)

{

    //const uint32_t friend_number = tox_event_friend_name_get_friend_number(event);

    const uint8_t *name = tox_event_friend_name_get_name(event);

    const uint32_t name_length = tox_event_friend_name_get_name_length(event);

    struct test_data *to_compare = (struct test_data *)user_data;

    set_string(&to_compare->name, name, name_length);

    to_compare->received_name = true;

}

static void statuschange_callback(Tox *tox, const Tox_Event_Friend_Status_Message *event, void *user_data)

{

    //const uint32_t friend_number = tox_event_friend_status_message_get_friend_number(event);

    const uint8_t *message = tox_event_friend_status_message_get_message(event);

    const uint32_t message_length = tox_event_friend_status_message_get_message_length(event);

    struct test_data *to_compare = (struct test_data *)user_data;

    set_string(&to_compare->status_message, message, message_length);

    to_compare->received_status_message = true;

}

int main(void)

{

    setvbuf(stdout, nullptr, _IONBF, 0);

    Tox *const tox1 = tox_new_log(nullptr, nullptr, nullptr);

    Tox *const tox2 = tox_new_log(nullptr, nullptr, nullptr);

    ck_assert(tox1 != nullptr);

    ck_assert(tox2 != nullptr);

    tox_events_init(tox1);

    Tox_Dispatch* dispatch1 = tox_dispatch_new(nullptr);

    ck_assert(dispatch1 != nullptr);

    printf("bootstrapping tox2 off tox1\n");

    uint8_t dht_key[TOX_PUBLIC_KEY_SIZE];

    tox_self_get_dht_id(tox1, dht_key);

    const uint16_t dht_port = tox_self_get_udp_port(tox1, nullptr);

    tox_bootstrap(tox2, "localhost", dht_port, dht_key, nullptr);

    struct test_data to_compare = {nullptr};

    uint8_t public_key[TOX_PUBLIC_KEY_SIZE];

    tox_self_get_public_key(tox1, public_key);

    tox_friend_add_norequest(tox2, public_key, nullptr);

    tox_self_get_public_key(tox2, public_key);

    tox_friend_add_norequest(tox1, public_key, nullptr);

    uint8_t *reference_name = (uint8_t *)malloc(tox_max_name_length());

    uint8_t *reference_status = (uint8_t *)malloc(tox_max_status_message_length());

    ck_assert(reference_name != nullptr);

    ck_assert(reference_status != nullptr);

    const Random *rng = os_random();

    ck_assert(rng != nullptr);

    set_random(tox1, rng, tox_self_set_name, tox_max_name_length());

    set_random(tox2, rng, tox_self_set_name, tox_max_name_length());

    set_random(tox1, rng, tox_self_set_status_message, tox_max_status_message_length());

    set_random(tox2, rng, tox_self_set_status_message, tox_max_status_message_length());

    tox_self_get_name(tox2, reference_name);

    tox_self_get_status_message(tox2, reference_status);

    tox_events_callback_friend_name(dispatch1, namechange_callback);

    tox_events_callback_friend_status_message(dispatch1, statuschange_callback);

    while (true) {

        if (tox_self_get_connection_status(tox1) &&

                tox_self_get_connection_status(tox2) &&

                tox_friend_get_connection_status(tox1, 0, nullptr) == TOX_CONNECTION_UDP) {

            printf("Connected.\n");

            break;

        }

        Tox_Err_Events_Iterate err = TOX_ERR_EVENTS_ITERATE_OK;

        Tox_Events *events = tox_events_iterate(tox1, true, &err);

        ck_assert(err == TOX_ERR_EVENTS_ITERATE_OK);

        tox_dispatch_invoke(dispatch1, events, tox1, &to_compare);

        tox_events_free(events);

        tox_iterate(tox2, nullptr);

        c_sleep(tox_iteration_interval(tox1));

    }

    while (true) {

        if (to_compare.received_name && to_compare.received_status_message) {

            printf("Exchanged names and status messages.\n");

            break;

        }

        Tox_Err_Events_Iterate err = TOX_ERR_EVENTS_ITERATE_OK;

        Tox_Events *events = tox_events_iterate(tox1, true, &err);

        ck_assert(err == TOX_ERR_EVENTS_ITERATE_OK);

        tox_dispatch_invoke(dispatch1, events, tox1, &to_compare);

        tox_events_free(events);

        tox_iterate(tox2, nullptr);

        c_sleep(tox_iteration_interval(tox1));

    }

    size_t save_size = tox_get_savedata_size(tox1);

    uint8_t *savedata = (uint8_t *)malloc(save_size);

    tox_get_savedata(tox1, savedata);

    struct Tox_Options *const options = tox_options_new(nullptr);

    tox_options_set_savedata_type(options, TOX_SAVEDATA_TYPE_TOX_SAVE);

    tox_options_set_savedata_data(options, savedata, save_size);

    Tox *const tox_to_compare = tox_new_log(options, nullptr, nullptr);

    alloc_string(&to_compare.name, tox_friend_get_name_size(tox_to_compare, 0, nullptr));

    tox_friend_get_name(tox_to_compare, 0, to_compare.name, nullptr);

    alloc_string(&to_compare.status_message, tox_friend_get_status_message_size(tox_to_compare, 0, nullptr));

    tox_friend_get_status_message(tox_to_compare, 0, to_compare.status_message, nullptr);

    ck_assert_msg(memcmp(reference_name, to_compare.name, tox_max_name_length()) == 0,

                  "incorrect name: should be all zeroes");

    ck_assert_msg(memcmp(reference_status, to_compare.status_message, tox_max_status_message_length()) == 0,

                  "incorrect status message: should be all zeroes");

    tox_options_free(options);

    tox_dispatch_free(dispatch1);

    tox_kill(tox1);

    tox_kill(tox2);

    tox_kill(tox_to_compare);

    free(savedata);

    free(to_compare.name);

    free(to_compare.status_message);

    free(reference_status);

    free(reference_name);

    return 0;

}