#include <catch2/catch_test_macros.hpp>

#include "matador/sql/column.hpp"
#include "matador/sql/condition.hpp"
#include "matador/sql/connection.hpp"
#include "matador/sql/query_builder.hpp"

#include "connection.hpp"

#include <list>

class QueryRecordFixture
{
public:
  QueryRecordFixture()
  : db(matador::test::connection::dns)
  , schema(db.dialect().default_schema_name())
  {
    db.open();
  }
  ~QueryRecordFixture() {
    drop_table_if_exists("flight");
    drop_table_if_exists("airplane");
    drop_table_if_exists("person");
    drop_table_if_exists("quotes");
  }

protected:
  matador::sql::connection db;
  matador::sql::schema schema;

private:
  void drop_table_if_exists(const std::string &table_name) {
    if (db.exists(table_name)) {
      db.query(schema).drop().table(table_name).execute();
    }
  }
};

using namespace matador::sql;

TEST_CASE_METHOD(QueryRecordFixture, "Create and drop table statement", "[session][record]")
{
  REQUIRE(!db.exists("person"));
  db.query(schema).create()
  .table("person", {
    make_pk_column<unsigned long>("id"),
    make_column<std::string>("name", 255),
    make_column<unsigned short>("age")
  })
  .execute();

  REQUIRE(db.exists("person"));

  db.query(schema).drop()
  .table("person")
  .execute();

  REQUIRE(!db.exists("person"));
}

TEST_CASE_METHOD(QueryRecordFixture, "Create and drop table statement with foreign key", "[session][record]")
{
  db.query(schema).create()
  .table("airplane", {
    make_pk_column<unsigned long>("id"),
    make_column<std::string>("brand", 255),
    make_column<std::string>("model", 255),
  })
  .execute();

  REQUIRE(db.exists("airplane"));

  db.query(schema).create()
  .table("flight", {
    make_pk_column<unsigned long>("id"),
    make_fk_column<unsigned long>("airplane_id", "airplane", "id"),
    make_column<std::string>("pilot_name", 255),
  })
  .execute();

  REQUIRE(db.exists("flight"));

  db.query(schema).drop()
  .table("flight")
  .execute();

  REQUIRE(!db.exists("flight"));

  db.query(schema).drop()
  .table("airplane")
  .execute();

  REQUIRE(!db.exists("airplane"));
}

TEST_CASE_METHOD(QueryRecordFixture, "Execute insert record statement", "[session][record]")
{
  db.query(schema).create()
  .table("person", {
    make_pk_column<unsigned long>("id"),
    make_column<std::string>("name", 255),
    make_column<unsigned short>("age")
  })
  .execute();

  auto res = db.query(schema).insert()
  .into("person", {"id", "name", "age"})
  .values({7, "george", 45})
  .execute();

  REQUIRE(res == 1);

  auto result = db.query(schema).select({"id", "name", "age"})
  .from("person")
  .fetch_all();

  for (const auto &i: result) {
    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).template as<long long>() == 7);
    REQUIRE(i.at(1).name() == "name");
    //    REQUIRE(i.at(1).type() == data_type_t::type_varchar);
    REQUIRE(i.at(1).template as<std::string>() == "george");
    REQUIRE(i.at(2).name() == "age");
    //    REQUIRE(i.at(2).type() == matador::sql::data_type_t::type_int);
    REQUIRE(i.at(2).template as<int>() == 45);
  }

  db.query(schema).drop()
  .table("person")
  .execute();
}

TEST_CASE_METHOD(QueryRecordFixture, "Execute insert record statement with foreign key", "[session][record]")
{
  db.query(schema).create()
  .table("airplane", {
    make_pk_column<unsigned long>("id"),
    make_column<std::string>("brand", 255),
    make_column<std::string>("model", 255),
  })
  .execute();

  db.query(schema).create()
  .table("flight", {
    make_pk_column<unsigned long>("id"),
    make_fk_column<unsigned long>("airplane_id", "airplane", "id"),
    make_column<std::string>("pilot_name", 255),
  })
  .execute();

  auto res = db.query(schema).insert().into("airplane", {"id", "brand", "model"}).values({1, "Airbus", "A380"}).execute();
  REQUIRE(res == 1);

  res = db.query(schema).insert().into("airplane", {"id", "brand", "model"}).values({2, "Boeing", "707"}).execute();
  REQUIRE(res == 1);

  res = db.query(schema).insert().into("airplane", {"id", "brand", "model"}).values({3, "Boeing", "747"}).execute();
  REQUIRE(res == 1);

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

  res = db.query(schema).insert().into("flight", {"id", "airplane_id", "pilot_name"}).values({4, 1, "George"}).execute();
  REQUIRE(res == 1);

  db.query(schema).drop().table("flight").execute();
  db.query(schema).drop().table("airplane").execute();

  REQUIRE(!db.exists("flight"));
  REQUIRE(!db.exists("airplane"));
}

TEST_CASE_METHOD(QueryRecordFixture, "Execute update record statement", "[session][record]")
{
  db.query(schema).create()
  .table("person", {
    make_pk_column<unsigned long>("id"),
    make_column<std::string>("name", 255),
    make_column<unsigned short>("age")
  })
  .execute();

  auto res = db.query(schema).insert()
  .into("person", {"id", "name", "age"})
  .values({7, "george", 45})
  .execute();

  REQUIRE(res == 1);

  res = db.query(schema).update("person")
  .set({{"id",   7},
        {"name", "jane"},
        {"age",  35}})
  .where("id"_col == 7)
  .execute();

  REQUIRE(res == 1);

  auto result = db.query(schema).select({"id", "name", "age"})
  .from("person")
  .fetch_all();

  for (const auto &i: result) {
    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<long 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>() == "jane");
    REQUIRE(i.at(2).name() == "age");
    REQUIRE(i.at(2).type() == matador::sql::data_type_t::type_int);
    REQUIRE(i.at(2).as<int>() == 35);
  }

  db.query(schema).drop().table("person").execute();
}

