From 7d1092db045ad5b20dbdde1ec1fb0bf2f5195267 Mon Sep 17 00:00:00 2001
From: Michael Carlberg <c@rlberg.se>
Date: Mon, 5 Dec 2016 04:45:01 +0100
Subject: [PATCH] refactor(date): Add label and date/time tokens

Ref #225
---
 examples/config.cmake    | 12 +++++-
 include/modules/date.hpp | 21 +++++++---
 src/modules/date.cpp     | 83 ++++++++++++++++++++++++++--------------
 3 files changed, 81 insertions(+), 35 deletions(-)

diff --git a/examples/config.cmake b/examples/config.cmake
index de0b09cd..4cff7b0f 100644
--- a/examples/config.cmake
+++ b/examples/config.cmake
@@ -231,12 +231,20 @@ label-disconnected-foreground = ${colors.foreground-alt}
 
 [module/date]
 type = internal/date
-date = %H:%M
 interval = 5
-format-prefix = " "
+
+date =
+date-alt = " %Y-%m-%d"
+
+time = %H:%M
+time-alt = %H:%M:%S
+
+format-prefix = 
 format-prefix-foreground = ${colors.foreground-alt}
 format-underline = #0a6cf5
 
+label = %date% %time%
+
 [module/volume]
 type = internal/volume
 
diff --git a/include/modules/date.hpp b/include/modules/date.hpp
index 8b1315d8..aab752d0 100644
--- a/include/modules/date.hpp
+++ b/include/modules/date.hpp
@@ -13,16 +13,27 @@ namespace modules {
     bool update();
     bool build(builder* builder, const string& tag) const;
     bool handle_event(string cmd);
-    bool receive_events() const;
+    bool receive_events() const {
+      return true;
+    }
 
    private:
-    static constexpr auto TAG_DATE = "<date>";
+    static constexpr auto TAG_LABEL = "<label>";
     static constexpr auto EVENT_TOGGLE = "datetoggle";
 
-    string m_format;
-    string m_formatalt;
+    // @deprecated: Use <label>
+    static constexpr auto TAG_DATE = "<date>";
+
+    label_t m_label;
+
+    string m_dateformat;
+    string m_dateformat_alt;
+    string m_timeformat;
+    string m_timeformat_alt;
+
+    string m_date;
+    string m_time;
 
-    char m_buffer[256] = {'\0'};
     stateflag m_toggled{false};
   };
 }
diff --git a/src/modules/date.cpp b/src/modules/date.cpp
index 0169383f..2c80ccec 100644
--- a/src/modules/date.cpp
+++ b/src/modules/date.cpp
@@ -1,4 +1,5 @@
 #include "modules/date.hpp"
+#include "drawtypes/label.hpp"
 
 #include "modules/meta/base.inl"
 #include "modules/meta/timer_module.inl"
@@ -14,58 +15,84 @@ namespace modules {
       setlocale(LC_TIME, m_bar.locale.c_str());
     }
 
-    m_interval = chrono::duration<double>(m_conf.get<float>(name(), "interval", 1));
+    m_dateformat = string_util::trim(m_conf.get<string>(name(), "date", ""), '"');
+    m_dateformat_alt = string_util::trim(m_conf.get<string>(name(), "date-alt", ""), '"');
+    m_timeformat = string_util::trim(m_conf.get<string>(name(), "time", ""), '"');
+    m_timeformat_alt = string_util::trim(m_conf.get<string>(name(), "time-alt", ""), '"');
 
-    m_formatter->add(DEFAULT_FORMAT, TAG_DATE, {TAG_DATE});
+    if (m_dateformat.empty() && m_timeformat.empty()) {
+      throw module_error("No date or time format specified");
+    }
 
-    m_format = m_conf.get<string>(name(), "date");
-    m_formatalt = m_conf.get<string>(name(), "date-alt", "");
+    m_interval = chrono::duration<double>(m_conf.get<double>(name(), "interval", 1.0));
+
+    m_formatter->add(DEFAULT_FORMAT, TAG_LABEL, {TAG_LABEL, TAG_DATE});
+
+    if (m_formatter->has(TAG_DATE)) {
+      m_log.warn("%s: The format tag `<date>` is deprecated, use `<label>` instead.", name());
+
+      m_formatter->get(DEFAULT_FORMAT)->value =
+          string_util::replace_all(m_formatter->get(DEFAULT_FORMAT)->value, TAG_DATE, TAG_LABEL);
+    }
+
+    if (m_formatter->has(TAG_LABEL)) {
+      m_label = load_optional_label(m_conf, name(), "label", "%date%");
+    }
   }
 
   bool date_module::update() {
-    if (!m_formatter->has(TAG_DATE)) {
+    auto time = std::time(nullptr);
+
+    auto date_format = m_toggled ? m_dateformat_alt : m_dateformat;
+    char date_buffer[64]{'\0'};
+    strftime(date_buffer, sizeof(date_buffer), date_format.c_str(), localtime(&time));
+
+    auto time_format = m_toggled ? m_timeformat_alt : m_timeformat;
+    char time_buffer[64]{'\0'};
+    strftime(time_buffer, sizeof(time_buffer), time_format.c_str(), localtime(&time));
+
+    bool date_changed{strncmp(date_buffer, m_date.c_str(), sizeof(date_buffer)) != 0};
+    bool time_changed{strncmp(time_buffer, m_time.c_str(), sizeof(time_buffer)) != 0};
+
+    if (!date_changed && !time_changed) {
       return false;
     }
 
-    auto time = std::time(nullptr);
-    auto date_format = m_toggled ? m_formatalt : m_format;
-    char buffer[256] = {'\0'};
+    m_date = string{date_buffer};
+    m_time = string{time_buffer};
 
-    std::strftime(buffer, sizeof(buffer), date_format.c_str(), std::localtime(&time));
-
-    if (std::strncmp(buffer, m_buffer, sizeof(buffer)) == 0) {
-      return false;
-    } else {
-      std::memmove(m_buffer, buffer, sizeof(buffer));
+    if (m_label) {
+      m_label->reset_tokens();
+      m_label->replace_token("%date%", m_date);
+      m_label->replace_token("%time%", m_time);
     }
 
     return true;
   }
 
   bool date_module::build(builder* builder, const string& tag) const {
-    if (tag != TAG_DATE) {
+    if (tag == TAG_LABEL) {
+      if (!m_dateformat_alt.empty() || !m_timeformat_alt.empty()) {
+        builder->cmd(mousebtn::LEFT, EVENT_TOGGLE);
+        builder->node(m_label);
+        builder->cmd_close();
+      } else {
+        builder->node(m_label);
+      }
+    } else {
       return false;
     }
 
-    if (!m_formatalt.empty()) {
-      m_builder->cmd(mousebtn::LEFT, EVENT_TOGGLE);
-    }
-
-    builder->node(m_buffer);
-
     return true;
   }
 
   bool date_module::handle_event(string cmd) {
-    if (cmd == EVENT_TOGGLE) {
-      m_toggled = !m_toggled;
-      wakeup();
+    if (cmd != EVENT_TOGGLE) {
+      return false;
     }
 
-    return cmd == EVENT_TOGGLE;
-  }
-
-  bool date_module::receive_events() const {
+    m_toggled = !m_toggled;
+    wakeup();
     return true;
   }
 }