From b87c19df845d661d3f7a7e82ee411470c20ec032 Mon Sep 17 00:00:00 2001
From: Moritz Ruth <dev@moritzruth.de>
Date: Tue, 21 Feb 2023 18:29:17 +0100
Subject: [PATCH] And again

---
 .../{Run.xml => Run__dry_.xml}                |  2 +-
 .idea/runConfigurations/Run__not_dry_.xml     | 19 +++++
 README.md                                     |  2 +-
 src/modules/power.rs                          | 84 ++++++++++++-------
 4 files changed, 77 insertions(+), 30 deletions(-)
 rename .idea/runConfigurations/{Run.xml => Run__dry_.xml} (84%)
 create mode 100644 .idea/runConfigurations/Run__not_dry_.xml

diff --git a/.idea/runConfigurations/Run.xml b/.idea/runConfigurations/Run__dry_.xml
similarity index 84%
rename from .idea/runConfigurations/Run.xml
rename to .idea/runConfigurations/Run__dry_.xml
index f1a7a17..4b2f4b8 100644
--- a/.idea/runConfigurations/Run.xml
+++ b/.idea/runConfigurations/Run__dry_.xml
@@ -1,5 +1,5 @@
 <component name="ProjectRunConfigurationManager">
-  <configuration default="false" name="Run" type="CargoCommandRunConfiguration" factoryName="Cargo Command" nameIsGenerated="true">
+  <configuration default="false" name="Run (dry)" type="CargoCommandRunConfiguration" factoryName="Cargo Command">
     <option name="command" value="run -F dry_run" />
     <option name="workingDirectory" value="file://$PROJECT_DIR$" />
     <option name="emulateTerminal" value="false" />
diff --git a/.idea/runConfigurations/Run__not_dry_.xml b/.idea/runConfigurations/Run__not_dry_.xml
new file mode 100644
index 0000000..08fc95d
--- /dev/null
+++ b/.idea/runConfigurations/Run__not_dry_.xml
@@ -0,0 +1,19 @@
+<component name="ProjectRunConfigurationManager">
+  <configuration default="false" name="Run (not dry)" type="CargoCommandRunConfiguration" factoryName="Cargo Command">
+    <option name="command" value="run" />
+    <option name="workingDirectory" value="file://$PROJECT_DIR$" />
+    <option name="emulateTerminal" value="false" />
+    <option name="channel" value="DEFAULT" />
+    <option name="requiredFeatures" value="false" />
+    <option name="allFeatures" value="false" />
+    <option name="withSudo" value="false" />
+    <option name="buildTarget" value="REMOTE" />
+    <option name="backtrace" value="SHORT" />
+    <envs>
+      <env name="RUST_LOG" value="hassliebe" />
+    </envs>
+    <option name="isRedirectInput" value="false" />
+    <option name="redirectInputPath" value="" />
+    <method v="2" />
+  </configuration>
+</component>
\ No newline at end of file
diff --git a/README.md b/README.md
index d8366b1..d8e53f9 100644
--- a/README.md
+++ b/README.md
@@ -3,7 +3,7 @@
 
 ## Features
 - [ ] Triggers
-  - [ ] shutdown, hibernate, suspend, reboot
+  - [ ] shutdown, reboot
   - [ ] Custom commands
 - [ ] Notifications
   - [ ] Actions
diff --git a/src/modules/power.rs b/src/modules/power.rs
index 0792977..9342e55 100644
--- a/src/modules/power.rs
+++ b/src/modules/power.rs
@@ -1,57 +1,54 @@
-use crate::modules::ModuleContext;
 use anyhow::Result;
 use rumqttc::QoS;
 use tokio::process::Command;
 
+use super::ModuleContext;
+
+const MODULE_ID: &str = "power";
+const BUTTON_TRIGGER_TEXT: &str = "press";
+
 pub(crate) async fn init(context: &mut ModuleContext<'_>) -> Result<()> {
     log::info!("Initializing…");
 
-    let module_id = "power";
-    let sub_id = "shutdown";
+    init_command_button(context, "shutdown", "Shutdown", "shutdown -h now").await?;
+    init_command_button(context, "reboot", "Reboot", "shutdown -r now").await?;
 
-    let trigger_text = "press";
+    Ok(())
+}
 
-    let entity_id = context.get_entity_id(module_id, sub_id);
-    let entity_id = entity_id.as_str();
+async fn init_command_button(
+    context: &mut ModuleContext<'_>,
+    sub_id: &str,
+    name: &str,
+    command: impl Into<String>,
+) -> Result<()> {
+    let command = command.into();
 
-    let command_topic = context.get_mqtt_topic("button", entity_id, "trigger");
+    let entity_id = context.get_entity_id(MODULE_ID, sub_id);
+    let command_topic = context.get_mqtt_topic("button", &entity_id, "trigger");
 
     context
         .mqtt_client
         .publish(
-            context.get_mqtt_topic("button", entity_id, "config"),
+            context.get_mqtt_topic("button", &entity_id, "config"),
             QoS::AtLeastOnce,
             true,
             json::stringify(json::object! {
                 "availability_topic": context.mqtt_availability_topic,
                 "command_topic": command_topic.as_str(),
                 "device": context.mqtt_discovery_device_object.clone(),
-                // "device_class": "restart",
                 "icon": "mdi:power",
-                "name": "Shutdown",
-                "payload_press": trigger_text,
-                "object_id": entity_id,
-                "unique_id": entity_id
+                "name": name,
+                "payload_press": BUTTON_TRIGGER_TEXT,
+                "object_id": entity_id.as_str(),
+                "unique_id": entity_id.as_str()
             }),
         )
         .await?;
 
     context.subscribe_mqtt_topic(command_topic, move |text| {
-        if text == trigger_text {
-            if cfg!(feature = "dry_run") {
-                log::info!("Received shutdown command (dry run)");
-            } else {
-                log::info!("Shutting down…");
-
-                tokio::spawn(async {
-                    Command::new("shutdown")
-                        .args(["-h", "now"])
-                        .status()
-                        .await?;
-
-                    anyhow::Ok(())
-                });
-            }
+        if text == BUTTON_TRIGGER_TEXT {
+            run_command(command.clone());
         }
 
         Ok(())
@@ -59,3 +56,34 @@ pub(crate) async fn init(context: &mut ModuleContext<'_>) -> Result<()> {
 
     Ok(())
 }
+
+fn run_command(command: String) {
+    tokio::spawn(async move {
+        let is_dry_run = cfg!(feature = "dry_run");
+
+        let status = {
+            log::info!(
+                "Executing command{}: {}",
+                if is_dry_run { "(dry run)" } else { "" },
+                command
+            );
+
+            let mut command_parts = command.split(' ').collect::<Vec<_>>();
+            let mut actual_command = Command::new(command_parts[0]);
+            command_parts.remove(0);
+            actual_command.args(command_parts);
+
+            if is_dry_run {
+                None
+            } else {
+                Some(actual_command.status())
+            }
+        };
+
+        if let Some(f) = status {
+            f.await?;
+        }
+
+        anyhow::Ok(())
+    });
+}