Implement file reader

src/Facility.cpp

@@ -40,8 +40,8 @@ const std::string Facility::readFromStream(std::istream& src) {
 }
 
 const uint8_t Facility::readFromString(const std::string& src) {
-    const auto &value = _values.find(boost::algorithm::to_lower_copy(src));
-    if (value != _values.end()) {
+    const auto& value = VALUES.find(boost::algorithm::to_lower_copy(src));
+    if (value != VALUES.end()) {
         return value->second;
     } else {
         throw "Illegal facility value: " + src;

src/Facility.h

@@ -27,6 +27,8 @@ SOFTWARE.
 
 #include <istream>
 
+class Severity;
+
 class Facility {
 public:
 
@@ -39,33 +41,31 @@ public:
     Facility(std::istream& source) : Facility(readFromStream(source)) {
     };
 
-    Facility(const Facility & orig) : _value(orig._value) {
+    Facility(const Facility& orig) : _value(orig._value) {
     };
 
     virtual ~Facility() {
     };
 
-    bool operator!=(const Facility & right) const {
+    bool operator!=(const Facility& right) const {
         bool result = !(*this == right); // Reuse equals operator
         return result;
     }
 
-    bool operator==(const Facility & right) const {
+    bool operator==(const Facility& right) const {
         return _value == right._value;
     }
 
     friend std::ostream& operator<<(std::ostream& os, const Facility& obj) {
-        os << obj._value;
+        os << std::to_string(obj._value);
         return os;
     }
 
-    uint8_t as_int() const {
-        return _value;
-    }
+    friend const uint8_t operator+(const Facility& f, const Severity& s);
 
 private:
 
-    const std::map<std::string, uint8_t> _values{
+    const std::map<std::string, uint8_t> VALUES{
         {"kern", 0},
         {"user", 1},
         {"mail", 2},

src/FileReader.h

@@ -0,0 +1,54 @@
+/*
+ The MIT License (MIT)
+
+Copyright (c) 2015 Marko Živanović
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+ */
+
+#ifndef FILEREADER_H
+#define	FILEREADER_H
+
+#include <fstream>
+#include "Reader.h"
+
+class FileReader : public Reader {
+public:
+
+    FileReader(const std::string filename) : _stream(std::ifstream(filename)) {
+        if (!_stream.is_open()) {
+            throw "File not found: " + filename;
+        }
+    }
+
+    virtual std::shared_ptr<const SyslogMessage> nextMessage() {
+        std::shared_ptr<const SyslogMessage> ret;
+        if (_stream.is_open() && !_stream.eof() && !_stream.fail()) {
+            ret.reset(new SyslogMessage(_stream));
+            for (auto c = _stream.peek(); c == '\n' || c == '\r'; _stream.get(), c = _stream.peek());
+        }
+        return ret;
+    }
+
+private:
+    std::ifstream _stream;
+};
+
+#endif	/* FILEREADER_H */
+

src/RFC3164FormattedSyslogMessage.h

@@ -41,7 +41,7 @@ public:
     virtual ~RFC3164FormattedSyslogMessage() {
     };
 
-    std::string operator()() {
+    const std::string operator()() {
         _stream.str("");
         _stream << "<" << std::to_string(_message.priority()) << ">";
         _stream << _message.timestamp() << " ";

src/Reader.h

@@ -32,7 +32,7 @@ class SyslogMessage;
 
 class Reader : private boost::noncopyable {
 public:
-    virtual std::shared_ptr<SyslogMessage> nextMessage() = 0;
+    virtual std::shared_ptr<const SyslogMessage> nextMessage() = 0;
 };
 
 #endif	/* READER_H */

src/Severity.cpp

@@ -40,8 +40,8 @@ const std::string Severity::readFromStream(std::istream& src) {
 }
 
 const uint8_t Severity::readFromString(const std::string& src) {
-    const auto &value = _values.find(boost::algorithm::to_lower_copy(src));
-    if (value != _values.end()) {
+    const auto &value = VALUES.find(boost::algorithm::to_lower_copy(src));
+    if (value != VALUES.end()) {
         return value->second;
     } else {
         throw "illegal severity: " + src;

src/Severity.h

@@ -25,6 +25,8 @@ SOFTWARE.
 #ifndef SEVERITY_H
 #define	SEVERITY_H
 
+class Facility;
+
 class Severity {
 public:
 
@@ -52,18 +54,16 @@ public:
         return _value == right._value;
     }
 
-    friend std::ostream& operator<<(std::ostream& os, const Severity& obj) {
-        os << obj._value;
-        return os;
-    }
+    friend const uint8_t operator+(const Facility& f, const Severity& s);
 
-    uint8_t as_int() const {
-        return _value;
+    friend std::ostream& operator<<(std::ostream& os, const Severity& obj) {
+        os << std::to_string(obj._value);
+        return os;
     }
 
 private:
 
-    const std::map<std::string, uint8_t> _values{
+    const std::map<std::string, uint8_t> VALUES{
         {"emergency", 0},
         {"alert", 1},
         {"critical", 2},

src/SyslogMessage.cpp

@@ -76,15 +76,14 @@ const std::string SyslogMessage::readSource(std::istream& src) {
     return ret;
 };
 
-const std::string SyslogMessage::readMessage(std::istream & src) {
+const std::string SyslogMessage::readMessage(std::istream& src) {
     std::string ret;
     skipWhitespace(src);
-    while (src) {
-        auto c = src.get();
-        if (!src.eof()) {
-            ret.push_back(c);
-        }
-    }
+    std::getline(src, ret);
     return ret;
 };
 
+const uint8_t operator+(const Facility& f, const Severity& s) {
+    return f._value * 8 + s._value;
+};
+

src/SyslogMessage.h

@@ -65,7 +65,7 @@ public:
     }
 
     const uint8_t priority() const {
-        return (_facility.as_int() * 8) +_severity.as_int();
+        return _facility + _severity;
     };
 
     friend std::ostream& operator<<(std::ostream& os, const SyslogMessage& obj) {

src/UDPWriter.cpp

@@ -26,7 +26,8 @@ SOFTWARE.
 #include "UDPWriter.h"
 #include "RFC3164FormattedSyslogMessage.h"
 
-using namespace boost::asio::ip;
+using boost::asio::ip::udp;
+using namespace boost::asio;
 
 UDPWriter::UDPWriter(const std::string& destination, const int port) {
     udp::resolver resolver(_ios);
@@ -40,10 +41,10 @@ UDPWriter::UDPWriter(const std::string& destination, const int port) {
     }
 };
 
-void UDPWriter::sendMessage(std::shared_ptr<SyslogMessage> message) {
+void UDPWriter::sendMessage(std::shared_ptr<const SyslogMessage> message) {
     if (_socket.is_open()) {
         auto formatted = RFC3164FormattedSyslogMessage{*message};
-        auto buff = ::boost::asio::buffer(formatted());
+        auto buff = buffer(formatted());
         _socket.send(buff);
     }
 }

src/UDPWriter.h

@@ -34,7 +34,7 @@ public:
 
     UDPWriter(const std::string&, const int);
 
-    virtual void sendMessage(std::shared_ptr<SyslogMessage>);
+    virtual void sendMessage(std::shared_ptr<const SyslogMessage>);
 
 private:
     boost::asio::io_service _ios;

src/Writer.h

@@ -31,7 +31,7 @@ SOFTWARE.
 
 class Writer : private boost::noncopyable {
 public:
-    virtual void sendMessage(std::shared_ptr<SyslogMessage>) = 0;
+    virtual void sendMessage(std::shared_ptr<const SyslogMessage>) = 0;
 };
 
 #endif	/* WRITER_H */