Allow sync up and down

2 files changed, 125 insertions, 0 deletions

diff --git a/src/remote/git.rs b/src/remote/git.rs
new file mode 100644
index 0000000..a4c1c28
--- /dev/null
+++ b/src/remote/git.rs
@@ -0,0 +1,60 @@
+use std::io::Result;
+use std::path::PathBuf;
+use std::process::Command;
+use std::time::{SystemTime, UNIX_EPOCH};
+
+pub struct Git;
+
+impl Git {
+    pub fn new() -> Self {
+        Git
+    }
+}
+
+impl super::Remote for Git {
+    fn can_handle(&self, _: &str) -> bool {
+        // If we ever implement another remote we will want a check strategy
+        true
+    }
+
+    fn sync_up(&self, remote: &str, directory: &PathBuf) -> Result<()> {
+        let timestamp = SystemTime::now().duration_since(UNIX_EPOCH)
+            .expect("Invalid time")
+            .as_millis();
+
+        let commands = vec![
+            format!("cd {} && git init -b main", directory.display()),
+            format!("cd {} && git add .", directory.display()),
+            format!("cd {} && git commit --allow-empty -m blog-sync-up-{}", directory.display(), timestamp),
+            format!("cd {} && git push {} main --force", directory.display(), remote),
+        ];
+
+        for command in commands {
+            Command::new("sh")
+                .arg("-c")
+                .arg(&command)
+                .status()
+                .expect("Failed while performing sync up with git");
+        }
+
+        Ok(())
+    }
+
+    fn sync_down(&self, remote: &str, directory: &PathBuf) -> Result<()> {
+        let commands = vec![
+            format!("cd {} && git init -b main", directory.display()),
+            format!("cd {} && git checkout .", directory.display()),
+            format!("cd {} && git clean . -f", directory.display()),
+            format!("cd {} && git pull {} main", directory.display(), remote),
+        ];
+
+        for command in commands {
+            Command::new("sh")
+                .arg("-c")
+                .arg(&command)
+                .status()
+                .expect("Failed while performing sync down with git");
+        }
+        Ok(())
+    }
+}
diff --git a/src/remote/mod.rs b/src/remote/mod.rs
index e69de29..de90ef5 100644
--- 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)
+}