#include "expresscpp/expresscpp.hpp"
#include "expresscpp/fetch.hpp"
#include "fmt/format.h"
#include "gtest/gtest.h"
#include "nlohmann/json.hpp"

using namespace expresscpp;

constexpr uint16_t port = 8081;

TEST(ErrorHandler, BasicErrorHandler) {
  ExpressCpp app;

  bool error_middleware_visited = false;
  std::string error_message = "Internal Error";

  auto ErrorMiddleware = [&](auto ec, auto /*req*/, auto res, auto /*next*/) {
    error_middleware_visited = true;
    std::cerr << "Internal Error: " << ec.message() << std::endl;
    res->SetStatus(500);
    res->Send(error_message);
  };

  app.Error(ErrorMiddleware);

  struct User {
    std::string name;
    uint8_t age;
  };

  std::vector<User> users{{"john", 30}, {"steve", 40}};

  app.Get("/users/:userId", [&](auto req, auto res, auto /*next*/) {
    auto params = req->GetParams();

    if ((params.find("userId") != params.end())) {
      auto id = params.at("userId");
      const int userId = std::stoi(id);

      // ops, possible the user does not exist -> exception
      auto user = users.at(userId);

      res->Json(fmt::format("{{\"name\":\"{}\", \"age\":{}}}", user.name, user.age));
    } else {
      res->SetStatus(404);
      res->Send(R"({"error":"user not found"})");
    }
  });

  app.Listen(port, [&](auto ec) {
    EXPECT_FALSE(ec);

    const auto should_work_response = fetch(fmt::format("localhost:{}/users/1", port));
    EXPECT_FALSE(error_middleware_visited);
    const auto expected = nlohmann::json::parse(should_work_response);
    EXPECT_TRUE(expected.count("name") > 0);

    const auto should_not_work_response = fetch(fmt::format("localhost:{}/users/2", port));
    EXPECT_TRUE(error_middleware_visited);

    EXPECT_EQ(should_not_work_response, error_message);
  });
}