WebKitDownloadClient.cpp [plain text]
#include "config.h"
#include "WebKitDownloadClient.h"
#include "APIDownloadClient.h"
#include "WebKitDownloadPrivate.h"
#include "WebKitURIResponsePrivate.h"
#include "WebKitWebContextPrivate.h"
#include "WebKitWebViewPrivate.h"
#include "WebProcessPool.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 didReceiveAuthenticationChallenge(WebProcessPool&, DownloadProxy& downloadProxy, AuthenticationChallengeProxy& authenticationChallenge) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
if (webkitDownloadIsCancelled(download.get()))
return;
if (auto* webView = webkit_download_get_web_view(download.get()))
webkitWebViewHandleAuthenticationChallenge(webView, &authenticationChallenge);
}
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);
}
void decideDestinationWithSuggestedFilename(WebProcessPool&, DownloadProxy& downloadProxy, const String& filename, Function<void(AllowOverwrite, String)>&& completionHandler) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
bool allowOverwrite = false;
String destination = webkitDownloadDecideDestinationWithSuggestedFilename(download.get(), filename.utf8(), allowOverwrite);
completionHandler(allowOverwrite ? AllowOverwrite::Yes : AllowOverwrite::No, destination);
}
void didCreateDestination(WebProcessPool&, DownloadProxy& downloadProxy, const String& path) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
webkitDownloadDestinationCreated(download.get(), path);
}
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)
{
webkitWebContextGetProcessPool(webContext).setDownloadClient(std::make_unique<DownloadClient>(webContext));
}