/*******************************************************************************
 * retroshare-gui/src/gui/msgs/RsMessageModel.cpp                              *
 *                                                                             *
 * Copyright 2019 by Cyril Soler <csoler@users.sourceforge.net>                *
 *                                                                             *
 * This program is free software: you can redistribute it and/or modify        *
 * it under the terms of the GNU Affero General Public License as              *
 * published by the Free Software Foundation, either version 3 of the          *
 * License, or (at your option) any later version.                             *
 *                                                                             *
 * This program is distributed in the hope that it will be useful,             *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of              *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the                *
 * GNU Affero General Public License for more details.                         *
 *                                                                             *
 * You should have received a copy of the GNU Affero General Public License    *
 * along with this program. If not, see <https://www.gnu.org/licenses/>.       *
 *                                                                             *
 *******************************************************************************/

#include <list>

#include <QApplication>
#include <QDateTime>
#include <QFontMetrics>
#include <QModelIndex>
#include <QIcon>

#include "gui/common/TagDefs.h"
#include "gui/common/FilesDefs.h"
#include "util/HandleRichText.h"
#include "util/DateTime.h"
#include "gui/gxs/GxsIdDetails.h"
#include "gui/gxs/GxsIdTreeWidgetItem.h"
#include "MessageModel.h"
#include "retroshare/rsexpr.h"
#include "retroshare/rsmsgs.h"

//#define DEBUG_MESSAGE_MODEL

#define IS_MESSAGE_UNREAD(flags) (flags &  (RS_MSG_NEW | RS_MSG_UNREAD_BY_USER))

#define IMAGE_STAR_ON          ":/images/star-on-16.png"
#define IMAGE_STAR_OFF         ":/images/star-off-16.png"
#define IMAGE_SPAM             ":/images/junk.png"
#define IMAGE_SPAM_ON          ":/images/junk_on.png"
#define IMAGE_SPAM_OFF         ":/images/junk_off.png"

std::ostream& operator<<(std::ostream& o, const QModelIndex& i);// defined elsewhere

const QString RsMessageModel::FilterString("filtered");

RsMessageModel::RsMessageModel(QObject *parent)
    : QAbstractItemModel(parent)
{
    mCurrentBox = Rs::Msgs::BoxName::BOX_NONE;
    mQuickViewFilter = QUICK_VIEW_ALL;
    mFilterType = FILTER_TYPE_NONE;
    mFilterStrings.clear();
    mFont = QApplication::font();
}

void RsMessageModel::preMods()
{
	emit layoutAboutToBeChanged();
}
void RsMessageModel::postMods()
{
	emit dataChanged(createIndex(0,0,(void*)NULL), createIndex(mMessages.size()-1,COLUMN_THREAD_NB_COLUMNS-1,(void*)NULL));
	emit layoutChanged();
}

int RsMessageModel::rowCount(const QModelIndex& parent) const
{
	if(!parent.isValid())
		return 0;

	if(parent.column() > 0)
		return 0;

	if(mMessages.empty())	// security. Should never happen.
		return 0;

	if(parent.internalPointer() == NULL)
		return mMessages.size();

	return 0;
}

int RsMessageModel::columnCount(const QModelIndex &/*parent*/) const
{
    return COLUMN_THREAD_NB_COLUMNS;
}

bool RsMessageModel::getMessageData(const QModelIndex& i,Rs::Msgs::MessageInfo& fmpe) const
{
	if(!i.isValid())
        return true;

    quintptr ref = i.internalId();
	uint32_t index = 0;

	if(!convertInternalIdToMsgIndex(ref,index) || index >= mMessages.size())
		return false ;

	return rsMsgs->getMessage(mMessages[index].msgId,fmpe);
}

bool RsMessageModel::hasChildren(const QModelIndex &parent) const
{
    if(!parent.isValid())
        return true;

    return false ;
}

bool RsMessageModel::convertMsgIndexToInternalId(uint32_t index,quintptr& id)
{
	// the internal id is set to the place in the table of items. We simply shift to allow 0 to mean something special.

    id = index + 1 ;

	return true;
}

