diff --git a/src/main.rs b/src/main.rs index da05a34..27a4a79 100644 --- a/src/main.rs +++ b/src/main.rs @@ -30,7 +30,7 @@ fn main() { for file in files { let desktop_file = format!("{}", file.as_ref().unwrap().file_name().into_string().unwrap()); let mut name = desktop_file.split("."); - if name.find(|x| *x == cli.search).is_some() { + if name.find(|x| *x == cli.search.strip_suffix(".desktop").unwrap_or(&cli.search)).is_some() { let path = file.as_ref().unwrap().path(); let out = if cli.show_tilde { format!("{}", path.display().to_string().replace(&home, "~"))