#include <catch.hpp>
#include "mock_curl.hpp"
#include "fixtures.hpp"
#include <leatherman/curl/client.hpp>
#include <leatherman/curl/request.hpp>
#include <leatherman/curl/response.hpp>
#include <boost/filesystem.hpp>
#include <boost/nowide/iostream.hpp>
#include <boost/nowide/fstream.hpp>
#include <boost/regex.hpp>
#include <sstream>
#include <cstdio>
#include <map>

using namespace std;
namespace fs = boost::filesystem;
namespace nw = boost::nowide;
using namespace leatherman::curl;

#define REQUIRE_THROWS_AS_WITH(expression, exception_type, msg_matcher) {\
    try {\
        expression;\
        REQUIRE(false);\
    } catch (exception_type& e) {\
        REQUIRE_THAT(e.what(), msg_matcher);\
    } catch (exception& e) {\
        REQUIRE(false);\
    }\
}

// TODO: Move non-test code to "fixtures.hpp" and "fixtures.cc".
fs::path find_matching_file(const boost::regex& re) {
    auto file = find_if(
        fs::recursive_directory_iterator(fs::current_path()),
        fs::recursive_directory_iterator(),
        [re](const fs::path& f) { return boost::regex_match(f.filename().string(), re); });
  
    // throw exception, as this means that the matching file does not exist.
    if (file == fs::recursive_directory_iterator()) {
      throw std::runtime_error("matching file not found");
    }

    return *file;
}

void remove_temp_file() {
    auto temp_path = find_matching_file(boost::regex(TEMP_FILE_REGEX));
    fs::remove(temp_path);
}

struct mock_client : client {
    curl_handle const& get_handle() { return client::get_handle(); }
};

TEST_CASE("curl::client HTTP methods") {
    mock_client test_client;
    request test_request {"http://valid.com/"};

    SECTION("GET succeeds on a given URL") {
        auto resp = test_client.get(test_request);
        REQUIRE(resp.status_code() == 200);
    }

    SECTION("POST succeeds on a given URL") {
        auto resp = test_client.post(test_request);
        REQUIRE(resp.status_code() == 200);
    }

    SECTION("PUT succeeds on a given URL") {
        auto resp = test_client.put(test_request);
        REQUIRE(resp.status_code() == 200);
    }

    SECTION("Request returns status code 404 on invalid URL") {
        request invalid_test_request {"http://invalid.com/"};
        auto resp = test_client.get(invalid_test_request);
        REQUIRE(resp.status_code() == 404);
    }
}

TEST_CASE("curl::client HTTP request setup") {
    mock_client test_client;
    request test_request {"http://valid.com"};

    SECTION("HTTP method is set to GET given a GET request") {
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->method == curl_impl::http_method::get);
    }

    SECTION("HTTP method is set to POST given a POST request") {
        auto resp = test_client.post(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->method == curl_impl::http_method::post);
    }

    SECTION("HTTP method is set to PUT given a PUT request") {
        auto resp = test_client.put(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->method == curl_impl::http_method::put);
    }

    SECTION("cURL should receive the URL specified in the request") {
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->request_url == "http://valid.com");
    }
}

TEST_CASE("curl::client header and body writing and reading") {
    mock_client test_client;

    /*
     * Header writing and reading tests
     */
    SECTION("Custom request headers should be honored in the request to the server") {
        request test_request {"http://valid.com"};
        test_request.add_header("header_name", "header_value");
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->header);
        REQUIRE(test_impl->header->data == string("header_name: header_value"));
    }

    SECTION("The header response delimiter should be ignored") {
        request test_request {"http://response-delimiter.com/"};
        auto resp = test_client.get(test_request);
        int headers = 0;
        resp.each_header([&](string const& name, string const& value) {
            ++headers;
            return true;
        });
        REQUIRE(headers == 0);
    }

    SECTION("Non-standard response header should be parsed for name and value") {
        request test_request {"http://nonstd-header.com/"};
        auto resp = test_client.get(test_request);
        REQUIRE(resp.header("nonstd_header_name"));
        REQUIRE(*(resp.header("nonstd_header_name")) == "nonstd_header_value");
    }

    SECTION("Invalid headers should not be parsed or returned in the response") {
        request test_request {"http://invalid-header.com/"};
        auto resp = test_client.get(test_request);
        int headers = 0;
        resp.each_header([&](string const& name, string const& value) {
            ++headers;
            return true;
        });
        REQUIRE(headers == 0);
    }

    /*
     * Body writing and reading tests
     */
    SECTION("Request body should be settable and readable") {
        request test_request {"http://valid.com"};
        test_request.body("Hello, I am a request body!", "message");
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->read_buffer == "Hello, I am a request body!");
    }

    SECTION("Response body should be what is in the data part of the cURL response") {
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        test_impl->resp_body = "Hello, I am a response body!";
        request test_request {"http://valid.com"};
        auto resp = test_client.get(test_request);
        REQUIRE(resp.body() == "Hello, I am a response body!");
    }
}

