WebSocketMessage.h

/* Copyright (C) 2026 by Marc Boris Duerner
   SPDX-License-Identifier: LGPL-2.1-or-later WITH mif-exception
*/
 
#ifndef PT_HTTP_WEBSOCKETMESSAGE_H
#define PT_HTTP_WEBSOCKETMESSAGE_H
 
#include <Pt/Http/Api.h>
#include <Pt/NonCopyable.h>
#include <iostream>
#include <cstddef>
 
namespace Pt {
 
namespace Http {
 
class WebSocketConnection;

class PT_HTTP_API WebSocketMessage : private NonCopyable
{
    friend class WebSocketConnection;
 
    public:
        enum Type
        {
            Unknown, 
            Text,    
            Binary   
        };

        Type type() const
        { return _type; }

        void setType(Type type);

        std::iostream& body()
        { return _body; }

        std::size_t available() const;

        std::size_t pending() const;

        void discard();

        void clear();
 
    private:
        class PayloadBuffer;
 
        WebSocketMessage();
 
        ~WebSocketMessage();
 
        void setTypeFromEngine(Type type);
 
        void append(const char* data, std::size_t n);
 
        void prepareRead();
 
        const char* sendData() const;
 
        std::size_t sendSize() const;
 
        void consume(std::size_t n);
 
    private:
        Type           _type;
        PayloadBuffer* _buffer;
        std::iostream  _body;
};
 
} // namespace Http
 
} // namespace Pt
 
#endif