/*
 * ConnectionManager.h
 *
 * 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/>.
 */

#ifndef MAD_SERVER_CONNECTIONMANAGER_H_
#define MAD_SERVER_CONNECTIONMANAGER_H_

#include "export.h"

#include <Core/Configurable.h>
#include <Core/Exception.h>
#include <Common/AuthContext.h>
#include <Common/Connection.h>
#include <Common/HostInfo.h>
#include <Common/RequestHandlerGroup.h>

#include <list>
#include <vector>
#include <map>
#include <set>

#include <boost/asio.hpp>

namespace Mad {

namespace Net {
class Connection;
class Listener;
class Packet;
}

namespace Server {

class Application;

class MAD_SERVER_EXPORT ConnectionManager : public Core::Configurable, private boost::noncopyable {
  private:
    class MAD_SERVER_EXPORT ServerConnection : public Common::Connection {
      public:
        enum ConnectionType {
          UNKNOWN = 0, DAEMON, CLIENT
        };

      private:
        Common::Application *application;

        boost::shared_ptr<Net::Connection> connection;
        ConnectionType type;
        Common::HostInfo *hostInfo;
        boost::shared_ptr<Common::AuthContext> authContext;

      protected:
        virtual bool send(const Net::Packet &packet);

      public:
        ServerConnection(Common::Application *application0, boost::shared_ptr<Net::Connection> connection0);

        bool isConnected() const;

        virtual bool disconnect();
        //virtual void* getCertificate(size_t *size) const;
        //virtual void* getPeerCertificate(size_t *size) const;

        ConnectionType getConnectionType() const {
          return type;
        }

        Common::HostInfo *getHostInfo() const {
          return hostInfo;
        }

        bool isIdentified() const {
          return (type != UNKNOWN);
        }

        void identify() {
          type = CLIENT;
        }

        void identify(Common::HostInfo *info) {
          type = DAEMON;
          hostInfo = info;
        }

        virtual bool isAuthenticated() const {
          return (authContext.get() != 0 && authContext->isAuthenticated());
        }

        boost::shared_ptr<const Common::AuthContext> authenticate(const std::string &method, const std::string &subMethod,
            const std::string &user, const std::vector<boost::uint8_t> &data, std::vector<boost::uint8_t> &response);
    };

    friend class Application;

    Application *application;

    std::string x509TrustFile, x509CrlFile, x509CertFile, x509KeyFile;

    std::vector<boost::asio::ip::tcp::endpoint> listenerAddresses;
    std::list<boost::shared_ptr<Net::Listener> > listeners;

    std::set<boost::shared_ptr<ServerConnection> > connections;

    std::map<std::string, Common::HostInfo> daemonInfo;

    boost::shared_ptr<Common::RequestHandlerGroup> connectionRequestHandlerGroup;
    boost::shared_ptr<Common::RequestHandlerGroup> daemonRequestHandlerGroup;
    boost::shared_ptr<Common::RequestHandlerGroup> userRequestHandlerGroup;

    static boost::asio::ip::tcp::endpoint parseAddress(const std::string &str) throw(Core::Exception);

    void updateState(Common::HostInfo *hostInfo, Common::HostInfo::State state);

    void handleNewConnection(boost::shared_ptr<Net::Connection> con);
    void handleDisconnect(boost::weak_ptr<ServerConnection> con);

    ConnectionManager(Application *application0);
    ~ConnectionManager();

  protected:
    virtual bool handleConfigEntry(const Core::ConfigEntry &entry, bool handled);
    virtual void configFinished();

  public:
    boost::shared_ptr<Common::Connection> getDaemonConnection(const std::string &name) const throw (Core::Exception);
    std::string getDaemonName(const Common::Connection *con) const throw (Core::Exception);

    void identifyDaemonConnection(Common::Connection *con, const std::string &name) throw (Core::Exception);

    boost::shared_ptr<const Common::AuthContext> authenticateConnection(Common::Connection *con, const std::string &method, const std::string &subMethod,
        const std::string &user, const std::vector<boost::uint8_t> &data, std::vector<boost::uint8_t> &response);

    std::vector<Common::HostInfo> getDaemonList() const;
};

}
}

#endif /*MAD_SERVER_CONNECTIONMANAGER_H_*/