#include <assert.h>
#include <string.h>

#include "control_event.h"

static void test_serialize_keycode_event(void) {
    struct control_event event = {
        .type = CONTROL_EVENT_TYPE_KEYCODE,
        .keycode_event = {
            .action = AKEY_EVENT_ACTION_UP,
            .keycode = AKEYCODE_ENTER,
            .metastate = AMETA_SHIFT_ON | AMETA_SHIFT_LEFT_ON,
        },
    };

    unsigned char buf[CONTROL_EVENT_SERIALIZED_MAX_SIZE];
    int size = control_event_serialize(&event, buf);
    assert(size == 10);

    const unsigned char expected[] = {
        0x00, // CONTROL_EVENT_TYPE_KEYCODE
        0x01, // AKEY_EVENT_ACTION_UP
        0x00, 0x00, 0x00, 0x42, // AKEYCODE_ENTER
        0x00, 0x00, 0x00, 0x41, // AMETA_SHIFT_ON | AMETA_SHIFT_LEFT_ON
    };
    assert(!memcmp(buf, expected, sizeof(expected)));
}

static void test_serialize_text_event(void) {
    struct control_event event = {
        .type = CONTROL_EVENT_TYPE_TEXT,
        .text_event = {
            .text = "hello, world!",
        },
    };

    unsigned char buf[CONTROL_EVENT_SERIALIZED_MAX_SIZE];
    int size = control_event_serialize(&event, buf);
    assert(size == 16);

    const unsigned char expected[] = {
        0x01, // CONTROL_EVENT_TYPE_TEXT
        0x00, 0x0d, // text length
        'h', 'e', 'l', 'l', 'o', ',', ' ', 'w', 'o', 'r', 'l', 'd', '!', // text
    };
    assert(!memcmp(buf, expected, sizeof(expected)));
}

static void test_serialize_long_text_event(void) {
    struct control_event event;
    event.type = CONTROL_EVENT_TYPE_TEXT;
    char text[CONTROL_EVENT_TEXT_MAX_LENGTH + 1];
    memset(text, 'a', sizeof(text));
    text[CONTROL_EVENT_TEXT_MAX_LENGTH] = '\0';
    event.text_event.text = text;

    unsigned char buf[CONTROL_EVENT_SERIALIZED_MAX_SIZE];
    int size = control_event_serialize(&event, buf);
    assert(size == 3 + CONTROL_EVENT_TEXT_MAX_LENGTH);

    unsigned char expected[3 + CONTROL_EVENT_TEXT_MAX_LENGTH];
    expected[0] = 0x01; // CONTROL_EVENT_TYPE_KEYCODE
    expected[1] = 0x01;
    expected[2] = 0x2c; // text length (16 bits)
    memset(&expected[3], 'a', CONTROL_EVENT_TEXT_MAX_LENGTH);

    assert(!memcmp(buf, expected, sizeof(expected)));
}

static void test_serialize_mouse_event(void) {
    struct control_event event = {
        .type = CONTROL_EVENT_TYPE_MOUSE,
        .mouse_event = {
            .action = AMOTION_EVENT_ACTION_DOWN,
            .buttons = AMOTION_EVENT_BUTTON_PRIMARY,
            .position = {
                .point = {
                    .x = 260,
                    .y = 1026,
                },
                .screen_size = {
                    .width = 1080,
                    .height = 1920,
                },
            },
        },
    };

    unsigned char buf[CONTROL_EVENT_SERIALIZED_MAX_SIZE];
    int size = control_event_serialize(&event, buf);
    assert(size == 18);

    const unsigned char expected[] = {
        0x02, // CONTROL_EVENT_TYPE_MOUSE
        0x00, // AKEY_EVENT_ACTION_DOWN
        0x00, 0x00, 0x00, 0x01, // AMOTION_EVENT_BUTTON_PRIMARY
        0x00, 0x00, 0x01, 0x04, 0x00, 0x00, 0x04, 0x02, // 260 1026
        0x04, 0x38, 0x07, 0x80, // 1080 1920
    };
    assert(!memcmp(buf, expected, sizeof(expected)));
}