bool RsMessageModel::convertInternalIdToMsgIndex(quintptr ref,uint32_t& index)
{
    if(ref == 0)
        return false ;

    index = ref - 1;
	return true;
}

QModelIndex RsMessageModel::index(int row, int column, const QModelIndex & parent) const
{
    if(row < 0 || column < 0 || column >= COLUMN_THREAD_NB_COLUMNS)
		return QModelIndex();

	quintptr ref ;

    if(parent.internalId() == 0 && convertMsgIndexToInternalId(row,ref))
		return createIndex(row,column,ref) ;

    return QModelIndex();
}

QModelIndex RsMessageModel::parent(const QModelIndex& index) const
{
	if(!index.isValid())
		return QModelIndex();

	return QModelIndex();
}

Qt::ItemFlags RsMessageModel::flags(const QModelIndex& index) const
{
    if (!index.isValid())
        return Qt::ItemFlags();

    return QAbstractItemModel::flags(index);
}

QVariant RsMessageModel::headerData(int section, Qt::Orientation /*orientation*/, int role) const
{
	if(role == Qt::DisplayRole)
		switch(section)
		{
		case COLUMN_THREAD_DATE:         return tr("Date");
		case COLUMN_THREAD_AUTHOR:       return tr("From");
        case COLUMN_THREAD_TO:           return tr("To");
        case COLUMN_THREAD_SUBJECT:      return tr("Subject");
		case COLUMN_THREAD_TAGS:         return tr("Tags");
		default:
			return QVariant();
		}

	if(role == Qt::DecorationRole)
		switch(section)
		{
		case COLUMN_THREAD_STAR:         return FilesDefs::getIconFromQtResourcePath(IMAGE_STAR_ON);
		case COLUMN_THREAD_SPAM:         return FilesDefs::getIconFromQtResourcePath(IMAGE_SPAM);
		case COLUMN_THREAD_READ:         return FilesDefs::getIconFromQtResourcePath(":/images/message-state-header.png");
		case COLUMN_THREAD_ATTACHMENT:   return FilesDefs::getIconFromQtResourcePath(":/icons/png/attachements.png");
		default:
			return QVariant();
		}

	if(role == Qt::ToolTipRole)
        switch(section)
        {
        case COLUMN_THREAD_ATTACHMENT: return tr("Click to sort by attachments");
        case COLUMN_THREAD_SUBJECT:    return tr("Click to sort by subject");
        case COLUMN_THREAD_READ:       return tr("Click to sort by read status");
        case COLUMN_THREAD_AUTHOR:     return tr("Click to sort by author");
        case COLUMN_THREAD_TO:         return tr("Click to sort by destination");
        case COLUMN_THREAD_DATE:       return tr("Click to sort by date");
        case COLUMN_THREAD_TAGS:       return tr("Click to sort by tags");
        case COLUMN_THREAD_STAR:       return tr("Click to sort by star");
        case COLUMN_THREAD_SPAM:       return tr("Click to sort by junk status");
		default:
			return QVariant();
        }
	return QVariant();
}

