/*
 * ConnectionManager.cpp
 *
 * Copyright (C) 2008 Matthias Schiffer <matthias@gamezock.de>
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "ConnectionManager.h"
#include "ConfigManager.h"
#include <Common/RequestHandlers/StatusRequestHandler.h>
#include "RequestHandlers/DaemonListRequestHandler.h"
#include "RequestHandlers/DaemonStatusRequestHandler.h"
#include "RequestHandlers/GSSAPIAuthRequestHandler.h"
#include "RequestHandlers/IdentifyRequestHandler.h"
#include <Net/ServerConnection.h>
#include <Net/Packet.h>
#include <Net/Listener.h>

#include <unistd.h>
#include <algorithm>

#include <iostream>

namespace Mad {
namespace Core {

std::auto_ptr<ConnectionManager> ConnectionManager::connectionManager;


void ConnectionManager::refreshPollfds() {
  pollfds.clear();
  pollfdMap.clear();

  for(std::list<Net::Listener*>::iterator listener = listeners.begin(); listener != listeners.end(); ++listener) {
    std::vector<struct pollfd> fds = (*listener)->getPollfds();

    for(std::vector<struct pollfd>::iterator fd = fds.begin(); fd != fds.end(); ++fd) {
      pollfds.push_back(*fd);
      pollfdMap.insert(std::make_pair(fd->fd, &pollfds.back().revents));
    }
  }

  for(std::list<Net::ServerConnection*>::iterator con = daemonConnections.begin(); con != daemonConnections.end(); ++con) {
    pollfds.push_back((*con)->getPollfd());
    pollfdMap.insert(std::make_pair(pollfds.back().fd, &pollfds.back().revents));
  }

  for(std::list<Net::ServerConnection*>::iterator con = clientConnections.begin(); con != clientConnections.end(); ++con) {
    pollfds.push_back((*con)->getPollfd());
    pollfdMap.insert(std::make_pair(pollfds.back().fd, &pollfds.back().revents));
  }
}

ConnectionManager::ConnectionManager() {
  Common::RequestManager::init(true);

  Common::RequestManager::getRequestManager()->registerPacketType<Common::RequestHandlers::StatusRequestHandler>(Net::Packet::STATUS);
  Common::RequestManager::getRequestManager()->registerPacketType<RequestHandlers::DaemonListRequestHandler>(Net::Packet::LIST_DAEMONS);
  Common::RequestManager::getRequestManager()->registerPacketType<RequestHandlers::DaemonStatusRequestHandler>(Net::Packet::DAEMON_STATUS);
  Common::RequestManager::getRequestManager()->registerPacketType<RequestHandlers::GSSAPIAuthRequestHandler>(Net::Packet::GSSAPI_AUTH);
  Common::RequestManager::getRequestManager()->registerPacketType<RequestHandlers::IdentifyRequestHandler>(Net::Packet::IDENTIFY);

  ConfigManager *configManager = ConfigManager::getConfigManager();

  Net::Connection::init();

  const std::vector<Net::IPAddress> &listenerAddresses = configManager->getListenerAddresses();

  if(listenerAddresses.empty()) {
    try {
      listeners.push_back(new Net::Listener(configManager->getX509CertFile(), configManager->getX509KeyFile()));
    }
    catch(Net::Exception &e) {
      // TODO Log error
    }
  }
  else {
    for(std::vector<Net::IPAddress>::const_iterator address = listenerAddresses.begin(); address != listenerAddresses.end(); ++address) {
      try {
        listeners.push_back(new Net::Listener(configManager->getX509CertFile(), configManager->getX509KeyFile(), *address));
      }
      catch(Net::Exception &e) {
        // TODO Log error
      }
    }
  }

  refreshPollfds();

  const std::vector<DaemonInfo>& daemons = configManager->getDaemonList();

  for(std::vector<DaemonInfo>::const_iterator daemon = daemons.begin(); daemon != daemons.end(); ++daemon) {
    daemonInfo.insert(std::make_pair(daemon->getName(), *daemon));
    identifiedDaemonConnections.insert(std::make_pair<std::string,Net::ServerConnection*>(daemon->getName(), 0));
  }
}

ConnectionManager::~ConnectionManager() {
  for(std::list<Net::ServerConnection*>::iterator con = daemonConnections.begin(); con != daemonConnections.end(); ++con)
    delete *con;

  for(std::list<Net::ServerConnection*>::iterator con = clientConnections.begin(); con != clientConnections.end(); ++con)
    delete *con;

  Net::Connection::deinit();
}

void ConnectionManager::handleConnections(std::list<Net::ServerConnection*>& connections) {
  for(std::list<Net::ServerConnection*>::iterator con = connections.begin(); con != connections.end();) {
    if((*con)->isConnected()) {
      std::map<int,const short*>::iterator events = pollfdMap.find((*con)->getSocket());

      if(events != pollfdMap.end() && *events->second)
        (*con)->sendReceive(*events->second);

      ++con;
    }
    else {
      if((*con)->isIdentified()) {
        for(std::map<std::string,Net::ServerConnection*>::iterator idCon = identifiedDaemonConnections.begin(); idCon != identifiedDaemonConnections.end(); ++idCon) {
          if(idCon->second == *con) {
            idCon->second = 0;
            break;
          }
        }
      }

      Common::RequestManager::getRequestManager()->unregisterConnection(*con);
      delete *con;
      connections.erase(con++);
    }
  }
}

void ConnectionManager::run() {
  // TODO Logging

  handleConnections(daemonConnections);
  handleConnections(clientConnections);

  for(std::list<Net::Listener*>::iterator listener = listeners.begin(); listener != listeners.end(); ++listener) {
    Net::ServerConnection *con;

    while((con = (*listener)->getConnection(pollfdMap)) != 0) {
      (con->isDaemonConnection() ? daemonConnections : clientConnections).push_back(con);
      Common::RequestManager::getRequestManager()->registerConnection(con);
    }
  }

  refreshPollfds();
}

Net::Connection* ConnectionManager::getDaemonConnection(const std::string &name) const throw (Common::Exception&) {
  std::map<std::string,Net::ServerConnection*>::const_iterator daemon = identifiedDaemonConnections.find(name);
  if(daemon == identifiedDaemonConnections.end())
    throw Common::Exception(Common::Exception::UNKNOWN_DAEMON);

  if(!daemon->second)
    throw Common::Exception(Common::Exception::DAEMON_INACTIVE);

  return daemon->second;
}

void ConnectionManager::identifyDaemonConnection(Net::Connection *connection, const std::string &name) throw (Common::Exception&) {
  // TODO Logging

  if(connection->isIdentified())
    throw Common::Exception(Common::Exception::ALREADY_IDENTIFIED);

  std::list<Net::ServerConnection*>::iterator con = std::find(daemonConnections.begin(), daemonConnections.end(), connection);
  if(con == daemonConnections.end())
    throw Common::Exception(Common::Exception::INVALID_ACTION);

  std::map<std::string,Net::ServerConnection*>::iterator idCon = identifiedDaemonConnections.find(name);
  if(idCon == identifiedDaemonConnections.end()) {
    throw Common::Exception(Common::Exception::UNKNOWN_DAEMON);
  }

  if(idCon->second) {
    idCon->second->disconnect();
    std::cerr << "Disconnecting old connection" << std::endl;
  }

  idCon->second = *con;
  connection->setIdentified();

  std::cerr << "Identified as '" << name << "'." << std::endl;
}

std::map<std::string,DaemonInfo> ConnectionManager::getDaemonList() const {
  std::map<std::string,DaemonInfo> ret;

  for(std::map<std::string,Net::ServerConnection*>::const_iterator con = identifiedDaemonConnections.begin(); con != identifiedDaemonConnections.end(); ++con) {
    if(con->second)
      ret.insert(std::make_pair(con->first, daemonInfo.at(con->first)));
  }

  return ret;
}

}
}