Allow sync up and down

[rbdr/blog] / src / remote / mod.rs

diff --git a/src/remote/mod.rs b/src/remote/mod.rs
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..de90ef57f4f2b6bb585ae38c996fe75cf849963f 100644 (file)
--- a/src/remote/mod.rs
+++ b/src/remote/mod.rs
@@ -0,0 +1,65 @@
+mod git;
+
+use std::fs::{create_dir_all, remove_file, write, File};
+use std::io::{Error, ErrorKind::Other, Read, Result};
+use std::path::PathBuf;
+
+use git::Git;
+
+pub trait Remote {
+    fn can_handle(&self, remote: &str) -> bool;
+    fn sync_up(&self, remote: &str, directory: &PathBuf) -> Result<()>;
+    fn sync_down(&self, remote: &str, directory: &PathBuf) -> Result<()>;
+}
+
+pub fn add(config_directory: &PathBuf, remote_config: &PathBuf, remote: &String) -> Result<()> {
+    create_dir_all(config_directory)?;
+    write(remote_config, remote)?;
+    Ok(())
+}
+
+pub fn remove(remote_config: &PathBuf) -> Result<()> {
+    if remote_config.exists() {
+        remove_file(remote_config)?
+    }
+    Ok(())
+}
+
+pub fn sync_up(data_directory: &PathBuf, remote_config: &PathBuf) -> Result<()> {
+    let remote_address = read_remote(remote_config)
+        .expect("No remote is configured");
+    create_dir_all(data_directory)?;
+    let remotes = available_remotes();
+    for remote in remotes {
+        if remote.can_handle(&remote_address) {
+            return remote.sync_up(&remote_address, data_directory);
+        }
+    }
+    Err(Error::new(Other, "No valid strategies found for your configured remote."))
+}
+
+pub fn sync_down(data_directory: &PathBuf, remote_config: &PathBuf) -> Result<()> {
+    let remote_address = read_remote(remote_config)
+        .expect("No remote is configured");
+    create_dir_all(data_directory)?;
+    let remotes = available_remotes();
+    for remote in remotes {
+        if remote.can_handle(&remote_address) {
+            return remote.sync_down(&remote_address, data_directory);
+        }
+    }
+    Err(Error::new(Other, "No valid strategies found for your configured remote."))
+}
+
+fn available_remotes() -> Vec<Box<dyn Remote>> {
+    vec![
+        Box::new(Git::new())
+    ]
+}
+
+fn read_remote(file_path: &PathBuf) -> Option<String> {
+    let mut file = File::open(file_path).ok()?;
+    let mut contents = String::new();
+    file.read_to_string(&mut contents).ok()?;
+    Some(contents)
+}