From cb2f5c154c1177da05e16f4c7c62312d6c80e352 Mon Sep 17 00:00:00 2001
From: Alex <alexisr245@gmail.com>
Date: Wed, 25 Mar 2020 22:25:27 +0100
Subject: [PATCH] feat(custon): restart_interval for continuous script

---
 man/waybar-custom.5.scd |  6 ++++++
 src/modules/custom.cpp  | 11 ++++++++++-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/man/waybar-custom.5.scd b/man/waybar-custom.5.scd
index 0ae7f4cd..92eda6cd 100644
--- a/man/waybar-custom.5.scd
+++ b/man/waybar-custom.5.scd
@@ -33,6 +33,12 @@ Addressed by *custom/<name>*
 	You can update it manually with a signal. If no *interval* is defined,
 	it is assumed that the out script loops it self.
 
+*restart_interval*: ++
+	typeof: integer ++
+	The restart interval (in seconds).
+	Can't be used with the *interval* option, so only with continuous scripts.
+	Once the scripts exit, it'll be re-executed after the *restart_interval*.
+
 *signal*: ++
 	typeof: integer ++
 	The signal number used to update the module.
diff --git a/src/modules/custom.cpp b/src/modules/custom.cpp
index ca095088..cbc865d6 100644
--- a/src/modules/custom.cpp
+++ b/src/modules/custom.cpp
@@ -55,12 +55,21 @@ void waybar::modules::Custom::continuousWorker() {
         exit_code = WEXITSTATUS(util::command::close(fp_, pid_));
         fp_ = nullptr;
       }
-      thread_.stop();
       if (exit_code != 0) {
         output_ = {exit_code, ""};
         dp.emit();
         spdlog::error("{} stopped unexpectedly, is it endless?", name_);
       }
+      if (config_["restart_interval"].isUInt()) {
+        pid_ = -1;
+        fp_ = util::command::open(cmd, pid_);
+        if (!fp_) {
+          throw std::runtime_error("Unable to open " + cmd);
+        }
+        thread_.sleep_for(std::chrono::seconds(config_["restart_interval"].asUInt());
+      } else {
+        thread_.stop();
+      }
       return;
     }
     std::string output = buff;