X-Git-Url: https://git.quassel-irc.org/?a=blobdiff_plain;f=src%2Fclient%2Fmessagemodel.cpp;h=76104bd241a2dcd0e84c21fd8463d356a6d7b52a;hb=39e2a78383295f86c5aa2dadbeac6f02b53eb7a4;hp=3be7601a551aa2f5322a8a615528014bde80d638;hpb=32b5e6447458a8e6e8074c40988b2d06f327094f;p=quassel.git

diff --git a/src/client/messagemodel.cpp b/src/client/messagemodel.cpp
index 3be7601a..76104bd2 100644
--- a/src/client/messagemodel.cpp
+++ b/src/client/messagemodel.cpp
@@ -34,8 +34,9 @@ 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);
 }
 
@@ -52,7 +53,7 @@ bool MessageModel::setData(const QModelIndex &index, const QVariant &value, int
 void MessageModel::insertMessage(const Message &msg) {
   MsgId id = msg.msgId();
   int idx = indexForId(id);
-  MessageItem *item = createMessageItem(msg);
+  MessageModelItem *item = createMessageModelItem(msg);
   beginInsertRows(QModelIndex(), idx, idx);
   _messageList.insert(idx, item);
   endInsertRows();
@@ -81,7 +82,7 @@ int MessageModel::indexForId(MsgId id) {
 
 /**********************************************************************************/
 
-MessageItem::MessageItem(const Message &msg) {
+MessageModelItem::MessageModelItem(const Message &msg) {
   _timestamp = msg.timestamp();
   _msgId = msg.msgId();
   _bufferId = msg.bufferInfo().bufferId();
@@ -90,12 +91,12 @@ MessageItem::MessageItem(const Message &msg) {
 
 }
 
-MessageItem::~MessageItem() {
+MessageModelItem::~MessageModelItem() {
 
 }
 
-QVariant MessageItem::data(int column, int role) const {
-  if(column < MessageModel::TimestampColumn || column > MessageModel::TextColumn) return QVariant();
+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);