1 files changed, 116 insertions, 0 deletions
diff --git a/libsowatch/watchletsmodel.cpp b/libsowatch/watchletsmodel.cpp
new file mode 100644
index 0000000..9e06928
--- /dev/null
+++ b/libsowatch/watchletsmodel.cpp
@@ -0,0 +1,116 @@
+#include <QtCore/QDebug>
+
+#include "registry.h"
+#include "watchletplugininterface.h"
+#include "watchletsmodel.h"
+
+using namespace sowatch;
+
+WatchletsModel::WatchletsModel(QObject *parent) :
+    QAbstractListModel(parent)
+{
+	QHash<int, QByteArray> roles = roleNames();
+	roles[TitleRole] = QByteArray("title");
+	roles[ObjectRole] = QByteArray("object");
+	setRoleNames(roles);
+}
+
+int WatchletsModel::rowCount(const QModelIndex &parent) const
+{
+	return parent.isValid() ? 0 : _list.count();
+}
+
+QVariant WatchletsModel::data(const QModelIndex &index, int role) const
+{
+	if (!index.isValid()) return QVariant();
+	const int row = index.row();
+	if (row >= _list.size()) {
+		qWarning() << "Could not seek to" << row;
+		return QVariant();
+	}
+
+	const Watchlet *watchlet = _list.at(row);
+	const WatchletInfo& info = _info.at(row);
+
+	switch (role) {
+	case Qt::DisplayRole:
+		return QVariant::fromValue(info.name);
+	case ObjectRole:
+		return QVariant::fromValue(const_cast<sowatch::Watchlet*>(watchlet));
+	}
+	return QVariant();
+}
+
+int WatchletsModel::size() const
+{
+	return _list.size();
+}
+
+Watchlet * WatchletsModel::at(int position) const
+{
+	Q_ASSERT(position < _list.size());
+	return const_cast<sowatch::Watchlet*>(_list.at(position));
+}
+
+void WatchletsModel::add(Watchlet *w)
+{
+	insert(_list.size(), w);
+}
+
+void WatchletsModel::insert(int position, Watchlet *w)
+{
+	Q_ASSERT(position >= 0 && position <= _list.size());
+
+	beginInsertRows(QModelIndex(), position, position);
+	_list.insert(position, w);
+	_info.insert(position, getInfoForWatchlet(w));
+	endInsertRows();
+
+	emit modelChanged();
+}
+
+void WatchletsModel::move(const Watchlet *w, int to)
+{
+	move(_list.indexOf(const_cast<Watchlet*>(w)), to);
+}
+
+void WatchletsModel::move(int position, int to)
+{
+	Q_ASSERT(position >= 0 && position < _list.size());
+	Q_ASSERT(to >= 0 && to < _list.size());
+
+	beginMoveRows(QModelIndex(), position, position, QModelIndex(), to);
+	_list.move(position, to);
+	_info.move(position, to);
+	endMoveRows();
+
+	emit modelChanged();
+}
+
+void WatchletsModel::remove(const Watchlet *w)
+{
+	remove(_list.indexOf(const_cast<Watchlet*>(w)));
+}
+
+void WatchletsModel::remove(int position)
+{
+	Q_ASSERT(position >= 0 && position < _list.size());
+
+	beginRemoveRows(QModelIndex(), position, position);
+	_list.removeAt(position);
+	_info.removeAt(position);
+	endRemoveRows();
+
+	emit modelChanged();
+}
+
+WatchletsModel::WatchletInfo WatchletsModel::getInfoForWatchlet(const Watchlet *w)
+{
+	QString id = w->id();
+	WatchletPluginInterface *plugin = Registry::registry()->getWatchletPlugin(id);
+	if (plugin) {
+		return plugin->describeWatchlet(id);
+	} else {
+		return WatchletInfo();
+	}
+}