#include "libhttpc.h"

#include <errno.h>
#include <stddef.h>
#include <string.h>

#include <sys/socket.h>
#include <sys/types.h>


struct LibHTTPC_Request *LibHTTPC_loadRequest(struct LibHTTPC_Request *request_buf, char *buf)
{
    char *next;
    int selfmalloc = 0;
    char *buf_end = strchr(buf, '\0');

    if (!buf)
        return NULL;

    if (!request_buf)
    {
        if (!LibHTTPC_malloc || !LibHTTPC_realloc)
            return NULL;
        request_buf = LibHTTPC_malloc(sizeof(struct LibHTTPC_Request));
        memset(request_buf, '\0', sizeof(struct LibHTTPC_Request));
        request_buf->selfalloc = 1;
        request_buf->header_count = -1;
    }

    if (request_buf->header_count == -1)
    {
        if (!LibHTTPC_malloc || !LibHTTPC_realloc)
            return NULL;
        selfmalloc = 1;
        request_buf->header_selfalloc = 1;
        request_buf->header_count = 0;
    }

    request_buf->method = request_buf->buf = buf;

    next = strchr(request_buf->method, ' ');
    if (!next)
        return NULL;
    *next = '\0';
    request_buf->uri = next + 1;

    next = strchr(request_buf->uri, ' ');
    if (!next)
        return NULL;
    *next = '\0';
    request_buf->version = next + 1;

    next = strstr(request_buf->version, "\r\n");
    if (!next)
        return NULL;
    *next++ = '\0';
    if (next > buf_end)
        return NULL;

    for (size_t i = 0; (next = strstr(next, "\r\n")); ++next, ++i)
    {
        if (!next)
            return NULL;
        *next = '\0';
        next += 2;
        if (next > buf_end)
            return NULL;

        if (strstr(next, "\r\n") == next)
            break;

        if (selfmalloc)
        {
            if (!LibHTTPC_malloc || !LibHTTPC_realloc)
                return NULL;
            ++request_buf->header_count;
            if (!request_buf->headers)
                request_buf->headers = LibHTTPC_malloc(sizeof(struct LibHTTPC_Header));
            else
                request_buf->headers = LibHTTPC_realloc(request_buf->headers, sizeof(struct LibHTTPC_Header));
        }
        if (i < request_buf->header_count)
        {
            request_buf->headers[i].name = next;
            next = strchr(next, ':');
            if (!next)
                return NULL;
            *next++ = '\0';
            if (next > buf_end)
                return NULL;
            request_buf->headers[i].value = next;
        }
    }
    request_buf->body = next + 2;

    return request_buf;
}

#ifdef LibHTTPC_SOCK
struct LibHTTPC_Request *LibHTTPC_readRequest(
    int sockfd,
    struct LibHTTPC_Request *request_buf,
    char *buf, size_t buf_len
)
{
    if (!buf)
    {
        /* TODO: Implement behavior when buf == NULL */
        return NULL;
    } else
    {
        if (!buf_len)
            return NULL;

        ssize_t received = recv(sockfd, buf, buf_len, 0);
        if (received < 0)
            return NULL;
        buf[buf_len - 1] = '\0';

        return LibHTTPC_loadRequest(request_buf, buf);
    }
}
#endif

int LibHTTPC_Request_(struct LibHTTPC_Request *request)
{
    if (!LibHTTPC_free)
        return -1;

    if (request->header_selfalloc)
        LibHTTPC_free(request->headers);
    if (request->selfalloc)
        LibHTTPC_free(request);
    return errno;
}