Migrate to tokio_rusqlite

1 files changed, 24 insertions, 0 deletions

diff --git a/src/db/utils.rs b/src/db/utils.rs
index 621a69c..f61d20a 100644
--- a/src/db/utils.rs
+++ b/src/db/utils.rs
@@ -1,3 +1,27 @@
+use crate::model::report::Report;
+use tokio_rusqlite::Row;
+
 pub fn format_key_for_search(value: &str) -> String {
     format!("replace(replace(replace(replace(replace(replace(replace(replace(replace(replace(replace(replace(replace(lower({}), 'à', 'a'), 'â', 'a'), 'ç', 'c'), 'è', 'e'), 'é', 'e'), 'ê', 'e'), 'ë', 'e'), 'î', 'i'), 'ï', 'i'), 'ô', 'o'), 'ù', 'u'), 'û', 'u'), 'ü', 'u')", value)
 }
+
+pub fn one<A, I: Iterator<Item = Result<A, rusqlite::Error>>>(
+    mut iter: I,
+) -> Result<A, tokio_rusqlite::Error> {
+    match iter.next() {
+        Some(Ok(user)) => Ok(user),
+        Some(Err(err)) => Err(tokio_rusqlite::Error::Rusqlite(err)),
+        None => Err(tokio_rusqlite::Error::Rusqlite(
+            rusqlite::Error::QueryReturnedNoRows,
+        )),
+    }
+}
+
+pub fn row_to_report(row: &Row) -> Result<Report, rusqlite::Error> {
+    Ok(Report {
+        date: row.get(0)?,
+        name: row.get(1)?,
+        amount: row.get(2)?,
+        action: row.get(3)?,
+    })
+}