QVariant RsMessageModel::data(const QModelIndex &index, int role) const
{
#ifdef DEBUG_MESSAGE_MODEL
    std::cerr << "calling data(" << index << ") role=" << role << std::endl;
#endif

	int coln = index.column();

	if(!index.isValid())
		return QVariant();

	switch(role)
	{
	case Qt::SizeHintRole: return sizeHintRole(index.column()) ;
    case Qt::StatusTipRole:return QVariant();
    default: break;
	}

	quintptr ref = (index.isValid())?index.internalId():0 ;
	uint32_t entry = 0;

#ifdef DEBUG_MESSAGE_MODEL
	std::cerr << "data(" << index << ")" ;
#endif

	if(!ref)
	{
#ifdef DEBUG_MESSAGE_MODEL
		std::cerr << " [empty]" << std::endl;
#endif
		return QVariant() ;
	}

	if(!convertInternalIdToMsgIndex(ref,entry) || entry >= mMessages.size())
	{
#ifdef DEBUG_MESSAGE_MODEL
		std::cerr << "Bad pointer: " << (void*)ref << std::endl;
#endif
		return QVariant() ;
	}

	const Rs::Msgs::MsgInfoSummary& fmpe(mMessages[entry]);

    if(role == Qt::FontRole)
    {
        QFont font = mFont;
		font.setBold(fmpe.msgflags & (RS_MSG_NEW | RS_MSG_UNREAD_BY_USER));

        return QVariant(font);
    }
	
	if (role == Qt::TextAlignmentRole)
	{
		if((coln == COLUMN_THREAD_ATTACHMENT))
			return int( Qt::AlignHCenter | Qt::AlignVCenter);
		else
			return QVariant();
	}

#ifdef DEBUG_MESSAGE_MODEL
	std::cerr << " [ok]" << std::endl;
#endif

	switch(role)
	{
	case Qt::DisplayRole:    return displayRole   (fmpe,index.column()) ;
	case Qt::DecorationRole: return decorationRole(fmpe,index.column()) ;
	case Qt::ToolTipRole:	 return toolTipRole   (fmpe,index.column()) ;
	case Qt::UserRole:	 	 return userRole      (fmpe,index.column()) ;
	case Qt::TextColorRole:  return textColorRole (fmpe,index.column()) ;
	case Qt::BackgroundRole: return backgroundRole(fmpe,index.column()) ;

	case FilterRole:         return filterRole    (fmpe,index.column()) ;
	case StatusRole:         return statusRole    (fmpe,index.column()) ;
	case SortRole:           return sortRole      (fmpe,index.column()) ;
	case MsgFlagsRole:       return fmpe.msgflags ;
	case UnreadRole: 		 return fmpe.msgflags & (RS_MSG_NEW | RS_MSG_UNREAD_BY_USER);
	case MsgIdRole:          return QString::fromStdString(fmpe.msgId) ;
    case SrcIdRole:          return QString::fromStdString(fmpe.from.toStdString()) ;
	default:
		return QVariant();
	}
}

QVariant RsMessageModel::textColorRole(const Rs::Msgs::MsgInfoSummary& fmpe,int /*column*/) const
{
	Rs::Msgs::MsgTagType tags;
	rsMsgs->getMessageTagTypes(tags);

	for(auto it(fmpe.msgtags.begin());it!=fmpe.msgtags.end();++it)
		for(auto it2(tags.types.begin());it2!=tags.types.end();++it2)
			if(it2->first == *it)
				return QColor(it2->second.second);

	return QVariant();
}

QVariant RsMessageModel::statusRole(const Rs::Msgs::MsgInfoSummary& /*fmpe*/,int /*column*/) const
{
// 	if(column != COLUMN_THREAD_DATA)
//        return QVariant();

    return QVariant();//fmpe.mMsgStatus);
}

