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