feat: Forget a birthday

README.md

@@ -67,4 +67,8 @@ $ birthday search --month 3
 ├────┼────────────┼──────────────┼───────────┼───────────────┤
 │ 1  │ Ben Dover  │ 3 may        | 34 (1990) │ today         │
 ╰────┴────────────┴──────────────┴───────────┴───────────────╯
+
+# Forget a birthday by ID
+$ birthday forget 3
+Birthday of 'Anita Bath' has been removed 🗑️
 ```

src/birthday.rs

@@ -1,5 +1,6 @@
 use chrono::{Datelike, NaiveDate};
 
+#[derive(Clone)]
 pub struct Birthday {
     pub id: i32,
     pub name: String,

src/birthday_store.rs

@@ -25,8 +25,27 @@ pub fn get_all() -> Result<Vec<Birthday>> {
         let date = from_timestamp(timestamp);
         Ok(Birthday { id, name, date })
     })?;
-    let birthdays: Result<Vec<Birthday>, rusqlite::Error> = birthday_iter.collect();
-    Ok(birthdays.unwrap())
+    let birthdays = birthday_iter.collect::<Result<Vec<Birthday>, rusqlite::Error>>()?;
+    Ok(birthdays)
+}
+
+pub fn remove(id: i32) -> Result<Option<Birthday>> {
+    let db = get_db()?;
+    let mut statement =
+        db.prepare("DELETE FROM birthdays WHERE id = :id RETURNING id, name, date_timestamp")?;
+    let birthday_iter = statement.query_map(&[(":id", id.to_string().as_str())], |row| {
+        let id = row.get(0)?;
+        let name = row.get(1)?;
+        let timestamp = row.get(2)?;
+        let date = from_timestamp(timestamp);
+        Ok(Birthday { id, name, date })
+    })?;
+    let birthdays = birthday_iter.collect::<Result<Vec<Birthday>, rusqlite::Error>>()?;
+    if birthdays.is_empty() {
+        Ok(None)
+    } else {
+        Ok(Some(birthdays[0].clone()))
+    }
 }
 
 fn get_db() -> Result<Connection> {

src/lib.rs

@@ -50,3 +50,7 @@ pub fn search(
 
     Ok(birthdays)
 }
+
+pub fn forget(id: i32) -> Result<Option<Birthday>> {
+    birthday_store::remove(id)
+}

src/main.rs

@@ -34,6 +34,8 @@ enum Command {
     },
     #[command(about = "Show today's birthdays")]
     Today {},
+    #[command(about = "Forget a birthday by ID")]
+    Forget { id: i32 },
 }
 
 fn main() -> Result<()> {
@@ -74,5 +76,13 @@ fn main() -> Result<()> {
             output::output(birthdays, today);
             Ok(())
         }
+        Command::Forget { id } => {
+            let maybe_birthday = birthday::forget(id)?;
+            match maybe_birthday {
+                Some(birthday) => output::birthday_forgotten(birthday),
+                None => output::no_birthday_found(),
+            }
+            Ok(())
+        }
     }
 }

src/output.rs

@@ -46,7 +46,7 @@ impl DisplayedBirthday {
 
 pub fn output(birthdays: Vec<Birthday>, today: NaiveDate) {
     if birthdays.is_empty() {
-        println!("No birthday found 🔎");
+        no_birthday_found();
         return;
     }
     let mut displayed_birthdays: Vec<DisplayedBirthday> = birthdays
@@ -59,3 +59,11 @@ pub fn output(birthdays: Vec<Birthday>, today: NaiveDate) {
         .to_string();
     println!("{table}")
 }
+
+pub fn no_birthday_found() {
+    println!("No birthday found 🔎");
+}
+
+pub fn birthday_forgotten(birthday: Birthday) {
+    println!("Birthday of '{}' has been forgotten 🗑️", birthday.name)
+}