/* Приглашение в другую группу через настоящую PM: отмена, ошибки, UDP-доставка и JOIN. */
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include "../lib/platform_compat.h"
#include "../lib/debug_config.h"
#include "../lib/mem.h"
#include "../src/utun_instance.h"
#include "../src/chat/chat_core.h"
#include "../src/chat/chat_event.h"
#include "../src/chat/chat_join.h"
#include "../src/chat/chat_sync.h"
#include "../src/chat/invite_build.h"
#include "../src/chat/invite_link.h"
#include "../src/dm/dm_core.h"
#include "../src/dm/dm_crypto.h"
#include "../src/routing_layer/topo_node_sqlite.h"
#include "../src/transport_layer/secure_channel.h"
#include "test_utils.h"

static struct UTUN_INSTANCE* nodes[2];
static int results[16], calls[16], joined;
static uint64_t joining_channel;

static void event(struct UTUN_INSTANCE* inst, int type, const uint8_t* data, int len) {
    if (type == CHAT_EVT_DM_INVITE_RESULT && len == 28) {
        uint64_t id, peer, conv; int32_t result;
        memcpy(&id, data, 8); memcpy(&result, data + 8, 4);
        memcpy(&peer, data + 12, 8); memcpy(&conv, data + 20, 8);
        if (id >= 16 || peer != nodes[1]->node_id || conv != dm_derive_conv_id(nodes[0]->node_id, peer)) abort();
        calls[id]++; results[id] = result;
    } else if (inst == nodes[1] && type == CHAT_EVT_CONNECT_RESULT && len == 20) {
        uint64_t channel; int32_t result;
        memcpy(&result, data + 8, 4); memcpy(&channel, data + 12, 8);
        if (channel == joining_channel) joined = result == 0 ? 1 : -1;
    }
}

struct link_result { int ready; char link[1024]; };
static void link_ready(void* arg, int result, const char* link) {
    struct link_result* state = arg;
    state->ready = result == CHAT_JOIN_OK ? 1 : -1;
    if (link) snprintf(state->link, sizeof(state->link), "%s", link);
}

static int scalar(struct UTUN_INSTANCE* inst, const char* sql, uint64_t* value) {
    sqlite3_stmt* statement = NULL;
    int rc = sqlite3_prepare_v2(inst->topo_sqlite_db, sql, -1, &statement, NULL);
    if (rc == SQLITE_OK && sqlite3_step(statement) == SQLITE_ROW) {
        *value = (uint64_t)sqlite3_column_int64(statement, 0); rc = 0;
    } else rc = -1;
    sqlite3_finalize(statement);
    return rc;
}

static int message_count(void) {
    uint64_t count = 0;
    return scalar(nodes[0], "SELECT count(*) FROM dm_messages WHERE dir=1", &count) ? -1 : (int)count;
}

static void request(uint64_t id, uint64_t group, const char* source, int cancel) {
    struct dm_invite_req* req = u_calloc(1, sizeof(*req));
    if (!req) abort();
    req->inst = nodes[0]; req->request_id = id; req->peer_node_id = nodes[1]->node_id;
    req->channel_id = group; req->cancel = cancel;
    snprintf(req->source_ch_id, sizeof(req->source_ch_id), "%s", source);
    snprintf(req->peer_name, sizeof(req->peer_name), "B");
    dm_invite_trampoline(req);
}

static int wait_result(struct UASYNC* ua, int id, int expected) {
    uint64_t deadline = get_time_tb() + 30000;
    while (!calls[id] && get_time_tb() < deadline) uasync_poll(ua, 100);
    return calls[id] == 1 && results[id] == expected;
}

static int join_link(struct UASYNC* ua, const char* link) {
    struct InviteData invite; char error[128]; uint8_t addresses[1024];
    if (invite_link_decode(link, strlen(link), &invite, error, sizeof(error))) return -1;
    int bytes = invite_serialize_addrs(&invite, addresses, sizeof(addresses));
    if (bytes < 0) return -1;
    joining_channel = invite.channelId; joined = 0;
    chat_sync_connect_from_invite(nodes[1], invite.channelId, invite.nodeId, invite.pubkey,
                                  addresses, invite.addrCount, bytes, NULL, invite.join_key);
    uint64_t deadline = get_time_tb() + 200000;
    while (!joined && get_time_tb() < deadline) uasync_poll(ua, 100);
    if (joined != 1) { DEBUG_ERROR(DEBUG_CATEGORY_DM, "invite test: JOIN failed ch=%llu result=%d",
        (unsigned long long)joining_channel, joined); return -1; }
    return 0;
}

