summaryrefslogtreecommitdiff
path: root/src/operations
diff options
context:
space:
mode:
authorBen Bridle <bridle.benjamin@gmail.com>2022-08-25 21:27:39 +1200
committerBen Bridle <bridle.benjamin@gmail.com>2022-08-25 21:27:39 +1200
commit8f410d1ead74b979481f1488a4dcddd33ea829c7 (patch)
tree2f22fd930c8d0cdb4de53fef473f7770073e14d5 /src/operations
downloadvagabond-1.0.zip
Initial commitv1.0
Diffstat (limited to 'src/operations')
-rw-r--r--src/operations/cp.rs62
-rw-r--r--src/operations/ls.rs51
-rw-r--r--src/operations/mkdir.rs20
-rw-r--r--src/operations/rm.rs23
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(())
+}