Allow to attach categories to events

1 files changed, 14 insertions, 0 deletions

diff --git a/src/db/event_color.rs b/src/db/event_color.rs
new file mode 100644
index 0000000..33d350b
--- /dev/null
+++ b/src/db/event_color.rs
@@ -0,0 +1,14 @@
+use anyhow::Result;
+use rusqlite::Connection;
+
+pub fn get_default_color(conn: &Connection) -> Result<String> {
+    let mut stmt = conn.prepare("SELECT * FROM event_color LIMIT 1")?;
+
+    let iter = stmt.query_map([], |row| row.get(0))?;
+
+    let mut res = vec![];
+    for color in iter {
+        res.push(color?)
+    }
+    Ok(res.first().unwrap_or(&"blue".to_string()).clone())
+}