fix: ensure the selected file list is always ordered (#2244)

This commit is contained in:
三咲雅 · Misaki Masa 2025-01-25 20:39:24 +08:00 committed by GitHub
parent a4f8e86471
commit 29284613a0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 45 additions and 44 deletions

2
Cargo.lock generated
View file

@ -3408,6 +3408,7 @@ dependencies = [
"crossterm",
"dirs",
"futures",
"indexmap",
"libc",
"notify",
"parking_lot",
@ -3473,6 +3474,7 @@ dependencies = [
"crossterm",
"fdlimit",
"futures",
"indexmap",
"libc",
"mlua",
"ratatui",

View file

@ -20,6 +20,7 @@ crossterm = { version = "0.28.1", features = [ "event-stream" ] }
dirs = "6.0.0"
futures = "0.3.31"
globset = "0.4.15"
indexmap = { version = "2.7.1", features = [ "serde" ] }
libc = "0.2.169"
md-5 = "0.10.6"
mlua = { version = "0.10.2", features = [ "anyhow", "async", "error-send", "lua54", "macros", "serialize" ] }

View file

@ -18,7 +18,7 @@ anyhow = { workspace = true }
bitflags = { workspace = true }
crossterm = { workspace = true }
globset = { workspace = true }
indexmap = { version = "2.7.1", features = [ "serde" ] }
indexmap = { workspace = true }
ratatui = { workspace = true }
regex = { workspace = true }
serde = { workspace = true }

View file

@ -27,6 +27,7 @@ bitflags = { workspace = true }
crossterm = { workspace = true }
dirs = { workspace = true }
futures = { workspace = true }
indexmap = { workspace = true }
notify = { version = "8.0.0", default-features = false, features = [ "macos_fsevent" ] }
parking_lot = { workspace = true }
ratatui = { workspace = true }

View file

@ -17,7 +17,7 @@ impl Manager {
return AppProxy::notify_warn("Bulk rename", "No text opener found");
};
let old: Vec<_> = self.selected_or_hovered(true).collect();
let old: Vec<_> = self.selected_or_hovered().collect();
let root = max_common_root(&old);
let old: Vec<_> = old.into_iter().map(|p| p.strip_prefix(&root).unwrap().to_owned()).collect();

View file

@ -33,9 +33,7 @@ impl Manager {
return;
};
let selected =
if opt.hovered { vec![&hovered] } else { self.selected_or_hovered(true).collect() };
let selected = if opt.hovered { vec![&hovered] } else { self.selected_or_hovered().collect() };
if Self::quit_with_selected(opt, &selected) {
return;
}

View file

@ -32,7 +32,7 @@ impl Manager {
opt.targets = if opt.hovered {
self.hovered().map_or(vec![], |h| vec![h.url.clone()])
} else {
self.selected_or_hovered(true).cloned().collect()
self.selected_or_hovered().cloned().collect()
};
if opt.targets.is_empty() {

View file

@ -18,7 +18,7 @@ impl Manager {
return;
}
self.yanked = Yanked::new(opt.cut, self.selected_or_hovered(false).cloned().collect());
self.yanked = Yanked::new(opt.cut, self.selected_or_hovered().cloned().collect());
render!(self.yanked.catchup_revision(true));
self.active_mut().escape_select();

View file

@ -77,12 +77,12 @@ impl Manager {
pub fn hovered_folder(&self) -> Option<&Folder> { self.active().hovered_folder() }
#[inline]
pub fn selected_or_hovered(&self, reorder: bool) -> Box<dyn Iterator<Item = &Url> + '_> {
self.tabs.active().selected_or_hovered(reorder)
pub fn selected_or_hovered(&self) -> Box<dyn Iterator<Item = &Url> + '_> {
self.tabs.active().selected_or_hovered()
}
#[inline]
pub fn hovered_and_selected(&self, reorder: bool) -> Box<dyn Iterator<Item = &Url> + '_> {
self.tabs.active().hovered_and_selected(reorder)
pub fn hovered_and_selected(&self) -> Box<dyn Iterator<Item = &Url> + '_> {
self.tabs.active().hovered_and_selected()
}
}

View file

@ -27,7 +27,7 @@ impl Tab {
}
let mut s = OsString::new();
let mut it = self.selected_or_hovered(true).peekable();
let mut it = self.selected_or_hovered().peekable();
while let Some(u) = it.next() {
s.push(match opt.type_.as_ref() {
"path" => opt.separator.transform(u),

View file

@ -53,7 +53,7 @@ impl Tab {
};
let cwd = opt.cwd.take().unwrap_or_else(|| self.cwd().clone());
let selected = self.hovered_and_selected(true).cloned().collect();
let selected = self.hovered_and_selected().cloned().collect();
tokio::spawn(async move {
if opt.interactive {
let mut result =

View file

@ -1,16 +1,17 @@
use std::{collections::HashMap, ops::Deref};
use indexmap::IndexMap;
use yazi_fs::FilesOp;
use yazi_shared::{timestamp_us, url::Url};
#[derive(Default)]
pub struct Selected {
inner: HashMap<Url, u64>,
inner: IndexMap<Url, u64>,
parents: HashMap<Url, usize>,
}
impl Deref for Selected {
type Target = HashMap<Url, u64>;
type Target = IndexMap<Url, u64>;
fn deref(&self) -> &Self::Target { &self.inner }
}
@ -66,21 +67,26 @@ impl Selected {
pub fn remove(&mut self, url: &Url) -> bool { self.remove_same(&[url]) == 1 }
pub fn remove_many(&mut self, urls: &[impl AsRef<Url>], same: bool) -> usize {
if same {
return self.remove_same(urls);
}
let mut grouped: HashMap<_, Vec<_>> = Default::default();
for u in urls {
if let Some(p) = u.as_ref().parent_url() {
grouped.entry(p).or_default().push(u);
let affected = if same {
self.remove_same(urls)
} else {
let mut grouped: HashMap<_, Vec<_>> = Default::default();
for u in urls {
if let Some(p) = u.as_ref().parent_url() {
grouped.entry(p).or_default().push(u);
}
}
grouped.into_values().map(|v| self.remove_same(&v)).sum()
};
if affected > 0 {
self.inner.sort_unstable_by(|_, a, _, b| a.cmp(b));
}
grouped.into_values().map(|v| self.remove_same(&v)).sum()
affected
}
fn remove_same(&mut self, urls: &[impl AsRef<Url>]) -> usize {
let count = urls.iter().filter_map(|u| self.inner.remove(u.as_ref())).count();
let count = urls.iter().filter_map(|u| self.inner.swap_remove(u.as_ref())).count();
if count == 0 {
return 0;
}

View file

@ -85,29 +85,20 @@ impl Tab {
}
}
pub fn selected_or_hovered(&self, reorder: bool) -> Box<dyn Iterator<Item = &Url> + '_> {
pub fn selected_or_hovered(&self) -> Box<dyn Iterator<Item = &Url> + '_> {
if self.selected.is_empty() {
Box::new(self.hovered().map(|h| vec![&h.url]).unwrap_or_default().into_iter())
} else if !reorder {
Box::new(self.selected.keys())
} else {
let mut vec: Vec<_> = self.selected.iter().collect();
vec.sort_unstable_by(|(_, a), (_, b)| a.cmp(b));
Box::new(vec.into_iter().map(|(k, _)| k))
Box::new(self.selected.keys())
}
}
pub fn hovered_and_selected(&self, reorder: bool) -> Box<dyn Iterator<Item = &Url> + '_> {
pub fn hovered_and_selected(&self) -> Box<dyn Iterator<Item = &Url> + '_> {
let Some(h) = self.hovered() else { return Box::new(iter::empty()) };
if self.selected.is_empty() {
Box::new([&h.url, &h.url].into_iter())
} else if !reorder {
Box::new([&h.url].into_iter().chain(self.selected.keys()))
} else {
let mut vec: Vec<_> = self.selected.iter().collect();
vec.sort_unstable_by(|(_, a), (_, b)| a.cmp(b));
Box::new([&h.url].into_iter().chain(vec.into_iter().map(|(k, _)| k)))
Box::new([&h.url].into_iter().chain(self.selected.keys()))
}
}

View file

@ -31,6 +31,7 @@ better-panic = "0.3.0"
crossterm = { workspace = true }
fdlimit = "0.3.0"
futures = { workspace = true }
indexmap = { workspace = true }
mlua = { workspace = true }
ratatui = { workspace = true }
scopeguard = { workspace = true }

View file

@ -1,5 +1,6 @@
use std::{collections::{HashMap, hash_map}, ops::Deref};
use std::ops::Deref;
use indexmap::{IndexMap, map::Keys};
use mlua::{AnyUserData, IntoLuaMulti, MetaMethod, UserData, UserDataMethods, UserDataRefMut};
use yazi_plugin::{bindings::Cast, url::Url};
@ -7,23 +8,23 @@ use super::{Iter, Lives};
#[derive(Clone, Copy)]
pub(super) struct Selected {
inner: *const HashMap<yazi_shared::url::Url, u64>,
inner: *const IndexMap<yazi_shared::url::Url, u64>,
}
impl Deref for Selected {
type Target = HashMap<yazi_shared::url::Url, u64>;
type Target = IndexMap<yazi_shared::url::Url, u64>;
fn deref(&self) -> &Self::Target { self.inner() }
}
impl Selected {
#[inline]
pub(super) fn make(inner: &HashMap<yazi_shared::url::Url, u64>) -> mlua::Result<AnyUserData> {
pub(super) fn make(inner: &IndexMap<yazi_shared::url::Url, u64>) -> mlua::Result<AnyUserData> {
Lives::scoped_userdata(Self { inner })
}
#[inline]
fn inner(&self) -> &'static HashMap<yazi_shared::url::Url, u64> { unsafe { &*self.inner } }
fn inner(&self) -> &'static IndexMap<yazi_shared::url::Url, u64> { unsafe { &*self.inner } }
}
impl UserData for Selected {
@ -32,7 +33,7 @@ impl UserData for Selected {
methods.add_meta_method(MetaMethod::Pairs, |lua, me, ()| {
let iter = lua.create_function(
|lua, mut iter: UserDataRefMut<Iter<hash_map::Keys<yazi_shared::url::Url, u64>, _>>| {
|lua, mut iter: UserDataRefMut<Iter<Keys<yazi_shared::url::Url, u64>, _>>| {
if let Some(next) = iter.next() {
(next.0, Url::cast(lua, next.1.clone())?).into_lua_multi(lua)
} else {