WebURLSchemeHandler.h [plain text]
#pragma once
#include "WebURLSchemeTask.h"
#include <wtf/HashMap.h>
#include <wtf/HashSet.h>
#include <wtf/RefCounted.h>
#include <wtf/RefPtr.h>
namespace WebCore {
class ResourceRequest;
}
namespace WebKit {
struct URLSchemeTaskParameters;
class WebPageProxy;
class WebProcessProxy;
using SyncLoadCompletionHandler = CompletionHandler<void(const WebCore::ResourceResponse&, const WebCore::ResourceError&, const Vector<char>&)>;
class WebURLSchemeHandler : public RefCounted<WebURLSchemeHandler> {
WTF_MAKE_NONCOPYABLE(WebURLSchemeHandler);
public:
virtual ~WebURLSchemeHandler();
uint64_t identifier() const { return m_identifier; }
void startTask(WebPageProxy&, WebProcessProxy&, WebCore::PageIdentifier, URLSchemeTaskParameters&&, SyncLoadCompletionHandler&&);
void stopTask(WebPageProxy&, uint64_t taskIdentifier);
void stopAllTasksForPage(WebPageProxy&, WebProcessProxy*);
void taskCompleted(WebURLSchemeTask&);
protected:
WebURLSchemeHandler();
private:
virtual void platformStartTask(WebPageProxy&, WebURLSchemeTask&) = 0;
virtual void platformStopTask(WebPageProxy&, WebURLSchemeTask&) = 0;
virtual void platformTaskCompleted(WebURLSchemeTask&) = 0;
void removeTaskFromPageMap(WebPageProxyIdentifier, uint64_t taskID);
WebProcessProxy* processForTaskIdentifier(uint64_t) const;
uint64_t m_identifier;
HashMap<uint64_t, Ref<WebURLSchemeTask>> m_tasks;
HashMap<WebPageProxyIdentifier, HashSet<uint64_t>> m_tasksByPageIdentifier;
SyncLoadCompletionHandler m_syncLoadCompletionHandler;
};
}