mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-20 01:37:39 +03:00
168d28c15f
LibWeb will need to use unbuffered requests to support server-sent events. Connection for such events remain open and the remote end sends data as HTTP bodies at its leisure. The browser needs to be able to handle this data as it arrives, as the request essentially never finishes. To support this, this make Protocol::Request operate in one of two modes: buffered or unbuffered. The existing mechanism for setting up a buffered request was a bit awkward; you had to set specific callbacks, but be sure not to set some others, and then set a flag. The new mechanism is to set the mode and the callbacks that the mode needs in one API.
105 lines
4.0 KiB
C++
105 lines
4.0 KiB
C++
/*
|
|
* Copyright (c) 2022, Dex♪ <dexes.ttp@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "WebSocketClientAdapter.h"
|
|
#include <LibProtocol/Request.h>
|
|
#include <LibProtocol/RequestClient.h>
|
|
#include <LibWebView/RequestServerAdapter.h>
|
|
|
|
namespace WebView {
|
|
|
|
ErrorOr<NonnullRefPtr<RequestServerRequestAdapter>> RequestServerRequestAdapter::try_create(NonnullRefPtr<Protocol::Request> request)
|
|
{
|
|
return adopt_nonnull_ref_or_enomem(new (nothrow) RequestServerRequestAdapter(move(request)));
|
|
}
|
|
|
|
RequestServerRequestAdapter::RequestServerRequestAdapter(NonnullRefPtr<Protocol::Request> request)
|
|
: m_request(request)
|
|
{
|
|
request->on_progress = [weak_this = make_weak_ptr()](Optional<u64> total_size, u64 downloaded_size) {
|
|
if (auto strong_this = weak_this.strong_ref())
|
|
if (strong_this->on_progress)
|
|
strong_this->on_progress(total_size, downloaded_size);
|
|
};
|
|
|
|
request->on_certificate_requested = [weak_this = make_weak_ptr()]() {
|
|
if (auto strong_this = weak_this.strong_ref()) {
|
|
if (strong_this->on_certificate_requested) {
|
|
auto certificate_and_key = strong_this->on_certificate_requested();
|
|
return Protocol::Request::CertificateAndKey {
|
|
.certificate = move(certificate_and_key.certificate),
|
|
.key = move(certificate_and_key.key),
|
|
};
|
|
}
|
|
}
|
|
|
|
return Protocol::Request::CertificateAndKey {};
|
|
};
|
|
}
|
|
|
|
RequestServerRequestAdapter::~RequestServerRequestAdapter() = default;
|
|
|
|
void RequestServerRequestAdapter::set_buffered_request_finished_callback(Protocol::Request::BufferedRequestFinished on_buffered_request_finished)
|
|
{
|
|
m_request->set_buffered_request_finished_callback(move(on_buffered_request_finished));
|
|
}
|
|
|
|
void RequestServerRequestAdapter::set_unbuffered_request_callbacks(Protocol::Request::HeadersReceived on_headers_received, Protocol::Request::DataReceived on_data_received, Protocol::Request::RequestFinished on_finished)
|
|
{
|
|
m_request->set_unbuffered_request_callbacks(move(on_headers_received), move(on_data_received), move(on_finished));
|
|
}
|
|
|
|
bool RequestServerRequestAdapter::stop()
|
|
{
|
|
return m_request->stop();
|
|
}
|
|
|
|
ErrorOr<NonnullRefPtr<RequestServerAdapter>> RequestServerAdapter::try_create(NonnullRefPtr<Protocol::RequestClient> protocol_client)
|
|
{
|
|
return try_make_ref_counted<RequestServerAdapter>(move(protocol_client));
|
|
}
|
|
|
|
ErrorOr<NonnullRefPtr<RequestServerAdapter>> RequestServerAdapter::try_create()
|
|
{
|
|
auto protocol_client = TRY(Protocol::RequestClient::try_create());
|
|
return try_make_ref_counted<RequestServerAdapter>(move(protocol_client));
|
|
}
|
|
|
|
RequestServerAdapter::RequestServerAdapter(NonnullRefPtr<Protocol::RequestClient> protocol_client)
|
|
: m_protocol_client(protocol_client)
|
|
{
|
|
}
|
|
|
|
RequestServerAdapter::~RequestServerAdapter() = default;
|
|
|
|
RefPtr<Web::ResourceLoaderConnectorRequest> RequestServerAdapter::start_request(ByteString const& method, URL::URL const& url, HashMap<ByteString, ByteString> const& headers, ReadonlyBytes body, Core::ProxyData const& proxy)
|
|
{
|
|
auto protocol_request = m_protocol_client->start_request(method, url, headers, body, proxy);
|
|
if (!protocol_request)
|
|
return {};
|
|
return RequestServerRequestAdapter::try_create(protocol_request.release_nonnull()).release_value_but_fixme_should_propagate_errors();
|
|
}
|
|
|
|
RefPtr<Web::WebSockets::WebSocketClientSocket> RequestServerAdapter::websocket_connect(URL::URL const& url, AK::ByteString const& origin, Vector<AK::ByteString> const& protocols)
|
|
{
|
|
auto underlying_websocket = m_protocol_client->websocket_connect(url, origin, protocols);
|
|
if (!underlying_websocket)
|
|
return {};
|
|
return WebSocketClientSocketAdapter::create(underlying_websocket.release_nonnull());
|
|
}
|
|
|
|
void RequestServerAdapter::prefetch_dns(URL::URL const& url)
|
|
{
|
|
m_protocol_client->ensure_connection(url, RequestServer::CacheLevel::ResolveOnly);
|
|
}
|
|
|
|
void RequestServerAdapter::preconnect(URL::URL const& url)
|
|
{
|
|
m_protocol_client->ensure_connection(url, RequestServer::CacheLevel::CreateConnection);
|
|
}
|
|
|
|
}
|