/*

* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:

*

*/

#include "mobcntmodel_p.h"

#include "mobcntmodel.h"

#include "mobcnticonmanager.h"

#include <qtcontacts.h>

#include <QSet>

#include <QTimerEvent>

#include <hbindexfeedback.h>

#include <hbframebackground.h>

#include <hbframedrawer.h>

const uint dummyMyCardId = 0;

/*!

 * Constuct a new contact model instance. The model will own its handle

 * to the default persisted store.

 */

MobCntModel::MobCntModel(const QContactFilter& contactFilter,

                         const QList<QContactSortOrder>& contactSortOrders,

                         bool showMyCard,

                         QObject *parent)

    : QAbstractListModel(parent),

    mIconManager(0)

{

    mIconManager = new MobCntIconManager();

    connect(mIconManager, SIGNAL(contactIconReady(int)), this, SLOT(updateContactIcon(int)));

    mDefaultIcon = QIcon(":/icons/qtg_large_avatar.svg");

    d = new MobCntModelData(contactFilter, contactSortOrders, showMyCard);

    d->m_contactManager = new QContactManager;

    d->ownedContactManager = true;

    d->mMyCardId = d->m_contactManager->selfContactId();

    if (doConstruct() != QContactManager::NoError) {

        throw("exception");

    }

}

/*!

 * Construct a new MobCntModel object using manager as the QContactManager

 * instance to communicate with the contacts database.

 *

 * \param manager A QContactManager instance to be used for

 *  communications with the contacts persistant store.

 */

MobCntModel::MobCntModel(QContactManager* manager,

                         const QContactFilter& contactFilter,

                         const QList<QContactSortOrder>& contactSortOrders,

                         bool showMyCard,

                         QObject *parent)

    : QAbstractListModel(parent),

    mIconManager(0)

{

    mIconManager = new MobCntIconManager();

    connect(mIconManager, SIGNAL(contactIconReady(int)), this, SLOT(updateContactIcon(int)));

    mDefaultIcon = QIcon(":/icons/qtg_large_avatar.svg");

    d = new MobCntModelData(contactFilter, contactSortOrders, showMyCard);

    d->m_contactManager = manager;

    d->mMyCardId = d->m_contactManager->selfContactId();

    if (doConstruct() != QContactManager::NoError) {

        throw("exception");

    }

}

MobCntModel::~MobCntModel()

{

    delete mIconManager;

}

/*!

 * Return the data to be used by the view or delegates for a particular

 * item and role.

 *

 * \param index The index of the item to return data about.

 * \param role The data should be relevant for this particular purpose.

 * \return QVariant The data for the specified index and role.

 */

QVariant MobCntModel::data(const QModelIndex &index, int role) const

{

    int row = index.row();

    // check that row is ok

    if (!validRowId(row)) {

        // invalid row

		return QVariant();

	}

    // update current contact if needed

    if (row != d->currentRow ) {

        if (d->contactIds[row] == dummyMyCardId) {

            // row contains dummy MyCard, so create dummy QContact

            d->currentContact = QContact();

        }

        else {

            // define fields we are interested in

            QStringList definitionRestrictions;

            definitionRestrictions.append(QContactName::DefinitionName);

            definitionRestrictions.append(QContactAvatar::DefinitionName);

            definitionRestrictions.append(QContactPhoneNumber::DefinitionName);

            definitionRestrictions.append(QContactOrganization::DefinitionName);

            d->currentContact = d->m_contactManager->contact(d->contactIds[row], definitionRestrictions);

        }

        d->currentRow = row;

    }

    if (role == Qt::DisplayRole) {

        return dataForDisplayRole(row);

    }

    else if (role == Hb::IndexFeedbackRole) {

        return dataForDisplayRole(row).toStringList().at(0).toUpper();

    }

	else if (role == Qt::BackgroundRole) {

	    if (d->mMyCardId == d->contactIds[row] ||

	        dummyMyCardId == d->contactIds[row]) {

                return HbFrameBackground("qtg_fr_groupbox", HbFrameDrawer::NinePieces);

	    }

	}

	else if (role == Qt::DecorationRole) {

	    QContactAvatar contactAvatar = d->currentContact.detail<QContactAvatar>();

	    QList<QVariant> icons;

	    if(!contactAvatar.imageUrl().isEmpty()) {

            QIcon icon = mIconManager->contactIcon(contactAvatar.imageUrl().toString(), row);

            if (icon.isNull()) {

                icons.append(mDefaultIcon);

            }

            else {

                icons.append(icon);

            }

	    }

	    else {

            icons.append(mDefaultIcon);

	    }        

	    return QVariant(icons);

	}

	return QVariant();

}

/*!

 * Get the number of rows (contacts) in this model.

 *

 * \param parent Optional parent index value.

 * \return Number of contacts in this model.

 */

int MobCntModel::rowCount(const QModelIndex& /*parent*/) const

{

   return d->contactIds.count();

}

