diff options
author | Ben Bridle <bridle.benjamin@gmail.com> | 2022-08-25 21:27:39 +1200 |
---|---|---|
committer | Ben Bridle <bridle.benjamin@gmail.com> | 2022-08-25 21:27:39 +1200 |
commit | 8f410d1ead74b979481f1488a4dcddd33ea829c7 (patch) | |
tree | 2f22fd930c8d0cdb4de53fef473f7770073e14d5 /src/operations | |
download | vagabond-1.0.zip |
Initial commitv1.0
Diffstat (limited to 'src/operations')
-rw-r--r-- | src/operations/cp.rs | 62 | ||||
-rw-r--r-- | src/operations/ls.rs | 51 | ||||
-rw-r--r-- | src/operations/mkdir.rs | 20 | ||||
-rw-r--r-- | src/operations/rm.rs | 23 |
4 files changed, 156 insertions, 0 deletions
diff --git a/src/operations/cp.rs b/src/operations/cp.rs new file mode 100644 index 0000000..be43826 --- /dev/null +++ b/src/operations/cp.rs @@ -0,0 +1,62 @@ +use crate::make_parent_directory; +use crate::{get_entry, get_optional_entry, list_directory}; +use crate::{EntryType, EntryWriteError}; +use std::path::Path; + +pub fn copy<P, Q>(source_path: P, target_path: Q) -> Result<(), EntryWriteError> +where + P: AsRef<Path>, + Q: AsRef<Path>, +{ + let source = get_entry(&source_path)?; + let target = get_optional_entry(&target_path)?; + let target_type = target.and_then(|e| Some(e.entry_type)); + + match (source.entry_type, target_type) { + (EntryType::File, Some(EntryType::File)) => { + copy_file(source_path, target_path)?; + } + (EntryType::File, Some(EntryType::Directory)) => { + let target_path = target_path.as_ref().join(source.name); + copy_file(source_path, target_path)?; + } + (EntryType::File, None) => { + make_parent_directory(&target_path)?; + copy_file(source_path, target_path)?; + } + (EntryType::Directory, Some(EntryType::File)) => { + std::fs::remove_file(&target_path)?; + copy_directory(&source_path, &target_path)?; + } + (EntryType::Directory, Some(EntryType::Directory)) => { + let target_path = target_path.as_ref().join(source.name); + copy_directory(source_path, target_path)?; + } + (EntryType::Directory, None) => { + make_parent_directory(&target_path)?; + copy_directory(source_path, target_path)?; + } + } + Ok(()) +} + +fn copy_file<P, Q>(source_path: P, target_path: Q) -> Result<(), EntryWriteError> +where + P: AsRef<Path>, + Q: AsRef<Path>, +{ + std::fs::copy(source_path, target_path)?; + Ok(()) +} + +fn copy_directory<P, Q>(source_path: P, target_path: Q) -> Result<(), EntryWriteError> +where + P: AsRef<Path>, + Q: AsRef<Path>, +{ + for entry in list_directory(&source_path)? { + let target_path = target_path.as_ref().join(entry.name); + copy(entry.path, &target_path)?; + } + Ok(()) +} diff --git a/src/operations/ls.rs b/src/operations/ls.rs new file mode 100644 index 0000000..2d2da5d --- /dev/null +++ b/src/operations/ls.rs @@ -0,0 +1,51 @@ +use crate::{Entry, EntryReadError, EntryType}; +use std::path::Path; + +pub fn get_entry<P>(path: P) -> Result<Entry, EntryReadError> +where + P: AsRef<Path>, +{ + Entry::from_path(path) +} + +pub fn get_optional_entry<P>(path: P) -> Result<Option<Entry>, EntryReadError> +where + P: AsRef<Path>, +{ + match get_entry(path) { + Ok(e) => Ok(Some(e)), + Err(EntryReadError::NotFound) => Ok(None), + Err(other) => Err(other), + } +} + +pub fn list_directory<P>(path: P) -> Result<Vec<Entry>, EntryReadError> +where + P: AsRef<Path>, +{ + let mut entries = Vec::new(); + for dir_entry in std::fs::read_dir(path)? { + let entry = match Entry::from_path(&dir_entry?.path()) { + Ok(v) => v, + Err(_) => continue, + }; + entries.push(entry); + } + return Ok(entries); +} + +/// Recursively descend into a directory and all sub-directories, +/// returning an [`Entry`](struct.Entry.html) for each discovered file. +pub fn traverse_directory<P>(path: P) -> Result<Vec<Entry>, EntryReadError> +where + P: AsRef<Path>, +{ + let mut file_entries = Vec::new(); + for entry in list_directory(path)? { + match entry.entry_type { + EntryType::File => file_entries.push(entry), + EntryType::Directory => file_entries.extend(traverse_directory(&entry.path)?), + } + } + return Ok(file_entries); +} diff --git a/src/operations/mkdir.rs b/src/operations/mkdir.rs new file mode 100644 index 0000000..011b8cf --- /dev/null +++ b/src/operations/mkdir.rs @@ -0,0 +1,20 @@ +use crate::EntryWriteError; +use std::path::Path; + +pub fn make_directory<P>(path: P) -> Result<(), EntryWriteError> +where + P: AsRef<Path>, +{ + std::fs::DirBuilder::new().recursive(true).create(path)?; + Ok(()) +} + +pub fn make_parent_directory<P>(path: P) -> Result<(), EntryWriteError> +where + P: AsRef<Path>, +{ + match path.as_ref().parent() { + Some(parent) => make_directory(parent), + None => Ok(()), + } +} diff --git a/src/operations/rm.rs b/src/operations/rm.rs new file mode 100644 index 0000000..846a094 --- /dev/null +++ b/src/operations/rm.rs @@ -0,0 +1,23 @@ +use crate::EntryWriteError; +use crate::{get_entry, EntryType}; +use std::path::Path; + +pub fn remove<P>(path: P) -> Result<(), EntryWriteError> +where + P: AsRef<Path>, +{ + let entry = get_entry(&path)?; + match entry.entry_type { + EntryType::File => std::fs::remove_file(&path)?, + EntryType::Directory => std::fs::remove_dir_all(&path)?, + } + Ok(()) +} + +pub fn remove_file<P>(path: P) -> Result<(), EntryWriteError> +where + P: AsRef<Path>, +{ + std::fs::remove_file(path)?; + Ok(()) +} |