overte-thingvellir/libraries/script-engine/src/WebSocketClass.cpp

//
//  WebSocketClass.cpp
//  libraries/script-engine/src/
//
//  Created by Thijs Wenker on 8/4/15.
//  Copyright (c) 2015 High Fidelity, Inc. All rights reserved.
//
//  This class is an implementation of the WebSocket object for scripting use.  It provides a near-complete implementation
//  of the class described in the Mozilla docs: https://developer.mozilla.org/en-US/docs/Web/API/WebSocket
//
//  Distributed under the Apache License, Version 2.0.
//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
//

#include "ScriptEngine.h"
#include "WebSocketClass.h"

WebSocketClass::WebSocketClass(QScriptEngine* engine, QString url) :
    _engine(engine),
    _webSocket(new QWebSocket())
{
    initialize();
    _webSocket->open(url);
}

WebSocketClass::WebSocketClass(QScriptEngine* engine, QWebSocket* qWebSocket) :
    _engine(engine),
    _webSocket(qWebSocket)
{
    initialize();
}

void WebSocketClass::initialize() {
    connect(_webSocket, &QWebSocket::disconnected, this, &WebSocketClass::handleOnClose);
    connect(_webSocket, &QWebSocket::textMessageReceived, this, &WebSocketClass::handleOnMessage);
    connect(_webSocket, &QWebSocket::connected, this, &WebSocketClass::handleOnOpen);
    _binaryType = "blob";
}

QScriptValue WebSocketClass::constructor(QScriptContext* context, QScriptEngine* engine) {
    QString url;
    QScriptValue callee = context->callee();
    if (context->argumentCount() == 1) {
        url = context->argument(0).toString();
    }
    return engine->newQObject(new WebSocketClass(engine, url));
}

WebSocketClass::~WebSocketClass() {

}

void WebSocketClass::send(QScriptValue message) {
    _webSocket->sendTextMessage(message.toString());
}

void WebSocketClass::close() {
    this->close(QWebSocketProtocol::CloseCode::CloseCodeNormal);
}

void WebSocketClass::close(QWebSocketProtocol::CloseCode closeCode) {
    this->close(closeCode, QStringLiteral(""));
}

void WebSocketClass::close(QWebSocketProtocol::CloseCode closeCode, QString reason) {
    _webSocket->close(closeCode, reason);
}

void WebSocketClass::handleOnClose() {
    bool hasError = false;
    if (_webSocket->error() != QAbstractSocket::SocketError::UnknownSocketError) {
        hasError = true;
        if (_onErrorEvent.isFunction()) {
            _onErrorEvent.call();
        }
    }
    if (_onCloseEvent.isFunction()) {
        QScriptValueList args;
        QScriptValue arg = _engine->newObject();
        arg.setProperty("code", hasError ? QWebSocketProtocol::CloseCode::CloseCodeAbnormalDisconnection : _webSocket->closeCode());
        arg.setProperty("reason", _webSocket->closeReason());
        arg.setProperty("wasClean", !hasError);
        args << arg;
        _onCloseEvent.call(QScriptValue(), args);
    }
}

void WebSocketClass::handleOnMessage(const QString& message) {
    if (_onMessageEvent.isFunction()) {
        QScriptValueList args;
        QScriptValue arg = _engine->newObject();
        arg.setProperty("data", message);
        args << arg;
        _onMessageEvent.call(QScriptValue(), args);
    }
}

void WebSocketClass::handleOnOpen() {
    if (_onOpenEvent.isFunction()) {
        _onOpenEvent.call();
    }
}

QScriptValue qWSCloseCodeToScriptValue(QScriptEngine* engine, const QWebSocketProtocol::CloseCode &closeCode) {
    return closeCode;
}

void qWSCloseCodeFromScriptValue(const QScriptValue &object, QWebSocketProtocol::CloseCode &closeCode) {
    closeCode = (QWebSocketProtocol::CloseCode)object.toUInt16();
}

QScriptValue webSocketToScriptValue(QScriptEngine* engine, WebSocketClass* const &in) {
    return engine->newQObject(in);
}

void webSocketFromScriptValue(const QScriptValue &object, WebSocketClass* &out) {
    out = qobject_cast<WebSocketClass*>(object.toQObject());
}