Volume plugin

dim_plugins/volume/src/config.rs

@@ -0,0 +1,37 @@
+use std::path::Path;
+use anyhow::Result;
+use serde::Deserialize;
+
+const DEFAULT_CFG: &'static str = include_str!("../config.template.toml");
+
+#[derive(Debug, Deserialize, Clone)]
+pub struct Config {
+    pub commands: ConfigCommands,
+}
+
+#[derive(Debug, Deserialize, Clone)]
+pub struct ConfigCommands {
+    pub get_volume: String,
+    pub get_volume_regex: String,
+}
+
+impl Config {
+    pub fn parse(p: &Path) -> Result<Self> {
+        if !p.exists() {
+            std::fs::write(p, DEFAULT_CFG)?;
+        }
+
+        let c = std::fs::read_to_string(p)?;
+        Ok(toml::from_str(c.as_str())?)
+    }
+
+    pub fn default() -> Self {
+        Self {
+            commands: ConfigCommands { 
+                get_volume: String::new(),
+                get_volume_regex: String::new()
+            } 
+        }
+    }
+
+}

dim_plugins/volume/src/lib.rs

@@ -0,0 +1,68 @@
+use std::{fmt::Write, path::PathBuf, process::Stdio};
+use dim_sdk::{plugin_info, Context, DimPlugin};
+use std::fs::read_to_string;
+
+mod config;
+
+plugin_info!(
+    Plug,                    // Your main global structs name that implements `DimPlugin`
+    "volume",                // Plugin name
+    "1.0.0",                 // Plugin Version (leave empty for none)
+    "GPLv3"                  // Plugin license (leave empty for none)
+    );
+
+#[derive(Debug, Clone)]
+struct Plug {
+    cfg: config::Config,
+}
+
+impl Plug {
+    pub fn new() -> Self {
+        Self {
+            cfg: config::Config::default()
+        }
+    }
+}
+
+impl DimPlugin for Plug {
+    fn init(&mut self, ctx: Context) {
+        match config::Config::parse(&ctx.config_dir.join("config.toml")) {
+            Ok(c) => self.cfg = c,
+            Err(e) => {
+                eprintln!("ERROR: Failed to open config file: {e}");
+                // TODO: Add function to disable the plugin
+                return;
+            }
+        }
+    }
+    fn poll(&mut self, f: &mut dim_sdk::CBuffer) -> dim_sdk::Result<()> {
+        let mut proc = {
+                let mut p = std::process::Command::new("sh");
+                p.arg("-c");
+                p.arg(&self.cfg.commands.get_volume);
+                p.stdout(Stdio::piped());
+                p
+            };
+
+        let output = String::from_utf8(proc.output()?.stdout)?;
+        let re = regex::Regex::new(&self.cfg.commands.get_volume_regex)?;
+    
+        if let Some(caps) = re.captures(&output) {
+            let volume = &caps["vol"];
+            write!(f, "Vol: {volume}%")?;
+        }
+        Ok(())
+    }
+    fn pre_reload(&mut self) {
+    }
+    fn post_reload(&mut self) {
+
+    }
+    fn free(&mut self) {
+
+    }
+}
+
+
+
+