/*!

 * Read a full contact entry from the database for the given model

 * index value. Only the row part of the index information will be

 * read. This is just an overload of MobCntModel::contact() that

 * supports old behaviour and calls:

 *      MobCntModel::contact(int row);

 *

 * The entry at the requested row will have its full contact information

 * (all fields) read from the database and returned as a QContact instance.

 *

 * \param index Index for the sought contact entry in this model.

 * \return A newly constructed QContact instance for this entry - ownership

 *  is transferred to the caller.

 *

 */

QContact MobCntModel::contact(const QModelIndex &index) const

{

    return contact(index.row());

}

/*!

 * Return an index that points to the row relating to the supplied contact.

 *  E.g. if the contact is at row 7, the index with the following properties

 *  is returned:

 *      index.row() == 7

 * \param contact The contact for whose row an index is required

 * \return a QModelIndex with the row set to match that of the contact.

 */

QModelIndex MobCntModel::indexOfContact(const QContact &contact) const

{

    return createIndex(rowId(contact.localId()), 0);

}

/*!

 * Return the contact manager used by this model.

 *

 * \return Reference to contact manager.

 */

QContactManager& MobCntModel::contactManager() const

{

    return *d->m_contactManager;

}

/*!

 * Set new filter and sort order for the model.

 *

 * \param contactFilter New contact filter.

 * \param contactSortOrders New sort order.

 */

void MobCntModel::setFilterAndSortOrder(const QContactFilter& contactFilter,

                           const QList<QContactSortOrder>& contactSortOrders)

{

    d->filter = contactFilter;

    // keep current sort order if new one isn't given

    if (contactSortOrders.count()) {

        d->sortOrders = contactSortOrders;

    }

    d->currentRow = -1;

    //refresh model

    updateContactIdsArray();

    beginResetModel();

    reset();

    endResetModel();

}

/*!

 * Enable/disable MyCard appearance in the model.

 *

 * \param enabled Status of MyCard appearance in the model.

 */

void MobCntModel::showMyCard(bool enabled)

{

    if (d->showMyCard == enabled) {

        return;

    }

    QContactLocalId myCardId = d->mMyCardId;

    if (enabled) {

        //add MyCard to the list

        if (myCardId <= 0) {

            // create a placeholder for MyCard

            d->contactIds.insert(0, dummyMyCardId);

        }

        else {

            d->contactIds.insert(0, myCardId);

        }

    }

    else {

        // remove MyCard from the list

        if (myCardId <= 0) {

            d->contactIds.removeOne(dummyMyCardId);

        }

        else {

            d->contactIds.removeOne(myCardId);

        }

    }

    d->showMyCard = enabled;

    d->currentRow = -1;

    beginResetModel();

    reset();

    endResetModel();

}

/*!

 * Returns MyCard status: shown or not.

 *

 * \return true if MyCard is shown, false otherwise.

 */

bool MobCntModel::myCardStatus() const

{

    return d->showMyCard;

}

/*!

 * Returns MyCard id.

 *

 * \return MyCard id.

 */

QContactLocalId MobCntModel::myCardId() const

{

    return d->mMyCardId;

}

/*!

 * Initializes model data and attaches database notifications to handlers.

 * 

 * \return Error status

 */

int MobCntModel::doConstruct()

{

    int error = initializeData();

    // Attach database notifications to handlers.

    connect(d->m_contactManager, SIGNAL(contactsAdded(const QList<QContactLocalId>&)), this, SLOT(handleAdded(const QList<QContactLocalId>&)));

    connect(d->m_contactManager, SIGNAL(contactsChanged(const QList<QContactLocalId>&)), this, SLOT(handleChanged(const QList<QContactLocalId>&)));

    connect(d->m_contactManager, SIGNAL(contactsRemoved(const QList<QContactLocalId>&)), this, SLOT(handleRemoved(const QList<QContactLocalId>&)));

    connect(d->m_contactManager, SIGNAL(relationshipsAdded(const QList<QContactLocalId>&)), this, SLOT(handleAdded(const QList<QContactLocalId>&)));

    connect(d->m_contactManager, SIGNAL(relationshipsRemoved(const QList<QContactLocalId>&)), this, SLOT(handleRemoved(const QList<QContactLocalId>&)));

    connect(d->m_contactManager, SIGNAL(selfContactIdChanged(const QContactLocalId&, const QContactLocalId&)), this, SLOT(handleMyCardChanged(const QContactLocalId&, const QContactLocalId&)));

    return error;

}

/*!

 * Initializes d->contactIds to contain contact Ids from the database.

 * 

 * \return Error status

 */

int MobCntModel::initializeData()

{

    int error(QContactManager::NoError);

    // Get a list of all contact IDs in the database.

    updateContactIdsArray();

    return error;

}

/*!

 * Gets the filtered list of the contact Ids in a sorted order  

 * 

 * \return Error status

 */

void MobCntModel::updateContactIdsArray()

