X-Git-Url: https://git.quassel-irc.org/?a=blobdiff_plain;f=src%2Fclient%2Fmessagemodel.cpp;h=940dda2016e44dfbc80169893bb7942f93ad5f5e;hb=eaba93b703ba5bca4edf09f4c076a00b529115cd;hp=11cd0245cd7df5352953889d5a8ea02c6882ad20;hpb=d52b78268ad6ab09f896508a94dd7be47a720153;p=quassel.git

diff --git a/src/client/messagemodel.cpp b/src/client/messagemodel.cpp
index 11cd0245..940dda20 100644
--- a/src/client/messagemodel.cpp
+++ b/src/client/messagemodel.cpp
@@ -20,52 +20,102 @@
 
 #include "messagemodel.h"
 
-MessageModel::MessageModel(QObject *parent) : QAbstractItemModel(parent) {
-  
-  
-  
-}
+#include "message.h"
 
-MessageModel::~MessageModel() {
-  
-  
+MessageModel::MessageModel(QObject *parent)
+  : QAbstractItemModel(parent)
+{
 }
 
 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);
 }
 
 bool MessageModel::setData(const QModelIndex &index, const QVariant &value, int role) {
   int row = index.row();
-  if(row < 0 || row >= _messageList.count()) return false;
-  if(_messageList[row]->setData(index.column(), role)) {
-    emit dataChanged(index, index); // FIXME make msg emit this (too)
+  if(row < 0 || row >= _messageList.count())
+    return false;
+
+  if(_messageList[row]->setData(index.column(), value, role)) {
+    emit dataChanged(index, index);
     return true;
   }
+
   return false;
 }
 
-void MessageModel::insertMessage(const Message &msg) {
+bool MessageModel::insertMessage(const Message &msg, bool fakeMsg) {
   MsgId id = msg.msgId();
-  MessageItem *item = createMessageItem(msg);
-  if(id > )
-    
+  int idx = indexForId(id);
+  if(!fakeMsg && idx < _messageList.count()) { // check for duplicate
+    if(_messageList[idx]->msgId() == id)
+      return false;
+  }
+
+  MessageModelItem *item = createMessageModelItem(msg);
+  beginInsertRows(QModelIndex(), idx, idx);
+  _messageList.insert(idx, item);
+  endInsertRows();
+  return true;
+}
+
+void MessageModel::insertMessages(const QList<Message> &msglist) {
+  if(msglist.isEmpty()) return;
+  // FIXME make this more efficient by grouping msgs
+  foreach(Message msg, msglist)
+    insertMessage(msg);
+}
+
+void MessageModel::clear() {
+  beginRemoveRows(QModelIndex(), 0, rowCount() - 1);
+  qDeleteAll(_messageList);
+  _messageList.clear();
+  endRemoveRows();
 }
 
 // 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.count() || id <= _messageList[0]->data(0, MsgIdRole).value<MsgId>()) return 0;
+  if(_messageList.isEmpty() || id <= _messageList.value(0)->data(0, MsgIdRole).value<MsgId>()) return 0;
   if(id > _messageList.last()->data(0, MsgIdRole).value<MsgId>()) return _messageList.count();
   // binary search
   int start = 0; int end = _messageList.count()-1;
-  int idx;
   while(1) {
-    if(start == end) return start;
-    idx = (end + start) / 2;
-    
+    if(end - start == 1) return end;
+    int pivot = (end + start) / 2;
+    if(id <= _messageList.value(pivot)->data(0, MsgIdRole).value<MsgId>()) end = pivot;
+    else start = pivot;
+  }
 }
 
 /**********************************************************************************/
 
+MessageModelItem::MessageModelItem(const Message &msg) :
+  _timestamp(msg.timestamp()),
+  _msgId(msg.msgId()),
+  _bufferId(msg.bufferInfo().bufferId()),
+  _type(msg.type()),
+  _flags(msg.flags())
+{
+}
+
+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();
+  }
+}
+