#include "../lib/http.hpp"

http_request::http_request(const string _method, const string _url, const string _body) {
    method = _method;
    url = _url;
    body = _body;
    mold();
}

http_request::http_request(const api *_api) {
    method = _api->method;
    url = _api->url;
    body = _api->body;
    mold();
}

http_request::http_request(const string _raw) {
    raw = _raw;
    parse();
}

void http_request::parse() {
    method = raw.substr(0, raw.find(" "));
    url = raw.substr(raw.find("/"), raw.find("HTTP/")-raw.find("/")-1);

    string _headers = raw.substr(raw.find("\r\n")+2, raw.find("\r\n\r\n")-raw.find("\r\n"));
    while (!_headers.empty()) {
        string key, value;
        key = _headers.substr(0, _headers.find(": "));
        _headers.erase(0, key.length()+2);
        value = _headers.substr(0, _headers.find("\r\n"));
        _headers.erase(0, value.length()+2);
        headers.insert(make_pair(key, value));
    }

    if ((size_t)raw.find("\r\n\r\n") == (size_t)raw.find("\r\n"))
        body = raw.substr(raw.find("\r\n\r\n")+4, raw.length()-raw.find("\r\n\r\n"));

    else if ((size_t)raw.find("\r\n\r\n") < raw.length())
        body = raw.substr(raw.find("\r\n\r\n")+4, raw.find("\r\n")-raw.find("\r\n\r\n"));

}

void http_request::putheader(const string _key, const string _value) {
    headers[_key] = _value;
    mold();
}

void http_request::putheaders(const map<string, string> _headers) {
    headers = _headers;
    mold();
}

/**
 * http zahtjev formiranje raw 
*/

void http_request::mold() {

    raw = method.empty() ? "GET" : method;
    raw += " " + url + " HTTP/1.1\r\n";

    if (!headers.empty()) {
        for (auto i : headers) {
            raw += i.first + ": " + i.second + "\r\n"; 
        }
    }

    raw += "\r\n" + body;
}

void http_response::send(const string _body) {
    body = _body;
    mold();        
}


void http_response::get(const string _raw) {
    raw = _raw;
    parse();
}

/**
 * 
*/


void http_response::mold() {
    raw = "HTTP/1.1 200 OK\r\n"; // implementirati status
    if (!headers.empty()) {
        raw += '?';
        for (auto i : headers) {
            raw += i.first + ": " + i.second + "\r\n"; 
        }
    }

    raw += "\r\n" + body;
}


void http_response::parse() {

    string protocol = raw.substr(0, raw.find(" "));
    status = raw.substr(raw.find(" ")+1, raw.find(" ",raw.find(" ")+1)-raw.find(" ")-1);

    if ((size_t)raw.find("\r\n") < raw.length()) {
        string _headers = raw.substr(raw.find("\r\n")+2, raw.find("\r\n\r\n")-raw.find("\r\n"));
        while (!_headers.empty()) {
            string key, value;
            key = _headers.substr(0, _headers.find(": "));
            _headers.erase(0, key.length()+2);
            value = _headers.substr(0, _headers.find("\r\n"));
            _headers.erase(0, value.length()+2);
            headers.insert(make_pair(key, value));
        }
    }
    // ne radi za specijalan slučaj kada nema zaglavlja
    if ((size_t)raw.find("\r\n\r\n") == (size_t)raw.find("\r\n"))
        body = raw.substr(raw.find("\r\n\r\n")+4, raw.length()-raw.find("\r\n\r\n"));

    else if ((size_t)raw.find("\r\n\r\n") < raw.length())
        body = raw.substr(raw.find("\r\n\r\n")+4, raw.find("\r\n")-raw.find("\r\n\r\n"));

}