websocketserver.h

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
 
#pragma once
 
// Local Includes
#include "iwebsocketserverendpoint.h"
#include "websocketinterface.h"
 
// External Includes
#include <rtti/factory.h>
#include <nap/resourceptr.h>
 
namespace nap
{
    class NAPAPI IWebSocketServer : public WebSocketInterface
    {
        template<typename config>
        friend class WebSocketServerEndPointSetup;
 
        RTTI_ENABLE(WebSocketInterface)
    public:
        IWebSocketServer(WebSocketService& service);
 
        virtual bool init(utility::ErrorState& errorState) override;
 
        virtual void onDestroy() override;
 
        ResourcePtr<IWebSocketServerEndPoint> mEndPoint;    
 
    protected:
        // Called by web-socket server endpoint when a new message is received
        virtual void onMessageReceived(const WebSocketConnection& connection, const WebSocketMessage& message) = 0;
 
        // Called by web-socket server endpoint when a client connection opened
        virtual void onConnectionOpened(const WebSocketConnection& connection) = 0;
 
        // Called by web-socket server endpoint when a client connection closed
        virtual void onConnectionClosed(const WebSocketConnection& connection, int code, const std::string& reason) = 0;
 
        // Called by web-socket server endpoint when a client connection failed to establish
        virtual void onConnectionFailed(const WebSocketConnection& connection, int code, const std::string& reason) = 0;
    };
 
 
    // WebSocketServer
 
    class NAPAPI WebSocketServer : public IWebSocketServer
    {
        friend class WebSocketService;
        RTTI_ENABLE(IWebSocketServer)
    public:
        WebSocketServer(WebSocketService& service);
 
        bool send(const WebSocketConnection& connection, const WebSocketMessage& message, nap::utility::ErrorState& error);
 
        bool send(const WebSocketConnection& connection, void const* payload, int length, EWebSocketOPCode code, nap::utility::ErrorState& error);
 
        bool send(const WebSocketConnection& connection, const std::string& message, EWebSocketOPCode code, nap::utility::ErrorState& error);
 
        bool broadcast(const WebSocketMessage& message, nap::utility::ErrorState& error);
 
        bool broadcast(void const* payload, int length, EWebSocketOPCode code, nap::utility::ErrorState& error);
 
    private:
        virtual void onConnectionOpened(const WebSocketConnection& connection) override;
 
        virtual void onConnectionClosed(const WebSocketConnection& connection, int code, const std::string& reason) override;
 
        virtual void onConnectionFailed(const WebSocketConnection& connection, int code, const std::string& reason) override;
 
        virtual void onMessageReceived(const WebSocketConnection& connection, const WebSocketMessage& message) override;
    };
 
    // Object creator used for constructing the web-socket server
    using WebSocketServerObjectCreator = rtti::ObjectCreator<WebSocketServer, WebSocketService>;
}