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 <streambuf>
#include <vector>
#include <cstddef>
 
namespace Pt {
 
namespace Http {
 
class WebSocketChannel;
 

class WebSocketMessageBuffer : public std::streambuf
{
    public:
        WebSocketMessageBuffer();

        ~WebSocketMessageBuffer();

        void discard();

        void discard(std::size_t n);

        std::size_t size() const;

        const char* data() const;
 
    protected:
        virtual int_type overflow(int_type ch);
 
        virtual std::streamsize xsputn(const char* data, std::streamsize n);
 
        virtual int_type underflow();
 
    private:
        std::vector<char> _payload;
};
 

class PT_HTTP_API WebSocketMessage : private NonCopyable
{
    friend class WebSocketChannel;
 
    public:
        WebSocketMessage();

        ~WebSocketMessage();

        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 discard(std::size_t n);

        void clear();

        WebSocketMessageBuffer& buffer()
        { return _buf; }
 
    private:
        void setTypeFromEngine(Type type);

        void setBuffer(std::streambuf& sb);
 
    private:
        Type                   _type;
        WebSocketMessageBuffer _buf;
        std::iostream          _body;
};
 
} // namespace Http
 
} // namespace Pt
 
#endif