bool RsMessageModel::passesFilter(const Rs::Msgs::MsgInfoSummary& fmpe,int /*column*/) const
{
	QString s ;
	bool passes_strings = true ;

	if(!mFilterStrings.empty())
	{
		switch(mFilterType)
		{
		case FILTER_TYPE_SUBJECT: 	s = displayRole(fmpe,COLUMN_THREAD_SUBJECT).toString();
			break;

		case FILTER_TYPE_FROM:      s = sortRole(fmpe,COLUMN_THREAD_AUTHOR).toString();
			if(s.isNull())
				passes_strings = false;
			break;
        case FILTER_TYPE_TO:        s = sortRole(fmpe,COLUMN_THREAD_TO).toString();
            if(s.isNull())
                passes_strings = false;
            break;

		case FILTER_TYPE_DATE:   	s = displayRole(fmpe,COLUMN_THREAD_DATE).toString();
			break;
		case FILTER_TYPE_CONTENT:   {
			Rs::Msgs::MessageInfo minfo;
			rsMsgs->getMessage(fmpe.msgId,minfo);
			s = QTextDocument(QString::fromUtf8(minfo.msg.c_str())).toPlainText();
		}
			break;
		case FILTER_TYPE_TAGS:		s = displayRole(fmpe,COLUMN_THREAD_TAGS).toString();
			break;

		case FILTER_TYPE_ATTACHMENTS:
		{
			Rs::Msgs::MessageInfo minfo;
			rsMsgs->getMessage(fmpe.msgId,minfo);

			for(auto it(minfo.files.begin());it!=minfo.files.end();++it)
				s += QString::fromUtf8((*it).fname.c_str())+" ";
		}
			break;
		case FILTER_TYPE_NONE:
			RS_ERR("None Type for Filter.");
		};
	}

    if(!s.isNull())
		for(auto iter(mFilterStrings.begin()); iter != mFilterStrings.end(); ++iter)
			passes_strings = passes_strings && s.contains(*iter,Qt::CaseInsensitive);

    bool passes_quick_view =
            (mQuickViewFilter==QUICK_VIEW_ALL)
            || (std::find(fmpe.msgtags.begin(),fmpe.msgtags.end(),mQuickViewFilter) != fmpe.msgtags.end())
            || (mQuickViewFilter==QUICK_VIEW_STARRED && (fmpe.msgflags & RS_MSG_STAR))
            || (mQuickViewFilter==QUICK_VIEW_SYSTEM && (fmpe.msgflags & RS_MSG_SYSTEM))
            || (mQuickViewFilter==QUICK_VIEW_SPAM && (fmpe.msgflags & RS_MSG_SPAM))
            || (mQuickViewFilter==QUICK_VIEW_ATTACHMENT && (fmpe.count >= 1));
#ifdef DEBUG_MESSAGE_MODEL
    std::cerr << "Passes filter: type=" << mFilterType << " s=\"" << s.toStdString() << "MsgFlags=" << fmpe.msgflags << " msgtags=" ;
    foreach(uint32_t i,fmpe.msgtags) std::cerr << i << " " ;
    std::cerr          << "\" strings:" << passes_strings << " quick_view:" << passes_quick_view << std::endl;
#endif

    return passes_quick_view && passes_strings;
}

QVariant RsMessageModel::filterRole(const Rs::Msgs::MsgInfoSummary& fmpe,int column) const
{
	if(passesFilter(fmpe,column))
		return QVariant(FilterString);

	return QVariant(QString());
}

uint32_t RsMessageModel::updateFilterStatus(ForumModelIndex /*i*/,int /*column*/,const QStringList& /*strings*/)
{
	return 0;
}


void RsMessageModel::setFilter(FilterType filter_type, const QStringList& strings)
{
#ifdef DEBUG_MESSAGE_MODEL
	std::cerr << "Setting filter to filter_type=" << int(filter_type) << " and strings to " ;
	foreach(const QString& str,strings)
		std::cerr << "\"" << str.toStdString() << "\" " ;
	std::cerr << std::endl;
#endif

	mFilterType = filter_type;
	mFilterStrings = strings;

        if(rowCount() > 0)
            emit dataChanged(createIndex(0,0),createIndex(rowCount()-1,RsMessageModel::columnCount()-1));
}

void RsMessageModel::setFont(const QFont &font)
{
	mFont = font;

	if (rowCount() > 0) {
		emit dataChanged(createIndex(0,0), createIndex(rowCount() - 1, RsMessageModel::columnCount() - 1));
	}
}

QVariant RsMessageModel::toolTipRole(const Rs::Msgs::MsgInfoSummary& fmpe,int column) const
{
    if(column == COLUMN_THREAD_AUTHOR || column == COLUMN_THREAD_TO)
	{
		QString str,comment ;
		QList<QIcon> icons;

        if(column == COLUMN_THREAD_AUTHOR && !GxsIdDetails::MakeIdDesc(RsGxsId(fmpe.from.toStdString()), true, str, icons, comment,GxsIdDetails::ICON_TYPE_AVATAR))
			return QVariant();

        if(column == COLUMN_THREAD_TO && !GxsIdDetails::MakeIdDesc(RsGxsId(fmpe.to.toStdString()), true, str, icons, comment,GxsIdDetails::ICON_TYPE_AVATAR))
            return QVariant();

        int S = QFontMetricsF(QApplication::font()).height();
		QImage pix( (*icons.begin()).pixmap(QSize(5*S,5*S)).toImage());

		QString embeddedImage;
		if(RsHtml::makeEmbeddedImage(pix.scaled(QSize(5*S,5*S), Qt::KeepAspectRatio, Qt::SmoothTransformation), embeddedImage, -1))
		{
			embeddedImage.insert(embeddedImage.indexOf("src="), "style=\"float:left\" ");
			comment = "<table><tr><td>" + embeddedImage + "</td><td>" + comment + "</td></table>";
		}

		return comment;
	}

    return QVariant();
}

