stream_server.h

Go to the documentation of this file.

#pragma once
 
#ifdef FASTLED_HAS_NETWORKING
 
#include "fl/net/http/stream_transport.h"
#include "fl/stl/asio/http/native_server.h"  // IWYU pragma: keep
#include "fl/stl/asio/http/http_parser.h"
#include "fl/stl/string.h"
#include "fl/stl/unique_ptr.h"
#include "fl/stl/flat_map.h"
#include "fl/stl/vector.h"
#include "fl/stl/noexcept.h"
 
namespace fl {
namespace net {
namespace http {

class HttpStreamServer : public HttpStreamTransport {
public:
    HttpStreamServer(u16 port = 8080, u32 heartbeatIntervalMs = 30000);

    ~HttpStreamServer() FL_NOEXCEPT override;
 
    // Connection Management (override from HttpStreamTransport)

    bool connect() override;

    void disconnect() override;

    bool isConnected() const override;
 
    // Server-specific methods

    u16 port() const;

    void acceptClients();

    size_t getClientCount() const;

    void disconnectClient(u32 clientId);

    fl::vector<u32> getClientIds() const;
 
protected:
    // Abstract methods implementation (from HttpStreamTransport)

    int sendData(fl::span<const u8> data) override;

    int recvData(fl::span<u8> buffer) override;

    void triggerReconnect() override;
 
private:
    struct ClientState {
        u32 clientId;
        bool httpHeaderReceived;
        bool httpHeaderSent;
        HttpRequestParser requestParser;
        fl::vector<u8> pendingData;  // Data waiting to be processed
        fl::string headerBuffer;     // Accumulates partial HTTP header reads
 
        // Default constructor (required for map operations)
        ClientState() FL_NOEXCEPT
            : clientId(0)
            , httpHeaderReceived(false)
            , httpHeaderSent(false)
            , requestParser()
            , pendingData()
        {
        }
 
        ClientState(u32 id)
            : clientId(id)
            , httpHeaderReceived(false)
            , httpHeaderSent(false)
            , requestParser()
            , pendingData()
        {
        }
    };

    bool readHttpRequestHeader(u32 clientId);

    bool sendHttpResponseHeader(u32 clientId);

    bool processClientData(u32 clientId);

    ClientState* getOrCreateClientState(u32 clientId);

    void removeClientState(u32 clientId);

    fl::unique_ptr<NativeHttpServer> mNativeServer;

    u16 mPort;

    fl::flat_map<u32, ClientState> mClientStates;

    fl::vector<u8> mRecvBuffer;

    u32 mLastProcessedClientId;
};
 
}  // namespace http
}  // namespace net
}  // namespace fl
 
#endif // FASTLED_HAS_NETWORKING