feat: back up and restore symlinks

1 files changed, 17 insertions, 0 deletions

diff --git a/src/cmd/restore.rs b/src/cmd/restore.rs
index 2b62ef5..27f0ce3 100644
--- a/src/cmd/restore.rs
+++ b/src/cmd/restore.rs
@@ -8,6 +8,7 @@ use log::{debug, error, info};
 use std::fs::File;
 use std::io::prelude::*;
 use std::io::Error;
+use std::os::unix::fs::symlink;
 use std::os::unix::io::AsRawFd;
 use std::path::{Path, PathBuf};
 use structopt::StructOpt;
@@ -81,6 +82,7 @@ fn restore_generation(
     to: &Path,
     progress: &ProgressBar,
 ) -> anyhow::Result<()> {
+    debug!("restoring {:?}", entry);
     progress.set_message(&format!("{}", entry.path().display()));
     progress.inc(1);
 
@@ -88,6 +90,7 @@ fn restore_generation(
     match entry.kind() {
         FilesystemKind::Regular => restore_regular(client, &gen, &to, fileid, &entry)?,
         FilesystemKind::Directory => restore_directory(&to)?,
+        FilesystemKind::Symlink => restore_symlink(&to, &entry)?,
     }
     Ok(())
 }
@@ -142,6 +145,20 @@ fn restore_regular(
     Ok(())
 }
 
+fn restore_symlink(path: &Path, entry: &FilesystemEntry) -> anyhow::Result<()> {
+    debug!("restoring symlink {}", path.display());
+    let parent = path.parent().unwrap();
+    debug!("  mkdir {}", parent.display());
+    if !parent.exists() {
+        std::fs::create_dir_all(parent)?;
+        {
+            symlink(path, entry.symlink_target().unwrap())?;
+        }
+    }
+    debug!("restored regular {}", path.display());
+    Ok(())
+}
+
 fn restore_metadata(path: &Path, entry: &FilesystemEntry) -> anyhow::Result<()> {
     debug!("restoring metadata for {}", entry.path().display());