fix some bugs

src-tauri/src/db.rs

@@ -20,6 +20,15 @@ pub struct Event {
     pub content: String,
 }
 
+#[derive(Serialize, Deserialize, Clone)]
+pub struct Reminder {
+    pub id: i64,
+    pub date: String,
+    pub minute: i32,
+    pub content: String,
+    pub is_completed: bool,
+}
+
 pub fn init_db(path: &str) -> anyhow::Result<()> {
     let conn = Connection::open(path)?;
     
@@ -49,6 +58,17 @@ pub fn init_db(path: &str) -> anyhow::Result<()> {
         [],
     )?;
 
+    conn.execute(
+        "CREATE TABLE IF NOT EXISTS reminders (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            date TEXT NOT NULL,
+            minute INTEGER NOT NULL,
+            content TEXT NOT NULL,
+            is_completed BOOLEAN NOT NULL DEFAULT 0
+        )",
+        [],
+    )?;
+
     Ok(())
 }
 
@@ -152,3 +172,51 @@ pub fn delete_event(path: &str, id: i64) -> anyhow::Result<()> {
     conn.execute("DELETE FROM events WHERE id = ?1", params![id])?;
     Ok(())
 }
+
+pub fn get_reminders(path: &str, date: &str) -> anyhow::Result<Vec<Reminder>> {
+    let conn = Connection::open(path)?;
+    let mut stmt = conn.prepare("SELECT id, date, minute, content, is_completed FROM reminders WHERE date = ?1 ORDER BY minute")?;
+    let iter = stmt.query_map(params![date], |row| {
+        Ok(Reminder {
+            id: row.get(0)?,
+            date: row.get(1)?,
+            minute: row.get(2)?,
+            content: row.get(3)?,
+            is_completed: row.get(4)?,
+        })
+    })?;
+    let mut reminders = Vec::new();
+    for r in iter {
+        reminders.push(r?);
+    }
+    Ok(reminders)
+}
+
+pub fn save_reminder(path: &str, reminder: Reminder) -> anyhow::Result<i64> {
+    let conn = Connection::open(path)?;
+    if reminder.id > 0 {
+        conn.execute(
+            "UPDATE reminders SET date=?1, minute=?2, content=?3, is_completed=?4 WHERE id=?5",
+            params![reminder.date, reminder.minute, reminder.content, reminder.is_completed, reminder.id],
+        )?;
+        Ok(reminder.id)
+    } else {
+        conn.execute(
+            "INSERT INTO reminders (date, minute, content, is_completed) VALUES (?1, ?2, ?3, ?4)",
+            params![reminder.date, reminder.minute, reminder.content, reminder.is_completed],
+        )?;
+        Ok(conn.last_insert_rowid())
+    }
+}
+
+pub fn delete_reminder(path: &str, id: i64) -> anyhow::Result<()> {
+    let conn = Connection::open(path)?;
+    conn.execute("DELETE FROM reminders WHERE id = ?1", params![id])?;
+    Ok(())
+}
+
+pub fn toggle_reminder(path: &str, id: i64, is_completed: bool) -> anyhow::Result<()> {
+    let conn = Connection::open(path)?;
+    conn.execute("UPDATE reminders SET is_completed=?1 WHERE id=?2", params![is_completed, id])?;
+    Ok(())
+}

src-tauri/src/engine.rs

@@ -72,6 +72,34 @@ pub fn delete_event(state: tauri::State<'_, AppState>, id: i64) -> Result<(), St
     crate::db::delete_event(path, id).map_err(|e| e.to_string())
 }
 
+#[tauri::command]
+pub fn get_reminders(state: tauri::State<'_, AppState>, date: String) -> Result<Vec<crate::db::Reminder>, String> {
+    let path = state.db_path.lock().unwrap();
+    let path = path.as_ref().ok_or("Database path not set")?;
+    crate::db::get_reminders(path, &date).map_err(|e| e.to_string())
+}
+
+#[tauri::command]
+pub fn save_reminder(state: tauri::State<'_, AppState>, reminder: crate::db::Reminder) -> Result<i64, String> {
+    let path = state.db_path.lock().unwrap();
+    let path = path.as_ref().ok_or("Database path not set")?;
+    crate::db::save_reminder(path, reminder).map_err(|e| e.to_string())
+}
+
+#[tauri::command]
+pub fn delete_reminder(state: tauri::State<'_, AppState>, id: i64) -> Result<(), String> {
+    let path = state.db_path.lock().unwrap();
+    let path = path.as_ref().ok_or("Database path not set")?;
+    crate::db::delete_reminder(path, id).map_err(|e| e.to_string())
+}
+
+#[tauri::command]
+pub fn toggle_reminder(state: tauri::State<'_, AppState>, id: i64, is_completed: bool) -> Result<(), String> {
+    let path = state.db_path.lock().unwrap();
+    let path = path.as_ref().ok_or("Database path not set")?;
+    crate::db::toggle_reminder(path, id, is_completed).map_err(|e| e.to_string())
+}
+
 #[tauri::command]
 pub fn update_interval(state: tauri::State<'_, AppState>, seconds: u64) {
     state.capture_interval_secs.store(seconds, Ordering::SeqCst);

src-tauri/src/lib.rs

@@ -11,6 +11,7 @@ pub fn run() {
     tauri::Builder::default()
         .plugin(tauri_plugin_dialog::init())
         .plugin(tauri_plugin_fs::init())
+        .plugin(tauri_plugin_notification::init())
         .plugin(tauri_plugin_store::Builder::new().build())
         .plugin(tauri_plugin_autostart::init(
             tauri_plugin_autostart::MacosLauncher::LaunchAgent,
@@ -31,6 +32,10 @@ pub fn run() {
             engine::get_events_range,
             engine::save_event,
             engine::delete_event,
+            engine::get_reminders,
+            engine::save_reminder,
+            engine::delete_reminder,
+            engine::toggle_reminder,
             engine::write_file
         ])
         .setup(|app| {