TEST_CASE("curl::client cookies") {
    mock_client test_client;
    request test_request {"http://valid.com"};

    SECTION("There should be no cookies in the request by default") {
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->cookie == "");
    }

    SECTION("Cookies should be present in the request when added") {
        test_request.add_cookie("cookie_name", "cookie_val");
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->cookie == "cookie_name=cookie_val");
    }

    SECTION("Cookies should be removable from the request") {
        test_request.add_cookie("cookie_0", "cookie_val_0");
        test_request.add_cookie("cookie_1", "cookie_val_1");
        test_request.remove_cookie("cookie_1");
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->cookie == "cookie_0=cookie_val_0");
    }

    SECTION("cURL should receieve cookies specified in the request") {
        test_request.add_cookie("cookie_0", "cookie_val_0");
        test_request.add_cookie("cookie_1", "cookie_val_1");
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->cookie == "cookie_0=cookie_val_0; cookie_1=cookie_val_1");
    }
}

TEST_CASE("curl::client CA bundle and SSL setup") {
    mock_client test_client;
    request test_request {"http://valid.com"};

    SECTION("Path to CA certificate should be unspecified by default") {
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->cacert == "");
    }

    SECTION("cURL should receive the path to the CA certificate specified in the request") {
        test_client.set_ca_cert("cacert");
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->cacert == "cacert");
    }

    SECTION("Client cert name should be unspecified by default") {
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->client_cert == "");
    }

    SECTION("cURL should receive the client cert name specified in the request") {
        test_client.set_client_cert("cert", "key");
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->client_cert == "cert");
    }

    SECTION("Private keyfile name should be unspecified by default") {
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->client_key == "");
    }

    SECTION("cURL should receive the private keyfile name specified in the request") {
        test_client.set_client_cert("cert", "key");
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->client_key == "key");
    }

    SECTION("cURL should make an HTTP request with the specified HTTP protocol") {
        test_client.set_supported_protocols(CURLPROTO_HTTP);
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->protocols == CURLPROTO_HTTP);
    }

    SECTION("cURL defaults to all protocols if no protocols are specified") {
        auto resp = test_client.get(test_request);
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        REQUIRE(test_impl->protocols == CURLPROTO_ALL);
    }
}