static int start_node(struct UASYNC* ua, const char* directory, int index, int port) {
    struct SC_MYKEYS keys;
    if (sc_generate_keypair(&keys) != SC_OK) return -1;
    char pub[65], priv[65], path[768], database[768];
    for (int i = 0; i < 32; i++) {
        snprintf(pub + i * 2, 3, "%02x", keys.public_key[i]);
        snprintf(priv + i * 2, 3, "%02x", keys.private_key[i]);
    }
    snprintf(database, sizeof(database), "%s/db%d", directory, index);
    if (utun_mkdir(database, 0700)) return -1;
    snprintf(path, sizeof(path), "%s/%d.conf", directory, index);
    FILE* config = fopen(path, "w");
    if (!config) return -1;
    fprintf(config, "[global]\nmy_public_key=%s\nmy_private_key=%s\ndb_path=%s\nmy_node_name=%c\n"
        "[server:s1]\naddr=127.0.0.1:%d\ntype=public\n[allowed_keys]\nallow_all=1\n"
        "[chatserver]\nstorage_autoload=0\n", pub, priv, database, 'A' + index, port);
    fclose(config);
    nodes[index] = utun_instance_create(ua, path);
    if (!nodes[index] || utun_instance_init(nodes[index])) return -1;
    chat_event_set_handler(nodes[index], event);
    return 0;
}

