#include "connectionregistry.h"

#include <iostream>
#include <utility>

void ConnectionRegistry::setId(ConnectionRegistry::connection c, const std::string& id)
{
 if (!m_connections.at(c).empty()) {
  auto& connections_set{m_ids.at(id)};
  connections_set.erase(c);
  if (connections_set.empty())
   m_ids.erase(id);
 }
 
 m_connections.at(c) = id;

 if (!id.empty()) {
  if (!m_ids.contains(id))
   m_ids[id] = {};
  m_ids.at(id).insert(c);
 }

}

void ConnectionRegistry::addConnection(ConnectionRegistry::connection c)
{
 if (m_connections.contains(c))
  throw std::runtime_error("ConnectionRegistry::addConnection: connection already exists");

 m_connections.emplace(c, "");

 std::cout << "Info: Added connection, now " << m_connections.size() << std::endl;
}

void ConnectionRegistry::delConnection(ConnectionRegistry::connection c)
{
 if (!m_connections.contains(c))
  throw std::runtime_error("ConnectionRegistry::delConnection: connection doesn't exist");

 std::string id {m_connections.at(c)};

 m_connections.erase(c);

 if (!id.empty()) {
  auto& connections_set{m_ids.at(id)};
  connections_set.erase(c);
  if (connections_set.empty())
   m_ids.erase(id);
 }
 
 std::cout << "Info: Deleted connection, now " << m_connections.size() << std::endl;
}

std::unordered_set<ConnectionRegistry::connection>::iterator ConnectionRegistry::begin(const std::string& id)
{
 return m_ids.at(id).begin();
}

std::unordered_set<ConnectionRegistry::connection>::iterator ConnectionRegistry::end(const std::string& id)
{
 return m_ids.at(id).end();
}

std::unordered_map<ConnectionRegistry::connection, std::string>::iterator ConnectionRegistry::begin()
{
 return m_connections.begin();
}

std::unordered_map<ConnectionRegistry::connection, std::string>::iterator ConnectionRegistry::end()
{
 return m_connections.end();
}

void ConnectionRegistry::dump() const
{
 std::cout << "Connection Registry:" << std::endl;
 
 std::cout << "Connections: " << std::endl;
 for (auto& i: m_connections) {
  std::cout << "    " << i.first << ": " << i.second << std::endl;
 }
 
 std::cout << "IDs: " << std::endl;
 for (auto& i: m_ids) {
  std::cout << "    " << i.first << ":";

  for (auto& j: i.second) {
   std::cout << " " << j;
  }
  std::cout << std::endl;
 }
}

size_t ConnectionRegistry::number_of_connections() const
{
 return m_connections.size();
}

ConnectionRegistry::RegistryGuard::RegistryGuard(ConnectionRegistry& registry, ConnectionRegistry::connection c):
 m_registry{registry},
 m_connection{c}
{
 m_registry.addConnection(m_connection);
}

ConnectionRegistry::RegistryGuard::~RegistryGuard()
{
 m_registry.delConnection(m_connection);
}