/*
*
* Copyright (C) 2010 Mattias Runge
*
* 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 2 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, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
*/
#include "Monitor.h"
#include <boost/lexical_cast.hpp>
#include <boost/algorithm/string/trim.hpp>
#include "net/Manager.h"
#include "Message.h"
namespace atom {
namespace can {
Monitor::Monitor(unsigned int port): broker::Subscriber(port), LOG("can::Monitor")
{
try
{
this->server_id_ = net::Manager::Instance()->StartServer(net::TRANSPORT_PROTOCOL_TCP, port);
LOG.Info("Started TCP server on port " + boost::lexical_cast<std::string>(port) + ".");
LOG.Debug("Server id is " + boost::lexical_cast<std::string>(this->server_id_) + ".");
}
catch (std::runtime_error& e)
{
LOG.Error(e.what());
}
}
Monitor::~Monitor()
{
net::Manager::Instance()->StopServer(this->server_id_);
this->server_id_ = 0;
}
void Monitor::SlotOnMessageHandler(broker::Message::Pointer message)
{
if (message->GetType() == broker::Message::CAN_MESSAGE)
{
Message* payload = static_cast<Message*>(message->GetPayload().get());
std::string line = "";
if (payload->GetClassName() == "nmt")
{
line += "NMT";
}
else
{
if (payload->GetDirectionName() == "To_Owner")
{
line += "RX ";
}
else if (payload->GetDirectionName() == "From_Owner")
{
line += "TX ";
}
else
{
line += "??";
}
}
line += " " + payload->GetCommandName() + " ";
while (line.length() < 20)
{
line += " ";
}
if (payload->GetClassName() == "nmt")
{
line += "-";
}
else
{
line += payload->GetClassName();
line += "_" + payload->GetModuleName();
line += ":" + boost::lexical_cast<std::string>(payload->GetId());
}
while (line.length() < 40)
{
line += " ";
}
while (line.length() < 20)
{
line += " ";
}
common::StringMap variables = payload->GetVariables();
for (common::StringMap::iterator it = variables.begin(); it != variables.end(); it++)
{
line += " " + it->first + "=" + it->second;
}
line += "\n";
net::Manager::Instance()->SendToAll(this->server_id_, common::Byteset(line.begin(), line.end()));
}
}
void Monitor::SlotOnNewDataHandler(net::SocketId id, common::Byteset data)
{
if (std::find(this->clients_.begin(), this->clients_.end(), id) == this->clients_.end())
{
return;
}
std::string str(data.begin(), data.end());
boost::algorithm::trim_right_if(str, boost::is_any_of("\r\n"));
LOG.Debug("Received: \"" + str + "\" from client " + boost::lexical_cast<std::string>(id) + " on server " + boost::lexical_cast<std::string>(this->server_id_));
if (str == "q" || str == "quit")
{
LOG.Info("Client " + boost::lexical_cast<std::string>(id) + " has requested to be disconnected.");
net::Manager::Instance()->Disconnect(id);
}
}
void Monitor::SlotOnNewClientHandler(net::SocketId id, net::SocketId server_id)
{
if (server_id == this->server_id_)
{
this->clients_.insert(id);
}
}
void Monitor::SlotOnNewStateHandler(net::SocketId id, net::ClientState client_state)
{
if (std::find(this->clients_.begin(), this->clients_.end(), id) == this->clients_.end())
{
return;
}
if (client_state == net::CLIENT_STATE_DISCONNECTED)
{
LOG.Info("Client " + boost::lexical_cast<std::string>(id) + " has disconnected.");
this->clients_.erase(id);
}
else if (client_state == net::CLIENT_STATE_CONNECTED)
{
LOG.Info("Client " + boost::lexical_cast<std::string>(id) + " has connected.");
std::string message = "Welcome to Atom CAN monitoring\n";
net::Manager::Instance()->SendTo(id, common::Byteset(message.begin(), message.end()));
}
else
{
LOG.Debug("Got state: " + boost::lexical_cast<std::string>((int)client_state));
}
}
}; // namespace can
}; // namespace atom