1 files changed, 223 insertions, 0 deletions

diff --git a/crates/vcs_data/src/data/local/config.rs b/crates/vcs_data/src/data/local/config.rs
index 338d01b..2fec457 100644
--- a/crates/vcs_data/src/data/local/config.rs
+++ b/crates/vcs_data/src/data/local/config.rs
@@ -1,12 +1,26 @@
 use cfg_file::ConfigFile;
+use cfg_file::config::ConfigFile;
 use serde::{Deserialize, Serialize};
 use std::net::SocketAddr;
+use std::path::Path;
+use std::path::PathBuf;
+use string_proc::format_processer::FormatProcesser;
+use string_proc::snake_case;
 
 use crate::constants::CLIENT_FILE_WORKSPACE;
+use crate::constants::CLIENT_FOLDER_WORKSPACE_ROOT_NAME;
+use crate::constants::CLIENT_PATH_LOCAL_DRAFT;
+use crate::constants::CLIENT_PATH_WORKSPACE_ROOT;
 use crate::constants::PORT;
+use crate::current::current_local_path;
+use crate::data::local::latest_info;
+use crate::data::local::latest_info::LatestInfo;
 use crate::data::member::MemberId;
+use crate::data::sheet::SheetName;
 use crate::data::vault::config::VaultUuid;
 
+const SHEET_NAME: &str = "{sheet_name}";
+
 #[derive(Serialize, Deserialize, ConfigFile)]
 #[cfg_file(path = CLIENT_FILE_WORKSPACE)]
 pub struct LocalConfig {
@@ -25,6 +39,9 @@ pub struct LocalConfig {
     /// If the value is None, it means not stained;
     /// otherwise, it contains the stain identifier (i.e., the upstream vault's unique ID)
     stained_uuid: Option<VaultUuid>,
+
+    /// The name of the sheet currently in use.
+    sheet_in_use: Option<SheetName>,
 }
 
 impl Default for LocalConfig {
@@ -36,6 +53,7 @@ impl Default for LocalConfig {
             )),
             using_account: "unknown".to_string(),
             stained_uuid: None,
+            sheet_in_use: None,
         }
     }
 }
@@ -56,6 +74,177 @@ impl LocalConfig {
         self.using_account = account;
     }
 
