pay-respects/src/modes.rs

use crate::shell::{command_output, get_shell, PRIVILEGE_LIST};
use crate::style::highlight_difference;
use crate::suggestions::{split_command, suggest_typo};
use crate::{shell, suggestions};
use colored::Colorize;

pub fn suggestion() {
	let shell = get_shell();
	let mut last_command = shell::last_command(&shell).trim().to_string();
	last_command = shell::expand_alias(&shell, &last_command);
	let mut error_msg = {
		let error_msg = std::env::var("_PR_ERROR_MSG");
		if let Ok(error_msg) = error_msg {
			error_msg
		} else {
			command_output(&shell, &last_command)
		}
	};

	error_msg = error_msg
		.split_whitespace()
		.collect::<Vec<&str>>()
		.join(" ");

	loop {
		let suggestion = {
			let command = suggestions::suggest_command(&shell, &last_command, &error_msg);
			if command.is_none() {
				break;
			};

			let mut command = command.unwrap();
			shell::shell_syntax(&shell, &mut command);
			command
		};

		let highlighted_suggestion = {
			let difference = highlight_difference(&shell, &suggestion, &last_command);
			if difference.is_none() {
				break;
			};
			difference.unwrap()
		};

		let execution =
			suggestions::confirm_suggestion(&shell, &suggestion, &highlighted_suggestion);
		if execution.is_ok() {
			return;
		} else {
			last_command = suggestion;
			error_msg = execution.err().unwrap();
			error_msg = error_msg
				.split_whitespace()
				.collect::<Vec<&str>>()
				.join(" ");

			let retry_message = format!("{}...", t!("retry"));

			eprintln!("\n{}\n", retry_message.cyan().bold());
		}
	}
	eprintln!("{}: {}\n", t!("no-suggestion"), last_command.red());
	eprintln!(
		"{}\n{}",
		t!("contribute"),
		option_env!("CARGO_PKG_REPOSITORY").unwrap_or("https://github.com/iffse/pay-respects/")
	);
}

pub fn cnf() {
	let shell = get_shell();
	let mut last_command = shell::last_command(&shell).trim().to_string();
	last_command = shell::expand_alias(&shell, &last_command);

	let mut split_command = split_command(&last_command);
	let executable = match PRIVILEGE_LIST.contains(&split_command[0].as_str()) {
		true => split_command.get(1).expect(&t!("no-command")).as_str(),
		false => split_command.first().expect(&t!("no-command")).as_str(),
	};

	let best_match = suggest_typo(&[executable.to_owned()], vec!["path".to_string()]);
	if best_match == executable {
		eprintln!("{}: command not found: {}", shell, executable);
		return;
	}
	match PRIVILEGE_LIST.contains(&split_command[0].as_str()) {
		true => {
			split_command[1] = best_match;
		}
		false => {
			split_command[0] = best_match;
		}
	}
	let suggestion = split_command.join(" ");

	let highlighted_suggestion = highlight_difference(&shell, &suggestion, &last_command).unwrap();
	let _ = suggestions::confirm_suggestion(&shell, &suggestion, &highlighted_suggestion);
}