1 files changed, 128 insertions, 0 deletions
diff --git a/sowatchui/providersmodel.cpp b/sowatchui/providersmodel.cpp
new file mode 100644
index 0000000..8e93e5c
--- /dev/null
+++ b/sowatchui/providersmodel.cpp
@@ -0,0 +1,128 @@
+#include <QtDebug>
+
+#include "providersmodel.h"
+
+using namespace sowatch;
+
+static const QString providersSubKey("/providers");
+
+ProvidersModel::ProvidersModel(QObject *parent) :
+    QAbstractListModel(parent),
+    _config(0)
+{
+	QHash<int, QByteArray> roles = roleNames();
+	roles[Qt::DisplayRole] = QByteArray("title");
+	roles[NameRole] = QByteArray("name");
+	roles[EnabledRole] = QByteArray("enabled");
+	setRoleNames(roles);
+}
+
+QString ProvidersModel::configKey() const
+{
+	if (_config) {
+		QString key = _config->key();
+		return key.left(key.length() - providersSubKey.length());
+	} else {
+		return QString();
+	}
+}
+
+void ProvidersModel::setConfigKey(const QString &configKey)
+{
+	QString oldConfigKey = this->configKey();
+	if (_config) {
+		delete _config;
+		_config = 0;
+	}
+	if (!configKey.isEmpty()) {
+		_config = new GConfKey(configKey + providersSubKey, this);
+		connect(_config, SIGNAL(changed()), SLOT(handleConfigChanged()));
+	}
+	if (this->configKey() != oldConfigKey) {
+		reload();
+		emit configKeyChanged();
+	}
+}
+
+int ProvidersModel::rowCount(const QModelIndex &parent) const
+{
+	Q_UNUSED(parent);
+	return _all_list.count();
+}
+
+QVariant ProvidersModel::data(const QModelIndex &index, int role) const
+{
+	switch (role) {
+	case Qt::DisplayRole:
+		return QVariant::fromValue(_info_list[index.row()].name);
+	case NameRole:
+		return QVariant::fromValue(_all_list[index.row()]);
+	case EnabledRole:
+		return QVariant::fromValue(_enabled.contains(_all_list[index.row()]));
+	}
+	return QVariant();
+}
+
+bool ProvidersModel::setData(const QModelIndex &index, const QVariant &value, int role)
+{
+	switch (role) {
+	case EnabledRole:
+		setProviderEnabled(_all_list[index.row()], value.toBool());
+		return true;
+	}
+	return false;
+}
+
+void ProvidersModel::setProviderEnabled(const QString &id, bool enabled)
+{
+	if (enabled) {
+		_enabled.insert(id);
+	} else {
+		_enabled.remove(id);
+	}
+
+	qDebug() << "Storing providers" << _enabled;
+
+	_config->set(QVariant::fromValue(QStringList(_enabled.toList())));
+}
+
+void ProvidersModel::reload()
+{
+	Registry *registry = Registry::registry();
+	beginResetModel();
+	_all_list.clear();
+	_info_list.clear();
+	_enabled.clear();
+
+	qDebug() << "Reloading providers";
+
+	_all_list = registry->allNotificationProviders();
+	_info_list.reserve(_all_list.size());
+	_all_list.sort();
+	foreach (const QString& s, _all_list) {
+		NotificationPluginInterface *plugin = registry->getNotificationPlugin(s);
+		if (plugin) {
+			_info_list.append(plugin->describeProvider(s));
+		} else {
+			NotificationPluginInterface::NotificationProviderInfo info;
+			info.name = s;
+			_info_list.append(info);
+		}
+	}
+
+	_enabled = _config->value().toStringList().toSet();
+	endResetModel();
+}
+
+void ProvidersModel::handleConfigChanged()
+{
+	QSet<QString> prevEnabled = _enabled;
+	_enabled = _config->value().toStringList().toSet();
+
+	for (int i = 0; i < _all_list.count(); i++) {
+		const QString& id = _all_list[i];
+		if (_enabled.contains(id) != prevEnabled.contains(id)) {
+			emit dataChanged(createIndex(i, 0), createIndex(i, 0));
+		}
+	}
+}