Clean up MessageModel on disconnect

[quassel.git] / src / client / messagemodel.cpp

diff --git a/src/client/messagemodel.cpp b/src/client/messagemodel.cpp
index d53449e..4032c64 100644 (file)
--- a/src/client/messagemodel.cpp
+++ b/src/client/messagemodel.cpp
@@ -23,19 +23,20 @@
 #include "message.h"
 
 MessageModel::MessageModel(QObject *parent) : QAbstractItemModel(parent) {
-  
-  
-  
+
+
+
 }
 
 MessageModel::~MessageModel() {
-  
-  
+
+
 }
 
 QVariant MessageModel::data(const QModelIndex &index, int role) const {
-  int row = index.row();
-  if(row < 0 || row >= _messageList.count()) return QVariant();
+  int row = index.row(); int column = index.column();
+  if(row < 0 || row >= _messageList.count() || column < 0) return QVariant();
+  if(role == ColumnTypeRole) return column;
   return _messageList[row]->data(index.column(), role);
 }
 
@@ -43,7 +44,7 @@ bool MessageModel::setData(const QModelIndex &index, const QVariant &value, int
   int row = index.row();
   if(row < 0 || row >= _messageList.count()) return false;
   if(_messageList[row]->setData(index.column(), value, role)) {
-    emit dataChanged(index, index); // FIXME make msg emit this (too)
+    emit dataChanged(index, index);
     return true;
   }
   return false;
@@ -51,8 +52,8 @@ bool MessageModel::setData(const QModelIndex &index, const QVariant &value, int
 
 void MessageModel::insertMessage(const Message &msg) {
   MsgId id = msg.msgId();
-  int idx = indexForId(id); qDebug() << "inserting at" << idx << msg.text();
-  MessageItem *item = createMessageItem(msg);
+  int idx = indexForId(id);
+  MessageModelItem *item = createMessageModelItem(msg);
   beginInsertRows(QModelIndex(), idx, idx);
   _messageList.insert(idx, item);
   endInsertRows();
@@ -65,6 +66,12 @@ void MessageModel::insertMessages(const QList<Message> &msglist) {
 
 }
 
+void MessageModel::clear() {
+  reset();
+  qDeleteAll(_messageList);
+  _messageList.clear();
+}
+
 // returns index of msg with given Id or of the next message after that (i.e., the index where we'd insert this msg)
 int MessageModel::indexForId(MsgId id) {
   if(_messageList.isEmpty() || id <= _messageList[0]->data(0, MsgIdRole).value<MsgId>()) return 0;
@@ -81,3 +88,28 @@ int MessageModel::indexForId(MsgId id) {
 
 /**********************************************************************************/
 
+MessageModelItem::MessageModelItem(const Message &msg) {
+  _timestamp = msg.timestamp();
+  _msgId = msg.msgId();
+  _bufferId = msg.bufferInfo().bufferId();
+  _type = msg.type();
+  _flags = msg.flags();
+
+}
+
+MessageModelItem::~MessageModelItem() {
+
+}
+
+QVariant MessageModelItem::data(int column, int role) const {
+  if(column < MessageModel::TimestampColumn || column > MessageModel::ContentsColumn) return QVariant();
+  switch(role) {
+    case MessageModel::MsgIdRole: return QVariant::fromValue<MsgId>(_msgId);
+    case MessageModel::BufferIdRole: return QVariant::fromValue<BufferId>(_bufferId);
+    case MessageModel::TypeRole: return _type;
+    case MessageModel::FlagsRole: return (int)_flags;
+    case MessageModel::TimestampRole: return _timestamp;
+    default: return QVariant();
+  }
+}
+