#include "catch2/catch_test_macros.hpp"

#include "matador/sql/column.hpp"
#include "matador/sql/condition.hpp"
#include "matador/sql/session.hpp"

#include "connection.hpp"

#include "models/airplane.hpp"
#include "models/flight.hpp"
#include "models/person.hpp"

using namespace matador::sql;
using namespace matador::test;

class SessionFixture
{
public:
  SessionFixture()
  : pool(matador::test::connection::dns, 4), ses(pool)
  {}
  ~SessionFixture() {
    drop_table_if_exists("flight");
    drop_table_if_exists("airplane");
    drop_table_if_exists("person");
  }

protected:
  matador::sql::connection_pool<matador::sql::connection> pool;
  matador::sql::session ses;

private:
  void drop_table_if_exists(const std::string &table_name) {
    if (ses.table_exists(table_name)) {
      ses.drop().table(table_name).execute();
    }
  }
};

TEST_CASE_METHOD(SessionFixture, " Create table with foreign key relation", "[session]") {
  ses.create()
  .table<airplane>("airplane")
  .execute();

  REQUIRE(ses.table_exists("airplane"));

  ses.create()
  .table<flight>("flight")
  .execute();

  REQUIRE(ses.table_exists("flight"));

  ses.drop().table("flight").execute();
  ses.drop().table("airplane").execute();

  REQUIRE(!ses.table_exists("flight"));
  REQUIRE(!ses.table_exists("airplane"));
}

TEST_CASE_METHOD(SessionFixture, " Execute select statement with where clause", "[session]") {
  ses.create()
    .table<person>("person")
    .execute();

  person george{7, "george", 45};
  george.image.push_back(37);

  auto res = ses.insert()
      .into("person", george)
      .execute();
      REQUIRE(res == 1);

  // fetch person as record
  auto result_record = ses.select<person>()
                        .from("person")
                        .where("id"_col == 7)
                        .fetch_all();

  for (const auto& i : result_record) {
    REQUIRE(i.size() == 4);
    REQUIRE(i.at(0).name() == "id");
    REQUIRE(i.at(0).type() == data_type_t::type_unsigned_long);
    REQUIRE(i.at(0).as<long long>() == george.id);
    REQUIRE(i.at(1).name() == "name");
    REQUIRE(i.at(1).type() == data_type_t::type_varchar);
    REQUIRE(i.at(1).as<std::string>() == george.name);
    REQUIRE(i.at(2).name() == "age");
    REQUIRE(i.at(2).type() == matador::sql::data_type_t::type_unsigned_int);
    REQUIRE(i.at(2).as<long long>() == george.age);
  }

  // fetch person as person
  auto result_person = ses.select<person>()
    .from("person")
    .where("id"_col == 7)
    .fetch_all<person>();

  for (const auto& i : result_person) {
    REQUIRE(i.id == 7);
    REQUIRE(i.name == "george");
    REQUIRE(i.age == 45);
  }

  ses.drop().table("person").execute();
}

TEST_CASE_METHOD(SessionFixture, " Execute insert statement", "[session]") {
  ses.create()
    .table("person", {
      make_pk_column<unsigned long>("id"),
      make_column<std::string>("name", 255),
      make_column<std::string>("color", 63)
    })
    .execute();

  auto res = ses.insert()
    .into("person", {"id", "name", "color"})
    .values({7, "george", "green"})
    .execute();

  REQUIRE(res == 1);

  // fetch person as record
  auto result_record = ses.select({"id", "name", "color"})
    .from("person")
    .where("id"_col == 7)
    .fetch_all();

  for (const auto& i : result_record) {
    REQUIRE(i.size() == 3);
    REQUIRE(i.at(0).name() == "id");
    REQUIRE(i.at(0).type() == data_type_t::type_long_long);
    REQUIRE(i.at(0).as<unsigned long>() == 7);
    REQUIRE(i.at(1).name() == "name");
    REQUIRE(i.at(1).type() == data_type_t::type_varchar);
    REQUIRE(i.at(1).as<std::string>() == "george");
    REQUIRE(i.at(2).name() == "color");
    REQUIRE(i.at(2).type() == matador::sql::data_type_t::type_varchar);
    REQUIRE(i.at(2).as<std::string>() == "green");
  }

  ses.drop().table("person").execute();
}

TEST_CASE_METHOD(SessionFixture, " Select statement with foreign key", "[session]") {
  ses.create()
  .table<airplane>("airplane")
  .execute();

  ses.create()
  .table<flight>("flight")
  .execute();

  std::vector<entity<airplane>> planes {
    make_entity<airplane>(1, "Airbus", "A380"),
    make_entity<airplane>(2, "Boeing", "707"),
    make_entity<airplane>(3, "Boeing", "747")
  };

  for (const auto &plane : planes) {
    auto res = ses.insert().into<airplane>("airplane").values(*plane).execute();
    REQUIRE(res == 1);
  }

  auto count = ses.select({count_all()}).from("airplane").fetch_value<int>();
  REQUIRE(count == 3);

  flight f4711{4, planes.at(1), "hans"};

  auto res = ses.insert().into<flight>("flight").values(f4711).execute();
  REQUIRE(res == 1);

  auto f = *ses.select<flight>().from("flight").fetch_all<flight>().begin();
  REQUIRE(f.id == 4);
  REQUIRE(f.pilot_name == "hans");
  REQUIRE(f.airplane.get() != nullptr);
  REQUIRE(f.airplane->id == 2);

  ses.drop().table("flight").execute();
  ses.drop().table("airplane").execute();
}