/* Copyright (c), Niclas Meyer <niclas@countingsort.com>
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at https://mozilla.org/MPL/2.0/.
 */

#include "luak/kiwmi_keyboard.h"

#include <stdint.h>

#include <lauxlib.h>
#include <wayland-server.h>
#include <wlr/types/wlr_input_device.h>
#include <wlr/types/wlr_keyboard.h>
#include <wlr/util/log.h>
#include <xkbcommon/xkbcommon.h>

#include "input/keyboard.h"
#include "luak/kiwmi_lua_callback.h"
#include "luak/lua_compat.h"

static int
l_kiwmi_keyboard_keymap(lua_State *L)
{
    struct kiwmi_object *obj =
        *(struct kiwmi_object **)luaL_checkudata(L, 1, "kiwmi_keyboard");
    luaL_checktype(L, 2, LUA_TTABLE);

    if (!obj->valid) {
        return luaL_error(L, "kiwmi_keyboard no longer valid");
    }

    struct kiwmi_keyboard *keyboard = obj->object;
    struct xkb_rule_names settings  = {0};

    lua_getfield(L, 2, "rules");
    if (lua_isstring(L, -1)) {
        settings.rules = luaL_checkstring(L, -1);
    }

    lua_getfield(L, 2, "model");
    if (lua_isstring(L, -1)) {
        settings.model = luaL_checkstring(L, -1);
    }

    lua_getfield(L, 2, "layout");
    if (lua_isstring(L, -1)) {
        settings.layout = luaL_checkstring(L, -1);
    }

    lua_getfield(L, 2, "variant");
    if (lua_isstring(L, -1)) {
        settings.variant = luaL_checkstring(L, -1);
    }

    lua_getfield(L, 2, "options");
    if (lua_isstring(L, -1)) {
        settings.options = luaL_checkstring(L, -1);
    }

    struct xkb_context *context = xkb_context_new(XKB_CONTEXT_NO_FLAGS);
    struct xkb_keymap *keymap   = xkb_keymap_new_from_names(
        context, &settings, XKB_KEYMAP_COMPILE_NO_FLAGS);

    wlr_keyboard_set_keymap(keyboard->device->keyboard, keymap);

    xkb_keymap_unref(keymap);
    xkb_context_unref(context);

    return 0;
}

static int
l_kiwmi_keyboard_modifiers(lua_State *L)
{
    struct kiwmi_object *obj =
        *(struct kiwmi_object **)luaL_checkudata(L, 1, "kiwmi_keyboard");

    if (!obj->valid) {
        return luaL_error(L, "kiwmi_keyboard no longer valid");
    }

    struct kiwmi_keyboard *keyboard = obj->object;

    uint32_t modifiers = wlr_keyboard_get_modifiers(keyboard->device->keyboard);

    lua_newtable(L);

    lua_pushboolean(L, modifiers & WLR_MODIFIER_SHIFT);
    lua_setfield(L, -2, "shift");

    lua_pushboolean(L, modifiers & WLR_MODIFIER_CAPS);
    lua_setfield(L, -2, "caps");

    lua_pushboolean(L, modifiers & WLR_MODIFIER_CTRL);
    lua_setfield(L, -2, "ctrl");

    lua_pushboolean(L, modifiers & WLR_MODIFIER_ALT);
    lua_setfield(L, -2, "alt");

    lua_pushboolean(L, modifiers & WLR_MODIFIER_MOD2);
    lua_setfield(L, -2, "mod2");

    lua_pushboolean(L, modifiers & WLR_MODIFIER_MOD3);
    lua_setfield(L, -2, "mod3");

    lua_pushboolean(L, modifiers & WLR_MODIFIER_LOGO);
    lua_setfield(L, -2, "super");

    lua_pushboolean(L, modifiers & WLR_MODIFIER_MOD5);
    lua_setfield(L, -2, "mod5");

    return 1;
}

static const luaL_Reg kiwmi_keyboard_methods[] = {
    {"keymap", l_kiwmi_keyboard_keymap},
    {"modifiers", l_kiwmi_keyboard_modifiers},
    {"on", luaK_callback_register_dispatch},
    {NULL, NULL},
};

