add possibility to lock file types or complete repository

Author: aawsome

crates/core/src/commands/lock.rs

@@ -23,6 +23,7 @@ use crate::{
     progress::{Progress, ProgressBars},
     repofile::{IndexFile, SnapshotFile},
     repository::{Open, Repository},
+    ALL_FILE_TYPES,
 };
 
 pub(super) mod constants {
@@ -163,6 +164,29 @@ impl LockOptions {
     }
 }
 
+pub fn lock_repo<P: ProgressBars, S>(
+    repo: &Repository<P, S>,
+    until: Option<DateTime<Local>>,
+) -> RusticResult<()> {
+    for file_type in ALL_FILE_TYPES {
+        lock_all_files(repo, file_type, until)?
+    }
+    Ok(())
+}
+
+pub fn lock_all_files<P: ProgressBars, S>(
+    repo: &Repository<P, S>,
+    file_type: FileType,
+    until: Option<DateTime<Local>>,
+) -> RusticResult<()> {
+    let p = &repo
+        .pb
+        .progress_spinner(format!("losting {file_type:?} files.."));
+    let ids: Vec<_> = repo.list(file_type)?.collect();
+    p.finish();
+    lock_files(repo, file_type, &ids, until)
+}
+
 fn lock_files<P: ProgressBars, S>(
     repo: &Repository<P, S>,
     file_type: FileType,

crates/core/src/repository.rs

@@ -10,7 +10,7 @@ use std::{
 };
 
 use bytes::Bytes;
-use chrono::Local;
+use chrono::{DateTime, Local};
 use derive_setters::Setters;
 use log::{debug, error, info};
 use serde_with::{serde_as, DisplayFromStr};
@@ -38,7 +38,7 @@ use crate::{
         copy::CopySnapshot,
         forget::{ForgetGroups, KeepOptions},
         key::KeyOptions,
-        lock::LockOptions,
+        lock::{lock_all_files, lock_repo, LockOptions},
         prune::{PruneOptions, PrunePlan},
         repair::{index::RepairIndexOptions, snapshots::RepairSnapshotsOptions},
         repoinfo::{IndexInfos, RepoFileInfos},
@@ -1060,6 +1060,37 @@ impl<P: ProgressBars, S: Open> Repository<P, S> {
         opts.get_plan(self)
     }
 
+    /// Lock the complete repository, i.e. everything in the storage backend.
+    ///
+    /// # Arguments
+    ///
+    /// * `until` - until when to lock. None means lock forever.
+    ///
+    /// # Errors
+    ///
+    // TODO: Document errors
+    pub fn lock_repo(&self, until: Option<DateTime<Local>>) -> RusticResult<()> {
+        lock_repo(self, until)
+    }
+
+    /// Lock all repository files of the given type
+    ///
+    /// # Arguments
+    ///
+    /// * `file_type` - the file type to lock
+    /// * `until` - until when to lock. None means lock forever.
+    ///
+    /// # Errors
+    ///
+    // TODO: Document errors
+    pub fn lock_repo_files(
+        &self,
+        file_type: FileType,
+        until: Option<DateTime<Local>>,
+    ) -> RusticResult<()> {
+        lock_all_files(self, file_type, until)
+    }
+
     /// Lock snapshot and pack files needed for the given snapshots
     ///
     /// # Arguments
@@ -1071,7 +1102,7 @@ impl<P: ProgressBars, S: Open> Repository<P, S> {
     /// # Errors
     ///
     // TODO: Document errors
-    pub fn lock(&self, opts: &LockOptions, snaps: &[SnapshotFile]) -> RusticResult<()> {
+    pub fn lock_snaphots(&self, opts: &LockOptions, snaps: &[SnapshotFile]) -> RusticResult<()> {
         let now = Local::now();
         opts.lock(self, snaps, now)
     }