From d0f054877622687a023951751fb387100a36663c Mon Sep 17 00:00:00 2001
From: Chris Rogus <rogusdev@gmail.com>
Date: Mon, 10 Jun 2024 21:48:36 -0600
Subject: [PATCH] add file size to FileEngine (#2323)

---
 packages/html/src/file_data.rs                      | 8 ++++++++
 packages/html/src/native_bind/native_file_engine.rs | 5 +++++
 packages/web/src/file_engine.rs                     | 5 +++++
 3 files changed, 18 insertions(+)

diff --git a/packages/html/src/file_data.rs b/packages/html/src/file_data.rs
index b15eaab2b..cb1492e30 100644
--- a/packages/html/src/file_data.rs
+++ b/packages/html/src/file_data.rs
@@ -20,6 +20,11 @@ impl FileEngine for SerializedFileEngine {
         self.files.keys().cloned().collect()
     }
 
+    async fn file_size(&self, file: &str) -> Option<u64> {
+        let file = self.files.get(file)?;
+        Some(file.len() as u64)
+    }
+
     async fn read_file(&self, file: &str) -> Option<Vec<u8>> {
         self.files.get(file).cloned()
     }
@@ -42,6 +47,9 @@ pub trait FileEngine {
     // get a list of file names
     fn files(&self) -> Vec<String>;
 
+    // get the size of a file
+    async fn file_size(&self, file: &str) -> Option<u64>;
+
     // read a file to bytes
     async fn read_file(&self, file: &str) -> Option<Vec<u8>>;
 
diff --git a/packages/html/src/native_bind/native_file_engine.rs b/packages/html/src/native_bind/native_file_engine.rs
index 81070bf0d..1301051b2 100644
--- a/packages/html/src/native_bind/native_file_engine.rs
+++ b/packages/html/src/native_bind/native_file_engine.rs
@@ -25,6 +25,11 @@ impl FileEngine for NativeFileEngine {
             .collect()
     }
 
+    async fn file_size(&self, file: &str) -> Option<u64> {
+        let file = File::open(file).await.ok()?;
+        Some(file.metadata().await.ok()?.len())
+    }
+
     async fn read_file(&self, file: &str) -> Option<Vec<u8>> {
         let mut file = File::open(file).await.ok()?;
 
diff --git a/packages/web/src/file_engine.rs b/packages/web/src/file_engine.rs
index f98f726e2..50968ffe8 100644
--- a/packages/web/src/file_engine.rs
+++ b/packages/web/src/file_engine.rs
@@ -42,6 +42,11 @@ impl FileEngine for WebFileEngine {
             .collect()
     }
 
+    async fn file_size(&self, file: &str) -> Option<u64> {
+        let file = self.find(file)?;
+        Some(file.size() as u64)
+    }
+
     // read a file to bytes
     async fn read_file(&self, file: &str) -> Option<Vec<u8>> {
         let file = self.find(file)?;