#include "framework/framework.h"

#include <stdio.h>

#include <string.h>

#include <stdlib.h>

#define FR_MESSAGE "Gentoo"

#define FR_TOX_COUNT 33

typedef struct {

    uint32_t requests_received;

} ReceiverState;

static void on_friend_request(const Tox_Event_Friend_Request *event, void *user_data)

{

    ToxNode *self = (ToxNode *)user_data;

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

    const uint8_t *public_key = tox_event_friend_request_get_public_key(event);

    const uint8_t *data = tox_event_friend_request_get_message(event);

    const size_t length = tox_event_friend_request_get_message_length(event);

    if (length == sizeof(FR_MESSAGE) && memcmp(FR_MESSAGE, data, sizeof(FR_MESSAGE)) == 0) {

        tox_friend_add_norequest(tox_node_get_tox(self), public_key, nullptr);

        state->requests_received++;

        tox_node_log(self, "Friend request received: %u", state->requests_received);

    }

}

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

{

    const ReceiverState *state = (const ReceiverState *)ctx;

    tox_events_callback_friend_request(tox_node_get_dispatch(self), on_friend_request);

    WAIT_UNTIL(tox_node_is_self_connected(self));

    // Wait for all requests to be received and friends connected

    WAIT_UNTIL(state->requests_received == FR_TOX_COUNT - 1);

    // Also wait until they are all connected as friends

    bool all_connected = false;

    while (!all_connected && tox_scenario_is_running(self)) {

        all_connected = true;

        for (uint32_t i = 0; i < FR_TOX_COUNT - 1; ++i) {

            if (!tox_node_is_friend_connected(self, i)) {

                all_connected = false;

                break;

            }

        }

        if (!all_connected) {

            tox_scenario_yield(self);

        }

    }

    tox_node_log(self, "All friends connected");

}

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

{

    WAIT_UNTIL(tox_node_is_self_connected(self));

    ToxNode *receiver = (ToxNode *)ctx;

    uint8_t address[TOX_ADDRESS_SIZE];

    tox_node_get_address(receiver, address);

    Tox_Err_Friend_Add err;

    tox_friend_add(tox_node_get_tox(self), address, (const uint8_t *)FR_MESSAGE, sizeof(FR_MESSAGE), &err);

    if (err != TOX_ERR_FRIEND_ADD_OK) {

        tox_node_log(self, "Failed to add friend: %u", err);

        return;

    }

    WAIT_UNTIL(tox_node_is_friend_connected(self, 0));

    tox_node_log(self, "Connected to receiver");

}

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

{

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

    ReceiverState receiver_state = {0};

    ToxNode *receiver = tox_scenario_add_node(s, "Receiver", receiver_script, &receiver_state, sizeof(ReceiverState));

    ToxNode *senders[FR_TOX_COUNT - 1];

    for (int i = 0; i < FR_TOX_COUNT - 1; ++i) {

        char alias[16];

        snprintf(alias, sizeof(alias), "Sender%d", i);

        senders[i] = tox_scenario_add_node(s, alias, sender_script, receiver, 0);

        tox_node_bootstrap(senders[i], receiver);

        tox_node_bootstrap(receiver, senders[i]);

    }

    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;

}

#undef FR_MESSAGE