QVariant RsMessageModel::backgroundRole(const Rs::Msgs::MsgInfoSummary &/*fmpe*/, int /*column*/) const
{
    return QVariant();
}

QVariant RsMessageModel::sizeHintRole(int col) const
{
	float factor = QFontMetricsF(mFont).height()/14.0f ;

	switch(col)
	{
	default:
	case COLUMN_THREAD_SUBJECT:      return QVariant( QSize(factor * 170, factor*14 ));
	case COLUMN_THREAD_DATE:         return QVariant( QSize(factor * 75 , factor*14 ));
	case COLUMN_THREAD_AUTHOR:       return QVariant( QSize(factor * 75 , factor*14 ));
    case COLUMN_THREAD_TO:           return QVariant( QSize(factor * 75 , factor*14 ));
    }
}

QVariant RsMessageModel::authorRole(const Rs::Msgs::MsgInfoSummary& /*fmpe*/,int /*column*/) const
{
    return QVariant();
}

QVariant RsMessageModel::sortRole(const Rs::Msgs::MsgInfoSummary& fmpe,int column) const
{
    switch(column)
    {
	case COLUMN_THREAD_DATE:  return QVariant(QString::number(fmpe.ts)); // we should probably have leading zeroes here

	case COLUMN_THREAD_READ:  return QVariant((bool)IS_MESSAGE_UNREAD(fmpe.msgflags));

	case COLUMN_THREAD_STAR:  return QVariant((fmpe.msgflags & RS_MSG_STAR)? 1:0);

	case COLUMN_THREAD_SPAM:  return QVariant((fmpe.msgflags & RS_MSG_SPAM)? 1:0);

    case COLUMN_THREAD_AUTHOR:{
			QString name;

            if(GxsIdTreeItemDelegate::computeName(RsGxsId(fmpe.from.toStdString()),name))
				return name;
			return ""; //Not Found
		}

    case COLUMN_THREAD_TO:    // fallthrough. In this case, the "to" field is not filled because the msg potentially has multiple destinations.
    default:
        return displayRole(fmpe,column);
	}
}

