From f1f68ad5a62211546a34d2cecb0b3c7285d5bcf8 Mon Sep 17 00:00:00 2001
From: dvkt <c@dvkt.io>
Date: Thu, 19 Dec 2019 18:31:13 -0800
Subject: [PATCH] load history

---
 src/main.rs |  1 +
 src/stub.rs | 29 +++++++++++++++++++++++++++++
 src/ui.rs   | 53 +++++++++++++++++++++++++++++++++++++++++++----------
 3 files changed, 73 insertions(+), 10 deletions(-)
 create mode 100644 src/stub.rs

diff --git a/src/main.rs b/src/main.rs
index 683e426..df3915d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -5,6 +5,7 @@ extern crate termion;
 mod gopher;
 mod help;
 mod menu;
+mod stub;
 mod text;
 mod ui;
 use std::process::exit;
diff --git a/src/stub.rs b/src/stub.rs
new file mode 100644
index 0000000..361ace5
--- /dev/null
+++ b/src/stub.rs
@@ -0,0 +1,29 @@
+use ui::{Action, Key, View};
+
+pub struct Stub {
+    url: String,
+}
+
+impl View for Stub {
+    fn respond(&mut self, key: Key) -> Action {
+        Action::Keypress(key)
+    }
+    fn render(&self) -> String {
+        self.url()
+    }
+    fn url(&self) -> String {
+        self.url.to_string()
+    }
+    fn raw(&self) -> String {
+        self.url()
+    }
+    fn term_size(&mut self, _cols: usize, _rows: usize) {}
+}
+
+impl Stub {
+    pub fn from(url: &str) -> Stub {
+        Stub {
+            url: url.to_string(),
+        }
+    }
+}
diff --git a/src/ui.rs b/src/ui.rs
index b84e95c..573d2a2 100644
--- a/src/ui.rs
+++ b/src/ui.rs
@@ -1,5 +1,5 @@
 use std::io;
-use std::io::{stdin, stdout, Write};
+use std::io::{stdin, stdout, BufRead, BufReader, Write};
 use std::process;
 use std::process::Stdio;
 use termion::color;
@@ -11,6 +11,7 @@ use gopher::io_error;
 use gopher::Type;
 use help;
 use menu::Menu;
+use stub::Stub;
 use text::Text;
 
 pub type Key = termion::event::Key;
@@ -62,6 +63,7 @@ impl UI {
     }
 
     pub fn run(&mut self) {
+        self.startup();
         while self.running {
             self.draw();
             self.update();
@@ -158,33 +160,64 @@ impl UI {
         }
     }
 
+    fn startup(&mut self) {
+        self.load_history();
+    }
+
     fn shutdown(&self) {
         self.save_history();
     }
 
-    fn save_history(&self) {
-        use std::fs::OpenOptions;
-        use std::path::Path;
-
+    fn config_dir_path(&self) -> Option<std::path::PathBuf> {
         let homevar = std::env::var("HOME");
         if homevar.is_err() {
-            return;
+            return None;
         }
 
         let dotdir = "~/.config/phetch".replace('~', &homevar.unwrap());
-        let dotdir = Path::new(&dotdir);
-        if !dotdir.exists() {
+        let dotdir = std::path::Path::new(&dotdir);
+        if dotdir.exists() {
+            Some(std::path::PathBuf::from(dotdir))
+        } else {
+            None
+        }
+    }
+
+    fn load_history(&mut self) {
+        let dotdir = self.config_dir_path();
+        if dotdir.is_none() {
             return;
         }
+        let history = dotdir.unwrap().join("history");
+        if let Ok(file) = std::fs::OpenOptions::new().read(true).open(history) {
+            let buffered = BufReader::new(file);
+            let mut lines = buffered.lines();
+            while let Some(Ok(url)) = lines.next() {
+                self.add_page(Stub::from(&url));
+            }
+        }
+    }
 
+    fn save_history(&self) {
+        let dotdir = self.config_dir_path();
+        if dotdir.is_none() {
+            return;
+        }
+        let dotdir = dotdir.unwrap();
         let mut out = String::new();
         for page in &self.pages {
             out.push_str(&page.url());
             out.push('\n');
         }
-        let history = dotdir.join("history");
-        if let Ok(mut file) = OpenOptions::new().append(true).create(true).open(history) {
+        let history_tmp = dotdir.join("history.tmp");
+        if let Ok(mut file) = std::fs::OpenOptions::new()
+            .write(true)
+            .create(true)
+            .open(history_tmp.clone())
+        {
             file.write_all(out.as_ref());
+            let history = dotdir.join("history");
+            std::fs::rename(history_tmp, history);
         }
     }