1 files changed, 153 insertions, 0 deletions

diff --git a/src/model/ThreadListModel.cpp b/src/model/ThreadListModel.cpp
new file mode 100644
index 0000000..02f0092
--- /dev/null
+++ b/src/model/ThreadListModel.cpp
@@ -0,0 +1,153 @@
+#include <sstream>
+#include <QQmlEngine>
+#include <QDebug>
+
+#include "ThreadListModel.h"
+
+
+ThreadListModel::ThreadListModel (const QList<Message *> &data, QObject * parent)
+  : QAbstractListModel (parent),
+    m_data (data)
+{
+  QListIterator<Message *> it (m_data);
+  while (it.hasNext ())
+  {
+    rowAdded (it.next ());
+  }
+}
+
+
+int
+ThreadListModel::rowCount (const QModelIndex & parent) const
+{
+  Q_UNUSED (parent);
+  return m_data.size ();
+}
+
+
+QVariant
+ThreadListModel::data (const QModelIndex & index, int role) const
+{
+  int idx = index.row ();
+  if (idx < 0 || m_data.count () <= idx)
+    return QVariant::Invalid;
+
+  Message *msg = m_data[idx];
+
+  switch (role)
+  {
+#if FIXME
+  case NAME:
+    return msg->name ();
+  case PUBKEY:
+    return msg->pubKeyString ();
+  case TYPE:
+    return msg->type ();
+  case DESC:
+    return msg->desc ();
+#endif
+  default:
+    return QVariant::Invalid;
+  }
+}
+
+
+QHash<int, QByteArray>
+ThreadListModel::roleNames () const
+{
+  QHash<int, QByteArray> roles;
+
+  roles[TITLE] = "title";
+  roles[TEXT] = "text";
+  roles[TYPE] = "type";
+  roles[PUBKEY] = "pubKey";
+
+  return roles;
+}
+
+
+/**
+ * A message was added to the source data.
+ */
+void
+ThreadListModel::rowAdded (Message *msg)
+{
+  int idx = m_data.count () - 1;
+  qDebug () << "ThreadListModel::rowAdded: " << idx;
+
+  m_thread += new ThreadModel (msg, this);
+
+  beginInsertRows (QModelIndex (), idx, idx);
+  endInsertRows ();
+
+  connect (msg, &Message::modified,
+           this, &ThreadListModel::rowModified);
+
+  connect (msg, &Message::childAdded,
+           m_thread[idx], &ThreadModel::rowAdded);
+
+}
+
+
+void
+ThreadListModel::rowModified (Message *msg)
+{
+  QModelIndex index = this->index (msg->index ());
+  emit dataChanged (index, index);
+}
+
+
+int
+ThreadListModel::count ()
+{
+  return m_data.count ();
+}
+
+
+Message *
+ThreadListModel::get (int index)
+{
+  if (index < 0 || count () < index)
+    return NULL;
+
+  Message *msg = m_data[index];
+
+  //Set the ownership so QML don't delete it.
+  QQmlEngine::setObjectOwnership (msg, QQmlEngine::CppOwnership);
+
+  return msg;
+}
+
+
+Message *
+ThreadListModel::get (QModelIndex index)
+{
+  if (count () < index.row ())
+    return NULL;
+
+  return m_data[index.row ()];
+}
+
+
+#if FIXME
+bool
+ThreadListModel::contains (QString id)
+{
+  return m_lookupIndex.contains (id);
+}
+
+
+Message *
+ThreadListModel::get (QString key)
+{
+  if (m_lookupIndex.contains (key))
+  {
+    Message *msg = m_data[m_lookupIndex[key]];
+
+    return msg;
+
+  }
+  else
+    return NULL;
+}
+#endif