summaryrefslogtreecommitdiff
path: root/crates/vcs_data/src/data/vault/service.rs
blob: 9fdce85a2bfadb868ff704b07f9f50f07431e37d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
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(())
    }
}