+    /// Set the currently used sheet
+    pub async fn use_sheet(&mut self, sheet: SheetName) -> Result<(), std::io::Error> {
+        let sheet = snake_case!(sheet);
+
+        // Check if the sheet is already in use
+        if self.sheet_in_use().is_some() {
+            return Err(std::io::Error::new(
+                std::io::ErrorKind::AlreadyExists,
+                "Sheet already in use",
+            ));
+        };
+
+        // Check if the local path exists
+        let local_path = self.get_local_path().await?;
+
+        // Get latest info
+        let Ok(latest_info) = LatestInfo::read().await else {
+            return Err(std::io::Error::new(
+                std::io::ErrorKind::NotFound,
+                "No latest info found",
+            ));
+        };
+
+        // Check if the sheet exists
+        if !latest_info.my_sheets.contains(&sheet) {
+            return Err(std::io::Error::new(
+                std::io::ErrorKind::NotFound,
+                "Sheet not found",
+            ));
+        }
+
+        // Check if there are any files or folders other than .jv
+        self.check_local_path_empty(&local_path).await?;
+
+        // Get the draft folder path
+        let draft_folder = self.draft_folder(&sheet, &local_path);
+
+        if draft_folder.exists() {
+            // Exists
+            // Move the contents of the draft folder to the local path
+            self.move_draft_to_local(&draft_folder, &local_path).await?;
+        }
+
+        self.sheet_in_use = Some(sheet);
+        LocalConfig::write(&self).await?;
+
+        Ok(())
+    }
+
+    /// Exit the currently used sheet
+    pub async fn exit_sheet(&mut self) -> Result<(), std::io::Error> {
+        // Check if the sheet is already in use
+        if self.sheet_in_use().is_none() {
+            return Ok(());
+        }
+
+        // Check if the local path exists
+        let local_path = self.get_local_path().await?;
+
+        // Get the current sheet name
+        let sheet_name = self.sheet_in_use().as_ref().unwrap().clone();
+
+        // Get the draft folder path
+        let draft_folder = self.draft_folder(&sheet_name, &local_path);
+
+        // Create the draft folder if it doesn't exist
+        if !draft_folder.exists() {
+            std::fs::create_dir_all(&draft_folder)
+                .map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e))?;
+        }
+
+        // Move all files and folders (except .jv folder) to the draft folder
+        self.move_local_to_draft(&local_path, &draft_folder).await?;
+
+        // Clear the sheet in use
+        self.sheet_in_use = None;
+        LocalConfig::write(&self).await?;
+
+        Ok(())
+    }
+
+    /// Get local path or return error
+    async fn get_local_path(&self) -> Result<PathBuf, std::io::Error> {
+        current_local_path().ok_or_else(|| {
+            std::io::Error::new(std::io::ErrorKind::NotFound, "Fail to get local path")
+        })
+    }
+
+    /// Check if local path is empty (except for .jv folder)
+    async fn check_local_path_empty(&self, local_path: &Path) -> Result<(), std::io::Error> {
+        let jv_folder = local_path.join(CLIENT_PATH_WORKSPACE_ROOT);
+        let mut entries = std::fs::read_dir(local_path)
+            .map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e))?;
+
+        if entries.any(|entry| {
+            if let Ok(entry) = entry {
+                let path = entry.path();
+                path != jv_folder
+                    && path.file_name().and_then(|s| s.to_str())
+                        != Some(CLIENT_FOLDER_WORKSPACE_ROOT_NAME)
+            } else {
+                false
+            }
+        }) {
+            return Err(std::io::Error::new(
+                std::io::ErrorKind::DirectoryNotEmpty,
+                "Local path is not empty!",
+            ));
+        }
+
+        Ok(())
+    }
+
+    /// Move contents from draft folder to local path
+    async fn move_draft_to_local(
+        &self,
+        draft_folder: &Path,
+        local_path: &Path,
+    ) -> Result<(), std::io::Error> {
+        let draft_entries = std::fs::read_dir(draft_folder)
+            .map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e))?;
+
+        for entry in draft_entries {
+            let entry = entry.map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e))?;
+            let entry_path = entry.path();
+            let target_path = local_path.join(entry_path.file_name().unwrap());
+
+            // Move each file/directory from draft folder to local path
+            std::fs::rename(&entry_path, &target_path)
+                .map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e))?;
+        }
+
+        // Remove the now-empty draft folder
+        std::fs::remove_dir(draft_folder)
+            .map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e))?;
+
+        Ok(())
+    }
+
+    /// Move contents from local path to draft folder (except .jv folder)
+    async fn move_local_to_draft(
+        &self,
+        local_path: &Path,
+        draft_folder: &Path,
+    ) -> Result<(), std::io::Error> {
+        let jv_folder = local_path.join(CLIENT_PATH_WORKSPACE_ROOT);
+        let entries = std::fs::read_dir(local_path)
+            .map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e))?;
+
+        for entry in entries {
+            let entry = entry.map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e))?;
+            let entry_path = entry.path();
+
+            // Skip the .jv folder
+            if entry_path == jv_folder
+                || entry_path.file_name().and_then(|s| s.to_str())
+                    == Some(CLIENT_FOLDER_WORKSPACE_ROOT_NAME)
+            {
+                continue;
+            }
+
+            let target_path = draft_folder.join(entry_path.file_name().unwrap());
+
+            // Move each file/directory from local path to draft folder
+            std::fs::rename(&entry_path, &target_path)
+                .map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e))?;
+        }
+
+        Ok(())
+    }
+
     /// Get the currently used account
     pub fn current_account(&self) -> MemberId {
         self.using_account.clone()
@@ -75,4 +264,38 @@ impl LocalConfig {
     pub fn unstain(&mut self) {
         self.stained_uuid = None;
     }
+
+    /// Get the upstream address.
+    pub fn upstream_addr(&self) -> SocketAddr {
+        self.upstream_addr
+    }
+
+    /// Get the currently used sheet
+    pub fn sheet_in_use(&self) -> &Option<SheetName> {
+        &self.sheet_in_use
+    }
+
+    /// Get draft folder
+    pub fn draft_folder(
+        &self,
+        sheet_name: &SheetName,
+        local_workspace_path: impl Into<PathBuf>,
+    ) -> PathBuf {
+        let sheet_name_str = snake_case!(sheet_name.as_str());
+        let draft_path = CLIENT_PATH_LOCAL_DRAFT.replace(SHEET_NAME, &sheet_name_str);
+        local_workspace_path.into().join(draft_path)
+    }
+
+    /// Get current draft folder
+    pub fn current_draft_folder(&self) -> Option<PathBuf> {
+        let Some(sheet_name) = self.sheet_in_use() else {
+            return None;
+        };
+
+        let Some(current_dir) = current_local_path() else {
+            return None;
+        };
+
+        Some(self.draft_folder(sheet_name, current_dir))
+    }
 }