perf: introduce URN to speed up large directory locating

Cargo.lock

@@ -135,9 +135,9 @@ dependencies = [
 
 [[package]]
 name = "anyhow"
-version = "1.0.87"
+version = "1.0.88"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "10f00e1f6e58a40e807377c75c6a7f97bf9044fab57816f2414e6f5f4499d7b8"
+checksum = "4e1496f8fb1fbf272686b8d37f523dab3e4a7443300055e74cdaa449f3114356"
 
 [[package]]
 name = "arbitrary"
@@ -1008,9 +1008,9 @@ checksum = "40ecd4077b5ae9fd2e9e169b102c6c330d0605168eb0e8bf79952b256dbefffd"
 
 [[package]]
 name = "globset"
-version = "0.4.14"
+version = "0.4.15"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "57da3b9b5b85bd66f31093f8c408b90a74431672542466497dcbdfdc02034be1"
+checksum = "15f1ce686646e7f1e19bf7d5533fe443a45dbfb990e00629110797578b42fb19"
 dependencies = [
  "aho-corasick",
  "bstr",

Cargo.toml

@@ -11,7 +11,7 @@ strip         = true
 
 [workspace.dependencies]
 ansi-to-tui   = "6.0.0"
-anyhow        = "1.0.86"
+anyhow        = "1.0.88"
 arc-swap      = "1.7.1"
 base64        = "0.22.1"
 bitflags      = "2.6.0"
@@ -19,7 +19,7 @@ clap          = { version = "4.5.17", features = [ "derive" ] }
 crossterm     = { version = "0.28.1", features = [ "event-stream" ] }
 dirs          = "5.0.1"
 futures       = "0.3.30"
-globset       = "0.4.14"
+globset       = "0.4.15"
 libc          = "0.2.158"
 md-5          = "0.10.6"
 mlua          = { version = "0.9.9", features = [ "lua54", "serialize", "macros", "async" ] }
@@ -27,7 +27,7 @@ parking_lot   = "0.12.3"
 ratatui       = { version = "0.28.1", features = [ "unstable-rendered-line-info" ] }
 regex         = "1.10.6"
 scopeguard    = "1.2.0"
-serde         = { version = "1.0.209", features = [ "derive" ] }
+serde         = { version = "1.0.210", features = [ "derive" ] }
 serde_json    = "1.0.128"
 shell-words   = "1.1.0"
 tokio         = { version = "1.40.0", features = [ "full" ] }

yazi-core/src/manager/commands/bulk_rename.rs

@@ -16,7 +16,7 @@ impl Manager {
 			return AppProxy::notify_warn("Bulk rename", "No text opener found");
 		};
 
-		let cwd = self.cwd().clone();
+		let cwd = self.cwd().url_owned();
 		let old: Vec<_> = self.selected_or_hovered(true).collect();
 
 		let root = max_common_root(&old);

yazi-core/src/manager/commands/hover.rs

@@ -1,7 +1,7 @@
-use std::collections::HashSet;
+use std::{collections::HashSet, path::PathBuf};
 
 use yazi_dds::Pubsub;
-use yazi_shared::{event::{Cmd, Data}, fs::Url, render};
+use yazi_shared::{event::{Cmd, Data}, fs::{Url, Urn}, render};
 
 use crate::manager::Manager;
 
@@ -25,13 +25,8 @@ impl From<Option<Url>> for Opt {
 impl Manager {
 	pub fn hover(&mut self, opt: impl Into<Opt>) {
 		let opt = opt.into() as Opt;
-
-		// Hover on the file
-		render!(self.current_or_mut(opt.tab).repos(opt.url.as_ref()));
-		if opt.url.zip(self.current_or(opt.tab).hovered()).is_some_and(|(u, f)| &u == f.url()) {
-			// `hover(Some)` occurs after user actions, such as create, rename, reveal, etc.
-			// At this point, it's intuitive to track the location of this file regardless.
-			self.current_or_mut(opt.tab).tracing = true;
+		if let Some(u) = opt.url {
+			self.hover_do(u, opt.tab);
 		}
 
 		// Repeek
@@ -40,7 +35,7 @@ impl Manager {
 		// Refresh watcher
 		let mut to_watch = HashSet::with_capacity(3 * self.tabs.len());
 		for tab in self.tabs.iter() {
-			to_watch.insert(tab.cwd());
+			to_watch.insert(tab.cwd().url());
 			if let Some(ref p) = tab.parent {
 				to_watch.insert(&p.loc);
 			}
@@ -53,4 +48,18 @@ impl Manager {
 		// Publish through DDS
 		Pubsub::pub_from_hover(self.active().idx, self.hovered().map(|h| h.url()));
 	}
+
+	fn hover_do(&mut self, url: Url, tab: Option<usize>) {
+		// Hover on the file
+		if let Some(p) = url.strip_prefix(&self.current_or(tab).loc).map(PathBuf::from) {
+			render!(self.current_or_mut(tab).repos(Some(Urn::new(&p))));
+		}
+
+		// Turn on tracing
+		if self.current_or(tab).hovered().is_some_and(|f| url == *f.url()) {
+			// `hover(Some)` occurs after user actions, such as create, rename, reveal, etc.
+			// At this point, it's intuitive to track the location of this file regardless.
+			self.current_or_mut(tab).tracing = true;
+		}
+	}
 }

yazi-core/src/manager/commands/refresh.rs

@@ -9,7 +9,7 @@ use crate::{manager::Manager, tasks::Tasks};
 impl Manager {
 	pub fn refresh(&mut self, _: Cmd, tasks: &Tasks) {
 		env::set_current_dir(self.cwd()).ok();
-		env::set_var("PWD", self.cwd());
+		env::set_var("PWD", self.cwd().url());
 
 		if !MANAGER.title_format.is_empty() {
 			execute!(std::io::stderr(), SetTitle(self.title())).ok();

yazi-core/src/manager/commands/tab_create.rs

@@ -46,7 +46,7 @@ impl Tabs {
 		} else {
 			tab.conf = self.active().conf.clone();
 			tab.apply_files_attrs();
-			tab.cd(self.active().cwd().clone());
+			tab.cd(self.active().cwd().url_owned());
 		}
 
 		self.items.insert(self.cursor + 1, tab);

yazi-core/src/manager/commands/update_files.rs

@@ -47,7 +47,7 @@ impl Manager {
 		let url = op.url();
 		tab.selected.apply_op(&op);
 
-		if url == tab.cwd() {
+		if url == tab.cwd().url() {
 			Self::update_current(tab, op, tasks);
 		} else if matches!(&tab.parent, Some(p) if url == &*p.loc) {
 			Self::update_parent(tab, op);
@@ -59,12 +59,15 @@ impl Manager {
 	}
 
 	fn update_parent(tab: &mut Tab, op: Cow<FilesOp>) {
-		let cwd = tab.cwd().clone();
-		let leave = matches!(*op, FilesOp::Deleting(_, ref urls) if urls.contains(&cwd));
+		let urn = tab.cwd().urn_owned();
+		// FIXME
+		let leave = false;
+		// let leave = matches!(*op, FilesOp::Deleting(_, ref urls) if
+		// urls.contains(&urn));
 
 		if let Some(f) = tab.parent.as_mut() {
 			render!(f.update(op.into_owned()));
-			render!(f.hover(&cwd));
+			render!(f.hover(urn._deref()));
 		}
 
 		if leave {
@@ -73,7 +76,7 @@ impl Manager {
 	}
 
 	fn update_current(tab: &mut Tab, op: Cow<FilesOp>, tasks: &Tasks) {
-		let hovered = tab.current.hovered().filter(|_| tab.current.tracing).map(|h| h.url_owned());
+		let hovered = tab.current.hovered().filter(|_| tab.current.tracing).map(|h| h.urn_owned());
 		let calc = !matches!(*op, FilesOp::Size(..) | FilesOp::Deleting(..));
 
 		let foreign = matches!(op, Cow::Borrowed(_));
@@ -81,7 +84,7 @@ impl Manager {
 			return;
 		}
 
-		tab.current.repos(hovered);
+		tab.current.repos(hovered.as_ref().map(|u| u._deref()));
 		if foreign {
 			return;
 		}
@@ -113,9 +116,9 @@ impl Manager {
 		);
 
 		let folder = tab.history.entry(op.url().clone()).or_insert_with(|| Folder::from(op.url()));
-		let hovered = folder.hovered().filter(|_| folder.tracing).map(|h| h.url_owned());
+		let hovered = folder.hovered().filter(|_| folder.tracing).map(|h| h.urn_owned());
 		if folder.update(op.into_owned()) {
-			folder.repos(hovered);
+			folder.repos(hovered.as_ref().map(|u| u._deref()));
 		}
 
 		if leave {

yazi-core/src/manager/commands/update_paged.rs

@@ -27,7 +27,7 @@ impl Manager {
 			return;
 		};
 
-		if opt.only_if.is_some_and(|u| u != *self.active().cwd()) {
+		if opt.only_if.is_some_and(|u| u != *self.active().cwd().url()) {
 			return;
 		}
 

yazi-core/src/manager/manager.rs

@@ -2,7 +2,7 @@ use ratatui::layout::Rect;
 use yazi_adapter::Dimension;
 use yazi_config::popup::{Origin, Position};
 use yazi_fs::Folder;
-use yazi_shared::fs::{File, Url};
+use yazi_shared::fs::{File, Loc, Url};
 
 use super::{Mimetype, Tabs, Watcher, Yanked};
 use crate::tab::Tab;
@@ -39,7 +39,7 @@ impl Manager {
 
 impl Manager {
 	#[inline]
-	pub fn cwd(&self) -> &Url { &self.current().loc }
+	pub fn cwd(&self) -> &Loc { self.active().cwd() }
 
 	#[inline]
 	pub fn active(&self) -> &Tab { self.tabs.active() }

yazi-core/src/tab/commands/cd.rs

@@ -39,7 +39,7 @@ impl Tab {
 			return self.cd_interactive();
 		}
 
-		if opt.target == *self.cwd() {
+		if opt.target == *self.cwd().url() {
 			return;
 		}
 

yazi-core/src/tab/commands/filter.rs

@@ -65,13 +65,13 @@ impl Tab {
 			ManagerProxy::update_paged(); // Update for paged files in next loop
 		}
 
-		let hovered = self.current.hovered().map(|f| f.url_owned());
+		let hovered = self.current.hovered().map(|f| f.urn_owned());
 		if !self.current.files.set_filter(filter) {
 			return;
 		}
 
-		self.current.repos(hovered.as_ref());
-		if self.current.hovered().map(|f| f.url()) != hovered.as_ref() {
+		self.current.repos(hovered.as_ref().map(|u| u._deref()));
+		if self.current.hovered().map(|f| f.urn()) != hovered.as_ref().map(|u| u._deref()) {
 			ManagerProxy::hover(None, self.idx);
 		}
 

yazi-core/src/tab/commands/leave.rs

@@ -16,7 +16,7 @@ impl Tab {
 			.current
 			.hovered()
 			.and_then(|h| h.parent())
-			.filter(|p| p != self.cwd())
+			.filter(|u| u != self.cwd().url())
 			.or_else(|| self.cwd().parent_url())
 			.map(|u| self.cd(u));
 	}

yazi-core/src/tab/tab.rs

@@ -6,7 +6,7 @@ use tokio::task::JoinHandle;
 use yazi_adapter::Dimension;
 use yazi_config::{popup::{Origin, Position}, LAYOUT};
 use yazi_fs::{Folder, FolderStage};
-use yazi_shared::{fs::Url, render};
+use yazi_shared::{fs::{Loc, Url}, render};
 
 use super::{Backstack, Config, Finder, History, Mode, Preview};
 use crate::tab::Selected;
@@ -39,10 +39,10 @@ impl Tab {
 impl Tab {
 	// --- Current
 	#[inline]
-	pub fn cwd(&self) -> &Url { &self.current.loc }
+	pub fn cwd(&self) -> &Loc { &self.current.loc }
 
 	pub fn hovered_rect(&self) -> Option<Rect> {
-		let y = self.current.files.position(self.current.hovered()?.url())? - self.current.offset;
+		let y = self.current.files.position(self.current.hovered()?.urn())? - self.current.offset;
 
 		let mut rect = LAYOUT.load().current;
 		rect.y = rect.y.saturating_sub(1) + y as u16;
@@ -98,12 +98,12 @@ impl Tab {
 				return render!();
 			}
 
-			let hovered = f.hovered().filter(|_| f.tracing).map(|h| h.url_owned());
+			let hovered = f.hovered().filter(|_| f.tracing).map(|h| h.urn_owned());
 			f.files.set_show_hidden(self.conf.show_hidden);
 			f.files.set_sorter(self.conf.sorter());
 
 			render!(f.files.catchup_revision());
-			render!(f.repos(hovered));
+			render!(f.repos(hovered.as_ref().map(|u| u._deref())));
 		};
 
 		apply(&mut self.current);
@@ -112,7 +112,7 @@ impl Tab {
 			apply(parent);
 
 			// The parent should always track the CWD
-			parent.hover(&self.current.loc);
+			parent.hover(self.current.loc.urn());
 			parent.tracing = parent.hovered().map(|h| h.url()) == Some(&self.current.loc);
 		}
 

yazi-fm/src/lives/tab.rs

@@ -24,7 +24,7 @@ impl Tab {
 	pub(super) fn register(lua: &Lua) -> mlua::Result<()> {
 		lua.register_userdata_type::<Self>(|reg| {
 			reg.add_method("name", |lua, me, ()| {
-				Some(lua.create_string(me.current.loc.name().as_encoded_bytes())).transpose()
+				lua.create_string(me.current.loc.name().as_encoded_bytes())
 			});
 
 			reg.add_field_method_get("mode", |_, me| Mode::make(&me.mode));

yazi-fs/src/files.rs

@@ -2,7 +2,7 @@ use std::{collections::{HashMap, HashSet}, mem, ops::Deref, sync::atomic::Orderi
 
 use tokio::{fs::{self, DirEntry}, select, sync::mpsc::{self, UnboundedReceiver}};
 use yazi_config::{manager::SortBy, MANAGER};
-use yazi_shared::fs::{maybe_exists, Cha, File, FilesOp, Url, FILES_TICKET};
+use yazi_shared::fs::{maybe_exists, Cha, File, FilesOp, Url, Urn, FILES_TICKET};
 
 use super::{FilesSorter, Filter};
 
@@ -345,8 +345,7 @@ impl Files {
 impl Files {
 	// --- Items
 	#[inline]
-	// TODO: use `name` instead of `url`
-	pub fn position(&self, url: &Url) -> Option<usize> { self.iter().position(|f| url == f.url()) }
+	pub fn position(&self, urn: &Urn) -> Option<usize> { self.iter().position(|f| urn == f.urn()) }
 
 	// --- Ticket
 	#[inline]

yazi-fs/src/folder.rs

@@ -2,7 +2,7 @@ use std::mem;
 
 use yazi_config::{LAYOUT, MANAGER};
 use yazi_proxy::ManagerProxy;
-use yazi_shared::fs::{Cha, File, FilesOp, Loc, Url};
+use yazi_shared::fs::{Cha, File, FilesOp, Loc, Url, Urn};
 
 use super::FolderStage;
 use crate::{Files, Step};
@@ -79,18 +79,18 @@ impl Folder {
 		b
 	}
 
-	pub fn hover(&mut self, url: &Url) -> bool {
-		if self.hovered().map(|h| h.url()) == Some(url) {
+	pub fn hover(&mut self, urn: &Urn) -> bool {
+		if self.hovered().map(|h| h.urn()) == Some(urn) {
 			return false;
 		}
 
-		let new = self.files.position(url).unwrap_or(self.cursor) as isize;
+		let new = self.files.position(urn).unwrap_or(self.cursor) as isize;
 		self.arrow(new - self.cursor as isize)
 	}
 
 	#[inline]
-	pub fn repos(&mut self, url: Option<impl AsRef<Url>>) -> bool {
-		if let Some(u) = url { self.hover(u.as_ref()) } else { self.arrow(0) }
+	pub fn repos(&mut self, url: Option<&Urn>) -> bool {
+		if let Some(u) = url { self.hover(u) } else { self.arrow(0) }
 	}
 
 	pub fn sync_page(&mut self, force: bool) {

yazi-plugin/src/file/file.rs

@@ -24,9 +24,11 @@ impl File {
 			me.as_ref().link_to.clone().map(|u| Url::cast(lua, u)).transpose()
 		});
 
-		// Extension
 		reg.add_field_method_get("name", |lua, me| {
-			me.as_ref().url().file_name().map(|n| lua.create_string(n.as_encoded_bytes())).transpose()
+			Some(me.as_ref().name())
+				.filter(|s| !s.is_empty())
+				.map(|s| lua.create_string(s.as_encoded_bytes()))
+				.transpose()
 		});
 
 		reg.add_method("icon", |lua, me, ()| {

yazi-shared/src/fs/file.rs

@@ -1,9 +1,9 @@
-use std::{cell::Cell, ffi::OsStr, fs::{FileType, Metadata}, ops::Deref, path::Path};
+use std::{cell::Cell, ffi::OsStr, fs::{FileType, Metadata}, ops::Deref};
 
 use anyhow::Result;
 use tokio::fs;
 
-use super::Loc;
+use super::{Loc, Urn, UrnBuf};
 use crate::{fs::{Cha, ChaKind, Url}, theme::IconCache};
 
 #[derive(Clone, Debug, Default)]
@@ -102,10 +102,13 @@ impl File {
 	pub fn url(&self) -> &Url { self.loc.url() }
 
 	#[inline]
-	pub fn url_owned(&self) -> Url { self.url().clone() }
+	pub fn url_owned(&self) -> Url { self.loc.url_owned() }
 
 	#[inline]
-	pub fn urn(&self) -> &Path { self.loc.urn() }
+	pub fn urn(&self) -> &Urn { self.loc.urn() }
+
+	#[inline]
+	pub fn urn_owned(&self) -> UrnBuf { self.loc.urn_owned() }
 
 	#[inline]
 	pub fn name(&self) -> &OsStr { self.loc.name() }

yazi-shared/src/fs/loc.rs

@@ -1,6 +1,6 @@
 use std::{ffi::OsStr, fmt::{self, Debug, Formatter}, ops::Deref, path::Path};
 
-use super::Url;
+use super::{Url, Urn, UrnBuf};
 
 pub struct Loc {
 	url:  Url,
@@ -73,7 +73,7 @@ impl Loc {
 	#[inline]
 	fn twin_urn<'a>(&self, new: &'a Url) -> &'a OsStr {
 		let total = new.components().count();
-		let take = self.urn().components().count();
+		let take = self.urn()._as_path().components().count();
 
 		let mut it = new.components();
 		for _ in 0..total - take {
@@ -89,7 +89,13 @@ impl Loc {
 	pub fn url(&self) -> &Url { &self.url }
 
 	#[inline]
-	pub fn urn(&self) -> &Path { Path::new(unsafe { &*self.urn }) }
+	pub fn url_owned(&self) -> Url { self.url.to_owned() }
+
+	#[inline]
+	pub fn urn(&self) -> &Urn { Urn::new(unsafe { &*self.urn }) }
+
+	#[inline]
+	pub fn urn_owned(&self) -> UrnBuf { self.urn().to_owned() }
 
 	#[inline]
 	pub fn name(&self) -> &OsStr { unsafe { &*self.name } }

yazi-shared/src/fs/mod.rs

@@ -5,6 +5,7 @@ mod loc;
 mod op;
 mod path;
 mod url;
+mod urn;
 
 pub use cha::*;
 pub use file::*;
@@ -13,3 +14,4 @@ pub use loc::*;
 pub use op::*;
 pub use path::*;
 pub use url::*;
+pub use urn::*;

yazi-shared/src/fs/urn.rs

@@ -0,0 +1,33 @@
+use std::{borrow::Borrow, path::{Path, PathBuf}};
+
+#[derive(Debug, Eq, PartialEq)]
+#[repr(transparent)]
+pub struct Urn(Path);
+
+impl Urn {
+	// TODO: clean this up
+	pub fn new<T: AsRef<Path> + ?Sized>(p: &T) -> &Self {
+		unsafe { &*(p.as_ref() as *const Path as *const Self) }
+	}
+
+	// FIXME: remove this
+	pub fn _as_path(&self) -> &Path { &self.0 }
+}
+
+impl ToOwned for Urn {
+	type Owned = UrnBuf;
+
+	fn to_owned(&self) -> Self::Owned { UrnBuf(self.0.to_owned()) }
+}
+
+// --- UrnBuf
+pub struct UrnBuf(PathBuf);
+
+impl Borrow<Urn> for UrnBuf {
+	fn borrow(&self) -> &Urn { Urn::new(&self.0) }
+}
+
+impl UrnBuf {
+	// FIXME: remove this
+	pub fn _deref(&self) -> &Urn { Urn::new(&self.0) }
+}