Merge pull request #26 from JustEnoughVCS/jvcs_dev

Jvcs dev

3 files changed, 45 insertions, 1 deletions

diff --git a/crates/vcs_data/src/data/local.rs b/crates/vcs_data/src/data/local.rs
index 1c99832..c93bd2b 100644
--- a/crates/vcs_data/src/data/local.rs
+++ b/crates/vcs_data/src/data/local.rs
@@ -93,7 +93,7 @@ Without these credentials, the server will reject all access requests.
     }
 
     /// Setup local workspace in current directory
-    pub async fn setup_local_workspacecurrent_dir() -> Result<(), std::io::Error> {
+    pub async fn setup_local_workspace_current_dir() -> Result<(), std::io::Error> {
         Self::setup_local_workspace(current_dir()?).await?;
         Ok(())
     }
diff --git a/crates/vcs_data/src/data/vault.rs b/crates/vcs_data/src/data/vault.rs
index 5d17a81..80ebe1d 100644
--- a/crates/vcs_data/src/data/vault.rs
+++ b/crates/vcs_data/src/data/vault.rs
@@ -17,6 +17,7 @@ use crate::{
 
 pub mod config;
 pub mod member;
+pub mod service;
 pub mod sheets;
 pub mod virtual_file;
 
diff --git a/crates/vcs_data/src/data/vault/service.rs b/crates/vcs_data/src/data/vault/service.rs
new file mode 100644
index 0000000..9fdce85
--- /dev/null
+++ b/crates/vcs_data/src/data/vault/service.rs
@@ -0,0 +1,43 @@
+use std::path::PathBuf;
+
+use crate::{constants::SERVER_FILE_LOCKFILE, data::vault::Vault};
+
+impl Vault {
+    /// Get the path of the lock file for the current Vault
+    pub fn lock_file_path(&self) -> PathBuf {
+        self.vault_path().join(SERVER_FILE_LOCKFILE)
+    }
+
+    /// Check if the current Vault is locked
+    pub fn is_locked(&self) ->  bool {
+        self.lock_file_path().exists()
+    }
+
+    /// Lock the current Vault
+    pub fn lock(&self) -> Result<(), std::io::Error> {
+        if self.is_locked() {
+            return Err(std::io::Error::new(
+                std::io::ErrorKind::AlreadyExists,
+                format!(
+                    "Vault is already locked at {}. \
+                    To unlock, please stop any running services. \
+                    If you are certain no services are running, \
+                    please delete this file",
+                    self.lock_file_path().display()
+                ),
+            ));
+        }
+        std::fs::File::create(self.lock_file_path())?;
+        Ok(())
+    }
+
+    /// Unlock the current Vault
+    pub fn unlock(&self) -> Result<(), std::io::Error> {
+        if let Err(e) = std::fs::remove_file(self.lock_file_path()) {
+            if e.kind() != std::io::ErrorKind::NotFound {
+                return Err(e);
+            }
+        }
+        Ok(())
+    }
+}