WebPreviewLoaderClient.cpp [plain text]
#include "config.h"
#include "WebPreviewLoaderClient.h"
#if USE(QUICK_LOOK)
#include "WebPageProxyMessages.h"
#include "WebProcess.h"
#include <WebCore/QuickLook.h>
#include <wtf/Function.h>
#include <wtf/HashMap.h>
#include <wtf/NeverDestroyed.h>
namespace WebKit {
using PasswordCallbackMap = HashMap<uint64_t, Function<void(const String&)>>;
static PasswordCallbackMap& passwordCallbacks()
{
static NeverDestroyed<PasswordCallbackMap> callbackMap;
return callbackMap.get();
}
WebPreviewLoaderClient::WebPreviewLoaderClient(const String& fileName, const String& uti, uint64_t pageID)
: m_fileName { fileName }
, m_uti { uti }
, m_pageID { pageID }
{
WebProcess::singleton().send(Messages::WebPageProxy::DidStartLoadForQuickLookDocumentInMainFrame(m_fileName, m_uti), m_pageID);
}
WebPreviewLoaderClient::~WebPreviewLoaderClient()
{
passwordCallbacks().remove(m_pageID);
}
void WebPreviewLoaderClient::didReceiveDataArray(CFArrayRef dataArray)
{
CFArrayApplyFunction(dataArray, CFRangeMake(0, CFArrayGetCount(dataArray)), [](const void* value, void* context) {
ASSERT(CFGetTypeID(value) == CFDataGetTypeID());
static_cast<QuickLookDocumentData*>(context)->append((CFDataRef)value);
}, &m_data);
}
void WebPreviewLoaderClient::didFinishLoading()
{
WebProcess::singleton().send(Messages::WebPageProxy::DidFinishLoadForQuickLookDocumentInMainFrame(m_data), m_pageID);
m_data.clear();
}
void WebPreviewLoaderClient::didFail()
{
m_data.clear();
}
void WebPreviewLoaderClient::didRequestPassword(Function<void(const String&)>&& completionHandler)
{
ASSERT(!passwordCallbacks().contains(m_pageID));
passwordCallbacks().add(m_pageID, WTFMove(completionHandler));
WebProcess::singleton().send(Messages::WebPageProxy::DidRequestPasswordForQuickLookDocumentInMainFrame(m_fileName), m_pageID);
}
void WebPreviewLoaderClient::didReceivePassword(const String& password, uint64_t pageID)
{
ASSERT(passwordCallbacks().contains(pageID));
auto completionHandler = passwordCallbacks().take(pageID);
completionHandler(password);
}
}
#endif // USE(QUICK_LOOK)