#pragma once
#include "DataReference.h"
#include "DownloadID.h"
#include "DownloadMap.h"
#include "NetworkDataTask.h"
#include "PendingDownload.h"
#include "PolicyDecision.h"
#include "SandboxExtension.h"
#include <WebCore/NotImplemented.h>
#include <wtf/Forward.h>
#include <wtf/HashMap.h>
#include <wtf/Noncopyable.h>
namespace PAL {
class SessionID;
}
namespace WebCore {
class BlobDataFileReference;
class ResourceHandle;
class ResourceRequest;
class ResourceResponse;
}
namespace IPC {
class Connection;
}
namespace WebKit {
class AuthenticationManager;
class Download;
class NetworkConnectionToWebProcess;
class NetworkLoad;
class PendingDownload;
enum class CallDownloadDidStart : bool { No, Yes };
class DownloadManager {
WTF_MAKE_NONCOPYABLE(DownloadManager);
public:
class Client {
public:
virtual ~Client() { }
virtual void didCreateDownload() = 0;
virtual void didDestroyDownload() = 0;
virtual IPC::Connection* downloadProxyConnection() = 0;
virtual IPC::Connection* parentProcessConnectionForDownloads() = 0;
virtual AuthenticationManager& downloadsAuthenticationManager() = 0;
virtual NetworkSession* networkSession(PAL::SessionID) const = 0;
virtual void ref() const = 0;
virtual void deref() const = 0;
};
explicit DownloadManager(Client&);
void startDownload(PAL::SessionID, DownloadID, const WebCore::ResourceRequest&, Optional<NavigatingToAppBoundDomain>, const String& suggestedName = { });
void dataTaskBecameDownloadTask(DownloadID, std::unique_ptr<Download>&&);
void continueWillSendRequest(DownloadID, WebCore::ResourceRequest&&);
void convertNetworkLoadToDownload(DownloadID, std::unique_ptr<NetworkLoad>&&, ResponseCompletionHandler&&, Vector<RefPtr<WebCore::BlobDataFileReference>>&&, const WebCore::ResourceRequest&, const WebCore::ResourceResponse&);
void downloadDestinationDecided(DownloadID, Ref<NetworkDataTask>&&);
void resumeDownload(PAL::SessionID, DownloadID, const IPC::DataReference& resumeData, const String& path, SandboxExtension::Handle&&, CallDownloadDidStart);
void cancelDownload(DownloadID, CompletionHandler<void(const IPC::DataReference&)>&&);
#if PLATFORM(COCOA)
void publishDownloadProgress(DownloadID, const URL&, SandboxExtension::Handle&&);
#endif
Download* download(DownloadID downloadID) { return m_downloads.get(downloadID); }
void downloadFinished(Download&);
bool isDownloading() const { return !m_downloads.isEmpty(); }
void applicationDidEnterBackground();
void applicationWillEnterForeground();
void didCreateDownload();
void didDestroyDownload();
IPC::Connection* downloadProxyConnection();
AuthenticationManager& downloadsAuthenticationManager();
Client& client() { return m_client; }
private:
Client& m_client;
HashMap<DownloadID, std::unique_ptr<PendingDownload>> m_pendingDownloads;
HashMap<DownloadID, RefPtr<NetworkDataTask>> m_downloadsAfterDestinationDecided;
DownloadMap m_downloads;
};
}