Working on cache

Cargo.toml

@@ -6,3 +6,11 @@ edition = "2018"
 
 [dependencies]
 colored = "1.7.0"
+
+[[bin]]
+name = "gclone"
+path = "src/gclone.rs"
+
+[[bin]]
+name = "cdg"
+path = "src/cdg.rs"

src/cdg.rs

@@ -0,0 +1,5 @@
+use gclone::Cache;
+
+fn main() {
+    let cache = Cache::read();
+}

src/gclone.rs

@@ -6,6 +6,8 @@ use std::path::PathBuf;
 
 use colored::*;
 
+use gclone::{git_dir, Cache};
+
 macro_rules! unwrap {
     ($e: expr) => {
         match $e {
@@ -55,17 +57,7 @@ fn help() {
 
 fn main() {
 
-    let git_dir = match env::var("GCLONE_PATH") {
+    let git_dir = git_dir();
-        Ok(path) => path,
-        Err(e) => {
-            eprintln!("{} {} {}",
-                "error:".bold().red(),
-                "environment variable GCLONE_PATH must be set:".bold(),
-                e.description());
-
-            exit(1);
-        },
-    };
 
     // Parse args
     let url = match env::args().nth(1) {
@@ -144,4 +136,18 @@ fn main() {
 
     eprintln!("{} {}", "info:".bold(), "done!");
 
+    // Append to cache
+    let mut cache = Cache::read();
+    cache.append(path.display().to_string());
+
+    match cache.write() {
+        Ok(_) => (),
+        Err(e) => {
+            eprintln!("{} {} {}",
+                "warning:".bold().yellow(),
+                "couldn't write cache:".bold(),
+                e.description());
+        },
+    }
+
 }

src/lib.rs

@@ -0,0 +1,66 @@
+use std::env;
+use std::process::exit;
+use std::error::Error;
+use std::path::PathBuf;
+use std::fs::File;
+use std::io::{Write, BufRead, BufReader};
+
+use colored::*;
+
+pub fn git_dir() -> String {
+    match env::var("GCLONE_PATH") {
+        Ok(path) => path,
+        Err(e) => {
+            eprintln!("{} {} {}",
+                "error:".bold().red(),
+                "environment variable GCLONE_PATH must be set:".bold(),
+                e.description());
+
+            exit(1);
+        },
+    }
+}
+
+pub struct Cache(Vec<String>);
+
+impl Cache {
+    pub fn read() -> Cache {
+
+        let mut path = PathBuf::from(git_dir());
+        path.push(".cdgcache");
+
+        let file = match File::open(&path) {
+            Ok(f) => f,
+            Err(_) => return Cache(vec![]),
+        };
+
+        let file = BufReader::new(file);
+
+        let mut values = vec![];
+
+        for line in file.lines() {
+            if let Ok(l) = line {
+                values.push(l);
+            }
+        }
+
+        Cache(values)
+    }
+
+    pub fn write(&self) -> Result<(), Box<Error>> {
+        let mut path = PathBuf::from(git_dir());
+        path.push(".cdgcache");
+        let mut file = File::create(path)?;
+
+        for line in &self.0 {
+            writeln!(file, "{}", line)?;
+        }
+
+        Ok(())
+    }
+
+    pub fn append(&mut self, value: String) {
+        self.0.push(value);
+    }
+}
+