int main(void) {
    debug_config_init(); debug_set_level(DEBUG_LEVEL_ERROR);
    if (getenv("UTUN_TEST_DEBUG")) {
        debug_set_category_level(DEBUG_CATEGORY_DM, DEBUG_LEVEL_DEBUG);
        debug_set_category_level(DEBUG_CATEGORY_CHAT_SYNC, DEBUG_LEVEL_DEBUG);
        debug_set_category_level(DEBUG_CATEGORY_MEMBER_SYNC, DEBUG_LEVEL_INFO);
    }
    utun_instance_set_tun_init_enabled(0);
    struct UASYNC* ua = uasync_create();
    char directory[512] = "/tmp/utun_dm_invite_XXXXXX";
    int failed = 1;
#define CHECK(condition, reason) do { if (!(condition)) { DEBUG_ERROR(DEBUG_CATEGORY_DM, "invite test: %s", reason); goto done; } } while (0)
    CHECK(ua && !test_mkdtemp(directory), "temporary directory/event loop");
    int port = 32000 + getpid() % 10000;
    CHECK(!start_node(ua, directory, 0, port) && !start_node(ua, directory, 1, port + 1), "start nodes");
    chat_core_create_channel_auto(nodes[0], "source");
    uint64_t source = 0, target = 0;
    CHECK(!scalar(nodes[0], "SELECT channel_id FROM channels WHERE name='source'", &source), "create source group");
    char source_id[64]; snprintf(source_id, sizeof(source_id), "%llu", (unsigned long long)source);
    struct link_result first = {0};
    CHECK(chat_invite_build_link(nodes[0], source, 0, NULL, link_ready, &first), "build initial invite");
    uint64_t deadline = get_time_tb() + 30000;
    while (!first.ready && get_time_tb() < deadline) uasync_poll(ua, 100);
    CHECK(first.ready == 1 && !join_link(ua, first.link), "join common source group");
    chat_core_create_channel_auto(nodes[0], "target");
    CHECK(!scalar(nodes[0], "SELECT channel_id FROM channels WHERE name='target'", &target), "create different target group");
    CHECK(target != source && !message_count(), "initial empty PM");

    request(1, target, source_id, 0);
    CHECK(!message_count() && !calls[1], "no PM before registration ACK");
    request(2, target, source_id, 0);
    CHECK(wait_result(ua, 2, DM_INVITE_BUSY), "busy request retains first operation");
    request(1, target, source_id, 1);
    CHECK(wait_result(ua, 1, DM_INVITE_CANCELLED) && !message_count(), "cancel before ACK sends nothing");
    request(3, 0, source_id, 0);
    CHECK(wait_result(ua, 3, DM_INVITE_BUILD_ERROR) && !message_count(), "invalid group sends nothing");

    CHECK(sqlite3_exec(nodes[0]->topo_sqlite_db, "UPDATE dm_conversations SET group_id=0", NULL, NULL, NULL) == SQLITE_OK,
          "simulate automatically created PM without group route");
    CHECK(sqlite3_exec(nodes[0]->topo_sqlite_db, "CREATE TEMP TRIGGER fail_invite BEFORE INSERT ON dm_outbox "
        "BEGIN SELECT RAISE(ABORT,'test invite commit failure'); END", NULL, NULL, NULL) == SQLITE_OK, "install failed commit");
    request(4, target, source_id, 0);
    CHECK(wait_result(ua, 4, DM_INVITE_SEND_ERROR) && !message_count(), "failed commit returns error and rolls back PM");
    CHECK(sqlite3_exec(nodes[0]->topo_sqlite_db, "DROP TRIGGER fail_invite", NULL, NULL, NULL) == SQLITE_OK, "remove failed commit");

    request(5, target, "", 0); /* из шапки существующей PM исходная группа не передаётся */
    request(4, target, "", 1); /* запоздалая отмена другой операции не отменяет текущую */
    CHECK(wait_result(ua, 5, DM_INVITE_OK) && message_count() == 1, "one invitation committed for fixed peer");
    uint64_t route = 0;
    CHECK(!scalar(nodes[0], "SELECT group_id FROM dm_conversations LIMIT 1", &route) && route == source,
          "successful PM restores its automatically resolved shared group route");
    char conv[64], messages[8192]; size_t length = 0;
    snprintf(conv, sizeof(conv), "%llu", (unsigned long long)dm_derive_conv_id(nodes[0]->node_id, nodes[1]->node_id));
    deadline = get_time_tb() + 100000;
    char* link = NULL;
    do {
        uasync_poll(ua, 100);
        if (!dm_list_messages_json(nodes[1], conv, 10, 0, messages, sizeof(messages), &length)) link = strstr(messages, "utun://");
    } while (!link && get_time_tb() < deadline);
    CHECK(link, "invite delivered to B over source group's UDP route");
    char* end = strchr(link, '"'); CHECK(end, "complete invite text in PM"); *end = '\0';
    struct InviteData received; char error[128];
    CHECK(!invite_link_decode(link, strlen(link), &received, error, sizeof(error)) && received.channelId == target,
          "PM invites target group rather than source group");
    CHECK(JOIN_KEY_TTL_SECONDS == 600 && chat_join_lookup_inviter(nodes[0], target, received.join_key) == nodes[0]->node_id,
          "delivered link is registered with ten minute lifetime");
    CHECK(!join_link(ua, link), "B joins target group using received PM link");
    char target_id[64]; snprintf(target_id, sizeof(target_id), "%llu", (unsigned long long)target);
    chat_core_connect_channel(nodes[1], target_id);
    deadline = get_time_tb() + 100000;
    while (!topo_node_sqlite_member_in_channel(nodes[1]->topo_sqlite_db, target_id, nodes[1]->node_id) &&
           get_time_tb() < deadline) uasync_poll(ua, 100);
    DEBUG_INFO(DEBUG_CATEGORY_DM, "invite test: membership after JOIN ch=%s A_has_B=%d B_has_B=%d", target_id,
               topo_node_sqlite_member_in_channel(nodes[0]->topo_sqlite_db, target_id, nodes[1]->node_id),
               topo_node_sqlite_member_in_channel(nodes[1]->topo_sqlite_db, target_id, nodes[1]->node_id));
    CHECK(topo_node_sqlite_member_in_channel(nodes[0]->topo_sqlite_db, target_id, nodes[1]->node_id) &&
          topo_node_sqlite_member_in_channel(nodes[1]->topo_sqlite_db, target_id, nodes[1]->node_id), "membership committed on both nodes");
    const char* text = "joined through PM invitation";
    struct chat_msg_submit message = {0};
    snprintf(message.channel_id, sizeof(message.channel_id), "%s", target_id);
    snprintf(message.content_type, sizeof(message.content_type), "text");
    message.data = (uint8_t*)text; message.data_len = (uint32_t)strlen(text);
    chat_core_submit_message(nodes[1], &message);
    deadline = get_time_tb() + 100000;
    while (!chat_core_count(nodes[0], target_id) && get_time_tb() < deadline) uasync_poll(ua, 100);
    CHECK(!chat_core_get_messages_json(nodes[0], target_id, 10, 0, messages, sizeof(messages), &length) && strstr(messages, text),
          "joined client can send a group message to the inviter");
    request(6, target, "", 0);
    dm_core_destroy(nodes[0]);
    CHECK(calls[6] == 1 && results[6] == DM_INVITE_CANCELLED && message_count() == 1, "teardown cancels pending invitation");
    failed = 0;
done:
    for (int i = 0; i < 2; i++) if (nodes[i]) utun_instance_destroy(nodes[i]);
    if (ua) uasync_destroy(ua, 0);
    printf("PM invite: cancellation, failure, delivery and JOIN: %s\n", failed ? "FAIL" : "OK");
    return failed;
}