1 files changed, 95 insertions, 0 deletions
diff --git a/src/common/scoped_message_writer.h b/src/common/scoped_message_writer.h
new file mode 100644
index 000000000..70db54eae
--- /dev/null
+++ b/src/common/scoped_message_writer.h
@@ -0,0 +1,95 @@
+#pragma once
+
+#include "misc_log_ex.h"
+#include <iostream>
+
+namespace tools
+{
+
+class scoped_message_writer
+{
+private:
+  bool m_flush;
+  std::stringstream m_oss;
+  epee::log_space::console_colors m_color;
+  bool m_bright;
+  int m_log_level;
+public:
+  scoped_message_writer(
+      epee::log_space::console_colors color = epee::log_space::console_color_default
+    , bool bright = false
+    , std::string&& prefix = std::string()
+    , int log_level = LOG_LEVEL_2
+    )
+    : m_flush(true)
+    , m_color(color)
+    , m_bright(bright)
+    , m_log_level(log_level)
+  {
+    m_oss << prefix;
+  }
+
+  scoped_message_writer(scoped_message_writer&& rhs)
+    : m_flush(std::move(rhs.m_flush))
+#if defined(_MSC_VER)
+    , m_oss(std::move(rhs.m_oss))
+#else
+      // GCC bug: http://gcc.gnu.org/bugzilla/show_bug.cgi?id=54316
+    , m_oss(rhs.m_oss.str(), std::ios_base::out | std::ios_base::ate)
+#endif
+    , m_color(std::move(rhs.m_color))
+    , m_log_level(std::move(rhs.m_log_level))
+  {
+    rhs.m_flush = false;
+  }
+
+  scoped_message_writer(scoped_message_writer& rhs) = delete;
+  scoped_message_writer& operator=(scoped_message_writer& rhs) = delete;
+  scoped_message_writer& operator=(scoped_message_writer&& rhs) = delete;
+
+  template<typename T>
+  std::ostream& operator<<(const T& val)
+  {
+    m_oss << val;
+    return m_oss;
+  }
+
+  ~scoped_message_writer()
+  {
+    if (m_flush)
+    {
+      m_flush = false;
+
+      LOG_PRINT(m_oss.str(), m_log_level)
+
+      if (epee::log_space::console_color_default == m_color)
+      {
+        std::cout << m_oss.str();
+      }
+      else
+      {
+        epee::log_space::set_console_color(m_color, m_bright);
+        std::cout << m_oss.str();
+        epee::log_space::reset_console_color();
+      }
+      std::cout << std::endl;
+    }
+  }
+};
+
+inline scoped_message_writer success_msg_writer()
+{
+  return scoped_message_writer(epee::log_space::console_color_green, false, std::string(), LOG_LEVEL_2);
+}
+
+inline scoped_message_writer msg_writer(epee::log_space::console_colors color = epee::log_space::console_color_default)
+{
+  return scoped_message_writer(color, false, std::string(), LOG_LEVEL_2);
+}
+
+inline scoped_message_writer fail_msg_writer()
+{
+  return scoped_message_writer(epee::log_space::console_color_red, true, "Error: ", LOG_LEVEL_0);
+}
+
+} // namespace tools