oxwm

use super::Block;

use anyhow::Result;

use std::process::Command;

use std::time::Duration;

pub struct ShellBlock {

    format: String,

    command: String,

    interval: Duration,

    color: u32,

}

impl ShellBlock {

    pub fn new(format: &str, command: &str, interval_secs: u64, color: u32) -> Self {

        Self {

            format: format.to_string(),

            command: command.to_string(),

            interval: Duration::from_secs(interval_secs),

            color,

        }

    }

}

impl Block for ShellBlock {

    fn content(&mut self) -> Result<String> {

        let output = Command::new("sh").arg("-c").arg(&self.command).output()?;

        let result = String::from_utf8_lossy(&output.stdout).trim().to_string();

        Ok(self.format.replace("{}", &result))

    }

    fn interval(&self) -> Duration {

        self.interval

    }

    fn color(&self) -> u32 {

        self.color

    }

}