pleme-fleet 0.1.3

NixOS fleet lifecycle CLI with DAG workflow orchestration
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

use anyhow::{bail, Result};
use std::process::Command;

use crate::commands::utils::{log_info, log_warning};
use crate::config::FleetConfig;
use crate::registry::Node;

/// Run the pre-hook for a command, if configured. Aborts on failure.
pub fn run_pre(
    config: &FleetConfig,
    command_name: &str,
    node_name: &str,
    node: &Node,
) -> Result<()> {
    if let Some(hook) = config.hooks.get(command_name) {
        if let Some(ref script) = hook.pre {
            log_info(&format!(
                "Running pre-{} hook for {}",
                command_name, node_name
            ));
            let status = Command::new("sh")
                .arg("-c")
                .arg(script)
                .env("FLEET_NODE", node_name)
                .env("FLEET_HOST", &node.hostname)
                .env("FLEET_USER", &node.ssh_user)
                .status()?;
            if !status.success() {
                bail!(
                    "Pre-{} hook failed for {} (exit {})",
                    command_name,
                    node_name,
                    status.code().unwrap_or(-1)
                );
            }
        }
    }
    Ok(())
}

/// Run the post-hook for a command, if configured. Warns on failure but does not abort.
pub fn run_post(
    config: &FleetConfig,
    command_name: &str,
    node_name: &str,
    node: &Node,
) {
    if let Some(hook) = config.hooks.get(command_name) {
        if let Some(ref script) = hook.post {
            log_info(&format!(
                "Running post-{} hook for {}",
                command_name, node_name
            ));
            match Command::new("sh")
                .arg("-c")
                .arg(script)
                .env("FLEET_NODE", node_name)
                .env("FLEET_HOST", &node.hostname)
                .env("FLEET_USER", &node.ssh_user)
                .status()
            {
                Ok(status) if !status.success() => {
                    log_warning(&format!(
                        "Post-{} hook failed for {} (exit {})",
                        command_name,
                        node_name,
                        status.code().unwrap_or(-1)
                    ));
                }
                Err(e) => {
                    log_warning(&format!(
                        "Post-{} hook error for {}: {}",
                        command_name, node_name, e
                    ));
                }
                _ => {}
            }
        }
    }
}