#pragma once
#include "ActiveDOMObject.h"
#include "EventTarget.h"
#include "ExceptionOr.h"
#include "Timer.h"
#include <wtf/URL.h>
#include "WebSocketChannelClient.h"
#include <wtf/Deque.h>
#include <wtf/HashSet.h>
#include <wtf/Lock.h>
namespace JSC {
class ArrayBuffer;
class ArrayBufferView;
}
namespace WebCore {
class Blob;
class ThreadableWebSocketChannel;
class WebSocket final : public RefCounted<WebSocket>, public EventTargetWithInlineData, public ActiveDOMObject, private WebSocketChannelClient {
public:
static const char* subprotocolSeparator();
static ExceptionOr<Ref<WebSocket>> create(ScriptExecutionContext&, const String& url);
static ExceptionOr<Ref<WebSocket>> create(ScriptExecutionContext&, const String& url, const String& protocol);
static ExceptionOr<Ref<WebSocket>> create(ScriptExecutionContext&, const String& url, const Vector<String>& protocols);
virtual ~WebSocket();
static HashSet<WebSocket*>& allActiveWebSockets(const LockHolder&);
static Lock& allActiveWebSocketsMutex();
enum State {
CONNECTING = 0,
OPEN = 1,
CLOSING = 2,
CLOSED = 3
};
ExceptionOr<void> connect(const String& url);
ExceptionOr<void> connect(const String& url, const String& protocol);
ExceptionOr<void> connect(const String& url, const Vector<String>& protocols);
ExceptionOr<void> send(const String& message);
ExceptionOr<void> send(JSC::ArrayBuffer&);
ExceptionOr<void> send(JSC::ArrayBufferView&);
ExceptionOr<void> send(Blob&);
ExceptionOr<void> close(Optional<unsigned short> code, const String& reason);
RefPtr<ThreadableWebSocketChannel> channel() const;
const URL& url() const;
State readyState() const;
unsigned bufferedAmount() const;
String protocol() const;
String extensions() const;
String binaryType() const;
ExceptionOr<void> setBinaryType(const String&);
ScriptExecutionContext* scriptExecutionContext() const final;
using RefCounted::ref;
using RefCounted::deref;
private:
explicit WebSocket(ScriptExecutionContext&);
void resumeTimerFired();
void dispatchOrQueueErrorEvent();
void dispatchOrQueueEvent(Ref<Event>&&);
void contextDestroyed() final;
bool canSuspendForDocumentSuspension() const final;
void suspend(ReasonForSuspension) final;
void resume() final;
void stop() final;
const char* activeDOMObjectName() const final;
EventTargetInterface eventTargetInterface() const final;
void refEventTarget() final { ref(); }
void derefEventTarget() final { deref(); }
void didConnect() final;
void didReceiveMessage(const String& message) final;
void didReceiveBinaryData(Vector<uint8_t>&&) final;
void didReceiveMessageError() final;
void didUpdateBufferedAmount(unsigned bufferedAmount) final;
void didStartClosingHandshake() final;
void didClose(unsigned unhandledBufferedAmount, ClosingHandshakeCompletionStatus, unsigned short code, const String& reason) final;
void didUpgradeURL() final;
size_t getFramingOverhead(size_t payloadSize);
enum class BinaryType { Blob, ArrayBuffer };
RefPtr<ThreadableWebSocketChannel> m_channel;
State m_state { CONNECTING };
URL m_url;
unsigned m_bufferedAmount { 0 };
unsigned m_bufferedAmountAfterClose { 0 };
BinaryType m_binaryType { BinaryType::Blob };
String m_subprotocol;
String m_extensions;
Timer m_resumeTimer;
bool m_shouldDelayEventFiring { false };
Deque<Ref<Event>> m_pendingEvents;
bool m_dispatchedErrorEvent { false };
};
}