MultiplayerWebSocket.h

Go to the documentation of this file.

// Copyright(c) 2023 PixoVR, LLC. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "IWebSocket.h"
 
enum MultiplayerWebSocketRequestType : uint8_t
{
    Invalid,
    MatchMake
};
 
struct MultiplayerRequestData
{
    uint8_t RequestType;
    FString Json;
};
 
class MultiplayerWebSocket
{
public:
 
    MultiplayerWebSocket();
    ~MultiplayerWebSocket();
 
    DECLARE_EVENT(MultiplayerWebSocket, FSocketConnected);
    FSocketConnected& OnWebSocketConnected() { return SocketConnectedEvent; }
 
    DECLARE_EVENT_OneParam(MultiplayerWebSocket, FSocketConnectFailed, const FString&);
    FSocketConnectFailed& OnWebSocketConnectFailed() { return SocketConnectFailedEvent; }
 
    DECLARE_EVENT_ThreeParams(MultiplayerWebSocket, FSocketDisconnected, int32 StatusCode, const FString& Reason, bool bWasClean);
    FSocketDisconnected& OnWebSocketDisconnected() { return SocketDisconnectedEvent; }
 
    DECLARE_EVENT_OneParam(MultiplayerWebSocket, FSocketMessageReceived, const FString&);
    FSocketMessageReceived& OnWebSocketMessageReceived() { return SocketMessageReceivedEvent; }
 
    bool AttemptConnect(FString Uri, FString Protocol = "wss");
    void Disconnect();
    bool IsWebsocketConnected();
    void RequestMatchMake(int OrgId, int ModuleId);
 
private:
    void SetupSocket(TSharedPtr<IWebSocket>& Socket, FString Uri, FString Protocol);
    void ProcessNextRequest();
    void CleanUpRequests();
 
    void SendData(FString Data);
 
public:
    // Bad Closes are when the underlying implementation of WebSockets cause new sockets
    // to close when a Read or Send occurs. The error code is 1006 and only happens in the editor.
    bool bShouldRestartOnBadClose;
    bool bShouldReplayLastRequestOnBadClose;
 
private:
    TSharedPtr<IWebSocket> WebSocket;
    TArray<MultiplayerRequestData> PendingWebsocketRequests;
    MultiplayerRequestData LastWebSocketRequest;
 
    FSocketConnected SocketConnectedEvent;
    FSocketConnectFailed SocketConnectFailedEvent;
    FSocketDisconnected SocketDisconnectedEvent;
    FSocketMessageReceived SocketMessageReceivedEvent;
 
    FString SocketUri;
    FString SocketProtocol;
};