WSWebSocketsProtocol.h

/*
╔══════════════════════════════════════════════════════════════════════════════╗
║                       SIMPLY POWERFUL TOOLKIT (SPTK)                         ║
╟──────────────────────────────────────────────────────────────────────────────╢
║  copyright            © 1999-2022 Alexey Parshin. All rights reserved.       ║
║  email                alexeyp@gmail.com                                      ║
╚══════════════════════════════════════════════════════════════════════════════╝
┌──────────────────────────────────────────────────────────────────────────────┐
│   This library is free software; you can redistribute it and/or modify it    │
│   under the terms of the GNU Library General Public License as published by  │
│   the Free Software Foundation; either version 2 of the License, or (at your │
│   option) any later version.                                                 │
│                                                                              │
│   This library is distributed in the hope that it will be useful, but        │
│   WITHOUT ANY WARRANTY; without even the implied warranty of                 │
│   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library   │
│   General Public License for more details.                                   │
│                                                                              │
│   You should have received a copy of the GNU Library General Public License  │
│   along with this library; if not, write to the Free Software Foundation,    │
│   Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.               │
│                                                                              │
│   Please report all bugs and problems to alexeyp@gmail.com.                  │
└──────────────────────────────────────────────────────────────────────────────┘
*/
 
#pragma once
 
#include "WSProtocol.h"
#include <sptk5/cnet>
 
namespace sptk {
 
 
class SP_EXPORT WSWebSocketsMessage
{
public:
    enum class OpCode : uint8_t
    {
        CONTINUATION = 0,
        TEXT = 1,
        BINARY = 2,
        CONNECTION_CLOSE = 8,
        PING1 = 9,
        PING2 = 10
    };
 
    WSWebSocketsMessage() = default;
 
    [[nodiscard]] const Buffer& payload() const;
 
    void decode(const char* incomingData);
 
    static void encode(const String& payload, OpCode opcode, bool finalMessage, Buffer& output);
 
    OpCode opcode() const;
 
    void opcode(OpCode code);
 
    uint32_t statusCode() const;
 
    bool isFinal() const;
 
private:
    OpCode m_opcode {0};
    uint32_t m_status {0};
    Buffer m_payload; 
    bool m_finalMessage {true};
};
 
class SP_EXPORT WSWebSocketsProtocol : public WSProtocol
{
public:
    WSWebSocketsProtocol(TCPSocket* socket, const HttpHeaders& headers);
 
    RequestInfo process() override;
 
    void replyCloseConnectionRequest(uint16_t statusCode, const String& closeReason);
};
 
class WSNotification
{
public:
    const std::map<String, String> m_headers;
    String m_data;
};
 
class WSNotificationManager
{
public:
    WSNotificationManager() = default;
 
private:
    Strings m_queues;
};
 
} // namespace sptk