WebSocketSession.h

/* Copyright (C) 2026 by Marc Boris Duerner
   SPDX-License-Identifier: LGPL-2.1-or-later WITH mif-exception
*/
 
#ifndef PT_HTTP_WEBSOCKETSESSION_H
#define PT_HTTP_WEBSOCKETSESSION_H
 
#include <Pt/Http/Api.h>
#include <Pt/Http/WebSocketMessage.h>
#include <Pt/Http/Message.h>
#include <Pt/Connectable.h>
#include <string>
#include <cstddef>
 
namespace Pt {
 
namespace System {
class EventLoop;
}
 
namespace Http {
 
class Stream;
class WebSocketServlet;
class WebSocketService;
class WebSocketConnection;

class PT_HTTP_API WebSocketSession : public Connectable
{
    friend class WebSocketServlet;
 
    public:
        WebSocketSession(WebSocketServlet& servlet,
                         System::EventLoop& loop,
                         Stream& stream);

        ~WebSocketSession();

        WebSocketService& service()
        { return *_service; }

        const WebSocketService& service() const
        { return *_service; }

        System::EventLoop& loop()
        { return *_loop; }

        const System::EventLoop& loop() const
        { return *_loop; }

        WebSocketMessage& incoming();

        WebSocketMessage& outgoing();

        void beginSend();

        MessageProgress endSend();

        void beginReceive();

        MessageProgress endReceive();

        void ping(const char* payload = 0, std::size_t n = 0);

        void close(unsigned code = 1000,
                   const std::string& reason = std::string());

        unsigned closeCode() const;

        const std::string& closeReason() const;
 
    protected:
        virtual void onInput() = 0;

        virtual void onOutput() = 0;

        virtual void onClose() = 0;
 
    private:
        void shutdown();
 
        void onInputReady();
 
        void onOutputReady();
 
        void onClosed();
 
    private:
        WebSocketService*    _service;
        WebSocketServlet*    _servlet;
        System::EventLoop*   _loop;
        WebSocketConnection* _connection;
};
 
} // namespace Http
 
} // namespace Pt
 
#endif