Subversion Repositories HomeAutomation

Rev

Rev 1606 | Rev 1956 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | SVN | RSS feed

  1. /*
  2.  *
  3.  *  Copyright (C) 2010  Mattias Runge
  4.  *
  5.  *  This program is free software; you can redistribute it and/or modify
  6.  *  it under the terms of the GNU General Public License as published by
  7.  *  the Free Software Foundation; either version 2 of the License, or
  8.  *  (at your option) any later version.
  9.  *
  10.  *  This program is distributed in the hope that it will be useful,
  11.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13.  *  GNU General Public License for more details.
  14.  *
  15.  *  You should have received a copy of the GNU General Public License along
  16.  *  with this program; if not, write to the Free Software Foundation, Inc.,
  17.  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18.  *
  19.  */
  20.  
  21. #include "Monitor.h"
  22.  
  23. #include <boost/lexical_cast.hpp>
  24. #include <boost/algorithm/string/trim.hpp>
  25.  
  26. #include "net/Manager.h"
  27.  
  28. #include "Message.h"
  29.  
  30. namespace atom {
  31. namespace can {
  32.    
  33. Monitor::Monitor(unsigned int port): broker::Subscriber(port), LOG("can::Monitor")
  34. {
  35.     try
  36.     {
  37.         this->server_id_ = net::Manager::Instance()->StartServer(net::PROTOCOL_TCP, port);
  38.         LOG.Info("Started TCP server on port " + boost::lexical_cast<std::string>(port) + ".");
  39.         LOG.Debug("Server id is " + boost::lexical_cast<std::string>(this->server_id_) + ".");
  40.     }
  41.     catch (std::runtime_error& e)
  42.     {
  43.         LOG.Error(e.what());
  44.     }
  45. }
  46.  
  47. Monitor::~Monitor()
  48. {
  49.     net::Manager::Instance()->StopServer(this->server_id_);
  50.     this->server_id_ = 0;
  51. }
  52.  
  53. void Monitor::SlotOnMessageHandler(broker::Message::Pointer message)
  54. {
  55.     if (message->GetType() == broker::Message::CAN_MESSAGE)
  56.     {
  57.         Message* payload = static_cast<Message*>(message->GetPayload().get());
  58.        
  59.         std::string line = "";
  60.        
  61.         if (payload->GetClassName() == "nmt")
  62.         {
  63.             line += "NMT";
  64.         }
  65.         else
  66.         {
  67.             if (payload->GetDirectionName() == "To_Owner")
  68.             {
  69.                 line += "RX ";
  70.             }
  71.             else if (payload->GetDirectionName() == "From_Owner")
  72.             {
  73.                 line += "TX ";
  74.             }
  75.             else
  76.             {
  77.                 line += "??";
  78.             }
  79.         }
  80.        
  81.         line += " " + payload->GetCommandName() + " ";
  82.        
  83.         while (line.length() < 20)
  84.         {
  85.             line += " ";
  86.         }
  87.        
  88.         if (payload->GetClassName() == "nmt")
  89.         {
  90.             line += "-";
  91.         }
  92.         else
  93.         {
  94.             line += payload->GetClassName();
  95.             line += "_" + payload->GetModuleName();
  96.             line += ":" + boost::lexical_cast<std::string>(payload->GetId());
  97.         }
  98.        
  99.         while (line.length() < 40)
  100.         {
  101.             line += " ";
  102.         }
  103.        
  104.         while (line.length() < 20)
  105.         {
  106.             line += " ";
  107.         }
  108.        
  109.         common::StringMap variables = payload->GetVariables();
  110.        
  111.         for (common::StringMap::iterator it = variables.begin(); it != variables.end(); it++)
  112.         {
  113.             if (it->second != "")
  114.             {
  115.                 line += " " + it->first + "=" + it->second;
  116.             }
  117.         }
  118.        
  119.         line += "\n";
  120.        
  121.         net::Manager::Instance()->SendToAll(this->server_id_, common::Byteset(line));
  122.     }
  123. }
  124.  
  125. void Monitor::SlotOnNewDataHandler(net::ClientId client_id, net::ServerId server_id, common::Byteset data)
  126. {
  127.     if (server_id != this->server_id_)
  128.     {
  129.         return;
  130.     }
  131.    
  132.     std::string str = data.ToCharString();
  133.    
  134.     boost::algorithm::trim_right_if(str, boost::is_any_of("\r\n"));
  135.    
  136.     LOG.Debug("Received: \"" + str + "\" from client " + boost::lexical_cast<std::string>(client_id) + " on server " + boost::lexical_cast<std::string>(server_id));
  137.    
  138.     if (str == "q" || str == "quit")
  139.     {
  140.         LOG.Info("Client " + boost::lexical_cast<std::string>(client_id) + " has requested to be disconnected.");
  141.         net::Manager::Instance()->Disconnect(client_id);
  142.     }
  143. }
  144.  
  145. void Monitor::SlotOnNewStateHandler(net::ClientId client_id, net::ServerId server_id, net::ClientState client_state)
  146. {
  147.     if (server_id != this->server_id_)
  148.     {
  149.         return;
  150.     }
  151.    
  152.     if (client_state == net::CLIENT_STATE_DISCONNECTED)
  153.     {
  154.         LOG.Info("Client " + boost::lexical_cast<std::string>(client_id) + " has disconnected.");
  155.     }
  156.     else if (client_state == net::CLIENT_STATE_CONNECTED)
  157.     {
  158.         LOG.Info("Client " + boost::lexical_cast<std::string>(client_id) + " has connected.");
  159.         net::Manager::Instance()->SendTo(client_id, common::Byteset("Welcome to Atom CAN monitoring\n"));
  160.     }
  161.     else
  162.     {
  163.         LOG.Debug("Got state: " + boost::lexical_cast<std::string>((int)client_state));
  164.     }
  165. }
  166.    
  167. }; // namespace can
  168. }; // namespace atom
  169.