{

    d->contactIds = d->m_contactManager->contactIds(d->filter,

            d->sortOrders);

    //find MyCard contact and move it to the first position

    QContactLocalId myCardId = d->mMyCardId;

    if (myCardId > 0) {

        // MyCard exists

        d->contactIds.removeOne(myCardId);

        if (d->showMyCard) {

            d->contactIds.insert(0, myCardId);

        }

    } 

    else if (d->showMyCard) {

        // create a placeholder for MyCard

        d->contactIds.insert(0, dummyMyCardId);

    }

}

/*!

 * Read a full contact entry from the database for the row number.

 *

 * The entry at the requested row will have its full contact information

 * (all fields) read from the database and returned as a QContact instance.

 *

 * \param row Row at which the sought contact entry is in this model.

 * \return A newly constructed QContact instance for this entry - ownership

 *  is transferred to the caller.

 *

 */

QContact MobCntModel::contact(int row) const

{

    if (!validRowId(row) || d->contactIds[row] == dummyMyCardId) {

        return QContact();

    }

    return d->m_contactManager->contact(d->contactIds[row]);

}

/*!

 * Verify specified row id is valid.

 *

 * \param row A row number

 * \return bool indicating validity of row id

 */

bool MobCntModel::validRowId(int row) const

{

	return (row >= 0 && row < rowCount());

}

/*!

 * Fetch the id of the row containing the contact of the specified id.

 *

 * \param contactId The id of the contact

 * \return the row id of the contact or -1 if no item matched.

 */

int MobCntModel::rowId(const QContactLocalId &contactId) const

{

    return d->contactIds.indexOf(contactId);

}

/*!

 * Return the data to be used by the view for a display role.

 *

 * \param row The row of the item to return data about.

 * \param column The column of the item to return data about.

 * \return QVariant The data for the specified index.

 */

QVariant MobCntModel::dataForDisplayRole(int row) const

{

    QStringList list;

    QString name;

    QString number;

    bool isNonEmptySelfContact = false;

    bool isSelfContact = false;

    QContactLocalId id = d->contactIds[row];

    if (d->mMyCardId == id ||

        dummyMyCardId == id) {

        isSelfContact = true;

        if (d->currentContact.details().count() <= 4) {    

            //empty card

            name = tr("My card");

            number = tr("Create my identity");

        }

        else {

            isNonEmptySelfContact = true;

        }

    }

    if (!isSelfContact || isNonEmptySelfContact) {

        name = d->currentContact.displayLabel();

        QContactPhoneNumber phoneNumber = d->currentContact.detail<QContactPhoneNumber>();

        number = phoneNumber.number();

    }

    if (!isNonEmptySelfContact) {

        list << name << number;

    }

    else {

        list << name;

    }

    return QVariant(list);

}

/*!

 * Handle adding of contacts. 

 *

 * \param contactIds Ids of contacts added.

 */

void MobCntModel::handleAdded(const QList<QContactLocalId>& contactIds)

{

    //invalidate cached contact

    d->currentRow = -1; 

    updateContactIdsArray();

    QList< QList<int> > indexSequences = findIndexes(contactIds);

    for (int i = (indexSequences.count() - 1);i >= 0; i--)

    {

        beginInsertRows(QModelIndex(), indexSequences.at(i).first(), indexSequences.at(i).last());

        endInsertRows();

    }

}

/*!

 * Handle changes in contacts. 

 *

 * \param contactIds Ids of contacts changed.

 */

void MobCntModel::handleChanged(const QList<QContactLocalId>& contactIds)

{

    //invalidate cached contact

    d->currentRow = -1; 

    updateContactIdsArray();

    QList< QList<int> > indexSequences = findIndexes(contactIds);

    for (int i = (indexSequences.count() - 1);i >= 0; i--)

    {

        QModelIndex top = index(indexSequences.at(i).first());

        QModelIndex bottom = index(indexSequences.at(i).last());

        emit dataChanged(top, bottom);

    }

}

/*!

 * Handle removing of contacts. 

 *

 * \param contactIds Ids of contacts removed.

 */

void MobCntModel::handleRemoved(const QList<QContactLocalId>& contactIds)

{

    QList< QList<int> > indexSequences = findIndexes(contactIds);

    // invalidate cached contact

    d->currentRow = -1;

    // if myCard has been removed, set myCard id to zero

    if (contactIds.contains(d->mMyCardId))

        d->mMyCardId = 0;

    updateContactIdsArray();

    for (int i = (indexSequences.count() - 1);i >= 0; i--)

    {

        if (indexSequences.at(i).first() == 0 && d->showMyCard) {

            // handle removed for MyCard

            d->mMyCardId = 0;

            QModelIndex index = createIndex(0, 0);

            emit dataChanged(index, index);

            // handle removed for other contacts deleted besides myCard

            if (indexSequences.at(i).count() > 1) {

                beginRemoveRows(QModelIndex(), indexSequences.at(i).first() + 1, indexSequences.at(i).last());

                endRemoveRows();

            }

        }

        else

        {

            beginRemoveRows(QModelIndex(), indexSequences.at(i).first(), indexSequences.at(i).last());

            endRemoveRows();