#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()) {
        QVBoxLayout *layout = new QVBoxLayout(this);
        statusLabel = new QLabel("Connecting to server...", this);
        layout->addWidget(statusLabel);
        setLayout(layout);

        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() {
        socket->close();
        delete socket;
    }

private slots:
    void onConnected() {
        statusLabel->setText("Connected to status server");
    }

    void onDisconnected() {
        statusLabel->setText("Disconnected from server");
    }

    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");
            return;
        }

        QJsonObject obj = doc.object();
        QString statusText = QString("Recording: %1\nCD: %2\nTrack: %3\nElapsed CD Time: %4\nElapsed Track Time: %5")
                                .arg(obj["recording"].toBool() ? "Yes" : "No")
                                .arg(obj["cd"].toInt())
                                .arg(obj["track"].toInt())
                                .arg(obj["cd_time"].toString())
                                .arg(obj["track_time"].toString());

        statusLabel->setText(statusText);
    }

private:
    QLabel *statusLabel;
    QWebSocket *socket;
};

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");
}