#define _POSIX_C_SOURCE 200809L
#include "yasl/yasl.h"

#include <assert.h>
#include <time.h>

struct state {
    struct yasl_display *display;
    struct yasl_surface *surface;
    unsigned draws;
};

static void draw(struct yasl_surface *surface, struct yasl_frame *frame, void *ud)
{
    struct state *state = ud;
    assert(surface == state->surface && frame->surface == surface);
    assert(yasl_surface_display(surface) == state->display);
    assert(frame->target.pixels);
    struct yasl_vector2i size = yasl_surface_size(surface);
    assert(size.x == frame->target.size.x && size.y == frame->target.size.y);
    for (int i = 0; i < size.x * size.y; i++) frame->target.pixels[i] = 0xff4080c0;
    struct yasl_damage_list damage = {.n = YASL_DAMAGE_MAX + 1};
    assert(!yasl_surface_present(surface, &frame->target, &damage));
    damage.n = -1;
    struct yasl_framebuffer invalid = frame->target;
    invalid.buffer++;
    assert(!yasl_surface_present(surface, &invalid, &damage));
    invalid = frame->target;
    invalid.generation--;
    assert(!yasl_surface_present(surface, &invalid, &damage));
    assert(!yasl_surface_present(surface, &frame->target, NULL));
    assert(yasl_surface_present(surface, &frame->target, &damage));
    assert(!yasl_surface_present(surface, &frame->target, &damage));
    state->draws++;
    yasl_surface_destroy(surface);
    state->surface = NULL;
}

static void test_window(enum yasl_backend request, bool null_spec)
{
    struct yasl_display_spec spec = {.backend = request};
    struct state state = {.display = yasl_display_open(NULL, null_spec ? NULL : &spec)};
    assert(state.display);
    assert(yasl_display_backend(state.display) == YASL_BACKEND_COCOA);
    struct yasl_surface_spec surface = {.title = "Native backend test", .size = {160, 100}};
    state.surface = yasl_surface_create_toplevel(state.display, &surface);
    assert(state.surface);
    struct yasl_surface_cbs callbacks = {.draw = draw};
    yasl_surface_set_callbacks(state.surface, &callbacks, &state);
    callbacks.draw = NULL;
    yasl_surface_schedule_redraw(state.surface);
    uint64_t deadline = yasl_loop_now_ns() + 3000000000ull;
    while (!state.draws && yasl_loop_now_ns() < deadline) {
        yasl_display_dispatch(state.display);
        yasl_display_flush(state.display);
        nanosleep(&(struct timespec){.tv_nsec = 1000000}, NULL);
    }
    assert(state.draws == 1 && !state.surface);
    yasl_display_close(state.display);
}

int main(void)
{
    test_window(YASL_BACKEND_AUTO, true);
    test_window(YASL_BACKEND_AUTO, false);
    test_window(YASL_BACKEND_COCOA, false);
    return 0;
}