static void
kiwmi_keyboard_on_destroy_notify(struct wl_listener *listener, void *data)
{
    struct kiwmi_lua_callback *lc   = wl_container_of(listener, lc, listener);
    struct kiwmi_server *server     = lc->server;
    lua_State *L                    = server->lua->L;
    struct kiwmi_keyboard *keyboard = data;

    lua_rawgeti(L, LUA_REGISTRYINDEX, lc->callback_ref);

    lua_pushcfunction(L, luaK_kiwmi_keyboard_new);
    lua_pushlightuserdata(L, server->lua);
    lua_pushlightuserdata(L, keyboard);
    if (lua_pcall(L, 2, 1, 0)) {
        wlr_log(WLR_ERROR, "%s", lua_tostring(L, -1));
        lua_pop(L, 1);
        return;
    }

    if (lua_pcall(L, 1, 1, 0)) {
        wlr_log(WLR_ERROR, "%s", lua_tostring(L, -1));
        lua_pop(L, 1);
        return;
    }
}

static bool
send_key_event(
    struct kiwmi_lua_callback *lc,
    xkb_keysym_t sym,
    uint32_t keycode,
    struct kiwmi_keyboard *keyboard,
    bool raw)
{
    struct kiwmi_server *server = lc->server;
    lua_State *L                = server->lua->L;

    static char keysym_name[64];
    size_t namelen = xkb_keysym_get_name(sym, keysym_name, sizeof(keysym_name));

    namelen = namelen > sizeof(keysym_name) ? sizeof(keysym_name) : namelen;

    lua_rawgeti(L, LUA_REGISTRYINDEX, lc->callback_ref);

    lua_newtable(L);

    lua_pushlstring(L, keysym_name, namelen);
    lua_setfield(L, -2, "key");

    lua_pushnumber(L, keycode);
    lua_setfield(L, -2, "keycode");

    lua_pushboolean(L, raw);
    lua_setfield(L, -2, "raw");

    lua_pushcfunction(L, luaK_kiwmi_keyboard_new);
    lua_pushlightuserdata(L, server->lua);
    lua_pushlightuserdata(L, keyboard);
    if (lua_pcall(L, 2, 1, 0)) {
        wlr_log(WLR_ERROR, "%s", lua_tostring(L, -1));
        lua_pop(L, 1);
        return false;
    }
    lua_setfield(L, -2, "keyboard");

    if (lua_pcall(L, 1, 1, 0)) {
        wlr_log(WLR_ERROR, "%s", lua_tostring(L, -1));
        lua_pop(L, 1);
        return false;
    }

    bool handled = lua_toboolean(L, -1);
    lua_pop(L, 1);
    return handled;
}

static void
kiwmi_keyboard_on_key_down_or_up_notify(
    struct wl_listener *listener,
    void *data)
{
    struct kiwmi_lua_callback *lc = wl_container_of(listener, lc, listener);
    struct kiwmi_keyboard_key_event *event = data;
    struct kiwmi_keyboard *keyboard        = event->keyboard;

    const xkb_keysym_t *raw_syms = event->raw_syms;
    int raw_syms_len             = event->raw_syms_len;

    const xkb_keysym_t *translated_syms = event->translated_syms;
    int translated_syms_len             = event->translated_syms_len;

    uint32_t keycode = event->keycode;

    bool handled = false;

    for (int i = 0; i < translated_syms_len; ++i) {
        xkb_keysym_t sym = translated_syms[i];
        handled |= send_key_event(lc, sym, keycode, keyboard, false);
    }

    if (!handled) {
        for (int i = 0; i < raw_syms_len; ++i) {
            xkb_keysym_t sym = raw_syms[i];
            handled |= send_key_event(lc, sym, keycode, keyboard, true);
        }
    }

    event->handled = handled;
}

static int
l_kiwmi_keyboard_on_destroy(lua_State *L)
{
    struct kiwmi_object *obj =
        *(struct kiwmi_object **)luaL_checkudata(L, 1, "kiwmi_keyboard");
    luaL_checktype(L, 2, LUA_TFUNCTION);

    if (!obj->valid) {
        return luaL_error(L, "kiwmi_keyboard no longer valid");
    }

    struct kiwmi_keyboard *keyboard = obj->object;
    struct kiwmi_server *server     = keyboard->server;

    lua_pushcfunction(L, luaK_kiwmi_lua_callback_new);
    lua_pushlightuserdata(L, server);
    lua_pushvalue(L, 2);
    lua_pushlightuserdata(L, kiwmi_keyboard_on_destroy_notify);
    lua_pushlightuserdata(L, &obj->events.destroy);
    lua_pushlightuserdata(L, obj);

    if (lua_pcall(L, 5, 0, 0)) {
        wlr_log(WLR_ERROR, "%s", lua_tostring(L, -1));
        return 0;
    }

    return 0;
}

