From f2f5e4c814f2897eb3b209b271edc230bd3db2d7 Mon Sep 17 00:00:00 2001
From: SolitudeSF <solitudesf@protonmail.com>
Date: Mon, 19 Aug 2024 16:51:44 +0300
Subject: [PATCH] Add entry count to directories

---
 yazi-plugin/src/file/file.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/yazi-plugin/src/file/file.rs b/yazi-plugin/src/file/file.rs
index 5334645b..4b1c3c32 100644
--- a/yazi-plugin/src/file/file.rs
+++ b/yazi-plugin/src/file/file.rs
@@ -42,6 +42,14 @@ impl File {
 				IconCache::Icon(cached) => Some(Icon::cast(lua, cached)).transpose(),
 			}
 		});
+
+		reg.add_method("count", |_, me, ()| {
+			if me.as_ref().cha.is_dir() {
+				return Ok(Some(std::fs::read_dir::<&std::path::Path>(me.as_ref().url.as_ref())?.count()));
+			} else {
+				return Ok(None);
+			}
+		});
 	}
 
 	pub fn install(lua: &Lua) -> mlua::Result<()> {