diff options
| -rw-r--r-- | src/main.rs | 17 |
1 files changed, 7 insertions, 10 deletions
diff --git a/src/main.rs b/src/main.rs index c21f7c2..1edece5 100644 --- a/src/main.rs +++ b/src/main.rs @@ -198,19 +198,14 @@ impl Bk { } } -fn save_path() -> String { - let home = std::env::var("HOME").unwrap(); - format!("{}/.local/share/bk", home) -} - -fn restore() -> (String, usize, usize) { - let save = std::fs::read_to_string(save_path()).unwrap(); +fn restore(save_path: &str) -> (String, usize, usize) { + let save = std::fs::read_to_string(save_path).unwrap(); let mut lines = save.lines(); let path = lines.next().unwrap().to_string(); if let Some(p) = std::env::args().nth(1) { if p != path { - return (p, 0, 0); + return (p, 0, 0) } } ( @@ -221,7 +216,9 @@ fn restore() -> (String, usize, usize) { } fn main() -> crossterm::Result<()> { - let (path, chapter, pos) = restore(); + let save_path = format!("{}/.local/share/bk", + std::env::var("HOME").unwrap()); + let (path, chapter, pos) = restore(&save_path); let (cols, rows) = terminal::size().unwrap(); let mut bk = Bk::new(&path, cols, rows as usize, chapter, pos, 2) @@ -260,7 +257,7 @@ fn main() -> crossterm::Result<()> { } std::fs::write( - save_path(), + save_path, format!("{}\n{}\n{}", path, bk.chapter_idx, bk.pos), ) .unwrap(); |