QtDownloadManager.cpp [plain text]
#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);
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);
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);
QWebDownloadItem* downloadItem = q->m_downloads.take(WKDownloadGetID(download));
ASSERT(downloadItem);
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);
QWebDownloadItem* downloadItem = q->m_downloads.value(WKDownloadGetID(download));
ASSERT(downloadItem);
downloadItem->d->totalBytesReceived += length;
emit downloadItem->totalBytesReceivedChanged(length);
}
}