#include "framework/framework.h"

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

typedef struct {

    bool connection_status_called;

    Tox_Connection last_status;

} State;

static void on_self_connection_status(const Tox_Event_Self_Connection_Status *event, void *user_data)

{

    ToxNode *self = (ToxNode *)user_data;

    State *state = (State *)tox_node_get_script_ctx(self);

    state->connection_status_called = true;

    state->last_status = tox_event_self_connection_status_get_connection_status(event);

    tox_node_log(self, "Self connection status: %s", tox_connection_to_string(state->last_status));

}

static void alice_script(ToxNode *self, void *ctx)

{

    State *state = (State *)ctx;

    // Register dispatch callback

    tox_events_callback_self_connection_status(tox_node_get_dispatch(self), on_self_connection_status);

    tox_node_wait_for_self_connected(self);

    ck_assert(state->connection_status_called);

    ck_assert(state->last_status != TOX_CONNECTION_NONE);

    Tox *tox = tox_node_get_tox(self);

    // Test ports

    Tox_Err_Get_Port err_port;

    uint16_t udp_port = tox_self_get_udp_port(tox, &err_port);

    ck_assert(err_port == TOX_ERR_GET_PORT_OK);

    ck_assert(udp_port > 0);

    tox_node_log(self, "UDP Port: %u", udp_port);

    // Friend list test

    tox_node_wait_for_friend_connected(self, 0);

    tox_node_wait_for_friend_connected(self, 1);

    size_t friend_count = tox_self_get_friend_list_size(tox);

    ck_assert(friend_count == 2);

    uint32_t friends[2];

    tox_self_get_friend_list(tox, friends);

    ck_assert(friends[0] == 0);

    ck_assert(friends[1] == 1);

    tox_node_log(self, "Self query tests passed!");

}

static void peer_script(ToxNode *self, void *ctx)

{

    tox_node_wait_for_self_connected(self);

    ToxNode *alice = tox_scenario_get_node(tox_node_get_scenario(self), 0);

    WAIT_UNTIL(tox_node_is_finished(alice));

}

int main(int argc, char *argv[])

{

    ToxScenario *s = tox_scenario_new(argc, argv, 60000);

    State alice_state = {false, TOX_CONNECTION_NONE};

    tox_scenario_add_node(s, "Alice", alice_script, &alice_state, sizeof(State));

    tox_scenario_add_node(s, "Bob", peer_script, nullptr, 0);

    tox_scenario_add_node(s, "Charlie", peer_script, nullptr, 0);

    ToxNode *alice = tox_scenario_get_node(s, 0);

    ToxNode *bob = tox_scenario_get_node(s, 1);

    ToxNode *charlie = tox_scenario_get_node(s, 2);

    tox_node_bootstrap(alice, bob);

    tox_node_bootstrap(charlie, bob);

    tox_node_friend_add(alice, bob);

    tox_node_friend_add(bob, alice);

    tox_node_friend_add(alice, charlie);

    tox_node_friend_add(charlie, alice);

    ToxScenarioStatus res = tox_scenario_run(s);

    if (res != TOX_SCENARIO_DONE) {

        fprintf(stderr, "Scenario failed with status %u\n", res);

        return 1;

    }

    tox_scenario_free(s);

    return 0;

}