/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the Qt Mobility Components.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include "qmessagecontentcontainer.h"

#include "qmessagecontentcontainer_maemo_p.h"

#include "modestengine_maemo_p.h"

#include <QFile>

#include <QFileInfo>

#include <QTextCodec>

QTM_BEGIN_NAMESPACE

QMessageContentContainer QMessageContentContainerPrivate::from(QString &messageId,

                                                               QString &attachmentId,

                                                               QByteArray &name,

                                                               QByteArray &mimeType,

                                                               QByteArray &mimeSubType,

                                                               int size)

{

    QMessageContentContainer result;

    result.d_ptr->_containingMessageId = messageId;

    result.d_ptr->_attachmentId = attachmentId;

    result.d_ptr->_name = name;

    result.d_ptr->_type = mimeType;

    result.d_ptr->_subType = mimeSubType;

    result.d_ptr->_size = size;

    result.d_ptr->_available = true;

    return result;

}

QMessageContentContainerPrivate& QMessageContentContainerPrivate::operator=(const QMessageContentContainerPrivate &other)

{

    q_ptr = other.q_ptr;

    _message = other._message;

    _available = other._available;

    _size = other._size;

    _attachments = other._attachments;

    _type = other._type;

    _subType = other._subType;

    _charset = other._charset;

    _name = other._name;

    _content = other._content;

    _textContent = other._textContent;

    _filename = other._filename;

    _messageId = other._messageId;

    _id = other._id;

    _header = other._header;

    _containingMessageId = other._containingMessageId;

    _attachmentId = other._attachmentId;

    return *this;

}

bool QMessageContentContainerPrivate::isMessage() const

{

    return (_message != 0);

}

void QMessageContentContainerPrivate::setDerivedMessage(QMessage *derived)

{

    _message = derived;

}

void QMessageContentContainerPrivate::clearContents()

{

    _type = QByteArray("text");

    _subType = QByteArray("plain");

    _charset = QByteArray();

    _name = QByteArray();

    _content = QByteArray();

    _textContent = QString();

    _filename = QByteArray();

    _available = false;

    _size = 0;

    _header.clear();

    _attachments.clear();

    _containingMessageId = QString();

    _attachmentId.clear();

}

void QMessageContentContainerPrivate::setContentType(const QByteArray &type, const QByteArray &subType, const QByteArray &charset)

{

    clearContents();

    _type = type;

    _subType = subType;

    _charset = charset;

}

void QMessageContentContainerPrivate::setContent(const QString &content, const QByteArray &type, const QByteArray &subType, const QByteArray &charset)

{

    setContentType(type, subType, charset);

    _textContent = content;

    _size = content.size();

    _available = true;

}

void QMessageContentContainerPrivate::setContent(const QByteArray &content, const QByteArray &type, const QByteArray &subType, const QByteArray &charset)

{

    setContentType(type, subType, charset);

    _content = content;

    _available = true;

}

void QMessageContentContainerPrivate::setHeaderField(const QByteArray &name, const QByteArray &value)

{

    _header.remove(name);

    _header.insert(name, value);

}

QMessageContentContainer* QMessageContentContainerPrivate::attachment(const QMessageContentContainerId &id)

{

    if (isMessage()) {

        if (id == bodyContentId()) {

            return _message;

        } else {

            foreach (const QMessageContentContainer &container, _attachments) {

                if (container.d_ptr->_id == id) {

                    return const_cast<QMessageContentContainer*>(&container);

                }

            }

        }

    }

    return 0;

}

const QMessageContentContainer* QMessageContentContainerPrivate::attachment(const QMessageContentContainerId &id) const

{

    if (isMessage()) {

        if (id == bodyContentId()) {

            return _message;

        } else {

            foreach (const QMessageContentContainer &container, _attachments) {

                if (container.d_ptr->_id == id) {

                    return &container;

                }

            }

        }

    }

    return 0;

}

bool QMessageContentContainerPrivate::createAttachment(const QString& attachmentPath)

{

    //set the attachment data

    if (!QFile::exists(attachmentPath)) {

        return false;

    }

    QFile attachmentFile(attachmentPath);

    if (!attachmentFile.open(QIODevice::ReadOnly)) {

        return false;

    }

    _content = attachmentFile.readAll();

    _available = true;

    _size = attachmentFile.size();

    attachmentFile.close();

    QFileInfo fi(attachmentPath);

    _name = fi.fileName().toLatin1();

    //set the mime-type

    QByteArray mimeType;

    QString type;

    // TODO:

    QString extension(fi.suffix());

    int index = mimeType.indexOf("/");

    if (index != -1) {

        _type = mimeType.left(index).trimmed();

        _subType = mimeType.mid(index + 1).trimmed();

    }

    // set the whole filepath to _filename

    _filename = fi.filePath().toLatin1();

    return true;

}

QMessageContentContainerId QMessageContentContainerPrivate::appendContent(QMessageContentContainer& container)

{

    container.d_ptr->_id = QMessageContentContainerId(QString::number(_attachments.count()+1));

    _attachments.append(container);

    return container.d_ptr->_id;

}

QMessageContentContainerId QMessageContentContainerPrivate::prependContent(QMessageContentContainer& container)

