HttpResponse.hpp

Go to the documentation of this file.

#pragma once
 
#include <vector>
#include <map>
#include <string>
#include <ostream>
#include "HttpStatus.hpp"
 
namespace http {
 
typedef std::map<std::string, std::string> HeaderMap;

enum BodySourceType {
    BODY_NONE,      
    BODY_IN_MEMORY, 
    BODY_FROM_FILE, 
    BODY_FROM_CGI   
};

class HttpResponse {
public:
    HttpResponse();
    HttpResponse(Status code, std::string const &httpVersion);
    HttpResponse(HttpResponse const &);
    HttpResponse const &operator=(HttpResponse const &);
    ~HttpResponse();
 
    void setStatus(Status s);
    void setNoBody();
    void setBodyInMemory(std::vector<char> const &, std::string const &mimeType);
    void setBodyFromFile(int fd, size_t size);
    void setBodyFromFile(int fd, size_t size, std::string const &mimeType);
    void setBodyFromCgi(int pipe_fd);
 
    Status getStatus() const;
    char const *getResponsePhrase() const;
    HeaderMap &getHeaders();
    HeaderMap const &getHeaders() const;
    std::string const &getVersion() const;
    BodySourceType getBodyType() const;
    std::string buildHeaders() const;
 
    union {
        struct {
            std::vector<char> *data; 
            size_t sent;             
        } inMemoryBody;

        struct {
            int fd;           
            size_t totalSize; 
            size_t sent;      
        } fileBody;

        struct {
            int pipe_fd; 
        } cgiBody;
    };
 
private:
    std::string httpVersion_;
    Status statusCode_;
    HeaderMap headers_;
    BodySourceType bodyType_;
 
    void cleanupBody();
};
 
std::ostream &operator<<(std::ostream &o, HttpResponse const &r);
 
} // namespace http