#include "treemap.h"
#include <map>

struct account_t {
  double balance;
  std::string note;

  account_t() : account_t { 0.0 } {
  }
  
  account_t(double balance) : balance { balance } {
  }
};

struct lanister_t {
  std::string name;

  lanister_t() {
  }
  
  lanister_t(const char* name) : name { name } {
  }

  bool operator < (const lanister_t& lanister) const {
    return name < lanister.name;
  }
  
  bool operator == (const lanister_t& lanister) const {
    return name == lanister.name;
  }
};

std::ostream& operator << (std::ostream& os, const lanister_t& lanister) {
  os << lanister.name << " Lanister";
  return os;
}

std::ostream& operator << (std::ostream& os, const account_t& account) {
  os << account.balance << "€";
  if(account.note.length() > 0)
    os << " (" << account.note << ")";
  return os;
}

int main(int argc, char* argv[]) {
  treemap_t<lanister_t, account_t> map;
  //std::map<lanister_t, account_t> map;

  map["Tyrion"] = 2;
  map["Jaime"] = 15;
  map["Cersei"] = 63;
  map["Tywin"] = 75;
  map["Kevan"] = 48;
  map["Lancel"] = 70;
  map["Alton"] = 54;
  map["Martyn"] = 32;
  map["Willem"] = 6;

  map["Tyrion"].note = "best client ever";
  
  std::cout << ":: " << map.at("Kevan") << std::endl;

  std::cout << "=====" << std::endl;
  map.display();
  std::cout << "=====" << std::endl;

  for(auto& [key, value]: map)
    std::cout << "(" << key << ", " << value << ")" << std::endl;

  std::cout << "-------------------------" << std::endl;
  // more tests
  treemap_t<int, std::string> map_int_str;
  treemap_t<int, int> map_int_int;
  treemap_t<std::string, std::string> map_str_str;

  //int tab[] = { 1, 2, 3, 4, 5, 6, 7 };
  int tab[] = { 4, 3, 1, 2, 5, 6, 7 };
  //int tab[] = { 7, 6, 5, 4, 3, 2, 1 };

  for(size_t i=0; i<sizeof(tab)/sizeof(int); i++) {
    map_int_str[tab[i]] = "hello-" + std::to_string(tab[i]);
    map_int_int[tab[i]] = tab[i];
    map_str_str["element-" + std::to_string(tab[i])] = "nice-" + std::to_string(tab[i]);
  }
  
  std::cout << ":: " << map_int_str[3] << std::endl;
  std::cout << ":: " << map_int_int[3] << std::endl;
  std::cout << ":: " << map_str_str["element-3"] << std::endl;

  map_int_str[3] = "just a test";
  map_int_int[4] = 42;
  map_str_str["element-5"] = "another test";
  
  std::cout << "=====" << std::endl;
  map_int_str.display();
  std::cout << "=====" << std::endl;
  map_int_int.display();
  std::cout << "=====" << std::endl;
  map_str_str.display();
  std::cout << "=====" << std::endl;
  
  for(auto& [key, value]: map_int_str)
    std::cout << "(" << key << ", " << value << ")" << std::endl;

  for(auto& [key, value]: map_int_int)
    std::cout << "(" << key << ", " << value << ")" << std::endl;

  for(auto& [key, value]: map_str_str)
    std::cout << "(" << key << ", " << value << ")" << std::endl;
  
  try {
    map_int_str.at(777);
  } catch(std::out_of_range& e) {
    std::cout << e.what() << std::endl;
  }
  
  return 0;
}