/*
    This file is part of Akonadi.

    Copyright (c) 2010 Stephen Kelly <steveire@gmail.com>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
    USA.
*/

// READ THE README FILE

#include "mixedtreemodel.h"

#include <KMime/Message>
#include <KContacts/Addressee>

MixedTreeModel::MixedTreeModel(Akonadi::Monitor *monitor, QObject *parent)
    : EntityTreeModel(monitor, parent)
{
}

int MixedTreeModel::entityColumnCount(Akonadi::EntityTreeModel::HeaderGroup headerGroup) const
{
    if (headerGroup == CollectionTreeHeaders) {
        return 1;
    }

    if (headerGroup == ItemListHeaders) {
        return 2;
    }
    return 0;
}

QVariant MixedTreeModel::entityHeaderData(int section, Qt::Orientation orientation, int role, Akonadi::EntityTreeModel::HeaderGroup headerGroup) const
{
    if (headerGroup == CollectionTreeHeaders || role != Qt::DisplayRole || orientation == Qt::Vertical) {
        return Akonadi::EntityTreeModel::entityHeaderData(section, orientation, role, headerGroup);
    }

    switch (section) {
    case 0:
        return QLatin1String("Name/Subject");
    case 1:
        return QLatin1String("Email");
    default:
        return QVariant();
    }
}

QVariant MixedTreeModel::entityData(const Akonadi::Item &item, int column, int role) const
{
    if (role == Qt::DisplayRole) {
        if (item.hasPayload<KMime::Message::Ptr>()) {
            KMime::Message::Ptr message = item.payload<KMime::Message::Ptr>();
            switch (column) {
            case 0:
                return message->subject()->asUnicodeString();
            case 1:
                if (item.mimeType() == KMime::Message::mimeType()) {
                    return message->from()->asUnicodeString();
                } else {
                    return QString(message->mainBodyPart()->decodedText().mid(0, 30) + QStringLiteral("..."));
                }
            default:
                return QVariant();
            }
        }

        if (item.hasPayload<KContacts::Addressee>()) {
            KContacts::Addressee addressee = item.payload<KContacts::Addressee>();
            switch (column) {
            case 0:
                return addressee.name();
            case 1:
                return addressee.preferredEmail();
            default:
                return QVariant();
            }
        }
    }
    return EntityTreeModel::entityData(item, column, role);
}

QVariant MixedTreeModel::entityData(const Akonadi::Collection &collection, int column, int role) const
{
    return Akonadi::EntityTreeModel::entityData(collection, column, role);
}
