use ignore::WalkBuilder;
use std::{fs, path};
#[derive(Debug)]
pub struct Move<'a> {
source: &'a path::Path,
temp: Option<&'a path::Path>,
}
impl<'a> Move<'a> {
pub fn from_source(source: &'a path::Path) -> Move<'a> {
Self { source, temp: None }
}
pub fn replace_using_temp(&mut self, temp: &'a path::Path) -> &mut Self {
self.temp = Some(temp);
self
}
pub fn to_dest(&self, dest: &path::Path) -> crate::api::Result<()> {
match self.temp {
None => {
fs::rename(self.source, dest)?;
}
Some(temp) => {
if dest.exists() {
fs::rename(dest, temp)?;
if let Err(e) = fs::rename(self.source, dest) {
fs::rename(temp, dest)?;
return Err(e.into());
}
} else {
fs::rename(self.source, dest)?;
}
}
};
Ok(())
}
pub fn walk_to_dest(&self, dest: &path::Path) -> crate::api::Result<()> {
match self.temp {
None => {
walkdir_and_copy(self.source, dest)?;
}
Some(temp) => {
if dest.exists() {
walkdir_and_copy(dest, temp)?;
if let Err(e) = walkdir_and_copy(self.source, dest) {
fs::rename(temp, dest)?;
return Err(e);
}
} else {
walkdir_and_copy(self.source, dest)?;
}
}
};
Ok(())
}
}
fn walkdir_and_copy(source: &path::Path, dest: &path::Path) -> crate::api::Result<()> {
let walkdir = WalkBuilder::new(source).hidden(false).build();
for entry in walkdir {
let element = entry?;
let metadata = element.metadata()?;
let destination = dest.join(element.path().strip_prefix(&source)?);
if metadata.is_dir() && !&destination.exists() {
fs::create_dir_all(&destination)?;
}
if metadata.is_file() {
fs::copy(element.path(), destination)?;
}
}
Ok(())
}