TEST_CASE_METHOD(QueryRecordFixture, "Execute select statement", "[session][record]")
{
  db.query(schema).create()
  .table("person", {
    make_pk_column<unsigned long>("id"),
    make_column<std::string>("name", 255),
    make_column<unsigned short>("age")
  })
  .execute();

  auto res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({1, "george", 45}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({2, "jane", 32}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({3, "michael", 67}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({4, "bob", 13}).execute();
  REQUIRE(res == 1);

  auto result = db.query(schema).select({"id", "name", "age"})
  .from("person")
  .fetch_all();

  std::list<std::string> expected_names{"george", "jane", "michael", "bob"};
  for (const auto &p: result) {
    REQUIRE(p.at(1).str() == expected_names.front());
    expected_names.pop_front();
  }
  REQUIRE(expected_names.empty());

  auto rec = db.query(schema).select({"id", "name", "age"})
  .from("person")
  .fetch_one();
  REQUIRE(rec.at(1).str() == "george");

  auto name = db.query(schema).select({"name"})
  .from("person")
  .fetch_value<std::string>();
  REQUIRE(name == "george");

  db.query(schema).drop().table("person").execute();
}

TEST_CASE_METHOD(QueryRecordFixture, "Execute select statement with order by", "[session][record]")
{
  db.query(schema).create()
  .table("person", {
    make_pk_column<unsigned long>("id"),
    make_column<std::string>("name", 255),
    make_column<unsigned short>("age")
  })
  .execute();

  auto res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({1, "george", 45}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({2, "jane", 32}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({3, "michael", 67}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({4, "bob", 13}).execute();
  REQUIRE(res == 1);

  auto result = db.query(schema).select({"id", "name", "age"})
  .from("person")
  .order_by("name").asc()
  .fetch_all();

  std::list<std::string> expected_names{"bob", "george", "jane", "michael"};
  for (const auto &p: result) {
    REQUIRE(p.at(1).str() == expected_names.front());
    expected_names.pop_front();
  }
  REQUIRE(expected_names.empty());

  db.query(schema).drop().table("person").execute();
}

TEST_CASE_METHOD(QueryRecordFixture, "Execute select statement with group by and order by", "[session][record]")
{
  db.query(schema).create()
  .table("person", {
    make_pk_column<unsigned long>("id"),
    make_column<std::string>("name", 255),
    make_column<unsigned short>("age")
  })
  .execute();

  auto res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({1, "george", 45}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({2, "jane", 45}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({3, "michael", 13}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({4, "bob", 13}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({5, "charlie", 67}).execute();
  REQUIRE(res == 1);

  auto result = db.query(schema).select({count("age").as("age_count"), "age"})
  .from("person")
  .group_by("age")
  .order_by("age_count").desc()
  .fetch_all();

  std::list<std::pair<int, int>> expected_values{{2, 45},
                                                 {2, 13},
                                                 {1, 67}};
  for (const auto &r: result) {
    REQUIRE(r.at(0).as<int>() == expected_values.front().first);
    REQUIRE(r.at(1).as<int>() == expected_values.front().second);
    expected_values.pop_front();
  }

  db.query(schema).drop().table("person").execute();
}

TEST_CASE_METHOD(QueryRecordFixture, "Execute delete statement", "[session][record]")
{
  db.query(schema).create()
  .table("person", {
    make_pk_column<unsigned long>("id"),
    make_column<std::string>("name", 255),
    make_column<unsigned short>("age")
  }).execute();

  auto res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({1, "george", 45}).execute();
  REQUIRE(res == 1);
  res = db.query(schema).insert().into("person", {"id", "name", "age"}).values({2, "jane", 45}).execute();
  REQUIRE(res == 1);

  auto count = db.query(schema).select({count_all()}).from("person").fetch_value<int>();
  REQUIRE(count == 2);

  res = db.query(schema).remove()
  .from("person")
  .where("id"_col == 1)
  .execute();

  REQUIRE(res == 1);

  count = db.query(schema).select({count_all()}).from("person").fetch_value<int>();
  REQUIRE(count == 1);

  db.query(schema).drop().table("person").execute();
}

TEST_CASE_METHOD(QueryRecordFixture, "Test quoted identifier", "[session][record]") {
  db.query(schema).create()
  .table("quotes", {
    make_column<std::string>("from", 255),
    make_column<std::string>("to", 255)
  }).execute();

  // check table description
  std::vector<std::string> columns = { "from", "to"};
  std::vector<data_type_t> types = {data_type_t::type_varchar, data_type_t::type_varchar};
  auto fields = db.describe("quotes");

  for (const auto &field : fields) {
    REQUIRE(field.name() == columns[field.index()]);
    REQUIRE(field.type() == types[field.index()]);
  }

  db.query(schema).insert().into("quotes", {"from", "to"}).values({"Berlin", "London"}).execute();

  auto res = db.query(schema).select({"from", "to"}).from("quotes").fetch_one();

  REQUIRE("Berlin" == res.at("from").str());
  REQUIRE("London" == res.at("to").str());

  db.query(schema).update("quotes").set({{"from", "Hamburg"}, {"to", "New York"}}).where("from"_col == "Berlin").execute();

  res = db.query(schema).select({"from", "to"}).from("quotes").fetch_one();

  REQUIRE("Hamburg" == res.at("from").str());
  REQUIRE("New York" == res.at("to").str());

  db.query(schema).drop().table("quotes").execute();
}