1 files changed, 132 insertions, 0 deletions
diff --git a/sowatchd/watchhandler.cpp b/sowatchd/watchhandler.cpp
new file mode 100644
index 0000000..e396c17
--- /dev/null
+++ b/sowatchd/watchhandler.cpp
@@ -0,0 +1,132 @@
+#include "watchhandler.h"
+
+using namespace sowatch;
+
+WatchHandler::WatchHandler(ConfigKey *config, QObject *parent)
+    : QObject(parent),
+      _config(config->getSubkey("", this))
+{
+	Registry *registry = Registry::registry();
+
+	qDebug() << "Starting watch handler on" << _config->key();
+
+	connect(_config, SIGNAL(subkeyChanged(QString)),
+	        SLOT(handleConfigSubkeyChanged(QString)));
+
+	const QString driver = _config->value("driver").toString();
+	if (driver.isEmpty()) {
+		qWarning() << "Watch" << _config->value("name") << "has no driver setting";
+		return;
+	}
+
+	WatchPluginInterface *watchPlugin = registry->getWatchPlugin(driver);
+	if (!watchPlugin) {
+		qWarning() << "Invalid driver" << driver;
+		return;
+	}
+
+	// Create the watch object from the plugin
+	_watch = watchPlugin->getWatch(driver, _config, this);
+	if (!_watch) {
+		qWarning() << "Driver" << driver << "failed to initiate watch";
+		return;
+	}
+
+	// Setup watch status connections
+	connect(_watch, SIGNAL(connected()),
+	        SIGNAL(statusChanged()));
+	connect(_watch, SIGNAL(connected()),
+	        SIGNAL(statusChanged()));
+
+	// Now create the UI server
+	_server = new WatchServer(_watch, this);
+
+	// Configure the server
+	_server->setNextWatchletButton(_config->value("next-watchlet-button").toString());
+
+	updateProviders();
+	updateWatchlets();
+}
+
+QString WatchHandler::status() const
+{
+	if (_watch->isConnected()) {
+		return "connected";
+	} else if (_config->value("enable").toBool()) {
+		return "enabled";
+	} else {
+		return "disabled";
+	}
+}
+
+void WatchHandler::updateWatchlets()
+{
+	Registry *registry = Registry::registry();
+	QStringList newWatchlets = _config->value("watchlets").toStringList();
+	QStringList curWatchlets = _watchlet_order;
+
+	// TODO: Something better than removing all and readding
+	foreach (const QString& s, curWatchlets) {
+		Watchlet* watchlet = _watchlets[s];
+		_server->removeWatchlet(watchlet);
+		delete watchlet;
+	}
+
+	_watchlet_order.clear();
+	_watchlets.clear();
+
+	foreach (const QString& s, newWatchlets) {
+		WatchletPluginInterface *plugin = registry->getWatchletPlugin(s);
+		if (!plugin) {
+			qWarning() << "Unknown watchlet" << s;
+			continue;
+		}
+		ConfigKey *subconfig = _config->getSubkey(s);
+		Watchlet* watchlet = plugin->getWatchlet(s, subconfig, _server);
+		_watchlet_order << s;
+		_watchlets[s] = watchlet;
+		_server->addWatchlet(watchlet);
+		delete subconfig;
+	}
+
+	qDebug() << "Watchlets reloaded";
+}
+
+void WatchHandler::updateProviders()
+{
+	Registry *registry = Registry::registry();
+	QSet<QString> curProviders = _providers.keys().toSet();
+	QSet<QString> newProviders = _config->value("providers").toStringList().toSet();
+	QSet<QString> removed = curProviders - newProviders;
+	QSet<QString> added = newProviders - curProviders;
+
+	foreach (const QString& s, removed) {
+		NotificationProvider *provider = _providers[s];
+		_server->removeProvider(provider);
+		delete provider;
+	}
+
+	foreach (const QString& s, added) {
+		NotificationPluginInterface *plugin = registry->getNotificationPlugin(s);
+		if (!plugin) {
+			qWarning() << "Unknown notification provider" << s;
+			continue;
+		}
+		ConfigKey *subconfig = _config->getSubkey(s);
+		NotificationProvider *provider = plugin->getProvider(s, subconfig, _server);
+		_server->addProvider(provider);
+		_providers[s] = provider;
+		delete subconfig;
+	}
+}
+
+void WatchHandler::handleConfigSubkeyChanged(const QString &subkey)
+{
+	if (subkey == "watchlets") {
+		updateWatchlets();
+	} else if (subkey == "providers") {
+		updateProviders();
+	} else if (subkey == "next-watchlet-button") {
+		_server->setNextWatchletButton(_config->value("next-watchlet-button").toString());
+	}
+}