#include "framework/framework.h"

#include <stdio.h>

#include <string.h>

#include <stdlib.h>

#define NUM_TOXES 90

#define NUM_FRIEND_PAIRS 50

#define FR_MESSAGE "Gentoo"

typedef struct {

    uint32_t friend_count;

} State;

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 == 6 && memcmp(msg, FR_MESSAGE, 6) == 0) {

        tox_friend_add_norequest(tox_node_get_tox(self), tox_event_friend_request_get_public_key(event), nullptr);

    }

}

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

{

    State *state = (State *)ctx;

    tox_events_callback_friend_request(tox_node_get_dispatch(self), on_friend_request);

    tox_node_wait_for_self_connected(self);

    // Wait until we have the expected number of friends connected

    while (1) {

        uint32_t connected_friends = 0;

        uint32_t total_friends = tox_self_get_friend_list_size(tox_node_get_tox(self));

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

            if (tox_node_is_friend_connected(self, i)) {

                connected_friends++;

            }

        }

        if (connected_friends >= state->friend_count) {

            break;

        }

        tox_scenario_yield(self);

    }

    tox_node_log(self, "All %u friends connected.", state->friend_count);

}

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

{

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

    State states[NUM_TOXES] = {0};

    ToxNode *nodes[NUM_TOXES];

    for (int i = 0; i < NUM_TOXES; ++i) {

        char alias[16];

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

        nodes[i] = tox_scenario_add_node(s, alias, peer_script, &states[i], sizeof(State));

    }

    for (int i = 0; i < NUM_TOXES; ++i) {

        tox_node_bootstrap(nodes[i], nodes[(i + 1) % NUM_TOXES]);

        tox_node_bootstrap(nodes[(i + 1) % NUM_TOXES], nodes[i]);

    }

    struct {

        uint16_t t1;

        uint16_t t2;

    } pairs[NUM_FRIEND_PAIRS];

    // Generate friend pairs

    for (int i = 0; i < NUM_FRIEND_PAIRS; ++i) {

        bool unique;

        do {

            unique = true;

            pairs[i].t1 = rand() % NUM_TOXES;

            pairs[i].t2 = (pairs[i].t1 + rand() % (NUM_TOXES - 1) + 1) % NUM_TOXES;

            // Avoid reciprocal or duplicate pairs

            for (int j = 0; j < i; ++j) {

                if ((pairs[j].t1 == pairs[i].t1 && pairs[j].t2 == pairs[i].t2) ||

                        (pairs[j].t1 == pairs[i].t2 && pairs[j].t2 == pairs[i].t1)) {

                    unique = false;

                    break;

                }

            }

        } while (!unique);

        const int t1 = pairs[i].t1;

        const int t2 = pairs[i].t2;

        uint8_t addr[TOX_ADDRESS_SIZE];

        tox_self_get_address(tox_node_get_tox(nodes[t1]), addr);

        Tox_Err_Friend_Add err;

        tox_friend_add(tox_node_get_tox(nodes[t2]), addr, (const uint8_t *)FR_MESSAGE, 6, &err);

        ck_assert(err == TOX_ERR_FRIEND_ADD_OK);

        ++states[t1].friend_count;

        ++states[t2].friend_count;

        // Bootstrap off each other

        tox_node_bootstrap(nodes[t2], nodes[t1]);

        tox_node_bootstrap(nodes[t1], nodes[t2]);

    }

    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

#undef NUM_TOXES

#undef NUM_FRIEND_PAIRS