src/libshared/src/Msg.cpp

   1 ////////////////////////////////////////////////////////////////////////////////
   2 //
   3 // Copyright 2006 - 2017, Paul Beckingham, Federico Hernandez.
   4 //
   5 // Permission is hereby granted, free of charge, to any person obtaining a copy
   6 // of this software and associated documentation files (the "Software"), to deal
   7 // in the Software without restriction, including without limitation the rights
   8 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   9 // copies of the Software, and to permit persons to whom the Software is
  10 // furnished to do so, subject to the following conditions:
  11 //
  12 // The above copyright notice and this permission notice shall be included
  13 // in all copies or substantial portions of the Software.
  14 //
  15 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  16 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  18 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21 // SOFTWARE.
  22 //
  23 // http://www.opensource.org/licenses/mit-license.php
  24 //
  25 ////////////////////////////////////////////////////////////////////////////////
  26
  27 #include <cmake.h>
  28 #include <Msg.h>
  29 #include <shared.h>
  30 #include <format.h>
  31
  32 ////////////////////////////////////////////////////////////////////////////////
  33 void Msg::set (const std::string& name, int value)
  34 {
  35   _header[name] = format (value);
  36 }
  37
  38 ////////////////////////////////////////////////////////////////////////////////
  39 void Msg::set (const std::string& name, const std::string& value)
  40 {
  41   _header[name] = value;
  42 }
  43
  44 ////////////////////////////////////////////////////////////////////////////////
  45 std::string Msg::get (const std::string& name) const
  46 {
  47   auto i = _header.find (name);
  48   if (i != _header.end ())
  49     return i->second;
  50
  51   return "";
  52 }
  53
  54 ////////////////////////////////////////////////////////////////////////////////
  55 void Msg::setPayload (const std::string& payload)
  56 {
  57   _payload = payload;
  58 }
  59
  60 ////////////////////////////////////////////////////////////////////////////////
  61 std::string Msg::getPayload () const
  62 {
  63   return _payload;
  64 }
  65
  66 ////////////////////////////////////////////////////////////////////////////////
  67 std::vector <std::string> Msg::all () const
  68 {
  69   std::vector <std::string> names;
  70   for (auto& i : _header)
  71     names.push_back (i.first);
  72
  73   return names;
  74 }
  75
  76 ////////////////////////////////////////////////////////////////////////////////
  77 std::string Msg::serialize () const
  78 {
  79   std::string output;
  80   for (auto& i : _header)
  81     output += i.first + ": " + i.second + '\n';
  82
  83   output += '\n' + _payload + '\n';
  84
  85   return output;
  86 }
  87
  88 ////////////////////////////////////////////////////////////////////////////////
  89 bool Msg::parse (const std::string& input)
  90 {
  91   _header.clear ();
  92   _payload = "";
  93
  94   auto separator = input.find ("\n\n");
  95   if (separator == std::string::npos)
  96     throw std::string ("Malformed message");
  97
  98   // Parse header.
  99   auto lines = split (input.substr (0, separator), '\n');
 100   for (auto& i : lines)
 101   {
 102     auto delimiter = i.find (':');
 103     if (delimiter == std::string::npos)
 104         throw std::string ("Malformed message header '") + i + '\'';
 105
 106     _header[trim (i.substr (0, delimiter))] = trim (i.substr (delimiter + 1));
 107   }
 108
 109   // Parse payload.
 110   _payload = input.substr (separator + 2);
 111
 112   return true;
 113 }
 114
 115 ////////////////////////////////////////////////////////////////////////////////