Source/WebKit2/UIProcess/qt/QtDownloadManager.cpp

/*
 * Copyright (C) 2010, 2011 Nokia Corporation and/or its subsidiary(-ies)
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public License
 * along with this program; see the file COPYING.LIB.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 *
 */

#include "config.h"
#include "QtDownloadManager.h"

#include "QtWebError.h"
#include "qwebdownloaditem_p.h"
#include "qwebdownloaditem_p_p.h"
#include <WKDownload.h>
#include <WKStringQt.h>
#include <WKURLQt.h>
#include <WKURLResponse.h>

namespace WebKit {

static inline QtDownloadManager* toQtDownloadManager(const void* clientInfo)
{
    ASSERT(clientInfo);
    return reinterpret_cast<QtDownloadManager*>(const_cast<void*>(clientInfo));
}

QtDownloadManager::QtDownloadManager(WKContextRef context)
{
    WKContextDownloadClient downloadClient;
    memset(&downloadClient, 0, sizeof(WKContextDownloadClient));
    downloadClient.version = kWKContextDownloadClientCurrentVersion;
    downloadClient.clientInfo = this;
    downloadClient.didReceiveResponse = didReceiveResponse;
    downloadClient.didReceiveData = didReceiveDataForDownload;
    downloadClient.didCreateDestination = didCreateDestination;
    downloadClient.didFinish = didFinishDownload;
    downloadClient.didFail = didFailDownload;
    WKContextSetDownloadClient(context, &downloadClient);
}

QtDownloadManager::~QtDownloadManager()
{
}

void QtDownloadManager::addDownload(WKDownloadRef download, QWebDownloadItem* downloadItem)
{
    m_downloads[WKDownloadGetID(download)] = downloadItem;
}

void QtDownloadManager::didReceiveResponse(WKContextRef, WKDownloadRef download, WKURLResponseRef response, const void* clientInfo)
{
    QtDownloadManager* q = toQtDownloadManager(clientInfo);

    // Will be called when the headers are read by WebProcess.
    QWebDownloadItem* downloadItem = q->m_downloads.value(WKDownloadGetID(download));
    ASSERT(downloadItem);

    downloadItem->d->sourceUrl = adoptToQUrl(WKURLResponseCopyURL(response));
    downloadItem->d->mimeType = adoptToQString(WKURLResponseCopyMIMEType(response));
    downloadItem->d->expectedContentLength = WKURLResponseGetExpectedContentLength(response);
    downloadItem->d->suggestedFilename = WKStringCopyQString(adoptWK(WKURLResponseCopySuggestedFilename(response)).get());

    downloadItem->d->didReceiveResponse(downloadItem);
}

void QtDownloadManager::didCreateDestination(WKContextRef, WKDownloadRef download, WKStringRef path, const void* clientInfo)
{
    QtDownloadManager* q = toQtDownloadManager(clientInfo);

    QWebDownloadItem* downloadItem = q->m_downloads.value(WKDownloadGetID(download));
    ASSERT(downloadItem);
    downloadItem->d->destinationPath = WKStringCopyQString(path);
    emit downloadItem->destinationFileCreated(downloadItem->d->destinationPath);
}

void QtDownloadManager::didFinishDownload(WKContextRef, WKDownloadRef download, const void *clientInfo)
{
    QtDownloadManager* q = toQtDownloadManager(clientInfo);

    // Will be called when download finishes with success.
    QWebDownloadItem* downloadItem = q->m_downloads.take(WKDownloadGetID(download));
    ASSERT(downloadItem);
    emit downloadItem->succeeded();
}

void QtDownloadManager::didFailDownload(WKContextRef, WKDownloadRef download, WKErrorRef error, const void* clientInfo)
{
    QtDownloadManager* q = toQtDownloadManager(clientInfo);

    // Will be called when download fails or is aborted.
    QWebDownloadItem* downloadItem = q->m_downloads.take(WKDownloadGetID(download));
    ASSERT(downloadItem);

    // If the parent is null at this point, the download failed before it
    // received a response and downloadRequested was emitted.
    // Due to this the item will never be parented and we have to delete it
    // manually at this point.
    if (!downloadItem->parent()) {
        delete downloadItem;
        return;
    }

    QtWebError qtError(error);
    emit downloadItem->failed(qtError.errorCodeAsDownloadError(), qtError.url(), qtError.description());
}

void QtDownloadManager::didReceiveDataForDownload(WKContextRef, WKDownloadRef download, uint64_t length, const void* clientInfo)
{
    QtDownloadManager* q = toQtDownloadManager(clientInfo);

    // Will be called everytime bytes were written to destination file by WebProcess.
    QWebDownloadItem* downloadItem = q->m_downloads.value(WKDownloadGetID(download));
    ASSERT(downloadItem);
    downloadItem->d->totalBytesReceived += length;
    emit downloadItem->totalBytesReceivedChanged(length);
}

} // namespace WebKit