wip: Separate source from definitions

2026-03-17 05:37:49 +00:00 · 2016-11-02 20:22:45 +01:00
parent 566f33b8b2
commit e7cb345cec
134 changed files with 10292 additions and 9181 deletions
--- a/src/modules/script.cpp
+++ b/src/modules/script.cpp
@ -0,0 +1,130 @@
+#include "modules/script.hpp"
+
+LEMONBUDDY_NS
+
+namespace modules {
+  void script_module::setup() {
+    m_formatter->add(DEFAULT_FORMAT, TAG_OUTPUT, {TAG_OUTPUT});
+
+    // Load configuration values
+
+    REQ_CONFIG_VALUE(name(), m_exec, "exec");
+    GET_CONFIG_VALUE(name(), m_tail, "tail");
+    GET_CONFIG_VALUE(name(), m_maxlen, "maxlen");
+    GET_CONFIG_VALUE(name(), m_ellipsis, "ellipsis");
+
+    m_actions[mousebtn::LEFT] = m_conf.get<string>(name(), "click-left", "");
+    m_actions[mousebtn::MIDDLE] = m_conf.get<string>(name(), "click-middle", "");
+    m_actions[mousebtn::RIGHT] = m_conf.get<string>(name(), "click-right", "");
+    m_actions[mousebtn::SCROLL_UP] = m_conf.get<string>(name(), "scroll-up", "");
+    m_actions[mousebtn::SCROLL_DOWN] = m_conf.get<string>(name(), "scroll-down", "");
+
+    m_interval = interval_t{m_conf.get<float>(name(), "interval", m_tail ? 0.0f : 2.0f)};
+  }
+
+  void script_module::stop() {
+    if (m_command && m_command->is_running()) {
+      m_log.warn("%s: Stopping shell command", name());
+      m_command->terminate();
+    }
+    wakeup();
+    event_module::stop();
+  }
+
+  void script_module::idle() {
+    if (!m_tail)
+      sleep(m_interval);
+    else if (!m_command || !m_command->is_running())
+      sleep(m_interval);
+  }
+
+  bool script_module::has_event() {
+    // Non tail commands should always run
+    if (!m_tail)
+      return true;
+
+    try {
+      if (!m_command || !m_command->is_running()) {
+        auto exec = string_util::replace_all(m_exec, "%counter%", to_string(++m_counter));
+        m_log.trace("%s: Executing '%s'", name(), exec);
+
+        m_command = command_util::make_command(exec);
+        m_command->exec(false);
+      }
+    } catch (const std::exception& err) {
+      m_log.err("%s: %s", name(), err.what());
+      throw module_error("Failed to execute tail command, stopping module...");
+    }
+
+    if (!m_command)
+      return false;
+
+    if ((m_output = m_command->readline()) == m_prev)
+      return false;
+
+    m_prev = m_output;
+
+    return true;
+  }
+
+  bool script_module::update() {
+    // Tailing commands always update
+    if (m_tail)
+      return true;
+
+    try {
+      if (m_command && m_command->is_running()) {
+        m_log.warn("%s: Previous shell command is still running...", name());
+        return false;
+      }
+
+      auto exec = string_util::replace_all(m_exec, "%counter%", to_string(++m_counter));
+      m_log.trace("%s: Executing '%s'", name(), exec);
+
+      m_command = command_util::make_command(exec);
+      m_command->exec();
+      m_command->tail([&](string output) { m_output = output; });
+    } catch (const std::exception& err) {
+      m_log.err("%s: %s", name(), err.what());
+      throw module_error("Failed to execute command, stopping module...");
+    }
+
+    if (m_output == m_prev)
+      return false;
+
+    m_prev = m_output;
+    return true;
+  }
+
+  string script_module::get_output() {
+    if (m_output.empty())
+      return " ";
+
+    // Truncate output to the defined max length
+    if (m_maxlen > 0 && m_output.length() > m_maxlen) {
+      m_output.erase(m_maxlen);
+      m_output += m_ellipsis ? "..." : "";
+    }
+
+    auto counter_str = to_string(m_counter);
+    OUTPUT_ACTION(mousebtn::LEFT);
+    OUTPUT_ACTION(mousebtn::MIDDLE);
+    OUTPUT_ACTION(mousebtn::RIGHT);
+    OUTPUT_ACTION(mousebtn::SCROLL_UP);
+    OUTPUT_ACTION(mousebtn::SCROLL_DOWN);
+    m_builder->node(module::get_output());
+
+    return m_builder->flush();
+  }
+
+  bool script_module::build(builder* builder, string tag) const {
+    if (tag == TAG_OUTPUT) {
+      builder->node(m_output);
+      return true;
+    } else {
+      return false;
+    }
+  }
+}
+
+LEMONBUDDY_NS_END