fix dmenu empty selection
This commit is contained in:
parent
2114403ae9
commit
c040a28539
3 changed files with 69 additions and 37 deletions
|
@ -1,15 +1,15 @@
|
|||
use enigo::{Enigo, Keyboard};
|
||||
use std::process::Command;
|
||||
use std::thread::sleep;
|
||||
use std::time::Duration;
|
||||
use enigo::{Enigo, Keyboard};
|
||||
|
||||
use worf_lib::{config, gui, Error};
|
||||
use worf_lib::config::Config;
|
||||
use worf_lib::gui::{KeyBinding, ItemProvider, MenuItem, Modifier, Key};
|
||||
use worf_lib::gui::{ItemProvider, Key, KeyBinding, MenuItem, Modifier};
|
||||
use worf_lib::{Error, config, gui};
|
||||
|
||||
#[derive(Clone)]
|
||||
struct PasswordProvider {
|
||||
items: Vec<MenuItem<String>>
|
||||
items: Vec<MenuItem<String>>,
|
||||
}
|
||||
|
||||
impl PasswordProvider {
|
||||
|
@ -22,20 +22,29 @@ impl PasswordProvider {
|
|||
let stdout = String::from_utf8_lossy(&output.stdout);
|
||||
|
||||
// todo the own solution should support images.
|
||||
let mut items: Vec<_>= stdout.lines().map(|line|
|
||||
MenuItem::new(line.to_owned(), None, None, vec![], None, 0.0, Some(String::new()))
|
||||
).collect();
|
||||
let mut items: Vec<_> = stdout
|
||||
.lines()
|
||||
.map(|line| {
|
||||
MenuItem::new(
|
||||
line.to_owned(),
|
||||
None,
|
||||
None,
|
||||
vec![],
|
||||
None,
|
||||
0.0,
|
||||
Some(String::new()),
|
||||
)
|
||||
})
|
||||
.collect();
|
||||
gui::apply_sort(&mut items, &config.sort_order());
|
||||
|
||||
Self {
|
||||
items
|
||||
}
|
||||
Self { items }
|
||||
}
|
||||
}
|
||||
|
||||
impl ItemProvider<String> for PasswordProvider {
|
||||
fn get_elements(&mut self, _: Option<&str>) -> (bool, Vec<MenuItem<String>>) {
|
||||
(false, self.items.clone())
|
||||
(false, self.items.clone())
|
||||
}
|
||||
|
||||
fn get_sub_elements(&mut self, _: &MenuItem<String>) -> (bool, Option<Vec<MenuItem<String>>>) {
|
||||
|
@ -52,10 +61,12 @@ fn rbw_get(name: &str, field: &str) -> String {
|
|||
.output()
|
||||
.expect("Failed to execute command");
|
||||
|
||||
String::from_utf8_lossy(&output.stdout).trim_end().to_string()
|
||||
String::from_utf8_lossy(&output.stdout)
|
||||
.trim_end()
|
||||
.to_string()
|
||||
}
|
||||
|
||||
fn rbw_get_user(name: &str) -> String{
|
||||
fn rbw_get_user(name: &str) -> String {
|
||||
rbw_get(name, "user")
|
||||
}
|
||||
|
||||
|
@ -77,7 +88,7 @@ fn main() -> anyhow::Result<()> {
|
|||
};
|
||||
|
||||
let type_user = KeyBinding {
|
||||
key: Key::Num2,
|
||||
key: Key::Num2,
|
||||
modifiers: Modifier::Alt,
|
||||
label: "<b>Alt+2</b> Type All".to_string(),
|
||||
};
|
||||
|
@ -89,25 +100,25 @@ fn main() -> anyhow::Result<()> {
|
|||
};
|
||||
|
||||
let reload = KeyBinding {
|
||||
key: Key::R,
|
||||
key: Key::R,
|
||||
modifiers: Modifier::Alt,
|
||||
label: "<b>Alt+r</b> Sync".to_string(),
|
||||
};
|
||||
|
||||
let urls = KeyBinding {
|
||||
key: Key::U, // switch view to urls
|
||||
key: Key::U, // switch view to urls
|
||||
modifiers: Modifier::Alt,
|
||||
label: "<b>Alt+u</b> Sync".to_string(),
|
||||
};
|
||||
|
||||
let names = KeyBinding {
|
||||
key: Key::N, // switch view to names
|
||||
key: Key::N, // switch view to names
|
||||
modifiers: Modifier::Alt,
|
||||
label: "<b>Alt+n</b> Sync".to_string(),
|
||||
};
|
||||
|
||||
let folders = KeyBinding {
|
||||
key: Key::C, // switch view to folders
|
||||
key: Key::C, // switch view to folders
|
||||
modifiers: Modifier::Alt,
|
||||
label: "<b>Alt+c</b> Sync".to_string(),
|
||||
};
|
||||
|
@ -124,23 +135,37 @@ fn main() -> anyhow::Result<()> {
|
|||
label: "<b>Alt+l</b> Sync".to_string(),
|
||||
};
|
||||
|
||||
match gui::show(config, provider, false, None, Some(vec![type_all.clone(), type_user, type_totp, reload, urls, names, folders, totp, lock])) {
|
||||
match gui::show(
|
||||
config,
|
||||
provider,
|
||||
false,
|
||||
None,
|
||||
Some(vec![
|
||||
type_all.clone(),
|
||||
type_user,
|
||||
type_totp,
|
||||
reload,
|
||||
urls,
|
||||
names,
|
||||
folders,
|
||||
totp,
|
||||
lock,
|
||||
]),
|
||||
) {
|
||||
Ok(selection) => {
|
||||
let mut enigo = Enigo::new(&enigo::Settings::default())?;
|
||||
let id = selection.menu.label.replace("\n", "");
|
||||
sleep(Duration::from_millis(250));
|
||||
if let Some(key) = selection.custom_key {
|
||||
if key.label == type_all.label {
|
||||
enigo.text(&rbw_get_user(&id))?;
|
||||
enigo.key(enigo::Key::Tab, enigo::Direction::Press)?;
|
||||
enigo.key(enigo::Key::Tab, enigo::Direction::Release)?;
|
||||
enigo.text(&rbw_get_password(&id))?;
|
||||
}
|
||||
if let Some(key) = selection.custom_key {
|
||||
if key.label == type_all.label {
|
||||
enigo.text(&rbw_get_user(&id))?;
|
||||
enigo.key(enigo::Key::Tab, enigo::Direction::Press)?;
|
||||
enigo.key(enigo::Key::Tab, enigo::Direction::Release)?;
|
||||
enigo.text(&rbw_get_password(&id))?;
|
||||
}
|
||||
}
|
||||
}
|
||||
Err(e) => {
|
||||
return Err(anyhow::anyhow!(e))
|
||||
}
|
||||
Err(e) => return Err(anyhow::anyhow!(e)),
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
|
|
@ -793,7 +793,7 @@ fn handle_key_press<T: Clone + 'static + Send>(
|
|||
Some(&search_lock),
|
||||
None,
|
||||
meta.new_on_empty,
|
||||
Some(&custom_key),
|
||||
Some(custom_key),
|
||||
) {
|
||||
log::error!("{e}");
|
||||
}
|
||||
|
@ -953,14 +953,16 @@ where
|
|||
T: Clone + Send,
|
||||
{
|
||||
if let Some(selected_item) = item {
|
||||
send_selected_item(&ui, meta, custom_key, &selected_item);
|
||||
send_selected_item(ui, meta, custom_key, &selected_item);
|
||||
return Ok(());
|
||||
} else if let Some(s) = ui.main_box.selected_children().into_iter().next() {
|
||||
let list_items = ui.menu_rows.lock().unwrap();
|
||||
let item = list_items.get(&s);
|
||||
if let Some(selected_item) = item {
|
||||
send_selected_item(&ui, meta, custom_key, selected_item);
|
||||
return Ok(());
|
||||
if selected_item.visible {
|
||||
send_selected_item(ui, meta, custom_key, selected_item);
|
||||
return Ok(());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -984,9 +986,13 @@ where
|
|||
}
|
||||
}
|
||||
|
||||
fn send_selected_item<T>(ui: &&UiElements<T>, meta: &MetaData<T>, custom_key: Option<&KeyBinding>, selected_item: &MenuItem<T>)
|
||||
where
|
||||
T: Clone + Send
|
||||
fn send_selected_item<T>(
|
||||
ui: &UiElements<T>,
|
||||
meta: &MetaData<T>,
|
||||
custom_key: Option<&KeyBinding>,
|
||||
selected_item: &MenuItem<T>,
|
||||
) where
|
||||
T: Clone + Send,
|
||||
{
|
||||
close_gui(&ui.app);
|
||||
if let Err(e) = meta.selected_sender.send(Ok(Selection {
|
||||
|
|
|
@ -38,6 +38,7 @@ fn main() -> anyhow::Result<()> {
|
|||
|
||||
Ok(())
|
||||
} else {
|
||||
Err(anyhow!("No mode provided"))
|
||||
log::error!("No mode provided");
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Add table
Reference in a new issue