static int
l_kiwmi_keyboard_on_key_down(lua_State *L)
{
    struct kiwmi_object *obj =
        *(struct kiwmi_object **)luaL_checkudata(L, 1, "kiwmi_keyboard");
    luaL_checktype(L, 2, LUA_TFUNCTION);

    if (!obj->valid) {
        return luaL_error(L, "kiwmi_keyboard no longer valid");
    }

    struct kiwmi_keyboard *keyboard = obj->object;
    struct kiwmi_server *server     = keyboard->server;

    lua_pushcfunction(L, luaK_kiwmi_lua_callback_new);
    lua_pushlightuserdata(L, server);
    lua_pushvalue(L, 2);
    lua_pushlightuserdata(L, kiwmi_keyboard_on_key_down_or_up_notify);
    lua_pushlightuserdata(L, &keyboard->events.key_down);
    lua_pushlightuserdata(L, obj);

    if (lua_pcall(L, 5, 0, 0)) {
        wlr_log(WLR_ERROR, "%s", lua_tostring(L, -1));
        return 0;
    }

    return 0;
}

static int
l_kiwmi_keyboard_on_key_up(lua_State *L)
{
    struct kiwmi_object *obj =
        *(struct kiwmi_object **)luaL_checkudata(L, 1, "kiwmi_keyboard");
    luaL_checktype(L, 2, LUA_TFUNCTION);

    if (!obj->valid) {
        return luaL_error(L, "kiwmi_keyboard no longer valid");
    }

    struct kiwmi_keyboard *keyboard = obj->object;
    struct kiwmi_server *server     = keyboard->server;

    lua_pushcfunction(L, luaK_kiwmi_lua_callback_new);
    lua_pushlightuserdata(L, server);
    lua_pushvalue(L, 2);
    lua_pushlightuserdata(L, kiwmi_keyboard_on_key_down_or_up_notify);
    lua_pushlightuserdata(L, &keyboard->events.key_up);
    lua_pushlightuserdata(L, obj);

    if (lua_pcall(L, 5, 0, 0)) {
        wlr_log(WLR_ERROR, "%s", lua_tostring(L, -1));
        return 0;
    }

    return 0;
}

static const luaL_Reg kiwmi_keyboard_events[] = {
    {"destroy", l_kiwmi_keyboard_on_destroy},
    {"key_down", l_kiwmi_keyboard_on_key_down},
    {"key_up", l_kiwmi_keyboard_on_key_up},
    {NULL, NULL},
};

int
luaK_kiwmi_keyboard_new(lua_State *L)
{
    luaL_checktype(L, 1, LUA_TLIGHTUSERDATA); // kiwmi_lua
    luaL_checktype(L, 2, LUA_TLIGHTUSERDATA); // kiwmi_keyboard

    struct kiwmi_lua *lua           = lua_touserdata(L, 1);
    struct kiwmi_keyboard *keyboard = lua_touserdata(L, 2);

    struct kiwmi_object *obj =
        luaK_get_kiwmi_object(lua, keyboard, &keyboard->events.destroy);

    struct kiwmi_object **keyboard_ud =
        lua_newuserdata(L, sizeof(*keyboard_ud));
    luaL_getmetatable(L, "kiwmi_keyboard");
    lua_setmetatable(L, -2);

    *keyboard_ud = obj;

    return 1;
}

int
luaK_kiwmi_keyboard_register(lua_State *L)
{
    luaL_newmetatable(L, "kiwmi_keyboard");

    lua_pushvalue(L, -1);
    lua_setfield(L, -2, "__index");
    luaC_setfuncs(L, kiwmi_keyboard_methods, 0);

    luaC_newlib(L, kiwmi_keyboard_events);
    lua_setfield(L, -2, "__events");

    lua_pushcfunction(L, luaK_usertype_ref_equal);
    lua_setfield(L, -2, "__eq");

    lua_pushcfunction(L, luaK_kiwmi_object_gc);
    lua_setfield(L, -2, "__gc");

    return 0;
}