ctry 1.1.9

Retry CLI commands
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

use std::process::{Command, Stdio};
use std::thread;
use std::time::Duration;

#[derive(Debug)]
pub struct RetryConfig<'a> {
	pub max: u32,
	pub interval: Duration,
	pub expected_exitcode: i32,
	pub quiet: bool,
	pub cmd: Vec<&'a str>,
}

pub fn retry(config: RetryConfig) {
	let mut i = 1;
	while i <= config.max || config.max == 0 {
		let status = match Command::new(config.cmd[0])
			.args(&config.cmd[1..config.cmd.len()])
			.stdin(if config.quiet { Stdio::null() } else { Stdio::inherit() })
			.stdout(if config.quiet { Stdio::null() } else { Stdio::inherit() })
			.stderr(if config.quiet { Stdio::null() } else { Stdio::inherit() })
			.status()
		{
			Ok(s) => s,
			Err(err) => panic!("Failed to execute command: {}", err),
		};

		match status.code() {
			Some(code) if code == config.expected_exitcode => {
				println!("Successfully ran command, finished ctry");
				break;
			}
			Some(code) => println!("[Try {}] Command failed with exit code {}", i, code),
			None => println!("[Try {}] Command was termianted by a signal", i),
		}

		if i != config.max {
			thread::sleep(config.interval);
		}
		i += 1;
	}
}