// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only

#include "JFJochViewerStatusBar.h"

#include <algorithm>

JFJochViewerStatusBar::JFJochViewerStatusBar(QWidget *parent) : QStatusBar(parent) {
    rate_label = new QLabel(this);
    rate_label->setFixedWidth(60);
    rate_label->setAlignment(Qt::AlignCenter);
    addPermanentWidget(rate_label);

    state_box = new QProgressBar(this);
    state_box->setFixedWidth(110);
    state_box->setRange(0, 1000);
    state_box->setValue(0);
    state_box->setTextVisible(true);
    state_box->setAlignment(Qt::AlignCenter);
    state_box->setFormat("");
    addPermanentWidget(state_box);
}

void JFJochViewerStatusBar::display(QString input) {
    showMessage(input, 60000);
}

void JFJochViewerStatusBar::setBrokerStatus(BrokerStatus status) {
    QString text;
    switch (status.state) {
        case JFJochState::Inactive:    text = "Inactive"; break;
        case JFJochState::Idle:        text = "Idle"; break;
        case JFJochState::Measuring:   text = "Measuring"; break;
        case JFJochState::Error:       text = "Error"; break;
        case JFJochState::Busy:        text = "Busy"; break;
        case JFJochState::Calibration: text = "Calibration"; break;
    }
    state_box->setFormat(text);

    if (status.progress.has_value())
        state_box->setValue(std::clamp(static_cast<int>(status.progress.value() * 1000.0f), 0, 1000));
    else
        state_box->setValue(1000.0);

    const QString chunk = (status.state == JFJochState::Error) ? "#d9534f" : "#5cb85c";
    state_box->setStyleSheet(QString(
        "QProgressBar { border: 1px solid #aaa; border-radius: 2px; background: transparent; }"
        "QProgressBar::chunk { background-color: %1; }").arg(chunk));

    if (status.message.has_value() && !status.message.value().empty())
        state_box->setToolTip(QString::fromStdString(status.message.value()));
    else
        state_box->setToolTip(text);
}

void JFJochViewerStatusBar::setHttpConnection(bool connected, QString addr) {
    if (addr.isEmpty()) {
        // No detector session at all -> blank the whole cluster (but keep the reserved space).
        state_box->setFormat("");
        state_box->setValue(0);
        state_box->setStyleSheet("");
        state_box->setToolTip("");
        rate_label->setText("");
        return;
    }

    if (!connected)  {
        // Connection lost: clear live readouts but keep the badge so the user sees why.
        state_box->setStyleSheet(QString(
       "QProgressBar { border: 1px solid #aaa; border-radius: 2px; background: transparent; }"
       "QProgressBar::chunk { background-color: #888888; }"));
        state_box->setFormat("Disconnected");
        state_box->setValue(0);
        state_box->setToolTip("");
        rate_label->setText("");
    }
}

void JFJochViewerStatusBar::setAutoloadMode(JFJochImageReadingWorker::AutoloadMode mode) {
    autoload_mode = mode;
}

void JFJochViewerStatusBar::setImageNumber(int64_t in_total_images, int64_t in_current_image) {
    total_images = in_total_images;
    current_image = in_current_image;
}

void JFJochViewerStatusBar::setLiveRate(double hz) {
    if (hz > 0.0)
        rate_label->setText(QString::number(hz, 'f', 1) + " Hz");
    else
        rate_label->setText("");
}