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 "WebsiteDataStore.h"
#include <WebCore/UserAgent.h>
#include <wtf/glib/GRefPtr.h>
#include <wtf/text/CString.h>
using namespace WebCore;
using namespace WebKit;
class LegacyDownloadClient final : public API::DownloadClient {
public:
explicit LegacyDownloadClient(WebKitWebContext* webContext)
: m_webContext(webContext)
{
}
private:
void legacyDidStart(DownloadProxy& downloadProxy) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
webkitDownloadStarted(download.get());
webkitWebContextDownloadStarted(m_webContext, download.get());
}
void willSendRequest(DownloadProxy& downloadProxy, ResourceRequest&& request, const ResourceResponse&, CompletionHandler<void(ResourceRequest&&)>&& completionHandler) override
{
if (!request.hasHTTPHeaderField(HTTPHeaderName::UserAgent)) {
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
auto* webView = webkit_download_get_web_view(download.get());
request.setHTTPUserAgent(webView ? webkitWebViewGetPage(webView).userAgentForURL(request.url()) : WebPageProxy::standardUserAgent());
}
completionHandler(WTFMove(request));
}
void didReceiveAuthenticationChallenge(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(DownloadProxy& downloadProxy, const ResourceResponse& resourceResponse)
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
if (webkitDownloadIsCancelled(download.get()))
return;
GRefPtr<WebKitURIResponse> response = adoptGRef(webkitURIResponseCreateForResourceResponse(resourceResponse));
webkitDownloadSetResponse(download.get(), response.get());
}
void didReceiveData(DownloadProxy& downloadProxy, uint64_t length, uint64_t, uint64_t) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
webkitDownloadNotifyProgress(download.get(), length);
}
void decideDestinationWithSuggestedFilename(DownloadProxy& downloadProxy, const ResourceResponse& resourceResponse, const String& filename, CompletionHandler<void(AllowOverwrite, String)>&& completionHandler) override
{
didReceiveResponse(downloadProxy, resourceResponse);
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(DownloadProxy& downloadProxy, const String& path) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
webkitDownloadDestinationCreated(download.get(), path);
}
void didFail(DownloadProxy& downloadProxy, const ResourceError& error, API::Data*) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
if (webkitDownloadIsCancelled(download.get())) {
webkitDownloadCancelled(download.get());
} else
webkitDownloadFailed(download.get(), error);
webkitWebContextRemoveDownload(&downloadProxy);
}
void legacyDidCancel(DownloadProxy& downloadProxy) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
webkitDownloadCancelled(download.get());
webkitWebContextRemoveDownload(&downloadProxy);
}
void didFinish(DownloadProxy& downloadProxy) override
{
GRefPtr<WebKitDownload> download = webkitWebContextGetOrCreateDownload(&downloadProxy);
webkitDownloadFinished(download.get());
webkitWebContextRemoveDownload(&downloadProxy);
}
WebKitWebContext* m_webContext;
};
void attachDownloadClientToContext(WebKitWebContext* webContext)
{
webkitWebContextGetProcessPool(webContext).setLegacyDownloadClient(adoptRef(*new LegacyDownloadClient(webContext)));
}