From 714ed292b37a0d9baee69186bca38cf0f0c77c89 Mon Sep 17 00:00:00 2001
From: Yann Herklotz <ymherklotz@gmail.com>
Date: Thu, 16 Nov 2017 21:33:21 +0000
Subject: Before setting license

---
 yage/core/logsink.cpp | 57 +++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 55 insertions(+), 2 deletions(-)

(limited to 'yage/core/logsink.cpp')

diff --git a/yage/core/logsink.cpp b/yage/core/logsink.cpp
index a8f4d7b4..081eb5b3 100644
--- a/yage/core/logsink.cpp
+++ b/yage/core/logsink.cpp
@@ -8,6 +8,10 @@
 
 #include "logsink.h"
 
+#include <chrono>
+#include <ctime>
+#include <fstream>
+#include <iomanip>
 #include <iostream>
 
 namespace yage
@@ -42,9 +46,58 @@ void LogSink::write(const LogMessage::Meta &meta, const std::string &msg) const
 LogSink makeConsoleSink()
 {
     return [](const LogMessage::Meta &meta, const std::string &msg) {
-        std::cout << msg << " (" << meta.fileName << ":" << meta.lineNo
-                  << ")\n";
+        std::cout << msg << "\n";
     };
 }
 
+namespace
+{
+
+class FileSink
+{
+public:
+    FileSink(std::string &&filename)
+        : fileHandle_(std::make_shared<std::ofstream>(filename))
+    {
+        if (!fileHandle_->good()) {
+            throw std::runtime_error("Could not open file: " + filename);
+        }
+    }
+
+    FileSink(const std::string filename)
+        : fileHandle_(std::make_shared<std::ofstream>(filename))
+    {
+        if (!fileHandle_->good()) {
+            throw std::runtime_error("Could not open file: " + filename);
+        }
+    }
+
+    ~FileSink() = default;
+
+    void operator()(const LogMessage::Meta &meta, const std::string &msg) const
+    {
+        using namespace std::chrono;
+
+        auto now = system_clock::now();
+        auto time_t = system_clock::to_time_t(now);
+        auto local_time = std::localtime(&time_t);
+
+        (*fileHandle_) << std::put_time(local_time, "[%H:%M:%S] ") << msg
+                       << " (" << meta.fileName << ":" << meta.lineNo << ")\n";
+    }
+
+private:
+    std::shared_ptr<std::ofstream> fileHandle_;
+};
+
+} // namespace
+
+LogSink makeFileSink(const std::string &filename) {
+    return FileSink(filename);
+}
+
+LogSink makeFileSink(std::string &&filename) {
+    return FileSink(filename);
+}
+
 } // namespace yage
-- 
cgit