Hyprland/src/protocols/IdleNotify.cpp

#include "IdleNotify.hpp"
#include "../managers/eventLoop/EventLoopManager.hpp"

#define LOGM PROTO::idle->protoLog

static int onTimer(std::shared_ptr<CEventLoopTimer> self, void* data) {

    const auto NOTIF = (CExtIdleNotification*)data;

    NOTIF->onTimerFired();

    return 0;
}

CExtIdleNotification::CExtIdleNotification(SP<CExtIdleNotificationV1> resource_, uint32_t timeoutMs_) : resource(resource_), timeoutMs(timeoutMs_) {
    if (!resource_->resource())
        return;

    resource->setDestroy([this](CExtIdleNotificationV1* r) { PROTO::idle->destroyNotification(this); });
    resource->setOnDestroy([this](CExtIdleNotificationV1* r) { PROTO::idle->destroyNotification(this); });

    timer = std::make_shared<CEventLoopTimer>(std::nullopt, onTimer, this);
    g_pEventLoopManager->addTimer(timer);

    updateTimer();

    LOGM(LOG, "Registered idle-notification for {}ms", timeoutMs_);
}

CExtIdleNotification::~CExtIdleNotification() {
    g_pEventLoopManager->removeTimer(timer);
    timer.reset();
}

bool CExtIdleNotification::good() {
    return resource->resource();
}

void CExtIdleNotification::updateTimer() {
    if (PROTO::idle->isInhibited)
        timer->updateTimeout(std::nullopt);
    else
        timer->updateTimeout(std::chrono::milliseconds(timeoutMs));
}

void CExtIdleNotification::onTimerFired() {
    resource->sendIdled();
    idled = true;
}

void CExtIdleNotification::onActivity() {
    if (idled)
        resource->sendResumed();

    idled = false;
    updateTimer();
}

CIdleNotifyProtocol::CIdleNotifyProtocol(const wl_interface* iface, const int& ver, const std::string& name) : IWaylandProtocol(iface, ver, name) {
    ;
}

void CIdleNotifyProtocol::bindManager(wl_client* client, void* data, uint32_t ver, uint32_t id) {
    const auto RESOURCE = m_vManagers.emplace_back(std::make_unique<CExtIdleNotifierV1>(client, ver, id)).get();
    RESOURCE->setOnDestroy([this](CExtIdleNotifierV1* p) { this->onManagerResourceDestroy(p->resource()); });

    RESOURCE->setDestroy([this](CExtIdleNotifierV1* pMgr) { this->onManagerResourceDestroy(pMgr->resource()); });
    RESOURCE->setGetIdleNotification([this](CExtIdleNotifierV1* pMgr, uint32_t id, uint32_t timeout, wl_resource* seat) { this->onGetNotification(pMgr, id, timeout, seat); });
}

void CIdleNotifyProtocol::onManagerResourceDestroy(wl_resource* res) {
    std::erase_if(m_vManagers, [&](const auto& other) { return other->resource() == res; });
}

void CIdleNotifyProtocol::destroyNotification(CExtIdleNotification* notif) {
    std::erase_if(m_vNotifications, [&](const auto& other) { return other.get() == notif; });
}

void CIdleNotifyProtocol::onGetNotification(CExtIdleNotifierV1* pMgr, uint32_t id, uint32_t timeout, wl_resource* seat) {
    const auto CLIENT = pMgr->client();
    const auto RESOURCE =
        m_vNotifications.emplace_back(std::make_unique<CExtIdleNotification>(std::make_shared<CExtIdleNotificationV1>(CLIENT, pMgr->version(), id), timeout)).get();

    if (!RESOURCE->good()) {
        pMgr->noMemory();
        m_vNotifications.pop_back();
        return;
    }
}

void CIdleNotifyProtocol::onActivity() {
    for (auto& n : m_vNotifications) {
        n->onActivity();
    }
}

void CIdleNotifyProtocol::setInhibit(bool inhibited) {
    isInhibited = inhibited;
    for (auto& n : m_vNotifications) {
        n->onActivity();
    }
}