ChunkedResponse.cpp

#include "ChunkedResponse.h"

#include <winsock2.h>

#include <sstream>

using std::string;
using std::stringstream;
using std::hex;

ChunkedResponse::ChunkedResponse(
    int socket)
{
    clientSocket = socket;
}

void ChunkedResponse::begin(
    const string& contentType)
{
    string header;

    header += "HTTP/1.1 200 OK\r\n";
    header += "Transfer-Encoding: chunked\r\n";
    header += "Content-Type: " + contentType + "\r\n";
    header += "Connection: keep-alive\r\n";
    header += "\r\n";

    send(
        clientSocket,
        header.c_str(),
        header.size(),
        0
    );
}

void ChunkedResponse::sendChunk(
    const string& data)
{
    stringstream ss;

    ss << hex << data.size();

    string chunk;

    chunk += ss.str();
    chunk += "\r\n";
    chunk += data;
    chunk += "\r\n";

    send(
        clientSocket,
        chunk.c_str(),
        chunk.size(),
        0
    );
}

void ChunkedResponse::finish()
{
    string endChunk =
        "0\r\n\r\n";

    send(
        clientSocket,
        endChunk.c_str(),
        endChunk.size(),
        0
    );
}