#include "framework/framework.h"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#define FRIEND_REQUEST_MSG "Gentoo"

static void on_friend_request(const Tox_Event_Friend_Request *event, void *user_data)
{
    ToxNode *self = (ToxNode *)user_data;
    const uint8_t *msg = tox_event_friend_request_get_message(event);
    size_t len = tox_event_friend_request_get_message_length(event);

    if (len == 7 && memcmp(msg, FRIEND_REQUEST_MSG, 7) == 0) {
        tox_friend_add_norequest(tox_node_get_tox(self), tox_event_friend_request_get_public_key(event), nullptr);
    }
}

static void relay_script(ToxNode *self, void *ctx)
{
    WAIT_UNTIL(tox_node_is_self_connected(self));

    ToxScenario *s = tox_node_get_scenario(self);
    bool peers_done = false;
    while (!peers_done && tox_scenario_is_running(self)) {
        peers_done = true;
        for (uint32_t i = 1; i < 3; ++i) {
            if (!tox_node_is_finished(tox_scenario_get_node(s, i))) {
                peers_done = false;
                break;
            }
        }
        if (!peers_done) {
            tox_scenario_yield(self);
        }
    }
}

static void peer_script(ToxNode *self, void *ctx)
{
    Tox *tox = tox_node_get_tox(self);
    uint32_t my_index = tox_node_get_index(self);

    if (my_index == 1) {
        tox_events_callback_friend_request(tox_node_get_dispatch(self), on_friend_request);
    }

    WAIT_UNTIL(tox_node_is_self_connected(self));

    if (my_index == 2) {
        ToxNode *peer1 = tox_scenario_get_node(tox_node_get_scenario(self), 1);
        uint8_t addr[TOX_ADDRESS_SIZE];
        tox_node_get_address(peer1, addr);
        tox_friend_add(tox, addr, (const uint8_t *)FRIEND_REQUEST_MSG, 7, nullptr);
    }

    WAIT_UNTIL(tox_node_is_friend_connected(self, 0));
    tox_node_log(self, "Connected to friend");

    // Reload test
    tox_node_log(self, "Reloading...");
    tox_node_reload(self);
    tox_node_log(self, "Reloaded");

    WAIT_UNTIL(tox_node_is_self_connected(self));
    WAIT_UNTIL(tox_node_is_friend_connected(self, 0));
    tox_node_log(self, "Connected to friend again after reload");
}

int main(int argc, char *argv[])
{
    ToxScenario *s = tox_scenario_new(argc, argv, 300000);

    ToxNode *relay = tox_scenario_add_node(s, "Relay", relay_script, nullptr, 0);
    ToxNode *peer1 = tox_scenario_add_node(s, "Peer1", peer_script, nullptr, 0);
    ToxNode *peer2 = tox_scenario_add_node(s, "Peer2", peer_script, nullptr, 0);

    // All peers bootstrap from relay
    tox_node_bootstrap(peer1, relay);
    tox_node_bootstrap(peer2, relay);

    ToxScenarioStatus res = tox_scenario_run(s);
    if (res != TOX_SCENARIO_DONE) {
        tox_scenario_log(s, "Test failed with status %u", res);
        return 1;
    }

    tox_scenario_free(s);
    return 0;
}