websocketmessage.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 "wspp.h"
#include "websocketutils.h"
 
// External Includes
#include <utility/dllexport.h>
 
namespace nap
{
    // Forward Declares
    template<typename T>
    class WebSocketServerEndPointSetup;
    template<typename T>
    class WebSocketClientEndPointSetup;
    template<typename T>
    class WebSocketClientWrapper;
 
    class NAPAPI WebSocketMessage final
    {
        template<typename T>
        friend class WebSocketServerEndPointSetup;
        template<typename T>
        friend class WebSocketClientEndPointSetup;
        template<typename T>
        friend class WebSocketClientWrapper;
 
    public:
        WebSocketMessage(const std::string& message, EWebSocketOPCode code, bool fin = true);
 
        WebSocketMessage(std::string&& message, EWebSocketOPCode code, bool fin = true);
 
        // no default constructor
        WebSocketMessage() = delete;
        // Move constructor
        WebSocketMessage(WebSocketMessage&& other);
        // Move assignment operator
        WebSocketMessage& operator=(WebSocketMessage&& other);
        // Default copy constructor
        WebSocketMessage(const WebSocketMessage& other) = default;
        // Default copy assignment operator
        WebSocketMessage& operator=(const WebSocketMessage& other) = default;
 
        const std::string& getPayload() const;
 
        void appendPayload(const std::string& payload);
 
        EWebSocketOPCode getCode() const;
 
        bool getFin() const;
 
    private:
        WebSocketMessage(wspp::MessagePtr message);
 
        std::string mMessage;                           
        EWebSocketOPCode mCode;                         
        bool mFin = true;                               
    };
}