obs-cd-rec-status/cd-rec-status.cpp

#include <obs-module.h>
#include <obs-frontend-api.h>
#include <QLabel>
#include <QTimer>
#include <QVBoxLayout>
#include <QWidget>
#include <QDockWidget>
#include <QWebSocket>
#include <QJsonDocument>
#include <QJsonObject>

OBS_DECLARE_MODULE()
OBS_MODULE_USE_DEFAULT_LOCALE("cd-rec-status", "en-US")

class CDStatusDock : public QWidget {
public:
    CDStatusDock(QWidget *parent = nullptr) : QWidget(parent), socket(new QWebSocket()), retryDotCount(0) {
        QVBoxLayout *layout = new QVBoxLayout(this);
        statusLabel = new QLabel("Connecting to websocket...", this);
        layout->addWidget(statusLabel);
        setLayout(layout);

        retryDotCount = 0;

        // Retry every 3s
        reconnectTimer = new QTimer(this);
        reconnectTimer->setInterval(3000);
        connect(reconnectTimer, &QTimer::timeout, this, &CDStatusDock::tryReconnect);

        // Animate retrying text every 500ms
        retryAnimationTimer = new QTimer(this);
        retryAnimationTimer->setInterval(500);
        connect(retryAnimationTimer, &QTimer::timeout, this, &CDStatusDock::updateRetryingLabel);

        connect(socket, &QWebSocket::connected, this, &CDStatusDock::onConnected);
        connect(socket, &QWebSocket::disconnected, this, &CDStatusDock::onDisconnected);
        connect(socket, &QWebSocket::textMessageReceived, this, &CDStatusDock::onMessageReceived);

        socket->open(QUrl(QStringLiteral("ws://localhost:8765")));
    }

    ~CDStatusDock() {
        reconnectTimer->stop();
	retryAnimationTimer->stop();
        socket->close();
        delete socket;
    }

private slots:
    void onConnected() {
        reconnectTimer->stop();
        retryAnimationTimer->stop();
        statusLabel->setText("Connected to websocket");
        updateBackgroundColor(Qt::gray);
    }


    void onDisconnected() {
        retryDotCount = 0;
        reconnectTimer->start();
        retryAnimationTimer->start();
        updateRetryingLabel();
        updateBackgroundColor(Qt::red);
    }

    void tryReconnect() {
        if (socket->state() == QAbstractSocket::ConnectedState ||
            socket->state() == QAbstractSocket::ConnectingState)
            return;

        socket->abort();  // force close if half-open
        socket->open(QUrl(QStringLiteral("ws://localhost:8765")));

        updateRetryingLabel();
    }

    void onMessageReceived(const QString &message) {
        QJsonParseError parseError;
        QJsonDocument doc = QJsonDocument::fromJson(message.toUtf8(), &parseError);
        if (parseError.error != QJsonParseError::NoError || !doc.isObject()) {
            statusLabel->setText("Invalid status message");
            updateBackgroundColor(Qt::red);
            return;
        }

        QJsonObject obj = doc.object();
        bool isRecording = obj["recording"].toBool();

        QString statusText;
        if (isRecording) {
            statusText =
                QString("Recording: active\nCD: %1\nTrack: %2\nElapsed CD Time: %3\nElapsed Track Time: %4")
                    .arg(obj["cd"].toInt())
                    .arg(obj["track"].toInt())
                    .arg(obj["cd_time"].toString())
                    .arg(obj["track_time"].toString());
        } else {
            statusText = QString("Recording is currently not active.");
        }

        statusLabel->setText(statusText);
        updateBackgroundColor(isRecording ? QColor("#228B22") : Qt::gray);
    }

private:
    void updateBackgroundColor(const QColor &bgColor) {
        this->setStyleSheet(QString(
            "background-color: %1;"
            "color: %2;"
        ).arg(bgColor.name())
         .arg(isColorDark(bgColor) ? "white" : "black"));
    }

    bool isColorDark(const QColor &color) {
        // Perceived brightness formula
        int brightness = (color.red() * 299 + color.green() * 587 + color.blue() * 114) / 1000;
        return brightness < 128;
    }

    void updateRetryingLabel() {
        retryDotCount = (retryDotCount % 3) + 1;
        QString dots = QString(".").repeated(retryDotCount);
        statusLabel->setText(QString("Disconnected from websocket\nRetrying%1").arg(dots));
    }



private:
    QLabel *statusLabel;
    QWebSocket *socket;

    QTimer *reconnectTimer;
    QTimer *retryAnimationTimer;
    int retryDotCount;
};

static void *create_cd_status_dock(obs_source_t *) {
    return new CDStatusDock();
}

bool obs_module_load(void)
{
    QDockWidget *dock = new QDockWidget("CD Rec Status");
    dock->setObjectName("cd-rec-status-dock");
    dock->setWidget(new CDStatusDock());

    obs_frontend_add_custom_qdock("cd_status_dock", (void *)dock);

    blog(LOG_INFO, "CD Rec Status Dock loaded!");
    return true;
}

MODULE_EXPORT const char *obs_module_description(void)
{
    return obs_module_text("Description");
}

MODULE_EXPORT const char *obs_module_name(void)
{
    return obs_module_text("CD Rec Status");
}