add webdav command

Author: aawsome

Cargo.toml

@@ -89,6 +89,11 @@ mimalloc = { version = "0.1.39", default_features = false, optional = true }
 rhai = { workspace = true }
 simplelog = { workspace = true }
 runtime-format = "0.1.3"
+webdav-handler = {version = "0.2.0", features = ["warp-compat"]}
+bytes = "1.5.0"
+futures = "0.3.30"
+tokio = "1.35.1"
+warp = "0.3.6"
 
 [dev-dependencies]
 aho-corasick = { workspace = true }

src/commands.rs

@@ -23,6 +23,7 @@ pub(crate) mod self_update;
 pub(crate) mod show_config;
 pub(crate) mod snapshots;
 pub(crate) mod tag;
+pub(crate) mod webdav;
 
 use std::fs::File;
 use std::path::PathBuf;
@@ -36,7 +37,7 @@ use crate::{
         init::InitCmd, key::KeyCmd, list::ListCmd, ls::LsCmd, merge::MergeCmd, mount::MountCmd,
         prune::PruneCmd, repair::RepairCmd, repoinfo::RepoInfoCmd, restore::RestoreCmd,
         self_update::SelfUpdateCmd, show_config::ShowConfigCmd, snapshots::SnapshotCmd,
-        tag::TagCmd,
+        tag::TagCmd, webdav::WebDavCmd,
     },
     config::{progress_options::ProgressOptions, RusticConfig},
     {Application, RUSTIC_APP},
@@ -129,6 +130,9 @@ enum RusticCmd {
 
     /// Change tags of snapshots
     Tag(TagCmd),
+
+    /// Mount repository
+    Webdav(WebDavCmd),
 }
 
 /// Entry point for the application. It needs to be a struct to allow using subcommands!

src/commands/webdav.rs

@@ -0,0 +1,86 @@
+//! `mount` subcommand
+mod webdavfs;
+
+use std::net::SocketAddr;
+use std::str::FromStr;
+
+use crate::{commands::open_repository, status_err, Application, RUSTIC_APP};
+use abscissa_core::{Command, Runnable, Shutdown};
+use anyhow::Result;
+use webdav_handler::{warp::dav_handler, DavHandler};
+use webdavfs::RusticWebDavFS;
+
+#[derive(clap::Parser, Command, Debug)]
+pub(crate) struct WebDavCmd {
+    /// The path template to use for snapshots. {id}, {id_long}, {time}, {username}, {hostname}, {label}, {tags}, {backup_start}, {backup_end} are replaced. [default: "[{hostname}]/[{label}]/{time}"]
+    #[clap(long)]
+    path_template: Option<String>,
+
+    /// The time template to use to display times in the path template. See https://docs.rs/chrono/latest/chrono/format/strftime/index.html for format options. [default: "%Y-%m-%d_%H-%M-%S"]
+    #[clap(long)]
+    time_template: Option<String>,
+
+    /// Use symlinks. This may not be supported by all WebDAV clients
+    #[clap(long)]
+    symlinks: bool,
+
+    /// Socket address to use
+    #[clap(value_name = "ADDR")]
+    socket: String,
+
+    /// Specify directly which path to mount
+    #[clap(value_name = "SNAPSHOT[:PATH]")]
+    snap: Option<String>,
+}
+
+impl Runnable for WebDavCmd {
+    fn run(&self) {
+        if let Err(err) = self.inner_run() {
+            status_err!("{}", err);
+            RUSTIC_APP.shutdown(Shutdown::Crash);
+        };
+    }
+}
+
+impl WebDavCmd {
+    fn inner_run(&self) -> Result<()> {
+        let config = RUSTIC_APP.config();
+
+        let repo = open_repository(&config)?.to_indexed()?;
+
+        let path_template = self
+            .path_template
+            .clone()
+            .unwrap_or("[{hostname}]/[{label}]/{time}".to_string());
+        let time_template = self
+            .time_template
+            .clone()
+            .unwrap_or("%Y-%m-%d_%H-%M-%S".to_string());
+
+        let sn_filter = |sn: &_| config.snapshot_filter.matches(sn);
+        let target_fs = if let Some(snap) = &self.snap {
+            let node = repo.node_from_snapshot_path(snap, sn_filter)?;
+            RusticWebDavFS::from_node(repo, node)
+        } else {
+            let snapshots = repo.get_matching_snapshots(sn_filter)?;
+            RusticWebDavFS::from_snapshots(
+                repo,
+                snapshots,
+                path_template,
+                time_template,
+                self.symlinks,
+            )?
+        };
+        let addr = SocketAddr::from_str(&self.socket)?;
+        let dav_server = DavHandler::builder().filesystem(target_fs).build_handler();
+
+        tokio::runtime::Builder::new_current_thread()
+            .enable_all()
+            .build()?
+            .block_on(async {
+                warp::serve(dav_handler(dav_server)).run(addr).await;
+            });
+
+        Ok(())
+    }
+}