/*****************************************************************************
* dtranslatebot Discord Translate Bot
* Copyright (C) 2026 Syping
*
* Redistribution and use in source and binary forms, with or without modification,
* are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* This software is provided as-is, no warranties are given to you, we are not
* responsible for anything with use of the software, you are self responsible.
*****************************************************************************/

#include <cstring>
#include <curl/curlver.h>
#include "curl_exception.h"
#include "http_request.h"
using namespace bot::exception;
using namespace bot::http;

http_request::http_request() {
    instance = curl_easy_init();
    if (!instance)
        throw std::bad_alloc();
}

const http_response http_request::get(const std::string &url, const http_headers &headers) {
    http_response response;
    char error[CURL_ERROR_SIZE]{};
    curl_easy_setopt(instance, CURLOPT_URL, url.c_str());
    curl_easy_setopt(instance, CURLOPT_HTTPHEADER, headers.data());
    curl_easy_setopt(instance, CURLOPT_WRITEDATA, &response.content);
    curl_easy_setopt(instance, CURLOPT_WRITEFUNCTION, &writer);
    curl_easy_setopt(instance, CURLOPT_ERRORBUFFER, error);
    CURLcode result = curl_easy_perform(instance);
    if (result == CURLE_OK)
        process_response(instance, response);
    curl_easy_reset(instance);
    return result == CURLE_OK ?
               response : throw curl_exception(error, result);
}

const http_response http_request::post(const std::string &url, const std::string &content, const http_headers &headers) {
    http_response response;
    char error[CURL_ERROR_SIZE]{};
    curl_easy_setopt(instance, CURLOPT_URL, url.c_str());
    curl_easy_setopt(instance, CURLOPT_HTTPHEADER, headers.data());
    curl_easy_setopt(instance, CURLOPT_POSTFIELDS, content.data());
    curl_easy_setopt(instance, CURLOPT_POSTFIELDSIZE, content.size());
    curl_easy_setopt(instance, CURLOPT_WRITEDATA, &response.content);
    curl_easy_setopt(instance, CURLOPT_WRITEFUNCTION, &writer);
    curl_easy_setopt(instance, CURLOPT_ERRORBUFFER, error);
    CURLcode result = curl_easy_perform(instance);
    if (result == CURLE_OK)
        process_response(instance, response);
    curl_easy_reset(instance);
    return result == CURLE_OK ?
               response : throw curl_exception(error, result);
}

std::string http_request::legacy_url(const std::string &hostname, uint16_t port, const std::string &url, bool tls) {
    return (tls ? "https://" : "http://") + hostname + ":" + std::to_string(port) + (url.empty() ? "/" : (url.front() != '/' ? "/" + url : url));
}

http_request::~http_request() {
    curl_easy_cleanup(instance);
}

void http_request::process_response(CURL* instance, http_response &response) {
    curl_easy_getinfo(instance, CURLINFO_RESPONSE_CODE, &response.status);
#if LIBCURL_VERSION_NUM >= CURL_VERSION_BITS(7, 83, 0)
    curl_header* content_type;
    CURLHcode result = curl_easy_header(instance, "Content-Type", 0, CURLH_HEADER, -1, &content_type);
    if (result == CURLHE_OK)
        response.content_type = content_type->value;
#else
    char* content_type = nullptr;
    CURLcode result = curl_easy_getinfo(instance, CURLINFO_CONTENT_TYPE, &content_type);
    if (result == CURLE_OK && content_type)
        response.content_type = content_type;
#endif
}

size_t http_request::writer(char* source, size_t size, size_t nmemb, std::string* target) {
    if (target == nullptr)
        return 0;
    size_t write_size = size * nmemb;
    target->append(source, write_size);
    return write_size;
}