QVariant RsMessageModel::displayRole(const Rs::Msgs::MsgInfoSummary& fmpe,int col) const
{
    switch(col)
    {
    case COLUMN_THREAD_SUBJECT:   return QVariant(QString::fromUtf8(fmpe.title.c_str()));
    case COLUMN_THREAD_ATTACHMENT:return QVariant(QString::number(fmpe.count));

    case COLUMN_THREAD_STAR:
    case COLUMN_THREAD_SPAM:
    case COLUMN_THREAD_READ:return QVariant();
    case COLUMN_THREAD_DATE:{
        QDateTime qtime;
        qtime.setTime_t(fmpe.ts);

        return QVariant(DateTime::formatDateTime(qtime));
    }

    case COLUMN_THREAD_TAGS:{
        // Tags
        Rs::Msgs::MsgTagInfo tagInfo;
        rsMsgs->getMessageTag(fmpe.msgId, tagInfo);

        Rs::Msgs::MsgTagType Tags;
        rsMsgs->getMessageTagTypes(Tags);

        QString text;

        // build tag names
        for (auto tag:tagInfo)
        {
            if (!text.isNull())
                text += ",";

            auto Tag = Tags.types.find(tag);

            if (Tag != Tags.types.end())
                text += TagDefs::name(Tag->first, Tag->second.first);
            else
                RS_WARN("Unknown tag ", (int)Tag->first, " in message ", fmpe.msgId);
        }
        return text;
    }
    case COLUMN_THREAD_TO: {
        QString name;

        switch(mCurrentBox)
        {
        case Rs::Msgs::BoxName::BOX_DRAFTS:	// in this case, we display the full list of destinations
        case Rs::Msgs::BoxName::BOX_TRASH:	// in this case, we display the full list of destinations
        case Rs::Msgs::BoxName::BOX_SENT:	// in this case, we display the full list of destinations
        {
            for(auto d:fmpe.destinations)
            {
                QString tmp;
                GxsIdTreeItemDelegate::computeName(RsGxsId(d.toStdString()),tmp);	// not nice, but works.
                if(tmp.isNull())
                    name += QString(tr("[Notification]") + ", ");
                else
                    name += tmp + ", " ;
            }
            name.chop(2);
            return name;
        }
            break;
        case Rs::Msgs::BoxName::BOX_NONE:	// in these cases, we display the actual destination
        case Rs::Msgs::BoxName::BOX_INBOX:	// in these cases, we display the actual destination
        case Rs::Msgs::BoxName::BOX_OUTBOX:
        {
            RsGxsId id = RsGxsId(fmpe.to.toStdString());	// use "to" field, which is populated in Outbox, .
            if(id.isNull())
                return QVariant(tr("[Notification]"));
            else
            {
                GxsIdTreeItemDelegate::computeName(id,name);
                return name;
            }
        }
            break;
        }
    }
        break;

    case COLUMN_THREAD_AUTHOR:{
        QString name;
        RsGxsId id = RsGxsId(fmpe.from.toStdString());

        if(id.isNull())
            return QVariant(tr("[Notification]"));
        if(GxsIdTreeItemDelegate::computeName(id,name))
            return name;
        return QVariant(tr("[Unknown]"));
    }

    default:
        return QVariant("[ TODO ]");
    }

    return QVariant("[ERROR]");
}

QVariant RsMessageModel::userRole(const Rs::Msgs::MsgInfoSummary& fmpe,int col) const
{
	switch(col)
    {
        case COLUMN_THREAD_AUTHOR:   return QVariant(QString::fromStdString(fmpe.from.toStdString()));
        case COLUMN_THREAD_MSGID:    return QVariant(QString::fromStdString(fmpe.msgId));
        case COLUMN_THREAD_TO:
        {
            // First check if the .to field is filled.

            if(!fmpe.to.toStdString().empty())
                return QVariant(QString::fromStdString(fmpe.to.toStdString()));

            // In the Send box, .to is never filled. In this case we look into destinations.

            if(fmpe.destinations.size()==1)
                return QVariant(QString::fromStdString((*fmpe.destinations.begin()).toStdString()));
        }
    default:
        return QVariant();
    }
}