static void test_serialize_scroll_event(void) {
    struct control_event event = {
        .type = CONTROL_EVENT_TYPE_SCROLL,
        .scroll_event = {
            .position = {
                .point = {
                    .x = 260,
                    .y = 1026,
                },
                .screen_size = {
                    .width = 1080,
                    .height = 1920,
                },
            },
            .hscroll = 1,
            .vscroll = -1,
        },
    };

    unsigned char buf[CONTROL_EVENT_SERIALIZED_MAX_SIZE];
    int size = control_event_serialize(&event, buf);
    assert(size == 21);

    const unsigned char expected[] = {
        0x03, // CONTROL_EVENT_TYPE_SCROLL
        0x00, 0x00, 0x01, 0x04, 0x00, 0x00, 0x04, 0x02, // 260 1026
        0x04, 0x38, 0x07, 0x80, // 1080 1920
        0x00, 0x00, 0x00, 0x01, // 1
        0xFF, 0xFF, 0xFF, 0xFF, // -1
    };
    assert(!memcmp(buf, expected, sizeof(expected)));
}

static void test_serialize_back_or_screen_on_event(void) {
    struct control_event event = {
        .type = CONTROL_EVENT_TYPE_BACK_OR_SCREEN_ON,
    };

    unsigned char buf[CONTROL_EVENT_SERIALIZED_MAX_SIZE];
    int size = control_event_serialize(&event, buf);
    assert(size == 1);

    const unsigned char expected[] = {
        0x04, // CONTROL_EVENT_TYPE_BACK_OR_SCREEN_ON
    };
    assert(!memcmp(buf, expected, sizeof(expected)));
}

static void test_serialize_expand_notification_panel_event(void) {
    struct control_event event = {
        .type = CONTROL_EVENT_TYPE_EXPAND_NOTIFICATION_PANEL,
    };

    unsigned char buf[CONTROL_EVENT_SERIALIZED_MAX_SIZE];
    int size = control_event_serialize(&event, buf);
    assert(size == 1);

    const unsigned char expected[] = {
        0x05, // CONTROL_EVENT_TYPE_EXPAND_NOTIFICATION_PANEL
    };
    assert(!memcmp(buf, expected, sizeof(expected)));
}

static void test_serialize_collapse_notification_panel_event(void) {
    struct control_event event = {
        .type = CONTROL_EVENT_TYPE_COLLAPSE_NOTIFICATION_PANEL,
    };

    unsigned char buf[CONTROL_EVENT_SERIALIZED_MAX_SIZE];
    int size = control_event_serialize(&event, buf);
    assert(size == 1);

    const unsigned char expected[] = {
        0x06, // CONTROL_EVENT_TYPE_COLLAPSE_NOTIFICATION_PANEL
    };
    assert(!memcmp(buf, expected, sizeof(expected)));
}

static void test_serialize_get_clipboard_event(void) {
    struct control_event event = {
        .type = CONTROL_EVENT_TYPE_GET_CLIPBOARD,
    };

    unsigned char buf[CONTROL_EVENT_SERIALIZED_MAX_SIZE];
    int size = control_event_serialize(&event, buf);
    assert(size == 1);

    const unsigned char expected[] = {
        0x07, // CONTROL_EVENT_TYPE_GET_CLIPBOARD
    };
    assert(!memcmp(buf, expected, sizeof(expected)));
}

int main(void) {
    test_serialize_keycode_event();
    test_serialize_text_event();
    test_serialize_long_text_event();
    test_serialize_mouse_event();
    test_serialize_scroll_event();
    test_serialize_back_or_screen_on_event();
    test_serialize_expand_notification_panel_event();
    test_serialize_collapse_notification_panel_event();
    test_serialize_get_clipboard_event();
    return 0;
}