e800605ad3
This URL library ends up being a relatively fundamental base library of the system, as LibCore depends on LibURL. This change has two main benefits: * Moving AK back more towards being an agnostic library that can be used between the kernel and userspace. URL has never really fit that description - and is not used in the kernel. * URL _should_ depend on LibUnicode, as it needs punnycode support. However, it's not really possible to do this inside of AK as it can't depend on any external library. This change brings us a little closer to being able to do that, but unfortunately we aren't there quite yet, as the code generators depend on LibCore.
120 lines
3.6 KiB
C++
120 lines
3.6 KiB
C++
/*
|
|
* Copyright (c) 2021-2022, Dex♪ <dexes.ttp@gmail.com>
|
|
* Copyright (c) 2023, Kenneth Myhra <kennethmyhra@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/ByteBuffer.h>
|
|
#include <LibCore/EventReceiver.h>
|
|
#include <LibURL/URL.h>
|
|
#include <LibWeb/Bindings/PlatformObject.h>
|
|
#include <LibWeb/DOM/EventTarget.h>
|
|
#include <LibWeb/Forward.h>
|
|
#include <LibWeb/HTML/Window.h>
|
|
#include <LibWeb/WebIDL/ExceptionOr.h>
|
|
|
|
#define ENUMERATE_WEBSOCKET_EVENT_HANDLERS(E) \
|
|
E(onerror, HTML::EventNames::error) \
|
|
E(onclose, HTML::EventNames::close) \
|
|
E(onopen, HTML::EventNames::open) \
|
|
E(onmessage, HTML::EventNames::message)
|
|
|
|
namespace Web::WebSockets {
|
|
|
|
class WebSocketClientSocket;
|
|
class WebSocketClientManager;
|
|
|
|
class WebSocket final : public DOM::EventTarget {
|
|
WEB_PLATFORM_OBJECT(WebSocket, DOM::EventTarget);
|
|
JS_DECLARE_ALLOCATOR(WebSocket);
|
|
|
|
public:
|
|
enum class ReadyState : u16 {
|
|
Connecting = 0,
|
|
Open = 1,
|
|
Closing = 2,
|
|
Closed = 3,
|
|
};
|
|
|
|
static WebIDL::ExceptionOr<JS::NonnullGCPtr<WebSocket>> construct_impl(JS::Realm&, String const& url, Optional<Variant<String, Vector<String>>> const& protocols);
|
|
|
|
virtual ~WebSocket() override;
|
|
|
|
WebIDL::ExceptionOr<String> url() const { return TRY_OR_THROW_OOM(vm(), m_url.to_string()); }
|
|
void set_url(URL::URL url) { m_url = move(url); }
|
|
|
|
#undef __ENUMERATE
|
|
#define __ENUMERATE(attribute_name, event_name) \
|
|
void set_##attribute_name(WebIDL::CallbackType*); \
|
|
WebIDL::CallbackType* attribute_name();
|
|
ENUMERATE_WEBSOCKET_EVENT_HANDLERS(__ENUMERATE)
|
|
#undef __ENUMERATE
|
|
|
|
ReadyState ready_state() const;
|
|
String extensions() const;
|
|
WebIDL::ExceptionOr<String> protocol() const;
|
|
|
|
String const& binary_type() { return m_binary_type; }
|
|
void set_binary_type(String const& type) { m_binary_type = type; }
|
|
|
|
WebIDL::ExceptionOr<void> close(Optional<u16> code, Optional<String> reason);
|
|
WebIDL::ExceptionOr<void> send(Variant<JS::Handle<WebIDL::BufferSource>, JS::Handle<FileAPI::Blob>, String> const& data);
|
|
|
|
private:
|
|
void on_open();
|
|
void on_message(ByteBuffer message, bool is_text);
|
|
void on_error();
|
|
void on_close(u16 code, String reason, bool was_clean);
|
|
|
|
WebSocket(JS::Realm&);
|
|
|
|
virtual void initialize(JS::Realm&) override;
|
|
|
|
ErrorOr<void> establish_web_socket_connection(URL::URL& url_record, Vector<String>& protocols, HTML::EnvironmentSettingsObject& client);
|
|
|
|
URL::URL m_url;
|
|
String m_binary_type { "blob"_string };
|
|
RefPtr<WebSocketClientSocket> m_websocket;
|
|
};
|
|
|
|
class WebSocketClientSocket : public RefCounted<WebSocketClientSocket> {
|
|
public:
|
|
virtual ~WebSocketClientSocket();
|
|
|
|
struct CertificateAndKey {
|
|
ByteString certificate;
|
|
ByteString key;
|
|
};
|
|
|
|
struct Message {
|
|
ByteBuffer data;
|
|
bool is_text { false };
|
|
};
|
|
|
|
enum class Error {
|
|
CouldNotEstablishConnection,
|
|
ConnectionUpgradeFailed,
|
|
ServerClosedSocket,
|
|
};
|
|
|
|
virtual Web::WebSockets::WebSocket::ReadyState ready_state() = 0;
|
|
virtual ByteString subprotocol_in_use() = 0;
|
|
|
|
virtual void send(ByteBuffer binary_or_text_message, bool is_text) = 0;
|
|
virtual void send(StringView text_message) = 0;
|
|
virtual void close(u16 code = 1005, ByteString reason = {}) = 0;
|
|
|
|
Function<void()> on_open;
|
|
Function<void(Message)> on_message;
|
|
Function<void(Error)> on_error;
|
|
Function<void(u16 code, ByteString reason, bool was_clean)> on_close;
|
|
Function<CertificateAndKey()> on_certificate_requested;
|
|
|
|
protected:
|
|
explicit WebSocketClientSocket() = default;
|
|
};
|
|
|
|
}
|