/*
 * Copyright (c) 2020 The ZMK Contributors
 *
 * SPDX-License-Identifier: MIT
 */
 
#pragma once

#include <stddef.h>
#include <kernel.h>
#include <zephyr/types.h>

struct zmk_event_type
{
    const char *name;
};

struct zmk_event_header {
    const struct zmk_event_type* event;
    u8_t last_listener_index;
};

#define ZMK_EV_EVENT_HANDLED 1
#define ZMK_EV_EVENT_CAPTURED 2

typedef int (*zmk_listener_callback_t)(const struct zmk_event_header *eh);
struct zmk_listener
{
    zmk_listener_callback_t callback;
};

struct zmk_event_subscription {
    const struct zmk_event_type *event_type;
    const struct zmk_listener *listener;
};

#define ZMK_EVENT_DECLARE(event_type) \
    struct event_type* new_##event_type(); \
    bool is_##event_type(const struct zmk_event_header *eh); \
    struct event_type* cast_##event_type(const struct zmk_event_header *eh); \
    extern const struct zmk_event_type zmk_event_##event_type;

#define ZMK_EVENT_IMPL(event_type) \
    const struct zmk_event_type zmk_event_##event_type = { \
        .name = STRINGIFY(event_type) \
    }; \
    const struct zmk_event_type* zmk_event_ref_##event_type __used __attribute__((__section__(".event_type"))) = &zmk_event_##event_type; \
    struct event_type* new_##event_type() { \
        struct event_type* ev = (struct event_type *) k_malloc(sizeof(struct event_type)); \
        ev->header.event = &zmk_event_##event_type; \
        return ev; \
    }; \
    bool is_##event_type(const struct zmk_event_header *eh) { \
        return eh->event == &zmk_event_##event_type; \
    }; \
    struct event_type* cast_##event_type(const struct zmk_event_header *eh) {\
        return (struct event_type*)eh; \
    };
    

#define ZMK_LISTENER(mod, cb) \
    const struct zmk_listener zmk_listener_##mod = { \
        .callback = cb \
    };

#define ZMK_SUBSCRIPTION(mod, ev_type) \
    const Z_DECL_ALIGN(struct zmk_event_subscription) _CONCAT(_CONCAT(zmk_event_sub_,mod),ev_type) __used __attribute__((__section__(".event_subscription"))) = { \
        .event_type = &zmk_event_##ev_type, \
        .listener = &zmk_listener_##mod, \
    };

#define ZMK_EVENT_RAISE(ev) \
    zmk_event_manager_raise((struct zmk_event_header *)ev);

#define ZMK_EVENT_RAISE_AFTER(ev, mod) \
    zmk_event_manager_raise_after((struct zmk_event_header *)ev, &zmk_listener_##mod);


#define ZMK_EVENT_RAISE_AT(ev, mod) \
    zmk_event_manager_raise_at((struct zmk_event_header *)ev, &zmk_listener_##mod);

#define ZMK_EVENT_RELEASE(ev) \
    zmk_event_manager_release((struct zmk_event_header *)ev);

int zmk_event_manager_raise(struct zmk_event_header *event);
int zmk_event_manager_raise_after(struct zmk_event_header *event, const struct zmk_listener *listener);
int zmk_event_manager_raise_at(struct zmk_event_header *event, const struct zmk_listener *listener);
int zmk_event_manager_release(struct zmk_event_header *event);