use std::process;
use crate::{
Choice, Error, FileSelection, FileSelectionMode, Input, Message, Password, Question, Result,
};
const OK: i32 = 0;
const CANCEL: i32 = 1;
#[derive(Debug, Default)]
pub struct KDialog {
icon: Option<String>,
}
impl KDialog {
pub fn new() -> KDialog {
Default::default()
}
pub fn set_icon(&mut self, icon: impl Into<String>) {
self.icon = Some(icon.into());
}
pub(crate) fn is_available() -> bool {
super::is_available("kdialog")
}
fn execute(&self, args: Vec<&str>, title: &Option<String>) -> Result<process::Output> {
let mut command = process::Command::new("kdialog");
if let Some(ref icon) = self.icon {
command.arg("--icon");
command.arg(icon);
}
if let Some(ref title) = title {
command.arg("--title");
command.arg(title);
}
command.args(args);
command.output().map_err(Error::IoError)
}
}
impl AsRef<KDialog> for KDialog {
fn as_ref(&self) -> &Self {
self
}
}
fn require_success(status: process::ExitStatus) -> Result<()> {
if status.success() {
Ok(())
} else if let Some(code) = status.code() {
match code {
CANCEL => Ok(()),
_ => Err(Error::from(("kdialog", status))),
}
} else {
Err(Error::from(("kdialog", status)))
}
}
fn get_choice(status: process::ExitStatus) -> Result<Choice> {
if let Some(code) = status.code() {
match code {
OK => Ok(Choice::Yes),
CANCEL => Ok(Choice::No),
_ => Err(Error::from(("kdialog", status))),
}
} else {
Err(Error::from(("kdialog", status)))
}
}
fn get_stdout(output: process::Output) -> Result<Option<String>> {
if output.status.success() {
String::from_utf8(output.stdout)
.map(|s| Some(s.trim_end_matches('\n').to_string()))
.map_err(Error::from)
} else if let Some(code) = output.status.code() {
match code {
OK => Ok(None),
CANCEL => Ok(None),
_ => Err(Error::from(("kdialog", output.status))),
}
} else {
Err(Error::from(("kdialog", output.status)))
}
}
impl super::Backend for KDialog {
fn show_input(&self, input: &Input) -> Result<Option<String>> {
let mut args = vec!["--inputbox", &input.text];
if let Some(ref default) = input.default {
args.push(default);
}
self.execute(args, &input.title).and_then(get_stdout)
}
fn show_message(&self, message: &Message) -> Result<()> {
let args = vec!["--msgbox", &message.text];
self.execute(args, &message.title)
.and_then(|output| require_success(output.status))
.map(|_| ())
}
fn show_password(&self, password: &Password) -> Result<Option<String>> {
let args = vec!["--password", &password.text];
self.execute(args, &password.title).and_then(get_stdout)
}
fn show_question(&self, question: &Question) -> Result<Choice> {
let args = vec!["--yesno", &question.text];
self.execute(args, &question.title)
.and_then(|output| get_choice(output.status))
}
fn show_file_selection(&self, file_selection: &FileSelection) -> Result<Option<String>> {
let dir = file_selection.path_to_string().ok_or("path not valid")?;
let option = match file_selection.mode {
FileSelectionMode::Open => "--getopenfilename",
FileSelectionMode::Save => "--getsavefilename",
};
let args = vec![option, &dir];
self.execute(args, &file_selection.title)
.and_then(get_stdout)
}
}