Add MQTT connection tab

data/locale/en-US.ini

@@ -132,6 +132,18 @@ AdvSceneSwitcher.twitchConnectionTab.no="No"
 AdvSceneSwitcher.twitchConnectionTab.removeSingleConnectionPopup.text="Are you sure you want to remove \"%1\"?"
 AdvSceneSwitcher.twitchConnectionTab.removeMultipleConnectionsPopup.text="Are you sure you want to remove %1 Twitch connections?"
 
+# MQTT Connections Tab
+AdvSceneSwitcher.mqttConnectionTab.title="MQTT Connections"
+AdvSceneSwitcher.mqttConnectionTab.help="MQTT connections can be used to communicate with other devices.\n\nClick on the highlighted plus symbol to add a new connection."
+AdvSceneSwitcher.mqttConnectionTab.mqttonnectionAddButton.tooltip="Add new MQTT connection"
+AdvSceneSwitcher.mqttConnectionTab.mqttConnectionRemoveButton.tooltip="Remove selected MQTT connections"
+AdvSceneSwitcher.mqttConnectionTab.name.header="Name"
+AdvSceneSwitcher.mqttConnectionTab.address.header="Address"
+AdvSceneSwitcher.mqttConnectionTab.topics.header="Topic subscription count"
+AdvSceneSwitcher.mqttConnectionTab.status.header="Status"
+AdvSceneSwitcher.mqttConnectionTab.removeSingleConnectionPopup.text="Are you sure you want to remove \"%1\"?"
+AdvSceneSwitcher.mqttConnectionTab.removeMultipleConnectionsPopup.text="Are you sure you want to remove %1 connections?"
+
 # Macro Tab
 AdvSceneSwitcher.macroTab.title="Macro"
 AdvSceneSwitcher.macroTab.macros="Macros"

plugins/mqtt/CMakeLists.txt

@@ -31,6 +31,8 @@ target_sources(
           macro-condition-mqtt.cpp
           mqtt-helpers.cpp
           mqtt-helpers.hpp
+          mqtt-tab.cpp
+          mqtt-tab.hpp
           topic-selection.cpp
           topic-selection.hpp)
 

plugins/mqtt/mqtt-helpers.hpp

@@ -40,6 +40,8 @@ public:
 	void Save(obs_data_t *data) const;
 	MqttMessageBuffer RegisterForEvents();
 	bool ConnectOnStartup() const { return _connectOnStart; }
+	QString GetURI() const { return QString::fromStdString(_uri); }
+	int GetTopicSubscriptionCount() const { return _topics.size(); }
 	QString GetStatus() const;
 
 private:

plugins/mqtt/mqtt-tab.cpp

