/*

 * 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 "conversationsmodel.h"

#include "conversationsenginedefines.h"

#include "conversationmsgstorehandler.h"

#include "convergedmessage.h"

#include <xqconversions.h>

#include "conversationsengineutility.h"

#include "unidatamodelloader.h"

#include "unidatamodelplugininterface.h"

#include "ringbc.h"

#include "msgcontacthandler.h"

#include "mmsconformancecheck.h"

#include <ccsconversationentry.h>

#include <fileprotectionresolver.h>

#include "debugtraces.h"

#include <QFile>

#include <QFileInfo>

#include <s32mem.h>

#include <s32strm.h>

#include <fbs.h>

#include <ccsdefs.h>

//CONSTANTS

_LIT(KDbFileName, "c:[2002A542]conversations.db");

// preview sql query

_LIT(KSelectConvMsgsStmt, "SELECT message_id, msg_processingstate, subject, body_text, preview_path, msg_property, preview_icon FROM conversation_messages WHERE message_id=:message_id ");

//selecet preview-icon query

_LIT(KSelectPreviewIconStmt,"SELECT  message_id, preview_icon FROM conversation_messages WHERE message_id = :message_id ");

//selecet vcard-path query

_LIT(KSelectVCardStmt,"SELECT  message_id, msg_processingstate, preview_path FROM conversation_messages WHERE message_id = :message_id ");

// preview-cache max cost (items)

const int CACHE_COST =  50;

//Preview thumbnail size

const int KWidth = 24 * 6.7;

const int KHeight = 24 * 6.7;

//---------------------------------------------------------------

// ConversationsModel::ConversationsModel

// Constructor

//---------------------------------------------------------------

ConversationsModel::ConversationsModel(ConversationMsgStoreHandler* msgStoreHandler,

    QObject* parent) :

    QStandardItemModel(parent), mMsgStoreHandler(msgStoreHandler), iSqlDbOpen(EFalse)

{

    //Open SQL DB

    if (KErrNone == iSqlDb.Open(KDbFileName))

    {

        iSqlDbOpen = ETrue;

    }

    previewIconCache.setMaxCost(CACHE_COST);

    int err = connect(this, SIGNAL(retrievePreviewIcon(int, QString&)), this,

        SLOT(updatePreviewIcon(int, QString&)));

    QCRITICAL_WRITE_FORMAT("Error from connect()", err)

    iDataModelPluginLoader = new UniDataModelLoader;

    iMmsDataPlugin = iDataModelPluginLoader->getDataModelPlugin(ConvergedMessage::Mms);

    iBioMsgPlugin = iDataModelPluginLoader->getDataModelPlugin(ConvergedMessage::BioMsg);

}

//---------------------------------------------------------------

// ConversationsModel::~ConversationsModel

// Destructor

//---------------------------------------------------------------

ConversationsModel::~ConversationsModel()

{

    //Close SQL-DB

    iSqlDb.Close();

	//clear preview-cache

    previewIconCache.clear();

    if (iDataModelPluginLoader) {

        delete iDataModelPluginLoader;

        iDataModelPluginLoader = NULL;

    }

}

//---------------------------------------------------------------

// ConversationsModel::data

// @see header

//---------------------------------------------------------------

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

{

    QVariant value;

    QStandardItem* item = itemFromIndex(index);

    switch (role) {

    case ConversationId:

    {

        value = item->data(ConversationId);

        break;

    }

    case UnReadStatus:

    {

        value = item->data(UnReadStatus);

        break;

    }

    case ContactId:

    {

        value = item->data(ContactId);

        break;

    }

    case TimeStamp:

    {

        value = item->data(TimeStamp);

        break;

    }

    case ConvergedMsgId:

    {

        value = item->data(ConvergedMsgId);

        break;

    }

    case MessageProperty:

    {

        value = item->data(MessageProperty);

        break;

    }

    case MessageType:

    {

        value = item->data(MessageType);

        break;

    }

    case MessageSubType:

    {

        value = item->data(MessageSubType);

        break;

    }

    case Subject:

    {

        value = item->data(Subject);

        break;

    }

    case BodyText:

    {

        value = item->data(BodyText);

        break;

    }

    case ConversationAddress:

    {

        value = item->data(ConversationAddress);

        break;

    }

    case Direction:

    {

        value = item->data(Direction);

        break;

    }

    case SendingState:

    {

        value = item->data(SendingState);

        break;

    }

    case PreviewIcon:

    {

        QString filepath(item->data(Attachments).toString());

        int msgId = item->data(ConvergedMsgId).toInt();

        HbIcon *icon = getPreviewIconItem(msgId, filepath);

        return *icon;

    }

    case MessagePriority:

    {

        value = item->data(MessagePriority);

        break;

    }

    case Attachments:

    {

        value = item->data(Attachments);

        break;

    }

    case MessageLocation:

    {

        value = item->data(MessageLocation);

        break;

    }

    case MessageStore:

    {

        value = item->data(MessageStore);

        break;

    }

    case ConversationAlias:

    {

        value = item->data(ConversationAlias);

        break;

    }

    case UnreadCount:

    {

        value = item->data(UnreadCount);

        break;

    }

    case DisplayName: // Fall through start

        value = item->data(DisplayName);

        break;

    case Avatar: // Fall througn end

        value = item->data(Avatar);

        break;

    case NotificationStatus:

        value = item->data(NotificationStatus);

        break;

    default:

    {

        //No matching role found, set invalid variant

        value = QVariant();

        break;

    }

    }

    return value;

}

//---------------------------------------------------------------

// ConversationsModel::addRow

// @see header

//---------------------------------------------------------------

void ConversationsModel::addRow(const CCsConversationEntry& entry, bool dynamicUpdate)

{

    int msgId = entry.EntryId();

    //match, if found update else add item

    QModelIndexList indexList = this->match(index(0, 0), ConvergedMsgId, msgId, -1,

        Qt::MatchExactly);

    // if not found, add new item

    if (indexList.count() == 0) {

        QStandardItem* item = new QStandardItem();

        populateItem(*item, entry);

        if (!dynamicUpdate) {

            insertRow(0, item);

        }

        else {

            int i;

            for (i = rowCount() - 1; i >= 0; --i) {

                QStandardItem* modelItem = this->item(i, 0);

                if (modelItem->data(ConvergedMsgId).toInt() < item->data(ConvergedMsgId).toInt()) {

                    if (i == rowCount() - 1) {

                        appendRow(item);

                    }

                    else {

                        insertRow(i + 1, item);

                    }

                    return;

                }

            }

            if (i == 0) {

                insertRow(0, item);

            }

        }

    }

    else {

        // Update an existing item

        QModelIndex index = indexList[0];

        QStandardItem* item = this->item(index.row(), 0);

        populateItem(*item, entry);

    }

}

//---------------------------------------------------------------

// ConversationsModel::deleteRow

// @see header

//---------------------------------------------------------------

void ConversationsModel::deleteRow(int msgId)

{

    //match, if found remove item

    QModelIndexList indexList =

        this->match(index(0, 0), ConvergedMsgId, msgId, 1, Qt::MatchExactly);

    if (indexList.count() == 1) {

        QModelIndex index = indexList[0];

        this->removeRow(index.row());

    }

}

//---------------------------------------------------------------

// ConversationsModel::populateItem

// @see header

//---------------------------------------------------------------

void ConversationsModel::populateItem(QStandardItem& item, const CCsConversationEntry& entry)

{

    QCRITICAL_WRITE("ConversationsModel::populateItem start.");

    int msgId = entry.EntryId();

    // id

    item.setData(msgId, ConvergedMsgId);

    // description

    HBufC* description = entry.Description();

    QString subject("");

    if (description && description->Length()) {

        subject = (XQConversions::s60DescToQString(*description));     

    }

    // time stamp

    TTime unixEpoch(KUnixEpoch);

    TTimeIntervalSeconds seconds;

    TTime timeStamp(entry.TimeStamp());

    timeStamp.SecondsFrom(unixEpoch, seconds);

    item.setData(seconds.Int(), TimeStamp);

    //contact details

    HBufC* contact = entry.Contact();

    if (contact && contact->Length()) {

        item.setData(XQConversions::s60DescToQString(*contact), ConversationAddress);

    }

    // message type.

    int msgType = ConversationsEngineUtility::messageType(entry.GetType());

    item.setData(msgType, MessageType);

    //message sub-type

    item.setData(ConversationsEngineUtility::messageSubType(entry.GetType()), MessageSubType);

    // unread status

    if (entry.IsAttributeSet(ECsAttributeUnread)) {

        item.setData(true, UnReadStatus);

    }

    else {

        item.setData(false, UnReadStatus);

    }

    //sending state

    item.setData(entry.GetSendState(), SendingState);

    // direction

    item.setData(entry.ConversationDir(), Direction);

    //location

    if (entry.ConversationDir() == ECsDirectionIncoming) {

        item.setData(ConvergedMessage::Inbox, MessageLocation);

    }

    else if (entry.IsAttributeSet(ECsAttributeDraft)) {

        item.setData(ConvergedMessage::Draft, MessageLocation);

    }

    else if (entry.IsAttributeSet(ECsAttributeSent)) {

        item.setData(ConvergedMessage::Sent, MessageLocation);

    }

    else {

        item.setData(ConvergedMessage::Outbox, MessageLocation);

    }

    //message specific handling

    if (msgType == ConvergedMessage::Mms) {

        QCRITICAL_WRITE("ConversationsModel::populateItem  MMS start.")

        handleMMS(item, entry);

        QCRITICAL_WRITE("ConversationsModel::populateItem MMS end.")

    }

    else if(msgType == ConvergedMessage::MmsNotification) {

        item.setData(subject, Subject);

        handleMMSNotification(item, entry);

    }

    else if (msgType == ConvergedMessage::BT) {

        handleBlueToothMessages(item, entry);

    }

    else if (msgType == ConvergedMessage::BioMsg) {

        handleBioMessages(item, entry);

    }

    else {

        // sms bodytext

        item.setData(subject, BodyText);

    }

    QCRITICAL_WRITE("ConversationsModel::populateItem end.");

}

//---------------------------------------------------------------

// ConversationsModel::validateMsgForForward

// @see header file

//---------------------------------------------------------------

bool ConversationsModel::validateMsgForForward(qint32 messageId)

{

    bool retValue = true;

    //Validate if the mms msg can be forwarded or not

    MmsConformanceCheck* mmsConformanceCheck = new MmsConformanceCheck;

    retValue = mmsConformanceCheck->validateMsgForForward(messageId);

    delete mmsConformanceCheck;

    return retValue;

}

//---------------------------------------------------------------

// ConversationsModel::handleMMS

// @see header

//---------------------------------------------------------------

void ConversationsModel::handleMMS(QStandardItem& item, const CCsConversationEntry& entry)

{

    //msg_id

    int msgId = entry.EntryId();

    bool isEntryInDb = false;

    TInt err = KErrNone;

    //check if db is open and query db

    if (iSqlDbOpen)

    {

        RSqlStatement sqlSelectStmt;

        err = sqlSelectStmt.Prepare(iSqlDb, KSelectConvMsgsStmt);

        // move to fallback option

        if (KErrNone == err)

            {

            TInt msgIdIndex = sqlSelectStmt.ParameterIndex(_L(":message_id"));

            TInt msgProcessingStateIndex = sqlSelectStmt.ColumnIndex(_L("msg_processingstate"));

            TInt subjectIndex = sqlSelectStmt.ColumnIndex(_L("subject"));

            TInt bodyIndex = sqlSelectStmt.ColumnIndex(_L("body_text"));

            TInt previewPathIndex = sqlSelectStmt.ColumnIndex(

                    _L("preview_path"));

            TInt msgpropertyIndex = sqlSelectStmt.ColumnIndex(

                _L("msg_property"));

            TInt previewIconIndex = sqlSelectStmt.ColumnIndex(

                _L("preview_icon"));

            err = sqlSelectStmt.BindInt(msgIdIndex, msgId);

            // populate item

            if ((KErrNone == err) && (sqlSelectStmt.Next() == KSqlAtRow))

                {

                int msgProcessingState = 0;

                msgProcessingState = sqlSelectStmt.ColumnInt(

                        msgProcessingStateIndex);

                if (msgProcessingState == EPreviewMsgProcessed)

                    {

                    // use entry to populate model only when,

                    // entry is present in DB and its processing is over.

                    RBuf subjectBuffer;

                    if( subjectBuffer.Create(

                            sqlSelectStmt.ColumnSize(

                                    subjectIndex)) == KErrNone)

                    {

                        sqlSelectStmt.ColumnText(subjectIndex, subjectBuffer);

                        item.setData(

                                XQConversions::s60DescToQString(

                                        subjectBuffer), Subject);

                        subjectBuffer.Close();

                    }

                    RBuf bodyBuffer;

                    if (bodyBuffer.Create(

                            sqlSelectStmt.ColumnSize(

                                    bodyIndex)) == KErrNone)

                    {

                       sqlSelectStmt.ColumnText(bodyIndex, bodyBuffer);

                       item.setData(

                               XQConversions::s60DescToQString(

                                       bodyBuffer), BodyText);

                       bodyBuffer.Close();

                    }

                    RBuf previewPathBuffer;

                    QString attachmentPath;

                    if (previewPathBuffer.Create(

                            sqlSelectStmt.ColumnSize(

                                    previewPathIndex)) == KErrNone)

                    {

                        sqlSelectStmt.ColumnText(

                                previewPathIndex,

                                previewPathBuffer);

                        //Rightnow set inside attachments

                        attachmentPath = XQConversions::s60DescToQString(

                                previewPathBuffer);

                        item.setData(attachmentPath, Attachments);

                        previewPathBuffer.Close();

                    }

                    int msgProperty = 0;

                    msgProperty = sqlSelectStmt.ColumnInt(msgpropertyIndex);

                    item.setData(msgProperty, MessageProperty);

                    RSqlColumnReadStream stream;

                    //Get data from binary column BLOB

                    TInt err = stream.ColumnBinary(sqlSelectStmt,

                            previewIconIndex);

                    QCRITICAL_WRITE_FORMAT("Error from ColumnBinary()", err)

                    if (err == KErrNone)

                        {

                        CFbsBitmap *bitmap = new CFbsBitmap;

                        TRAPD(err,bitmap->InternalizeL(stream));

                        QCRITICAL_WRITE_FORMAT("Error from bitmap InternalizeL()", err)

                        //convert bitmap to pixmap

                        if (err == KErrNone)

                            {

                            TSize size = bitmap->SizeInPixels();

                            int bytesPerLine = bitmap->ScanLineLength(

                                    size.iWidth, bitmap->DisplayMode());

                            const uchar* dataPtr =

                                    (const uchar*) bitmap->DataAddress();

                            QPixmap pixmap = QPixmap::fromImage(QImage(

                                    dataPtr, size.iWidth, size.iHeight,

                                    bytesPerLine, QImage::Format_RGB16));

                            setPreviewIcon(pixmap, attachmentPath, msgId,

                                    true);

                            }

                        else