jrnl: adding entries now works!!

programs/jrnl/src/commands/add_entry.rs

@@ -0,0 +1,68 @@
+use std::{
+    env,
+    fs::{self, OpenOptions},
+    io::{self, Write},
+    path::PathBuf,
+    process::Command,
+};
+
+use temp_file::{TempFile, TempFileBuilder};
+
+use crate::md::{Entry, ToMd};
+
+// TODO: the usual (better error handling)
+pub fn add_entry(path: PathBuf, title: Option<String>) -> io::Result<()> {
+    if !path.exists() {
+        eprintln!("Journal file does not exist at {path:?}, exiting...");
+        std::process::exit(1);
+    }
+
+    let title = prompt("Title")?;
+
+    let tmp = TempFileBuilder::new()
+        .suffix(".jrnl-entry.md")
+        .build()
+        .unwrap();
+
+    let editor = match env::var("EDITOR") {
+        Ok(val) => val,
+        Err(env::VarError::NotPresent) => {
+            eprintln!("EDITOR not set, exiting...");
+            std::process::exit(1);
+        }
+        _ => unreachable!(),
+    };
+
+    let mut editor_cmd = Command::new(&editor);
+    editor_cmd.arg(tmp.path());
+    editor_cmd.status().unwrap();
+
+    let content = fs::read_to_string(tmp.path()).unwrap();
+
+    let now = chrono::offset::Local::now();
+
+    let entry = Entry {
+        timestamp: now.fixed_offset(),
+        title: &title,
+        content: &content,
+    };
+
+    let mut file = OpenOptions::new()
+        .write(true)
+        .append(true)
+        .open(path)
+        .unwrap();
+
+    write!(file, "{}", entry.to_md())?;
+
+    Ok(())
+}
+
+fn prompt(title: &str) -> io::Result<String> {
+    print!("{}: ", title);
+    let _ = io::stdout().flush();
+    let mut buf = String::new();
+    let stdin = io::stdin();
+    stdin.read_line(&mut buf)?;
+    Ok(buf)
+}

programs/jrnl/src/commands/list_entries.rs

@@ -1,22 +1,23 @@
 use owo_colors::OwoColorize;
-use std::{fs, path::PathBuf};
+use std::{fs, io, path::PathBuf};
 
 use crate::md::Doc;
 
-pub fn list_entries(path: PathBuf) {
-    let file = fs::read_to_string(path).unwrap();
+pub fn list_entries(path: PathBuf) -> io::Result<()> {
+    let file = fs::read_to_string(path)?;
 
     if let Some(doc) = Doc::new(&file) {
+        let termsize::Size { cols, .. } = termsize::get().unwrap();
         for (i, entry) in doc.entries.into_iter().enumerate() {
             let n = format!("{:>2}", i + 1);
             let r = format!(". {}", entry.title,);
             let l = format!(" {} ", crate::utils::format_datetime(entry.timestamp));
-            let termsize::Size { cols, .. } = termsize::get().unwrap();
 
             let padding = " ".repeat(cols as usize - (n.len() + r.len() + l.len()));
 
             println!("{}{r}{padding}{}", n.cyan(), l.white())
         }
+        Ok(())
     } else {
         eprintln!("Parsing error...");
         std::process::exit(1);