git.delta.rocks / remowt / refs/commits / 61cfb317f078

1use std::process::Stdio;23use tokio::io::AsyncWriteExt;4use tokio::process::Command;5use tracing::trace;67use crate::{Error, Prompter, Result, Source};89#[derive(Clone)]10pub struct RofiPrompter;1112fn fixup_prompt(prompt: &str) -> &str {13	// Rofi always appends such suffix14	prompt.strip_suffix(": ").unwrap_or(prompt)15}1617fn rofi_command() -> Command {18	Command::new(option_env!("ROFI").unwrap_or("rofi"))19}2021impl Prompter for RofiPrompter {22	async fn prompt_enum(23		&self,24		prompt: &str,25		description: &str,26		variants: &[&str],27		source: &[Source],28	) -> Result<u32> {29		trace!("rofi radio");30		let mut cmd = rofi_command();31		let mesg = if source.is_empty() {32			description.to_owned()33		} else {34			let mut out = format!("{description}\n\n<b>Requested on ",);35			for (i, s) in source.iter().enumerate() {36				if i != 0 {37					out.push_str(" -> ");38				}39				out.push_str(&s.to_string());40			}41			out.push_str("</b>");42			out43		};44		cmd.args([45			"-dmenu",46			"-mesg",47			&mesg,48			"-sync",49			"-no-custom",50			"-p",51			fixup_prompt(prompt),52			"-format",53			"i",54			"-markup-rows",55		]);56		cmd.stdin(Stdio::piped());57		cmd.stdout(Stdio::piped());58		cmd.kill_on_drop(true);59		let mut child = cmd60			.spawn()61			.map_err(|e| Error::InputError(format!("failed to spawn rofi: {e}")))?;6263		let mut stdin = child.stdin.take().expect("stdin is piped");64		for var in variants {65			stdin66				.write_all(var.replace('\n', " ").as_bytes())67				.await68				.map_err(|e| Error::InputError(format!("failed to write rofi variants: {e}")))?;69			stdin70				.write_all(b"\n")71				.await72				.map_err(|e| Error::InputError(format!("failed to write rofi variants: {e}")))?;73		}74		// write_all already flushes, just to be sure.75		let _ = stdin.flush().await;76		drop(stdin);7778		let out = child79			.wait_with_output()80			.await81			.map_err(|e| Error::InputError(format!("failed to wait for rofi: {e}")))?;82		match out.status.code() {83			Some(0) => {}84			Some(1) => return Err(Error::Cancel),85			other => {86				return Err(Error::InputError(format!(87					"rofi exited with status {other:?}"88				)));89			}90		}91		let stdout = out92			.stdout93			.strip_suffix(b"\n")94			.unwrap_or(&out.stdout)95			.to_owned();9697		let id: u32 = String::from_utf8(stdout)98			.map_err(|e| Error::InputError(format!("rofi produced invalid output: {e}")))?99			.parse()100			.map_err(|e| Error::InputError(format!("rofi produced invalid output: {e}")))?;101		if id as usize >= variants.len() {102			return Err(Error::InputError("invalid rofi response".to_owned()));103		}104105		Ok(id)106	}107108	async fn prompt_text(109		&self,110		echo: bool,111		prompt: &str,112		description: &str,113		source: &[Source],114	) -> Result<String> {115		trace!("rofi text");116		let mut cmd = rofi_command();117		let mesg = if source.is_empty() {118			description.to_owned()119		} else {120			let mut out = format!("{description}\n\n<b>Requested on ",);121			for (i, s) in source.iter().enumerate() {122				if i != 0 {123					out.push_str(" -> ");124				}125				out.push_str(&s.to_string());126			}127			out.push_str("</b>");128			out129		};130		cmd.args(["-dmenu", "-mesg", &mesg, "-p", fixup_prompt(prompt)]);131		if !echo {132			cmd.arg("-password");133		}134		cmd.stdin(Stdio::null());135		cmd.stdout(Stdio::piped());136		cmd.kill_on_drop(true);137		let child = cmd138			.spawn()139			.map_err(|e| Error::InputError(format!("failed to spawn rofi: {e}")))?;140141		let out = child142			.wait_with_output()143			.await144			.map_err(|e| Error::InputError(format!("failed to wait for rofi: {e}")))?;145		match out.status.code() {146			Some(0) => {}147			Some(1) => return Err(Error::Cancel),148			other => {149				return Err(Error::InputError(format!(150					"rofi exited with status {other:?}"151				)));152			}153		}154		let stdout = out155			.stdout156			.strip_suffix(b"\n")157			.unwrap_or(&out.stdout)158			.to_owned();159160		Ok(String::from_utf8_lossy(&stdout).to_string())161	}162163	async fn display_text(&self, error: bool, description: &str, source: &[Source]) -> Result<()> {164		trace!("rofi display");165		let mut cmd = rofi_command();166		let mut mesg = if source.is_empty() {167			description.to_owned()168		} else {169			let mut out = format!("{description}\n\n<b>Coming from ",);170			for s in source.iter() {171				out.push_str(&s.to_string());172			}173			out.push_str("</b>");174			out175		};176		if error {177			mesg.insert_str(0, "<span color=\"red\">");178			mesg.push_str("</span>");179		}180		cmd.args(["-e", &mesg, "-markup"]);181		cmd.stdin(Stdio::null());182		cmd.stdout(Stdio::null());183		cmd.kill_on_drop(true);184		let mut child = cmd185			.spawn()186			.map_err(|e| Error::InputError(format!("failed to spawn rofi: {e}")))?;187188		child189			.wait()190			.await191			.map_err(|e| Error::InputError(format!("failed to wait for rofi: {e}")))?;192193		Ok(())194	}195}196197#[cfg(test)]198mod tests {199	use std::borrow::Cow;200201	use crate::rofi::RofiPrompter;202	use crate::{PrependSourcePrompter, Prompter as _, Source};203204	// #[tokio::test]205	#[tokio::test]206	#[ignore = "interactive"]207	async fn test() {208		let prompter = PrependSourcePrompter {209			prompter: RofiPrompter,210			description: "test".to_owned(),211			source: vec![Source(Cow::Borrowed("ssh"))],212		};213		prompter214			.prompt_radio("Enable", "Polkit needs access", &[])215			.await216			.expect("rofi");217		prompter218			.prompt_text(false, "Password", "Polkit needs access", &[])219			.await220			.expect("rofi");221		prompter222			.display_text(true, "Polkit needs access", &[])223			.await224			.expect("rofi");225	}226}