#pragma once
#include "MessageReceiver.h"
#include "MessageSender.h"
#include <WebCore/SocketStreamHandle.h>
#include <pal/SessionID.h>
#include <wtf/Identified.h>
namespace IPC {
class Connection;
class Decoder;
class DataReference;
}
namespace WebCore {
class SocketStreamError;
}
namespace WebKit {
class WebSocketStream : public IPC::MessageSender, public IPC::MessageReceiver, public WebCore::SocketStreamHandle, public Identified<WebSocketStream> {
public:
static Ref<WebSocketStream> create(const WebCore::URL&, WebCore::SocketStreamHandleClient&, PAL::SessionID, const String& credentialPartition);
static void networkProcessCrashed();
static WebSocketStream* streamWithIdentifier(uint64_t);
void didReceiveMessage(IPC::Connection&, IPC::Decoder&);
void platformSend(const char*, size_t, Function<void(bool)>&&) final;
void platformClose() final;
size_t bufferedAmount() final;
void didOpenSocketStream();
void didCloseSocketStream();
void didReceiveSocketStreamData(const IPC::DataReference&);
void didFailToReceiveSocketStreamData();
void didUpdateBufferedAmount(uint64_t);
void didFailSocketStream(WebCore::SocketStreamError&&);
void didSendData(uint64_t, bool);
private:
IPC::Connection* messageSenderConnection() final;
uint64_t messageSenderDestinationID() final;
WebSocketStream(const WebCore::URL&, WebCore::SocketStreamHandleClient&, PAL::SessionID, const String& credentialPartition);
~WebSocketStream();
size_t m_bufferedAmount { 0 };
WebCore::SocketStreamHandleClient& m_client;
HashMap<uint64_t, Function<void(bool)>> m_sendDataCallbacks;
};
}