diff options
Diffstat (limited to 'src/Client/InformationManager.cpp')
-rw-r--r-- | src/Client/InformationManager.cpp | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/src/Client/InformationManager.cpp b/src/Client/InformationManager.cpp new file mode 100644 index 0000000..8a3227c --- /dev/null +++ b/src/Client/InformationManager.cpp @@ -0,0 +1,90 @@ +/* + * InformationManager.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 "InformationManager.h" +#include "Requests/DaemonListRequest.h" +#include <Common/Logger.h> +#include <Common/RequestManager.h> +#include <Net/Packets/ErrorPacket.h> +#include <Net/Packets/HostStatePacket.h> + + +namespace Mad { +namespace Client { + +std::auto_ptr<InformationManager> InformationManager::informationManager; + + +void InformationManager::DaemonStateUpdateRequest::handlePacket(Net::Connection *connection, const Net::Packet &packet) { + if(packet.getType() != Net::Packet::DAEMON_STATE_UPDATE) { + Common::Logger::log(Common::Logger::ERROR, "Received an unexpected packet."); + connection->send(Net::Packets::ErrorPacket(Net::Packet::ERROR, packet.getRequestId(), Common::Exception(Common::Exception::UNEXPECTED_PACKET))); + + signalFinished().emit(); + return; + } + + // TODO Require authentication + + Net::Packets::HostStatePacket hostStatePacket(packet); + + std::map<std::string, Common::HostInfo>::iterator host = informationManager->daemons.find(hostStatePacket.getName()); + if(host != informationManager->daemons.end()) + host->second.setState(hostStatePacket.getState()); + else + Common::Logger::log(Common::Logger::WARNING, "Received a state update for an unknown host."); + + connection->send(Net::Packet(Net::Packet::OK, packet.getRequestId())); + + signalFinished().emit(); +} + + +InformationManager::InformationManager(Net::Connection *connection) : initFinished(false) { + Common::RequestManager::getRequestManager()->sendRequest(connection, + std::auto_ptr<Common::RequestBase>( + new Requests::DaemonListRequest(sigc::mem_fun(this, &InformationManager::daemonListRequestFinished)) + ) + ); + + Common::RequestManager::getRequestManager()->registerPacketType<DaemonStateUpdateRequest>(Net::Packet::DAEMON_STATE_UPDATE); +} + +InformationManager::~InformationManager() { + Common::RequestManager::getRequestManager()->unregisterPacketType(Net::Packet::DAEMON_STATE_UPDATE); +} + +void InformationManager::daemonListRequestFinished(const Common::Request<Net::Packets::HostListPacket> &request) { + try { + const std::vector<Common::HostInfo> &hostInfo = request.getResult().getHostInfo(); + + for(std::vector<Common::HostInfo>::const_iterator daemon = hostInfo.begin(); daemon != hostInfo.end(); ++daemon) { + daemons.clear(); + daemons.insert(std::make_pair(daemon->getName(), *daemon)); + } + + initFinished = true; + } + catch(Common::Exception &e) { + Common::Logger::logf(Common::Logger::CRITICAL, "Host list request failed: %s", e.strerror().c_str()); + } +} + +} +} |