mythwebsocketevent.cpp

Go to the documentation of this file.

#include "http/mythwebsocketevent.h"
#include "mythcorecontext.h"
#include "mythlogging.h"
 
MythWebSocketEvent::MythWebSocketEvent()
{
    setObjectName("MythWebSocketEvent");
    gCoreContext->addListener(this);
}
 
MythWebSocketEvent::~MythWebSocketEvent()
{
    gCoreContext->removeListener(this);
}
 
bool MythWebSocketEvent::HandleRawTextMessage(const DataPayloads& Payloads)
{
    // TODO: Payloads potentially has multiple messages to process
    // We only handle the first one here.
    StringPayload p_message = MythSharedString::CreateString();
    (*p_message).append(*(Payloads.front()));
    return HandleTextMessage(p_message);
}
 
bool MythWebSocketEvent::HandleTextMessage(const StringPayload& Text)
{
    const QString& message = *Text;
 
    if (message.isEmpty())
        return false;
 
    QStringList tokens = message.split(" ", Qt::SkipEmptyParts);
    if (tokens[0] == "WS_EVENT_ENABLE") // Only send events if asked
    {
        m_sendEvents = true;
        LOG(VB_HTTP, LOG_NOTICE, "WebSocketMythEvent: Enabled");
    }
    else if (tokens[0] == "WS_EVENT_DISABLE")
    {
        m_sendEvents = false;
        LOG(VB_HTTP, LOG_NOTICE, "WebSocketMythEvent: Disabled");
    }
    else if (tokens[0] == "WS_EVENT_SET_FILTER")
    {
        m_filters.clear();
 
        if (tokens.length() == 1)
            return true;
 
        m_filters = tokens.mid(1);
 
        QString filterString = m_filters.join(", ");
        LOG(VB_HTTP, LOG_NOTICE, QString("WebSocketMythEvent: Updated filters (%1)").arg(filterString));
    }
 
    return false;
}
 
void MythWebSocketEvent::customEvent(QEvent* event)
{
    if (event->type() == MythEvent::kMythEventMessage)
    {
        if (!m_sendEvents)
            return;
 
        auto *me = dynamic_cast<MythEvent *>(event);
        if (me == nullptr)
            return;
        QString message = me->Message();
 
        if (message.startsWith("SYSTEM_EVENT"))
            message.remove(0, 13); // Strip SYSTEM_EVENT from the frontend, it's not useful
 
        QStringList tokens = message.split(" ", Qt::SkipEmptyParts);
        if (tokens.isEmpty())
            return;
 
        // If no-one is listening for this event, then ignore it
        if (!m_filters.contains("ALL") && !m_filters.contains(tokens[0]))
            return;
 
        emit SendTextMessage(message);
    }
}