Serenity Operating System
1/*
2 * Copyright (c) 2021, Dex♪ <dexes.ttp@gmail.com>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#pragma once
8
9#include <AK/HashMap.h>
10#include <LibIPC/ConnectionFromClient.h>
11#include <LibWebSocket/WebSocket.h>
12#include <WebSocket/WebSocketClientEndpoint.h>
13#include <WebSocket/WebSocketServerEndpoint.h>
14
15namespace WebSocket {
16
17class ConnectionFromClient final
18 : public IPC::ConnectionFromClient<WebSocketClientEndpoint, WebSocketServerEndpoint> {
19 C_OBJECT(ConnectionFromClient);
20
21public:
22 ~ConnectionFromClient() override = default;
23
24 virtual void die() override;
25
26private:
27 explicit ConnectionFromClient(NonnullOwnPtr<Core::LocalSocket>);
28
29 virtual Messages::WebSocketServer::ConnectResponse connect(URL const&, DeprecatedString const&, Vector<DeprecatedString> const&, Vector<DeprecatedString> const&, IPC::Dictionary const&) override;
30 virtual Messages::WebSocketServer::ReadyStateResponse ready_state(i32) override;
31 virtual Messages::WebSocketServer::SubprotocolInUseResponse subprotocol_in_use(i32) override;
32 virtual void send(i32, bool, ByteBuffer const&) override;
33 virtual void close(i32, u16, DeprecatedString const&) override;
34 virtual Messages::WebSocketServer::SetCertificateResponse set_certificate(i32, DeprecatedString const&, DeprecatedString const&) override;
35
36 void did_connect(i32);
37 void did_receive_message(i32, Message);
38 void did_error(i32, i32 message);
39 void did_close(i32, u16 code, DeprecatedString reason, bool was_clean);
40 void did_request_certificates(i32);
41
42 i32 m_connection_ids { 0 };
43 HashMap<i32, RefPtr<WebSocket>> m_connections;
44};
45
46}