knope 0.22.4

A command line tool for automating common development tasks
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94

use std::borrow::Cow;

use indexmap::IndexMap;
use knope_config::{Template, Variable};
use miette::Diagnostic;
use tracing::info;

use crate::{RunType, state::State, variables, variables::replace_variables};

/// Run the command string `command` in the current shell after replacing the keys of `variables`
/// with the values that the [`Variable`]s represent.
pub(crate) fn run_command(
    state: RunType<State>,
    mut command: String,
    shell: bool,
    variables: Option<IndexMap<Cow<'static, str>, Variable>>,
) -> Result<RunType<State>, Error> {
    let (run_type, mut state) = state.take();
    command = replace_variables(&Template::new(command, variables), &mut state)?;
    if let RunType::DryRun(()) = run_type {
        info!("Would run {command}");
        return Ok(run_type.of(state));
    }
    let status = if shell {
        execute::shell(command).status()?
    } else {
        execute::command(command).status()?
    };
    if status.success() {
        return Ok(run_type.of(state));
    }
    Err(Error::Command(status))
}

#[derive(Debug, Diagnostic, thiserror::Error)]
pub(crate) enum Error {
    #[error("Command returned non-zero exit code")]
    #[diagnostic(
        code(command::failed),
        help("The command failed to execute. Try running it manually to get more information.")
    )]
    Command(std::process::ExitStatus),
    #[error("I/O error: {0}")]
    #[diagnostic(code(command::io))]
    Io(#[from] std::io::Error),
    #[error(transparent)]
    #[diagnostic(transparent)]
    Variables(#[from] variables::Error),
}

#[cfg(test)]
#[allow(clippy::unwrap_used)]
mod test_run_command {

    use super::*;
    use crate::State;

    #[test]
    fn test() {
        let command = "echo \"hello\"";
        let result = run_command(
            RunType::Real(State::new(
                None,
                None,
                None,
                Vec::new(),
                Vec::new(),
                Vec::new(),
                false,
            )),
            command.to_string(),
            false,
            None,
        );

        assert!(result.is_ok());

        let result = run_command(
            RunType::Real(State::new(
                None,
                None,
                None,
                Vec::new(),
                Vec::new(),
                Vec::new(),
                false,
            )),
            String::from("exit 1"),
            false,
            None,
        );
        assert!(result.is_err());
    }
}