TEST_CASE("curl::client errors") {
    mock_client test_client;
    request test_request {"http://valid.com/"};
    CURL* const& handle = test_client.get_handle();
    auto test_impl = reinterpret_cast<curl_impl* const>(handle);

    /*
     * Note: we do not currently test the case where cURL errors
     * on curl_global_init, as the global init is done as part of
     * static initialization in the cURL helper, and there is
     * currently no way to force it to be reinitialized.
     */

    SECTION("client fails to initialize a libcurl easy session") {
        curl_fail_init mock_error {easy_init_error};
        REQUIRE_THROWS_AS(mock_client(), http_exception);
    }

    SECTION("client fails to perform a cURL request") {
        test_impl->test_failure_mode = curl_impl::error_mode::easy_perform_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_request_exception);
    }

    SECTION("client fails to set HTTP method to POST") {
        test_impl->test_failure_mode = curl_impl::error_mode::http_post_error;
        REQUIRE_THROWS_AS(test_client.post(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set HTTP method to PUT") {
        test_impl->test_failure_mode = curl_impl::error_mode::http_put_error;
        REQUIRE_THROWS_AS(test_client.put(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set the request URL") {
        test_impl->test_failure_mode = curl_impl::error_mode::set_url_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set the request headers") {
        test_impl->test_failure_mode = curl_impl::error_mode::set_header_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set cookies in the request") {
        test_impl->test_failure_mode = curl_impl::error_mode::set_cookie_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set the header callback function") {
        test_impl->test_failure_mode = curl_impl::error_mode::header_function_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set the header write location") {
        test_impl->test_failure_mode = curl_impl::error_mode::header_context_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set the body writing callback function") {
        test_impl->test_failure_mode = curl_impl::error_mode::write_body_function_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set the body write location") {
        test_impl->test_failure_mode = curl_impl::error_mode::write_body_context_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }
    SECTION("client fails to set the read_body callback function") {
        test_impl->test_failure_mode = curl_impl::error_mode::read_body_function_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set the read_body data source") {
        test_impl->test_failure_mode = curl_impl::error_mode::read_body_context_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set the connection timeout") {
        test_impl->test_failure_mode = curl_impl::error_mode::connect_timeout_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set the request timeout") {
        test_impl->test_failure_mode = curl_impl::error_mode::request_timeout_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set certificate authority info") {
        test_client.set_ca_cert("certfile");
        test_impl->test_failure_mode = curl_impl::error_mode::ca_bundle_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set SSL cert info") {
        test_client.set_client_cert("cert", "key");
        test_impl->test_failure_mode = curl_impl::error_mode::ssl_cert_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to set SSL key info") {
        test_client.set_client_cert("cert", "key");
        test_impl->test_failure_mode = curl_impl::error_mode::ssl_key_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }

    SECTION("client fails to make http call with https protocol only enabled") {
        test_client.set_supported_protocols(CURLPROTO_HTTPS);
        test_impl->test_failure_mode = curl_impl::error_mode::protocol_error;
        REQUIRE_THROWS_AS(test_client.get(test_request), http_curl_setup_exception);
    }
}

    TEST_CASE("curl::client download_file") {
        mock_client test_client;
        temp_directory temp_dir;
        fs::path temp_dir_path = fs::path(temp_dir.get_dir_name()); 
        CURL* const& handle = test_client.get_handle();
        auto test_impl = reinterpret_cast<curl_impl* const>(handle);
        std::string url = "https://download.com";

        SECTION("when a response is not passed in") {
            SECTION("successfully downloads the file to the specified location") {
                std::string ca_file = "ca";
                std::string cert_file = "client";
                std::string key_file = "key";

                test_client.set_ca_cert(ca_file);
                test_client.set_client_cert(cert_file, key_file);
                test_client.set_supported_protocols(CURLPROTO_HTTPS);

                std::string file_path = (temp_dir_path / "test_file").string();
                std::string token = "token";
                long connect_timeout = 300000;

                request req(url);
                req.add_header("X-Authentication", token);
                req.connection_timeout(connect_timeout);
                test_client.download_file(req, file_path);

                // ensure that the correct curl parameters were used.
                REQUIRE(test_impl->cacert == ca_file);
                REQUIRE(test_impl->client_cert == cert_file);
                REQUIRE(test_impl->client_key == key_file);
                REQUIRE(test_impl->protocols == CURLPROTO_HTTPS);
                REQUIRE(test_impl->connect_timeout == connect_timeout);
                REQUIRE(std::string(test_impl->header->data) == ("X-Authentication: " + token));
                if (test_impl->header->next) {
                  FAIL("X-Authentication should be the only header");
                }

                // now check that the file was actually downloaded and written with the right
                // contents.
                REQUIRE(fs::exists(file_path));
                nw::ifstream in(file_path);
                stringstream stream;
                stream << in.rdbuf();
                REQUIRE(stream.str() == "successfully downloaded file");
          }

#ifndef _WIN32
          SECTION("sets permissions if requested") {
              auto file_path = (temp_dir_path / "other_test_file").string();
              request req(url);
              auto perms = boost::filesystem::owner_read | boost::filesystem::owner_write;
              test_client.download_file(req, file_path, perms);

              REQUIRE(fs::exists(file_path));
              REQUIRE(fs::status(file_path).permissions() == perms);
          }
#endif

          SECTION("downloads the response body for a 400+ status") {
              std::string url = "https://download_trigger_404.com";
              auto file_path = (temp_dir_path / "404_test_file").string();
              request req(url);
              test_client.download_file(req, file_path); 

              // now check that the file was actually downloaded and written with the right
              // contents.
              REQUIRE(fs::exists(file_path));
              nw::ifstream in(file_path);
              stringstream stream;
              stream << in.rdbuf();
              REQUIRE(stream.str() == "Not found");
          }
      }

      SECTION("when a response is passed in") {
          SECTION("successfully downloads the file to the specified location, and includes the response") {
              std::string ca_file = "ca";
              std::string cert_file = "client";
              std::string key_file = "key";

              test_client.set_ca_cert(ca_file);
              test_client.set_client_cert(cert_file, key_file);
              test_client.set_supported_protocols(CURLPROTO_HTTPS);

              std::string file_path = (temp_dir_path / "test_file").string();
              std::string token = "token";
              long connect_timeout = 300000;

              request req(url);
              req.add_header("X-Authentication", token);
              req.connection_timeout(connect_timeout);
              response res;
              test_client.download_file(req, file_path, res);

              // ensure that the correct curl parameters were used.
              REQUIRE(test_impl->cacert == ca_file);
              REQUIRE(test_impl->client_cert == cert_file);
              REQUIRE(test_impl->client_key == key_file);
              REQUIRE(test_impl->protocols == CURLPROTO_HTTPS);
              REQUIRE(test_impl->connect_timeout == connect_timeout);
              REQUIRE(std::string(test_impl->header->data) == ("X-Authentication: " + token));
              if (test_impl->header->next) {
                FAIL("X-Authentication should be the only header");
              }

              // now check that the file was actually downloaded and written with the right
              // contents.
              REQUIRE(fs::exists(file_path));
              nw::ifstream in(file_path);
              stringstream stream;
              stream << in.rdbuf();
              REQUIRE(stream.str() == "successfully downloaded file");

              // now check the response
              REQUIRE(res.status_code() == 200);
              REQUIRE(res.body().empty());
          }

#ifndef _WIN32
          SECTION("sets permissions if requested") {
              auto file_path = (temp_dir_path / "other_test_file").string();
              request req(url);
              response res;
              auto perms = boost::filesystem::owner_read | boost::filesystem::owner_write;
              test_client.download_file(req, file_path, res, perms);

              REQUIRE(fs::exists(file_path));
              REQUIRE(fs::status(file_path).permissions() == perms);
          }
#endif

          SECTION("does not download anything for a 400+ status") {
              std::string url = "https://download_trigger_404.com";
              auto file_path = (temp_dir_path / "404_test_file").string();
              request req(url);
              response res;
              test_client.download_file(req, file_path, res); 

              REQUIRE(res.status_code() == 404);
              REQUIRE(res.body() == "Not found");
              // check that the file was not downloaded 
              REQUIRE(!fs::exists(file_path));
          }
      }
    }

TEST_CASE("curl::client download_file errors") {
    mock_client test_client;
    temp_directory temp_dir;
    fs::path temp_dir_path = fs::path(temp_dir.get_dir_name()); 
    CURL* const& handle = test_client.get_handle();
    auto test_impl = reinterpret_cast<curl_impl* const>(handle);

    SECTION("when fopen fails, an http_file_operation_exception is thrown") {
        fs::path parent_dir = temp_dir_path / "parent";
        std::string file_path = (parent_dir / "child").string();
        request req("");
        REQUIRE_THROWS_AS_WITH(
            test_client.download_file(req, file_path),
            http_file_operation_exception,
            Catch::Equals("File operation error: failed to open temporary file for writing"));
    }

    SECTION("when curl_easy_setopt fails, an http_curl_setup_exception is thrown and the temporary file is removed") {
        request req("");
        std::string file_path = (temp_dir_path / "file").string();
        test_impl->test_failure_mode = curl_impl::error_mode::set_url_error;
        REQUIRE_THROWS_AS(test_client.download_file(req, file_path), http_curl_setup_exception);
        // Ensure that the temp file was removed
        REQUIRE(fs::is_empty(temp_dir_path));
    }

    SECTION("when curl_easy_perform fails due to a CURLE_WRITE_ERROR, but the temporary file is removed, an http_file_operation_exception is thrown") {
        std::string file_path = (temp_dir_path / "file").string();
        request req("");
        test_impl->test_failure_mode = curl_impl::error_mode::easy_perform_write_error; 
        REQUIRE_THROWS_AS_WITH(
            test_client.download_file(req, file_path),
            http_file_operation_exception,
            Catch::StartsWith("File operation error: failed to write to the temporary file during download"));
    }

    SECTION("when curl_easy_perform fails for reasons other than a CURLE_WRITE_ERROR, but the temporary file is removed, only the errbuf message is contained in the thrown http_file_download_exception") {
        std::string file_path = (temp_dir_path / "file").string();
        request req("");
        test_impl->test_failure_mode = curl_impl::error_mode::easy_perform_error; 
        REQUIRE_THROWS_AS_WITH(
            test_client.download_file(req, file_path),
            http_file_download_exception,
            Catch::Equals("File download server side error: easy perform failed"));

        // Ensure that the temp file was removed
        REQUIRE(fs::is_empty(temp_dir_path));
    }

    SECTION("when renaming the temporary file to the user-provided file path fails, an http_file_operation_exception is thrown") {
        std::string file_path = (temp_dir_path / "file").string();
        request req("https://download.com");
        test_impl->trigger_external_failure = remove_temp_file; 
        REQUIRE_THROWS_AS_WITH(
            test_client.download_file(req, file_path),
            http_file_operation_exception,
            Catch::StartsWith("File operation error: failed to move over the temporary file's downloaded contents"));
    }

    SECTION("when writing the temporary file's contents to the response body fails, an http_file_operation_exception is thrown") {
        std::string file_path = (temp_dir_path / "file").string();
        request req("https://download_trigger_404.com");
        test_impl->trigger_external_failure = remove_temp_file;
        response res;
        REQUIRE_THROWS_AS_WITH(
            test_client.download_file(req, file_path, res),
            http_file_operation_exception,
            Catch::StartsWith("File operation error: failed to write the temporary file's contents to the response body"));
    }
}