/*
 * 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 <Net/ServerConnection.h>
#include <Net/Packet.h>
#include <Net/Listener.h>
#include <unistd.h>

#include <iostream>

namespace Mad {
namespace Core {

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));
  }
}

void ConnectionManager::receiveHandler(Net::Connection *connection, const Net::Packet &packet) {
  switch(packet.getType()) {
    case Net::Packet::TYPE_UNKNOWN:
      break;
    case Net::Packet::TYPE_DEBUG:
      std::cout << "Received debug packet." << std::endl;
      std::cout << "  Request ID: 0x" << std::hex << std::uppercase << packet.getRequestId() << std::dec << std::endl;
      break;
    case Net::Packet::TYPE_PING:
      connection->send(Net::Packet(Net::Packet::TYPE_PONG, packet.getRequestId(), packet.getData(), packet.getLength()));
      break;
    case Net::Packet::TYPE_PONG:
      // TODO: Pong!
      break;
    case Net::Packet::TYPE_DISCONNECT_REQ:
      connection->send(Net::Packet(Net::Packet::TYPE_DISCONNECT_REP, packet.getRequestId()));
      connection->disconnect();
      break;
    case Net::Packet::TYPE_DISCONNECT_REP:
      connection->disconnect();
  }
}

ConnectionManager::ConnectionManager() {
  try {
    // TODO: Get listener addresses from config
    listeners.push_back(new Net::Listener(Net::IPAddress("0.0.0.0", 6666)));
  }
  catch(Net::Exception &e) {
    // TODO: Log error
  }
  
  refreshPollfds();
}

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;
}

void ConnectionManager::run() {
  // TODO: Logging
  
  for(std::list<Net::ServerConnection*>::iterator con = daemonConnections.begin(); con != daemonConnections.end();) {
    if((*con)->isConnected()) {
      std::map<int,const short*>::iterator events = pollfdMap.find((*con)->getSocket());
      
      if(events != pollfdMap.end())
        (*con)->sendReceive(*events->second);
      else
        (*con)->sendReceive();
      
      ++con;
    }
    else {
      std::cout << "A daemon connection was dropped." << std::endl;
      delete *con;
      daemonConnections.erase(con++);
    }
  }
  
  for(std::list<Net::ServerConnection*>::iterator con = clientConnections.begin(); con != clientConnections.end();) {
    if((*con)->isConnected()) {
      std::map<int,const short*>::iterator events = pollfdMap.find((*con)->getSocket());
      
      if(events != pollfdMap.end())
        (*con)->sendReceive(*events->second);
      else
        (*con)->sendReceive();
      
      ++con;
    }
    else {
      std::cout << "A client connection was dropped." << std::endl;
      delete *con;
      clientConnections.erase(con++);
    }
  }
  
  for(std::list<Net::Listener*>::iterator listener = listeners.begin(); listener != listeners.end(); ++listener) {
    Net::ServerConnection *con;
    
    while((con = (*listener)->getConnection(pollfdMap)) != 0) {
      if(con->isDaemonConnection()) {
        daemonConnections.push_back(con);
        con->signalReceive().connect(sigc::mem_fun(this, &ConnectionManager::receiveHandler));
      }
      else {
        clientConnections.push_back(con);
        con->signalReceive().connect(sigc::mem_fun(this, &ConnectionManager::receiveHandler));
      }
    }
  }
  
  refreshPollfds();
}

}
}