{

    _attachments.prepend(container);

    for (int i = 0; i < _attachments.count(); ++i) {

        _attachments[i].d_ptr->_id = QMessageContentContainerId(QString::number(i+1));

    }

    return _attachments[0].d_ptr->_id;

}

QMessageContentContainerId QMessageContentContainerPrivate::bodyContentId()

{

    return QMessageContentContainerId(QString::number(0));

}

QByteArray QMessageContentContainerPrivate::attachmentFilename(const QMessageContentContainer& container)

{

    return container.d_ptr->_filename;

}

QMessageContentContainerPrivate* QMessageContentContainerPrivate::implementation(const QMessageContentContainer &container)

{

    return container.d_ptr;

}

QMessageContentContainer::QMessageContentContainer()

: d_ptr(new QMessageContentContainerPrivate(this))

{

}

QMessageContentContainer::QMessageContentContainer(const QMessageContentContainer &other)

: d_ptr(new QMessageContentContainerPrivate(this))

{

    this->operator=(other);

}

QMessageContentContainer& QMessageContentContainer::operator=(const QMessageContentContainer& other)

{

    if (&other != this) {

        *d_ptr = *other.d_ptr;

    }

    return *this;

}

QMessageContentContainer::~QMessageContentContainer()

{

    delete d_ptr;

}

QByteArray QMessageContentContainer::contentType() const

{

    return d_ptr->_type;

}

QByteArray QMessageContentContainer::contentSubType() const

{

    return d_ptr->_subType;

}

QByteArray QMessageContentContainer::contentCharset() const

{

    return d_ptr->_charset;

}

QByteArray QMessageContentContainer::suggestedFileName() const

{

    return d_ptr->_name;

}

bool QMessageContentContainer::isContentAvailable() const

{

    return d_ptr->_available;

}

int QMessageContentContainer::size() const

{

    return d_ptr->_size;

}

QString QMessageContentContainer::textContent() const

{

    QString retString;

    if (!d_ptr->_textContent.isEmpty()) {

        retString = d_ptr->_textContent;

    } else {

        if (d_ptr->_content.isEmpty() && d_ptr->_attachmentId != 0) {

            ModestEngine *engine = ModestEngine::instance();

            d_ptr->_content = engine->getMimePart(QMessageId(d_ptr->_containingMessageId),

                                                  d_ptr->_attachmentId);

            d_ptr->_size = d_ptr->_content.size();

        }

        if (!d_ptr->_content.isEmpty()) {

            if (d_ptr->_subType.toLower() == "html") {

                QTextCodec *codec = QTextCodec::codecForName(d_ptr->_charset.data());

                if (codec) {

                    retString = codec->toUnicode(d_ptr->_content);

                }

            } else {

                // Modest plugin automatically decodes plain text content to UTF-8 format

                retString = QString::fromUtf8(d_ptr->_content);

            }

        }

    }

    return retString;

}

QByteArray QMessageContentContainer::content() const

{

    if (d_ptr->_content.isEmpty() && d_ptr->_attachmentId != 0) {

        ModestEngine *engine = ModestEngine::instance();

        d_ptr->_content = engine->getMimePart(

                QMessageId (d_ptr->_containingMessageId),

                d_ptr->_attachmentId);

        d_ptr->_size = d_ptr->_content.size();

    }

    return d_ptr->_content;

}

void QMessageContentContainer::writeTextContent(QTextStream& out) const

{

    out << textContent();

}

void QMessageContentContainer::writeContent(QDataStream& out) const

{

    QByteArray data(content());

    out.writeRawData(data.constData(), data.length());

}

QMessageContentContainerIdList QMessageContentContainer::contentIds() const

{

    QMessageContentContainerIdList ids;

    if (d_ptr->isMessage()) {

        foreach (const QMessageContentContainer &container, d_ptr->_attachments) {

            ids.append(container.d_ptr->_id);

        }

    }

    return ids;

}

QMessageContentContainer QMessageContentContainer::find(const QMessageContentContainerId &id) const

{

    if (d_ptr->isMessage()) {

        if (const QMessageContentContainer *container = d_ptr->attachment(id)) {

            return *container;

        }

    }

    return QMessageContentContainer();

}

bool QMessageContentContainer::contains(const QMessageContentContainerId &id) const

{

    if (d_ptr->isMessage()) {

        return (d_ptr->attachment(id) != 0);

    }

    return false;

}

QString QMessageContentContainer::headerFieldValue(const QByteArray &name) const

{

    QMultiMap<QByteArray, QString>::const_iterator it = d_ptr->_header.find(name);

    if (it != d_ptr->_header.end()) {

        return it.value();

    }

    return QString();

}

QStringList QMessageContentContainer::headerFieldValues(const QByteArray &name) const

{

    QStringList values;

    QMultiMap<QByteArray, QString>::const_iterator it = d_ptr->_header.find(name);

    while ((it != d_ptr->_header.end()) && (it.key() == name)) {

        values.append(it.value());

        ++it;

    }

    return values;

}

QList<QByteArray> QMessageContentContainer::headerFields() const

{

    return d_ptr->_header.keys();

}

void QMessageContentContainer::setDerivedMessage(QMessage *derived)

{

    d_ptr->setDerivedMessage(derived);

}

QTM_END_NAMESPACE