/**
   This file is part of CWRT package **

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

   This program is free software: you can redistribute it and/or modify
   it under the terms of the GNU (Lesser) General Public License as 
   published by the Free Software Foundation, version 2.1 of the License. 
   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 
   (Lesser) General Public License for more details. You should have 
   received a copy of the GNU (Lesser) General Public License along 
   with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#include "clientdownload.h"
#include "downloadmanager.h"
#include "downloadcore.h"
#include "downloadstore.h"
#include "httpdownloadbackend.h"
#include "filestorage.h"
#include "dmcommoninternal.h"
#include <QFileInfo>
#include <QString>
#include <QMap>
#include <QIODevice>

// private implementation
class HttpDownloadBackendPrivate
{
    DM_DECLARE_PUBLIC(HttpDownloadBackend);
public:
    HttpDownloadBackendPrivate();
    ~HttpDownloadBackendPrivate();
    DownloadCore *m_downloadCore; // for network operations
    DownloadStore *m_storage; // responsible for handling the storage
    ClientDownload *m_download; // not owned
};  

HttpDownloadBackendPrivate::HttpDownloadBackendPrivate()
{
    m_downloadCore = 0; 
    m_storage = 0;
    m_download = 0;
}

HttpDownloadBackendPrivate::~HttpDownloadBackendPrivate()
{
    if(m_storage)
    {
        delete m_storage;
        m_storage = 0;
    }
}

HttpDownloadBackend::HttpDownloadBackend(DownloadCore *dlCore, ClientDownload *dl, DownloadStore* store)
    :DownloadBackend(dlCore, dl)
{
    DM_INITIALIZE(HttpDownloadBackend);
    priv->m_downloadCore = dlCore;
    priv->m_storage = store; 
    priv->m_download = dl;
    setValue(DownloadInfo::EFinalPath, dl->attributes().value(DlDestPath).toString());

    QString fileName;
    if(dl->isCreatedByDlInfo())
    {
        getValue(DownloadInfo::EFileName, fileName);
        dl->attributes().insert(DlFileName, fileName);
        priv->m_storage->open(QIODevice::Append);           
    }
    else
    {
        fileName = dl->attributes().value(DlFileName).toString();
        // if filename is not set, take it from url
        if(fileName.length() == 0) {
            fileName = priv->m_downloadCore->fileNameFromContentDispositionHeader();
            if(fileName.length() == 0) {
                QUrl url(priv->m_downloadCore->url());
                QFileInfo fileUrl(url.path());
                fileName = fileUrl.fileName();     
            }         
        }
        dl->attributes().insert(DlFileName, fileName);
        // create the new storage
        priv->m_storage->createStore();      
    }
    setValue(DownloadInfo::EFileName, download()->attributes().value(DlFileName).toString());
}

HttpDownloadBackend::~HttpDownloadBackend()
{
    DM_UNINITIALIZE(HttpDownloadBackend);
}

int HttpDownloadBackend::resume()
{
    DM_PRIVATE(HttpDownloadBackend);
    // Open the file in append mode as we need to append the received chunks    
    if(downloadState() != DlCancelled)
    {
        priv->m_storage->open(QIODevice::Append);               
    }
    else
    {
        QString fileName = download()->attributes().value(DlFileName).toString();
        // if filename is not set, take it from url 
        if(fileName.length() == 0) {
            fileName = priv->m_downloadCore->fileNameFromContentDispositionHeader();
            if(fileName.length() == 0) {
                QUrl url(priv->m_downloadCore->url());
                QFileInfo fileUrl(url.path());
                fileName = fileUrl.fileName();     
            }         
        }
        download()->attributes().insert(DlFileName, fileName);
        // create the new storage
        priv->m_storage->createStore(); 
    }
    setValue(DownloadInfo::EFileName, download()->attributes().value(DlFileName).toString());
    return DownloadBackend::resume();
}

void HttpDownloadBackend::store(QByteArray data, bool lastChunk)
{
    DM_PRIVATE(HttpDownloadBackend); 
    // write the chunks to the storage
    priv->m_storage->write(data, lastChunk); 
}

void HttpDownloadBackend::deleteStore()
{
    DM_PRIVATE(HttpDownloadBackend);
    if(priv->m_storage)
    {
        priv->m_storage->deleteStore();
    }
}

qint64 HttpDownloadBackend::storedDataSize()
{
    DM_PRIVATE(HttpDownloadBackend);
    // size of stored data chunk
    return priv->m_storage->storedDataSize();
}

QVariant HttpDownloadBackend::getAttribute(DownloadAttribute attr)
{
    DM_PRIVATE(HttpDownloadBackend);
    switch(attr)
    {
        case DlFileName:
        {
            if(priv->m_download)
                return priv->m_download->attributes().value(attr);
        }
        case DlDestPath:
        {
            if(priv->m_download)
                return priv->m_download->attributes().value(attr);
        }
        default:
            break;
    }  
    return DownloadBackend::getAttribute(attr);
}

int HttpDownloadBackend::setAttribute(DownloadAttribute attr, const QVariant& value)
{
    return DownloadBackend::setAttribute(attr, value);
}

void HttpDownloadBackend::headerReceived()
{
    DM_PRIVATE(HttpDownloadBackend);
    DlEventAttributeMap* attrMap = NULL;
    HttpStatusCode status = (HttpStatusCode) priv->m_downloadCore->reply()->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
    switch(status)
    {
        case HttpOK:                    // OK
        case HttpCreated:               // Created
        case HttpAccepted:              // Accepted
        case HttpNonAuthorativeInfo:    // Non-Authorative Information
        case HttpNoContent:             // No Content
        case HttpResetContent:          // Reset Conetent
        {
            // get the content type and save it in dl info
            QString contentType = priv->m_downloadCore->contentType();
            setValue(DownloadInfo::EContentType, contentType);
            setDownloadState(DlInprogress);
            break;
        }

        case HttpPartialContent:        // Partial Conetent Download
        {
            setDownloadState(DlInprogress);
            break;
        }
        
        case HttpPreconditionFailed:    // Precondition Failed
        {
            // attrMap will be deleted in destructor of DownloadEventPrivate class
            attrMap = new DlEventAttributeMap ;
            attrMap->insert(HeaderReceivedStatusCode, QVariant(status));
            break;
        }

        default:
           break;
    }

    postEvent(HeaderReceived, attrMap);
}