QVariant RsMessageModel::decorationRole(const Rs::Msgs::MsgInfoSummary& fmpe,int col) const
{
	bool exist=false;
	switch(col)
	{
		case COLUMN_THREAD_READ:
		return (fmpe.msgflags & (RS_MSG_NEW | RS_MSG_UNREAD_BY_USER))
		        ? FilesDefs::getIconFromQtResourcePath(":/images/message-state-unread.png")
		        : FilesDefs::getIconFromQtResourcePath(":/images/message-state-read.png");
		case COLUMN_THREAD_SUBJECT:
		{
			if(fmpe.msgflags & RS_MSG_NEW         )  return FilesDefs::getIconFromQtResourcePath(":/images/message-state-new.png");
			if(fmpe.msgflags & RS_MSG_USER_REQUEST)  return FilesDefs::getIconFromQtResourcePath(":/images/user/user_request48.png");
			if(fmpe.msgflags & RS_MSG_FRIEND_RECOMMENDATION) return FilesDefs::getIconFromQtResourcePath(":/images/user/invite24.png");
			if(fmpe.msgflags & RS_MSG_PUBLISH_KEY) return FilesDefs::getIconFromQtResourcePath(":/images/share-icon-16.png");

			if(fmpe.msgflags & RS_MSG_UNREAD_BY_USER)
			{
				if((fmpe.msgflags & (RS_MSG_REPLIED | RS_MSG_FORWARDED)) == RS_MSG_REPLIED)    return FilesDefs::getIconFromQtResourcePath(":/images/message-mail-replied.png");
				if((fmpe.msgflags & (RS_MSG_REPLIED | RS_MSG_FORWARDED)) == RS_MSG_FORWARDED)  return FilesDefs::getIconFromQtResourcePath(":/images/message-mail-forwarded.png");
				if((fmpe.msgflags & (RS_MSG_REPLIED | RS_MSG_FORWARDED)) == (RS_MSG_REPLIED | RS_MSG_FORWARDED)) return FilesDefs::getIconFromQtResourcePath(":/images/message-mail-replied-forw.png");

				return FilesDefs::getIconFromQtResourcePath(":/images/message-mail.png");
			}
			if((fmpe.msgflags & (RS_MSG_REPLIED | RS_MSG_FORWARDED)) == RS_MSG_REPLIED)    return FilesDefs::getIconFromQtResourcePath(":/images/message-mail-replied-read.png");
			if((fmpe.msgflags & (RS_MSG_REPLIED | RS_MSG_FORWARDED)) == RS_MSG_FORWARDED)  return FilesDefs::getIconFromQtResourcePath(":/images/message-mail-forwarded-read.png");
			if((fmpe.msgflags & (RS_MSG_REPLIED | RS_MSG_FORWARDED)) == (RS_MSG_REPLIED | RS_MSG_FORWARDED)) return FilesDefs::getIconFromQtResourcePath(":/images/message-mail-replied-forw-read.png");

			return FilesDefs::getIconFromQtResourcePath(":/images/message-mail-read.png");
		}

		case COLUMN_THREAD_STAR:
		return FilesDefs::getIconFromQtResourcePath((fmpe.msgflags & RS_MSG_STAR) ? (IMAGE_STAR_ON ): (IMAGE_STAR_OFF));

		case COLUMN_THREAD_SPAM:
		return FilesDefs::getIconFromQtResourcePath((fmpe.msgflags & RS_MSG_SPAM) ? (IMAGE_SPAM_ON ): (IMAGE_SPAM_OFF));

        case COLUMN_THREAD_TO://Return icon as place holder.
            return FilesDefs::getIconFromGxsIdCache(RsGxsId(fmpe.to.toStdString()),QIcon(), exist);
        case COLUMN_THREAD_AUTHOR://Return icon as place holder.
            return FilesDefs::getIconFromGxsIdCache(RsGxsId(fmpe.from.toStdString()),QIcon(), exist);
	}
	return QVariant();
}

void RsMessageModel::clear()
{
	preMods();

	beginResetModel();
	mMessages.clear();
	mMessagesMap.clear();
	endResetModel();

	postMods();

	emit messagesLoaded();
}

void RsMessageModel::setMessages(const std::list<Rs::Msgs::MsgInfoSummary>& msgs)
{

	clear();

	for(auto it(msgs.begin());it!=msgs.end();++it)
	{
		mMessagesMap[(*it).msgId] = mMessages.size();
        mMessages.push_back(*it);
	}

	// now update prow for all posts

#ifdef DEBUG_MESSAGE_MODEL
	debug_dump();
#endif

	if (mMessages.size()>0)
	{
		beginInsertRows(QModelIndex(),0,mMessages.size()-1);
		endInsertRows();
	}

	emit messagesLoaded();
}

void RsMessageModel::setCurrentBox(Rs::Msgs::BoxName bn)
{
	if(mCurrentBox != bn)
	{
		mCurrentBox = bn;
		updateMessages();
	}
}

Rs::Msgs::BoxName RsMessageModel::currentBox() const
{
    return mCurrentBox;
}
void RsMessageModel::setQuickViewFilter(QuickViewFilter fn)
{
	if(fn != mQuickViewFilter)
	{
#ifdef DEBUG_MESSAGE_MODEL
		std::cerr << "Changing new quickview filter to " << fn << std::endl;
#endif

        mQuickViewFilter = fn ;

        if(rowCount() > 0)
            emit dataChanged(createIndex(0,0),createIndex(rowCount()-1,RsMessageModel::columnCount()-1));
	}
}

