WebKitDownloadClient.cpp [plain text]
#include "config.h"
#include "WebKitDownloadClient.h"
#include "APIDownloadClient.h"
#include "WebKitDownloadPrivate.h"
#include "WebKitURIResponsePrivate.h"
#include "WebKitWebContextPrivate.h"
#include "WebProcessPool.h"
#include <WebKit/WKString.h>
#include <wtf/glib/GRefPtr.h>
#include <wtf/text/CString.h>
using namespace WebCore;
using namespace WebKit;
class DownloadClient final : public API::DownloadClient {
public:
explicit DownloadClient(WebKitWebContext* webContext)
: m_webContext(webContext)
{
}
private:
void didStart(WebProcessPool*, DownloadProxy* downloadProxy) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(downloadProxy);
webkitWebContextDownloadStarted(m_webContext, download.get());
}
void didReceiveResponse(WebProcessPool*, DownloadProxy* downloadProxy, const ResourceResponse& resourceResponse) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(downloadProxy);
if (webkitDownloadIsCancelled(download.get()))
return;
GRefPtr<WebKitURIResponse> response = adoptGRef(webkitURIResponseCreateForResourceResponse(resourceResponse));
webkitDownloadSetResponse(download.get(), response.get());
}
void didReceiveData(WebProcessPool*, DownloadProxy* downloadProxy, uint64_t length) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(downloadProxy);
webkitDownloadNotifyProgress(download.get(), length);
}
String decideDestinationWithSuggestedFilename(WebProcessPool*, DownloadProxy* downloadProxy, const String& filename, bool& allowOverwrite) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(downloadProxy);
return String::fromUTF8(webkitDownloadDecideDestinationWithSuggestedFilename(download.get(), filename.utf8(), allowOverwrite));
}
void didCreateDestination(WebProcessPool*, DownloadProxy* downloadProxy, const String& path) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(downloadProxy);
webkitDownloadDestinationCreated(download.get(), path.utf8());
}
void didFail(WebProcessPool*, DownloadProxy* downloadProxy, const ResourceError& error) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(downloadProxy);
if (webkitDownloadIsCancelled(download.get())) {
webkitDownloadCancelled(download.get());
} else
webkitDownloadFailed(download.get(), error);
webkitWebContextRemoveDownload(downloadProxy);
}
void didCancel(WebProcessPool*, DownloadProxy* downloadProxy) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(downloadProxy);
webkitDownloadCancelled(download.get());
webkitWebContextRemoveDownload(downloadProxy);
}
void didFinish(WebProcessPool*, DownloadProxy* downloadProxy) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(downloadProxy);
webkitDownloadFinished(download.get());
webkitWebContextRemoveDownload(downloadProxy);
}
WebKitWebContext* m_webContext;
};
void attachDownloadClientToContext(WebKitWebContext* webContext)
{
auto* processPool = webkitWebContextGetProcessPool(webContext);
processPool->setDownloadClient(std::make_unique<DownloadClient>(webContext));
}