crabtalk-command 0.0.19

//! Linux systemd service management.

use anyhow::Result;
use wcore::paths::LOGS_DIR;

/// Embedded systemd service template.
pub const TEMPLATE: &str = include_str!("systemd.service");

/// Check if a systemd service is installed.
pub fn is_installed(label: &str) -> bool {
    dirs::home_dir()
        .map(|h| {
            h.join(format!(".config/systemd/user/{label}.service"))
                .exists()
        })
        .unwrap_or(false)
}

/// Install a systemd service.
pub fn install(rendered: &str, label: &str) -> Result<()> {
    let unit_name = format!("{label}.service");

    let unit_dir = dirs::home_dir()
        .ok_or_else(|| anyhow::anyhow!("cannot determine home directory"))?
        .join(".config/systemd/user");
    std::fs::create_dir_all(&unit_dir)?;
    std::fs::create_dir_all(&*LOGS_DIR)?;

    let unit_path = unit_dir.join(&unit_name);
    std::fs::write(&unit_path, rendered)?;
    println!("wrote {}", unit_path.display());

    let status = std::process::Command::new("systemctl")
        .args(["--user", "enable", "--now", &unit_name])
        .status()?;
    if status.success() {
        println!("service enabled and started");
    } else {
        anyhow::bail!("systemctl enable failed (exit {})", status);
    }
    Ok(())
}

/// Uninstall a systemd service.
pub fn uninstall(label: &str) -> Result<()> {
    let unit_name = format!("{label}.service");

    let unit_path = dirs::home_dir()
        .ok_or_else(|| anyhow::anyhow!("cannot determine home directory"))?
        .join(".config/systemd/user")
        .join(&unit_name);

    if !unit_path.exists() {
        anyhow::bail!("service not installed ({})", unit_path.display());
    }

    let status = std::process::Command::new("systemctl")
        .args(["--user", "disable", "--now", &unit_name])
        .status()?;
    if !status.success() {
        eprintln!("warning: systemctl disable exited with {}", status);
    }

    std::fs::remove_file(&unit_path)?;
    println!("service uninstalled");
    Ok(())
}