From f4689cb6840df07747e59d6dc7728c29d7d58dd0 Mon Sep 17 00:00:00 2001
From: George Miao <gm@miao.dev>
Date: Thu, 20 Jun 2024 16:08:47 +0900
Subject: [PATCH] implement aux fn's

---
 core/src/services/compfs/backend.rs | 81 +++++++++++++++++++++++++++--
 core/src/services/compfs/core.rs    |  4 ++
 2 files changed, 80 insertions(+), 5 deletions(-)

diff --git a/core/src/services/compfs/backend.rs b/core/src/services/compfs/backend.rs
index f6559fa4c8c0..9926e04f977f 100644
--- a/core/src/services/compfs/backend.rs
+++ b/core/src/services/compfs/backend.rs
@@ -15,7 +15,7 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use compio::dispatcher::Dispatcher;
+use compio::{dispatcher::Dispatcher, fs::OpenOptions};
 
 use super::{core::CompfsCore, lister::CompfsLister, reader::CompfsReader, writer::CompfsWriter};
 
@@ -112,7 +112,6 @@ impl Access for CompfsBackend {
 
                 copy: true,
                 rename: true,
-                blocking: true,
 
                 ..Default::default()
             });
@@ -120,8 +119,79 @@ impl Access for CompfsBackend {
         am
     }
 
+    async fn create_dir(&self, path: &str, _: OpCreateDir) -> Result<RpCreateDir> {
+        let path = self.core.prepare_path(path);
+
+        self.core
+            .exec(move || async move { compio::fs::create_dir_all(path).await })
+            .await?;
+
+        Ok(RpCreateDir::default())
+    }
+
+    async fn stat(&self, path: &str, _: OpStat) -> Result<RpStat> {
+        let path = self.core.prepare_path(path);
+
+        let meta = self
+            .core
+            .exec(move || async move { compio::fs::metadata(path).await })
+            .await?;
+        let ty = meta.file_type();
+        let mode = if ty.is_dir() {
+            EntryMode::DIR
+        } else if ty.is_file() {
+            EntryMode::FILE
+        } else {
+            EntryMode::Unknown
+        };
+        let last_mod = meta.modified().map_err(new_std_io_error)?.into();
+        let ret = Metadata::new(mode).with_last_modified(last_mod);
+
+        Ok(RpStat::new(ret))
+    }
+
+    async fn delete(&self, path: &str, _: OpDelete) -> Result<RpDelete> {
+        let path = self.core.prepare_path(path);
+
+        self.core
+            .exec(move || async move { compio::fs::remove_file(path).await })
+            .await?;
+
+        Ok(RpDelete::default())
+    }
+
+    async fn copy(&self, from: &str, to: &str, _: OpCopy) -> Result<RpCopy> {
+        let from = self.core.prepare_path(from);
+        let to = self.core.prepare_path(to);
+
+        self.core
+            .exec(move || async move {
+                let from = OpenOptions::new().read(true).open(from).await?;
+                let to = OpenOptions::new().write(true).create(true).open(to).await?;
+
+                let (mut from, mut to) = (Cursor::new(from), Cursor::new(to));
+                compio::io::copy(&mut from, &mut to).await?;
+
+                Ok(())
+            })
+            .await?;
+
+        Ok(RpCopy::default())
+    }
+
+    async fn rename(&self, from: &str, to: &str, _: OpRename) -> Result<RpRename> {
+        let from = self.core.prepare_path(from);
+        let to = self.core.prepare_path(to);
+
+        self.core
+            .exec(move || async move { compio::fs::rename(from, to).await })
+            .await?;
+
+        Ok(RpRename::default())
+    }
+
     async fn read(&self, path: &str, op: OpRead) -> Result<(RpRead, Self::Reader)> {
-        let path = self.core.root.join(path.trim_end_matches('/'));
+        let path = self.core.prepare_path(path);
 
         let file = self
             .core
@@ -133,7 +203,7 @@ impl Access for CompfsBackend {
     }
 
     async fn write(&self, path: &str, args: OpWrite) -> Result<(RpWrite, Self::Writer)> {
-        let path = self.core.root.join(path.trim_end_matches('/'));
+        let path = self.core.prepare_path(path);
         let append = args.append();
         let file = self
             .core
@@ -153,7 +223,8 @@ impl Access for CompfsBackend {
     }
 
     async fn list(&self, path: &str, _: OpList) -> Result<(RpList, Self::Lister)> {
-        let path = self.core.root.join(path.trim_end_matches('/'));
+        let path = self.core.prepare_path(path);
+
         let read_dir = match self
             .core
             .exec_blocking(move || std::fs::read_dir(path))
diff --git a/core/src/services/compfs/core.rs b/core/src/services/compfs/core.rs
index 60c68106f154..e600edbe7d2e 100644
--- a/core/src/services/compfs/core.rs
+++ b/core/src/services/compfs/core.rs
@@ -44,6 +44,10 @@ pub(super) struct CompfsCore {
 }
 
 impl CompfsCore {
+    pub fn prepare_path(&self, path: &str) -> PathBuf {
+        self.root.join(path.trim_end_matches('/'))
+    }
+
     pub async fn exec<Fn, Fut, R>(&self, f: Fn) -> crate::Result<R>
     where
         Fn: FnOnce() -> Fut + Send + 'static,