MqttMonitor/mainwindow.cpp at main · johnjiamzhong-project/MqttMonitor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
#include "mainwindow.h"
#include "ui_mainwindow.h"

#include <QWidget>
#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QButtonGroup>
#include <QIcon>
#include <QSize>
#include <iostream>

#ifdef Q_OS_WIN
#  include <windows.h>
#  include <dwmapi.h>
#  ifndef DWMWA_USE_IMMERSIVE_DARK_MODE
#    define DWMWA_USE_IMMERSIVE_DARK_MODE 20
#  endif
#endif

MainWindow::MainWindow(QWidget* parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    ui->setupUi(this);
    setWindowTitle("MqttMonitor");
    setWindowIcon(QIcon(":/icon/star.png"));

#ifdef Q_OS_WIN
    // 通知 DWM 对本窗口使用暗色标题栏（Win10 18985+ / Win11）
    HWND hwnd = reinterpret_cast<HWND>(winId());
    BOOL dark = TRUE;
    DwmSetWindowAttribute(hwnd, DWMWA_USE_IMMERSIVE_DARK_MODE, &dark, sizeof(dark));
#endif

    // --- Nav bar ---
    configBtn_ = new QPushButton("");
    deviceBtn_ = new QPushButton("");

    configBtn_->setIcon(QIcon(":/icon/tool.png"));
    deviceBtn_->setIcon(QIcon(":/icon/connect.png"));

    configBtn_->setIconSize(QSize(40, 40));
    deviceBtn_->setIconSize(QSize(40, 40));

    configBtn_->setCheckable(true);
    deviceBtn_->setCheckable(true);
    configBtn_->setChecked(true);   // default page

    // 互斥分组，保证同一时刻只有一个按钮处于 checked 状态
    auto* navGroup = new QButtonGroup(this);
    navGroup->setExclusive(true);
    navGroup->addButton(configBtn_);
    navGroup->addButton(deviceBtn_);

    auto* navBar    = new QWidget(this);
    navBar->setObjectName("navBar");   // QSS selector: QWidget#navBar
    auto* navLayout = new QVBoxLayout(navBar);
    navLayout->setContentsMargins(6, 10, 6, 10);
    navLayout->setSpacing(4);
    navLayout->addWidget(configBtn_);
    navLayout->addWidget(deviceBtn_);
    navLayout->addStretch();
    navBar->setFixedWidth(76);

    // --- Pages ---
    configPanel_ = new ConfigPanel;
    deviceView_  = new DeviceView;

    stack_ = new QStackedWidget(this);
    stack_->addWidget(configPanel_);   // index 0
    stack_->addWidget(deviceView_);    // index 1

    // --- Main layout ---
    auto* container = new QWidget(this);
    auto* hLayout   = new QHBoxLayout(container);
    hLayout->setContentsMargins(0, 0, 0, 0);
    hLayout->setSpacing(0);
    hLayout->addWidget(navBar);
    hLayout->addWidget(stack_);
    setCentralWidget(container);

    // --- Nav connections ---
    connect(configBtn_, &QPushButton::clicked, [this]{ stack_->setCurrentIndex(0); });
    connect(deviceBtn_, &QPushButton::clicked, [this]{ stack_->setCurrentIndex(1); });

    // --- ConfigPanel connections ---
    connect(configPanel_, &ConfigPanel::connectRequested,
            this, &MainWindow::onConnectRequested);
    connect(configPanel_, &ConfigPanel::disconnectRequested,
            this, &MainWindow::onDisconnectRequested);
}

MainWindow::~MainWindow()
{
    mqttBridge_.reset();
    mqttClient_.reset();
    delete ui;
}

void MainWindow::onConnectRequested(const QString& brokerUrl,
                                    const QString& clientId,
                                    const QString& username,
                                    const QString& password,
                                    const QString& topic)
{
    mqttBridge_.reset();
    mqttClient_ = std::make_unique<MqttClient>(brokerUrl.toStdString(),
                                               clientId.toStdString());

    const bool ok = mqttClient_->connect(username.toStdString(),
                                         password.toStdString());
    if (!ok) {
        ui->statusbar->showMessage("Connection failed: " + brokerUrl, 5000);
        mqttClient_.reset();
        return;
    }

    mqttClient_->subscribe(topic.toStdString());

    mqttBridge_ = std::make_unique<MqttBridge>(*mqttClient_);
    connect(mqttBridge_.get(), &MqttBridge::messageReceived,
            configPanel_, &ConfigPanel::appendMessage);
    connect(mqttBridge_.get(), &MqttBridge::messageReceived,
            deviceView_, &DeviceView::updateDevice);
    connect(mqttBridge_.get(), &MqttBridge::connectionLost,
            this, &MainWindow::onMqttConnectionLost);
    connect(deviceView_, &DeviceView::publishRequested,
            this, &MainWindow::onPublishRequested);

    configPanel_->onConnected();
    ui->statusbar->showMessage("Connected: " + brokerUrl);
}

void MainWindow::onPublishRequested(const QString& topic, const QString& payload, int qos)
{
    if (mqttClient_ && mqttClient_->isConnected())
        mqttClient_->publish(topic.toStdString(), payload.toStdString(), qos);
}

void MainWindow::onDisconnectRequested()
{
    deviceView_->stopDisconnectDetection();
    mqttBridge_.reset();
    if (mqttClient_) {
        mqttClient_->disconnect();
        mqttClient_.reset();
    }
    configPanel_->onDisconnected();
    ui->statusbar->showMessage("Disconnected", 3000);
}

void MainWindow::onMqttConnectionLost()
{
    std::cerr << "[MainWindow] onMqttConnectionLost called" << std::endl;
    // 标记所有设备为断联
    deviceView_->markAllDevicesDisconnected();
    ui->statusbar->showMessage("MQTT Connection Lost", 5000);
}