feat: timeout if the execution takes too long

src/shell.rs

@@ -1,16 +1,31 @@
 use std::io::prelude::*;
 use std::process::exit;
 
+
+use std::sync::mpsc::channel;
+use std::thread;
+use std::time::Duration;
+
 pub const PRIVILEGE_LIST: [&str; 2] = ["sudo", "doas"];
 
 pub fn command_output(shell: &str, command: &str) -> String {
-	let output = std::process::Command::new(shell)
+
+	let (sender, receiver) = channel();
+
+	let _shell = shell.to_owned();
+	let _command = command.to_owned();
+	thread::spawn(move || {
+		sender.send(std::process::Command::new(_shell)
 			.arg("-c")
-		.arg(command)
+			.arg(_command)
 			.env("LC_ALL", "C")
 			.output()
-		.expect("failed to execute process");
+			.expect("failed to execute process"))
+			.expect("failed to send output");
+	});
 
+	match receiver.recv_timeout(Duration::from_secs(3)) {
+		Ok(output) => {
 			String::from_utf8_lossy(&output.stderr)
 				.to_string()
 				.split_whitespace()
@@ -18,6 +33,13 @@ pub fn command_output(shell: &str, command: &str) -> String {
 				.join(" ")
 				.to_lowercase()
 		}
+		Err(_) => {
+			use colored::*;
+			eprintln!("Timeout while executing command: {}", command.red());
+			exit(1);
+		}
+	}
+}
 
 fn last_command(shell: &str) -> String {
 	let last_command = std::env::var("_PR_LAST_COMMAND").expect("No _PR_LAST_COMMAND in environment. Did you aliased the command with the correct argument?");