#include "framework/framework.h"

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

typedef struct {

    bool nospam_changed;

} AliceState;

static void set_checksum(uint8_t *address)

{

    uint8_t checksum[2] = {0};

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

        checksum[i % 2] ^= address[i];

    }

    address[36] = checksum[0];

    address[37] = checksum[1];

}

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

{

    AliceState *state = (AliceState *)ctx;

    tox_node_wait_for_self_connected(self);

    Tox *tox = tox_node_get_tox(self);

    uint32_t old_nospam = tox_self_get_nospam(tox);

    uint32_t new_nospam = old_nospam + 1;

    tox_node_log(self, "Old nospam: %u, setting new nospam: %u", old_nospam, new_nospam);

    tox_self_set_nospam(tox, new_nospam);

    ck_assert(tox_self_get_nospam(tox) == new_nospam);

    // Signal to Bob that we changed nospam

    state->nospam_changed = true;

    tox_node_log(self, "Waiting for friend request from Bob...");

    WAIT_FOR_EVENT(TOX_EVENT_FRIEND_REQUEST);

    tox_node_log(self, "Received friend request from Bob!");

}

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

{

    tox_node_wait_for_self_connected(self);

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

    const AliceState *alice_view = (const AliceState *)tox_node_get_peer_ctx(alice);

    tox_node_log(self, "Waiting for Alice to change nospam...");

    WAIT_UNTIL(alice_view->nospam_changed);

    uint8_t alice_addr[TOX_ADDRESS_SIZE];

    tox_node_get_address(alice, alice_addr);

    // Alice's address in alice_addr is already the NEW one.

    // Manually construct an old address.

    uint8_t old_alice_addr[TOX_ADDRESS_SIZE];

    memcpy(old_alice_addr, alice_addr, TOX_ADDRESS_SIZE);

    uint32_t nospam;

    memcpy(&nospam, old_alice_addr + 32, 4);

    nospam--; // Revert to old nospam

    memcpy(old_alice_addr + 32, &nospam, 4);

    set_checksum(old_alice_addr);

    tox_node_log(self, "Trying to add Alice with OLD nospam...");

    Tox_Err_Friend_Add err;

    uint32_t friend_num = tox_friend_add(tox_node_get_tox(self), old_alice_addr, (const uint8_t *)"Hi", 2, &err);

    ck_assert(err == TOX_ERR_FRIEND_ADD_OK);

    // Wait some time, but not too long.

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

        tox_scenario_yield(self);

    }

    tox_node_log(self, "Deleting Alice and trying with NEW nospam...");

    tox_friend_delete(tox_node_get_tox(self), friend_num, nullptr);

    tox_friend_add(tox_node_get_tox(self), alice_addr, (const uint8_t *)"Hi", 2, &err);

    ck_assert(err == TOX_ERR_FRIEND_ADD_OK);

    tox_node_log(self, "Friend request sent!");

}

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

{

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

    AliceState alice_state = {false};

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

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

    ToxNode *alice = tox_scenario_get_node(s, 0);

    ToxNode *bob = tox_scenario_get_node(s, 1);

    tox_node_bootstrap(alice, bob);

    tox_node_bootstrap(bob, 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;

}