/*
 * 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 Lesser 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>
#include "ConnectionManager.h"
#include "Application.h"
#include <Core/ConfigEntry.h>
#include <Core/ConfigManager.h>
#include <Common/AuthManager.h>
#include <Common/RequestManager.h>
#include <Common/RequestHandlers/FSInfoRequestHandler.h>
#include <Common/RequestHandlers/StatusRequestHandler.h>
#include "Requests/DaemonStateUpdateRequest.h"
#include "RequestHandlers/ConnectionRequestHandlerGroup.h"
#include "RequestHandlers/DaemonRequestHandlerGroup.h"
#include "RequestHandlers/UserRequestHandlerGroup.h"
#include <Net/Packet.h>
#include <Net/Listener.h>

#include <algorithm>

namespace Mad {
namespace Server {

bool ConnectionManager::ServerConnection::send(const Net::Packet &packet) {
  return connection->send(packet);
}

ConnectionManager::ServerConnection::ServerConnection(Common::Application *application0, boost::shared_ptr<Net::Connection> connection0)
: Common::Connection(application0), application(application0), connection(connection0), type(UNKNOWN), hostInfo(0) {
  connection->connectSignalReceive(boost::bind(&ServerConnection::receive, this, _1));
}

bool ConnectionManager::ServerConnection::isConnected() const {
  return connection->isConnected();
}

bool ConnectionManager::ServerConnection::disconnect() {
  connection->disconnect();

  return true;
}

boost::shared_ptr<const Common::AuthContext> ConnectionManager::ServerConnection::authenticate(const Core::String &method,
    const Core::String &subMethod, const Core::String &user, const std::vector<boost::uint8_t> &data, std::vector<boost::uint8_t> &response) {
  if(!isIdentified())
    type = CLIENT;

  authContext = application->getAuthManager()->authenticate(method, subMethod, user, data, response, authContext);

  if(authContext->isAuthenticated())
    connection->setReceiveLimit(0xFFFFFFFF); // 4 Gigs

  return authContext;
}

boost::asio::ip::tcp::endpoint ConnectionManager::parseAddress(const std::string &str) throw(Core::Exception) {
  try {
    if(str == "*")
      return boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v6(), DEFAULT_PORT);

    boost::smatch match;

    static const boost::regex r1("\\*:(\\d+)", boost::regex_constants::perl);
    if(boost::regex_match(str, match, r1))
      return boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v6(), std::atoi(match[1].str().c_str()));

    static const boost::regex r2("\\[(.+)\\]:(\\d+)", boost::regex_constants::perl);
    if(boost::regex_match(str, match, r2))
      return boost::asio::ip::tcp::endpoint(boost::asio::ip::address_v6::from_string(match[1].str()),
          std::atoi(match[2].str().c_str()));

    static const boost::regex r3("\\[(.+)\\]", boost::regex_constants::perl);
    if(boost::regex_match(str, match, r3))
      return boost::asio::ip::tcp::endpoint(boost::asio::ip::address_v6::from_string(match[1].str()), DEFAULT_PORT);

    static const boost::regex r4("((?:\\d{1,3}\\.){3}\\d{1,3}):(\\d+)", boost::regex_constants::perl);
    if(boost::regex_match(str, match, r4))
      return boost::asio::ip::tcp::endpoint(boost::asio::ip::address_v4::from_string(match[1].str()), std::atoi(match[2].str().c_str()));

    static const boost::regex r5("((?:\\d{1,3}\\.){3}\\d{1,3})", boost::regex_constants::perl);
    if(boost::regex_match(str, match, r5))
      return boost::asio::ip::tcp::endpoint(boost::asio::ip::address_v4::from_string(match[1].str()), DEFAULT_PORT);
  }
  catch(...) {}

  throw Core::Exception(Core::Exception::INVALID_INPUT);
}

void ConnectionManager::updateState(Common::HostInfo *hostInfo, Common::HostInfo::State state) {
  hostInfo->setState(state);

  for(std::set<boost::shared_ptr<ServerConnection> >::iterator con = connections.begin(); con != connections.end(); ++con) {
    if((*con)->getConnectionType() == ServerConnection::CLIENT) {
      boost::shared_ptr<Requests::DaemonStateUpdateRequest> request(new Requests::DaemonStateUpdateRequest(application, hostInfo->getName(), state));
      application->getRequestManager()->sendRequest(con->get(), request);
    }
  }
}

void ConnectionManager::configure() {
  x509TrustFile = application->getConfigManager()->get("X509TrustFile");
  x509CrlFile = application->getConfigManager()->get("X509CrlFile");
  x509CertFile = application->getConfigManager()->get("X509CertFile");
  x509KeyFile = application->getConfigManager()->get("X509KeyFile");

  std::vector<const Core::ConfigEntry*> listenEntries = application->getConfigManager()->getEntries("Listen");
  if(!listenEntries.empty()) {
    for(std::vector<const Core::ConfigEntry*>::iterator listenEntry = listenEntries.begin(); listenEntry != listenEntries.end(); ++listenEntry) {
      try {
        boost::shared_ptr<Net::Listener> listener(new Net::Listener(application, x509CertFile, x509KeyFile,
            parseAddress((*listenEntry)->getValue().toString())));
        listener->connectSignalNewConnection(boost::bind(&ConnectionManager::handleNewConnection, this, _1));
        listeners.push_back(listener);
      }
      catch(Core::Exception &e) {
        application->logf(Core::Logger::LOG_WARNING, "ConnectionManager: Invalid listen address '%s'", (*listenEntry)->getValue().toLocale().c_str());
      }
    }
  }
  else {
    boost::shared_ptr<Net::Listener> listener(new Net::Listener(application, x509CertFile, x509KeyFile,
        boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v6(), DEFAULT_PORT)));
    listener->connectSignalNewConnection(boost::bind(&ConnectionManager::handleNewConnection, this, _1));
    listeners.push_back(listener);
  }

  std::vector<const Core::ConfigEntry*> daemonEntries = application->getConfigManager()->getEntries("Daemon");
  for(std::vector<const Core::ConfigEntry*>::iterator daemonEntry = daemonEntries.begin(); daemonEntry != daemonEntries.end(); ++daemonEntry) {
    Core::String name = (*daemonEntry)->getValue();
    if(name.isEmpty())
      continue;

    Common::HostInfo daemon(name);
    daemon.setIP((*daemonEntry)->get("IpAddress"));

    daemonInfo.insert(std::make_pair(name, daemon));
  }
}

void ConnectionManager::handleNewConnection(boost::shared_ptr<Net::Connection> con) {
  boost::shared_ptr<ServerConnection> connection(new ServerConnection(application, con));
  con->connectSignalDisconnected(boost::bind(&ConnectionManager::handleDisconnect, this, boost::weak_ptr<ServerConnection>(connection)));
  connections.insert(connection);

  application->getRequestManager()->registerConnection(connection.get());

  con->startReceive();
}

void ConnectionManager::handleDisconnect(boost::weak_ptr<ServerConnection> con) {
  boost::shared_ptr<ServerConnection> connection = con.lock();
  if(!connection)
    return;

  if(connection->getHostInfo())
    updateState(connection->getHostInfo(), Common::HostInfo::INACTIVE);

  connections.erase(connection);

  application->getRequestManager()->unregisterConnection(connection.get());
}

ConnectionManager::ConnectionManager(Application *application0) : application(application0),
connectionRequestHandlerGroup(new RequestHandlers::ConnectionRequestHandlerGroup(application)),
daemonRequestHandlerGroup(new RequestHandlers::DaemonRequestHandlerGroup),
userRequestHandlerGroup(new RequestHandlers::UserRequestHandlerGroup(application)) {
  application->getRequestManager()->registerPacketType<Common::RequestHandlers::FSInfoRequestHandler>("FSInfo");
  application->getRequestManager()->registerPacketType<Common::RequestHandlers::StatusRequestHandler>("GetStatus");

  application->getRequestManager()->registerRequestHandlerGroup(connectionRequestHandlerGroup);
  application->getRequestManager()->registerRequestHandlerGroup(daemonRequestHandlerGroup);
  application->getRequestManager()->registerRequestHandlerGroup(userRequestHandlerGroup);

  application->getConfigManager()->registerConfigurable(this);
}

ConnectionManager::~ConnectionManager() {
  application->getConfigManager()->unregisterConfigurable(this);

  connections.clear();

  application->getRequestManager()->unregisterRequestHandlerGroup(userRequestHandlerGroup);
  application->getRequestManager()->unregisterRequestHandlerGroup(daemonRequestHandlerGroup);
  application->getRequestManager()->unregisterRequestHandlerGroup(connectionRequestHandlerGroup);

  application->getRequestManager()->unregisterPacketType("FSInfo");
  application->getRequestManager()->unregisterPacketType("GetStatus");
}

boost::shared_ptr<Common::Connection> ConnectionManager::getDaemonConnection(const Core::String &name) const throw (Core::Exception) {
  
  
  std::map<Core::String, Common::HostInfo>::const_iterator hostIt = daemonInfo.find(name);
  if(hostIt == daemonInfo.end())
    throw Core::Exception(Core::Exception::UNKNOWN_DAEMON);
  
  const Common::HostInfo *hostInfo = &hostIt->second;

  if(hostInfo->getState() != Common::HostInfo::INACTIVE) {
    for(std::set<boost::shared_ptr<ServerConnection> >::const_iterator it = connections.begin(); it != connections.end(); ++it) {
      if((*it)->getHostInfo() == hostInfo) {
        return *it;
      }
    }
  }

  throw Core::Exception(Core::Exception::NOT_AVAILABLE);
}

Core::String ConnectionManager::getDaemonName(const Common::Connection *con) const throw (Core::Exception) {
  const ServerConnection *connection = dynamic_cast<const ServerConnection*>(con);

  if(connection && connection->getConnectionType() == ServerConnection::DAEMON)
    return connection->getHostInfo()->getName();

  throw Core::Exception(Core::Exception::UNKNOWN_DAEMON);
}

void ConnectionManager::identifyDaemonConnection(Common::Connection *con, const Core::String &name) throw (Core::Exception) {
  // TODO Logging

  ServerConnection *connection = dynamic_cast<ServerConnection*>(con);

  if(!connection)
    throw Core::Exception(Core::Exception::INVALID_INPUT);

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

  if(daemonInfo.count(name) == 0)
    throw Core::Exception(Core::Exception::UNKNOWN_DAEMON);

  Common::HostInfo *hostInfo = &daemonInfo[name];

  if(hostInfo->getState() != Common::HostInfo::INACTIVE) {
    try {
      getDaemonConnection(name)->disconnect();
      application->log(Core::Logger::LOG_WARNING, "Disconnecting old connection.");
    }
    catch(Core::Exception&) {}
  }

  connection->identify(hostInfo);
  updateState(hostInfo, Common::HostInfo::RUNNING);
}

boost::shared_ptr<const Common::AuthContext> ConnectionManager::authenticateConnection(Common::Connection *con, const Core::String &method,
    const Core::String &subMethod, const Core::String &user, const std::vector<boost::uint8_t> &data, std::vector<boost::uint8_t> &response) {
  // TODO Logging

  ServerConnection *connection = dynamic_cast<ServerConnection*>(con);

  if(!connection)
    throw Core::Exception(Core::Exception::INVALID_INPUT);

  if(!connection->isIdentified())
    connection->identify();

  return connection->authenticate(method, subMethod, user, data, response);
}

std::vector<Common::HostInfo> ConnectionManager::getDaemonList() const {
  std::vector<Common::HostInfo> ret;

  for(std::map<Core::String,Common::HostInfo>::const_iterator daemon = daemonInfo.begin(); daemon != daemonInfo.end(); ++daemon) {
    ret.push_back(daemon->second);
  }

  return ret;
}

}
}