ClientHandler.hpp

Go to the documentation of this file.

#pragma once
 
#include "IEventHandler.hpp"
#include "http/Headers.hpp"
#include "http/HttpStatus.hpp"
#include "http/RequestParser.hpp"
#include "http/Response.hpp"
#include "http/Router.hpp"
#include <ctime>
 
namespace network {

class ClientHandler : public IEventHandler {
public:
    // =========================================================================
    // Lifecycle & Interface
    // =========================================================================
public:
    ClientHandler(int clientFd, int port, std::string const &clientAddr, http::Router const &);
    virtual ~ClientHandler();
 
    // IEventHandler implementation
    virtual void handleEvent(uint32_t events);
    virtual int getFd() const;
 
    // =========================================================================
    // CGI / Buffer Interaction (Called by external helpers)
    // =========================================================================
public:
    void pushToSendBuffer(const char *data, size_t length);

    bool isSendBufferFull() const;

    void onCgiComplete();

    void onCgiHeadersParsed(http::Headers const &);

    void handleError(http::HttpStatus);
 
    // =========================================================================
    // Internal Helper Structures
    // =========================================================================
private:
    // Helper to manage the output buffer logic
    struct SendBuffer {
        std::vector<char> buffer;
        size_t sent;
 
        explicit SendBuffer(size_t initialCapacity);
        void reset();
        bool isFullySent() const;
 
        enum SendStatus { SEND_DONE, SEND_AGAIN, SEND_ERROR };
        SendStatus send(int clientFd);
    };
 
    // Helper to group CGI/Event Source state
    struct CgiState {
        IEventHandler *handler; // The worker handling the CGI script
        bool isDone;
 
        CgiState();
        void clear();
        void remove();
    };
 
private:
    // Core Socket State
    int clientFd_;
    int port_;
    std::string clientAddr_;
    bool headersSent_;
 
    // HTTP State
    http::Router const &router_;
    http::Request request_;
    http::Response response_;
    http::RequestParser reqParser_;
 
    // Response State
    bool isKeepAlive_;
    bool isDraining_;
    std::time_t drainStartTime_;
    SendBuffer rspBuffer_;
    CgiState cgiState_;
 
    static const size_t IO_BUFFER_SIZE = 8192;
 
private:
    void handleRead();
    void handleDraining();
    void handleStaticResponseWrite();
    void handleCgiResponseWrite();
 
    // Logic Steps
    void generateResponse();
    void handleRequestParsingState(http::RequestParser::State state);
    bool setupCgiHandler(http::IResponseBody *body);
    void setupStaticResponse();
 
    // Connection Lifecycle
    void resetForNewRequest();
    void closeConnection();
    void initiateDraining();
    void finalizeConnection();
 
    ClientHandler(const ClientHandler &);
    ClientHandler &operator=(const ClientHandler &);
 
    std::string getLogSignature() const;
};
 
} // namespace network