void RsMessageModel::updateMessages()
{
    emit messagesAboutToLoad();

    std::list<Rs::Msgs::MsgInfoSummary> msgs;

    rsMsgs->getMessageSummaries(mCurrentBox,msgs);
	setMessages(msgs);

    emit messagesLoaded();
}

void RsMessageModel::setMsgReadStatus(const QModelIndex& i,bool read_status)
{
	if(!i.isValid())
		return ;

	preMods();
	rsMsgs->MessageRead(i.data(MsgIdRole).toString().toStdString(),!read_status);

	emit dataChanged(i,i);
}

void RsMessageModel::setMsgsReadStatus(const QModelIndexList& mil,bool read_status)
{
	//Get all msgId before changing model else Index are invalid and provoc SIGSEGV
	QVector<std::string> list;
	int start = rowCount(), stop = 0;
	for(auto& it : mil)
		if (it.isValid())
		{
			list.append(it.data(MsgIdRole).toString().toStdString());
			start = std::min(start, it.row());
			stop = std::max(stop, it.row());
		}

	preMods();
	for(auto& it : list)
		rsMsgs->MessageRead(it,!read_status);

	emit dataChanged(createIndex(start,0),createIndex(stop,RsMessageModel::columnCount()-1));
}

void RsMessageModel::setMsgStar(const QModelIndex& i,bool star)
{
	if(!i.isValid())
		return ;

	preMods();
	rsMsgs->MessageStar(i.data(MsgIdRole).toString().toStdString(),star);

	emit dataChanged(i,i);
}

void RsMessageModel::setMsgsStar(const QModelIndexList& mil,bool star)
{
	//Get all msgId before changing model else Index are invalid and provoc SIGSEGV
	QVector<std::string> list;
	int start = rowCount(), stop = 0;
	for(auto& it : mil)
		if (it.isValid())
		{
			list.append(it.data(MsgIdRole).toString().toStdString());
			start = std::min(start, it.row());
			stop = std::max(stop, it.row());
		}

	preMods();
	for(auto& it : list)
		rsMsgs->MessageStar(it,star);

	emit dataChanged(createIndex(start,0),createIndex(stop,RsMessageModel::columnCount()-1));
}

void RsMessageModel::setMsgJunk(const QModelIndex& i,bool junk)
{
	if(!i.isValid())
		return ;

	preMods();
	rsMsgs->MessageJunk(i.data(MsgIdRole).toString().toStdString(),junk);

	emit dataChanged(i,i);
}

void RsMessageModel::setMsgsJunk(const QModelIndexList& mil,bool junk)
{
	//Get all msgId before changing model else Index are invalid and provoc SIGSEGV
	QVector<std::string> list;
	int start = rowCount(), stop = 0;
	for(auto& it : mil)
		if (it.isValid())
		{
			list.append(it.data(MsgIdRole).toString().toStdString());
			start = std::min(start, it.row());
			stop = std::max(stop, it.row());
		}

	preMods();
	for(auto& it : list)
		rsMsgs->MessageJunk(it,junk);

	emit dataChanged(createIndex(start,0),createIndex(stop,RsMessageModel::columnCount()-1));
}

QModelIndex RsMessageModel::getIndexOfMessage(const std::string& mid) const
{
	// Brutal search. This is not so nice, so dont call that in a loop! If too costly, we'll use a map.

	auto it = mMessagesMap.find(mid);

	if(it == mMessagesMap.end() || it->second >= mMessages.size())
		return QModelIndex();

	quintptr ref ;
	convertMsgIndexToInternalId(it->second,ref);

	return createIndex(it->second,0,ref);
}

#ifdef DEBUG_MESSAGE_MODEL
void RsMessageModel::debug_dump() const
{
	for(auto& it : mMessages)
		std::cerr << "Id: " << it.msgId << ": from " << it.srcId << ": flags=" << it.msgflags << ": title=\"" << it.title << "\"" << std::endl;
}
#endif