Hyprland/src/config/lua/objects/LuaEventSubscription.cpp

#include "LuaEventSubscription.hpp"

#include <format>
#include <string_view>

using namespace Config::Lua;

static constexpr const char* MT = "HL.EventSubscription";

namespace {
    struct SEventSubscriptionRef {
        CLuaEventHandler* handler = nullptr;
        uint64_t          handle  = 0;
        bool              active  = false;
    };
}

static int eventSubscriptionEq(lua_State* L) {
    const auto* lhs = sc<SEventSubscriptionRef*>(luaL_checkudata(L, 1, MT));
    const auto* rhs = sc<SEventSubscriptionRef*>(luaL_checkudata(L, 2, MT));

    lua_pushboolean(L, lhs->handler == rhs->handler && lhs->handle == rhs->handle);
    return 1;
}

static int eventSubscriptionToString(lua_State* L) {
    const auto* ref = sc<SEventSubscriptionRef*>(luaL_checkudata(L, 1, MT));

    const auto  str = std::format("HL.EventSubscription({},{})", ref->handle, ref->active ? "active" : "inactive");
    lua_pushstring(L, str.c_str());
    return 1;
}

static int eventSubscriptionRemove(lua_State* L) {
    auto* ref = sc<SEventSubscriptionRef*>(luaL_checkudata(L, 1, MT));

    if (!ref->active || !ref->handler)
        return 0;

    ref->handler->unregisterEvent(ref->handle);
    ref->active = false;
    return 0;
}

static int eventSubscriptionIsActive(lua_State* L) {
    auto* ref = sc<SEventSubscriptionRef*>(luaL_checkudata(L, 1, MT));

    lua_pushboolean(L, ref->active);
    return 1;
}

static int eventSubscriptionIndex(lua_State* L) {
    luaL_checkudata(L, 1, MT);
    const std::string_view key = luaL_checkstring(L, 2);

    if (key == "remove")
        lua_pushcfunction(L, eventSubscriptionRemove);
    else if (key == "is_active")
        lua_pushcfunction(L, eventSubscriptionIsActive);
    else
        lua_pushnil(L);

    return 1;
}

void Objects::CLuaEventSubscription::setup(lua_State* L) {
    registerMetatable(L, MT, eventSubscriptionIndex, gcRef<SEventSubscriptionRef>, eventSubscriptionEq, eventSubscriptionToString);
}

void Objects::CLuaEventSubscription::push(lua_State* L, CLuaEventHandler* handler, uint64_t handle) {
    new (lua_newuserdata(L, sizeof(SEventSubscriptionRef))) SEventSubscriptionRef{.handler = handler, .handle = handle, .active = true};
    luaL_getmetatable(L, MT);
    lua_setmetatable(L, -2);
}