@@ -0,0 +1,232 @@
+#include "mqtt-tab.hpp"
+#include "mqtt-helpers.hpp"
+#include "obs-module-helper.hpp"
+#include "sync-helpers.hpp"
+#include "tab-helpers.hpp"
+#include "ui-helpers.hpp"
+
+#include <QTabWidget>
+#include <QTimer>
+
+namespace advss {
+
+static bool registerTab();
+static void setupTab(QTabWidget *);
+static bool registerTabDone = registerTab();
+
+static MqttConnectionsTable *tabWidget = nullptr;
+
+static bool registerTab()
+{
+	AddSetupTabCallback("mqttConnectionTab", MqttConnectionsTable::Create,
+			    setupTab);
+	return true;
+}
+
+static void setTabVisible(QTabWidget *tabWidget, bool visible)
+{
+	SetTabVisibleByName(
+		tabWidget, visible,
+		obs_module_text("AdvSceneSwitcher.mqttConnectionTab.title"));
+}
+
+MqttConnectionsTable *MqttConnectionsTable::Create()
+{
+	tabWidget = new MqttConnectionsTable();
+	return tabWidget;
+}
+
+void MqttConnectionsTable::Add()
+{
+	auto newConnection = std::make_shared<MqttConnection>();
+	auto accepted = MqttConnectionSettingsDialog::AskForSettings(
+		GetSettingsWindow(), *newConnection);
+	if (!accepted) {
+		return;
+	}
+
+	{
+		auto lock = LockContext();
+		auto &connections = GetMqttConnections();
+		connections.emplace_back(newConnection);
+	}
+
+	MqttConnectionSignalManager::Instance()->Add(
+		QString::fromStdString(newConnection->Name()));
+}
+
+void MqttConnectionsTable::Remove()
+{
+	auto selectedRows =
+		tabWidget->Table()->selectionModel()->selectedRows();
+	if (selectedRows.empty()) {
+		return;
+	}
+
+	QStringList connectionNames;
+	for (auto row : selectedRows) {
+		auto cell = tabWidget->Table()->item(row.row(), 0);
+		if (!cell) {
+			continue;
+		}
+
+		connectionNames << cell->text();
+	}
+
+	int connectionNameCount = connectionNames.size();
+	if (connectionNameCount == 1) {
+		QString deleteWarning = obs_module_text(
+			"AdvSceneSwitcher.mqttConnectionTab.removeSingleConnectionPopup.text");
+		if (!DisplayMessage(deleteWarning.arg(connectionNames.at(0)),
+				    true)) {
+			return;
+		}
+	} else {
+		QString deleteWarning = obs_module_text(
+			"AdvSceneSwitcher.mqttConnectionTab.removeMultipleConnectionsPopup.text");
+		if (!DisplayMessage(deleteWarning.arg(connectionNameCount),
+				    true)) {
+			return;
+		}
+	}
+
+	{
+		auto lock = LockContext();
+		RemoveItemsByName(GetMqttConnections(), connectionNames);
+	}
+
+	for (const auto &name : connectionNames) {
+		MqttConnectionSignalManager::Instance()->Remove(name);
+	}
+}
+
+static QStringList getCellLabels(MqttConnection *connection,
+				 bool addName = true)
+{
+	assert(connection);
+
+	auto result = QStringList();
+	if (addName) {
+		result << QString::fromStdString(connection->Name());
+	}
+	result << connection->GetURI()
+	       << QString::number(connection->GetTopicSubscriptionCount())
+	       << connection->GetStatus();
+	return result;
+}
+
+static void openSettingsDialog()
+{
+	auto selectedRows =
+		tabWidget->Table()->selectionModel()->selectedRows();
+	if (selectedRows.empty()) {
+		return;
+	}
+
+	auto cell = tabWidget->Table()->item(selectedRows.last().row(), 0);
+	if (!cell) {
+		return;
+	}
+
+	auto weakConnection = GetWeakMqttConnectionByQString(cell->text());
+	auto connection = weakConnection.lock();
+	if (!connection) {
+		return;
+	}
+
+	auto oldName = connection->Name();
+	bool accepted = MqttConnectionSettingsDialog::AskForSettings(
+		GetSettingsWindow(), *connection.get());
+	if (accepted && oldName != connection->Name()) {
+		MqttConnectionSignalManager::Instance()->Rename(
+			QString::fromStdString(oldName),
+			QString::fromStdString(connection->Name()));
+	}
+}
+
+static const QStringList headers =
+	QStringList()
+	<< obs_module_text("AdvSceneSwitcher.mqttConnectionTab.name.header")
+	<< obs_module_text("AdvSceneSwitcher.mqttConnectionTab.address.header")
+	<< obs_module_text("AdvSceneSwitcher.mqttConnectionTab.topics.header")
+	<< obs_module_text("AdvSceneSwitcher.mqttConnectionTab.status.header");
+
+MqttConnectionsTable::MqttConnectionsTable(QTabWidget *parent)
+	: ResourceTable(
+		  parent,
+		  obs_module_text("AdvSceneSwitcher.mqttConnectionTab.help"),
+		  obs_module_text(
+			  "AdvSceneSwitcher.mqttConnectionTab.websocketConnectionAddButton.tooltip"),
+		  obs_module_text(
+			  "AdvSceneSwitcher.mqttConnectionTab.websocketConnectionRemoveButton.tooltip"),
+		  headers, openSettingsDialog)
+{
+	for (const auto &connection : GetMqttConnections()) {
+		auto c = std::static_pointer_cast<MqttConnection>(connection);
+		AddItemTableRow(Table(), getCellLabels(c.get()));
+	}
+
+	SetHelpVisible(GetMqttConnections().empty());
+}
+
+static void updateConnectionStatus(QTableWidget *table)
+{
+	for (int row = 0; row < table->rowCount(); row++) {
+		auto item = table->item(row, 0);
+		if (!item) {
+			continue;
+		}
+
+		auto weakConnection =
+			GetWeakMqttConnectionByQString(item->text());
+		auto connection = weakConnection.lock();
+		if (!connection) {
+			continue;
+		}
+
+		UpdateItemTableRow(table, row,
+				   getCellLabels(connection.get(), false));
+	}
+}
+
+static void setupTab(QTabWidget *tab)
+{
+	if (GetMqttConnections().empty()) {
+		setTabVisible(tab, false);
+	}
+
+	QWidget::connect(MqttConnectionSignalManager::Instance(),
+			 &MqttConnectionSignalManager::Rename, tab,
+			 [](const QString &oldName, const QString &newName) {
+				 RenameItemTableRow(tabWidget->Table(), oldName,
+						    newName);
+			 });
+	QWidget::connect(
+		MqttConnectionSignalManager::Instance(),
+		&MqttConnectionSignalManager::Add, tab,
+		[tab](const QString &name) {
+			AddItemTableRow(
+				tabWidget->Table(),
+				getCellLabels(GetMqttConnectionByName(name)));
+			tabWidget->SetHelpVisible(false);
+			tabWidget->HighlightAddButton(false);
+			setTabVisible(tab, true);
+		});
+	QWidget::connect(MqttConnectionSignalManager::Instance(),
+			 &MqttConnectionSignalManager::Remove, tab,
+			 [](const QString &name) {
+				 RemoveItemTableRow(tabWidget->Table(), name);
+				 if (tabWidget->Table()->rowCount() == 0) {
+					 tabWidget->SetHelpVisible(true);
+					 tabWidget->HighlightAddButton(true);
+				 }
+			 });
+
+	auto timer = new QTimer(tabWidget);
+	timer->setInterval(1000);
+	QWidget::connect(timer, &QTimer::timeout,
+			 []() { updateConnectionStatus(tabWidget->Table()); });
+	timer->start();
+}
+
+} // namespace advss

plugins/mqtt/mqtt-tab.hpp

@@ -0,0 +1,20 @@
+#pragma once
+#include "resource-table.hpp"
+
+namespace advss {
+
+class MqttConnectionsTable final : public ResourceTable {
+	Q_OBJECT
+
+public:
+	static MqttConnectionsTable *Create();
+
+private slots:
+	void Add();
+	void Remove();
+
+private:
+	MqttConnectionsTable(QTabWidget *parent = nullptr);
+};
+
+} // namespace advss