From d00364bbb90358d1e5383f9e4889bf2106eed36b Mon Sep 17 00:00:00 2001
From: yggverse <yggverse@project>
Date: Thu, 30 Jan 2025 00:07:43 +0200
Subject: [PATCH] add custom search providers db model

---
 src/profile/search/database.rs | 156 +++++++++++++++++++++++++++++++++
 1 file changed, 156 insertions(+)
 create mode 100644 src/profile/search/database.rs

diff --git a/src/profile/search/database.rs b/src/profile/search/database.rs
new file mode 100644
index 00000000..b033f4ef
--- /dev/null
+++ b/src/profile/search/database.rs
@@ -0,0 +1,156 @@
+use sqlite::{Connection, Error, Transaction};
+use std::{rc::Rc, sync::RwLock};
+
+pub struct Row {
+    pub id: i64,
+    //pub profile_id: i64,
+    pub is_default: bool,
+    pub query: String,
+    pub name: String,
+}
+
+pub struct Database {
+    connection: Rc<RwLock<Connection>>,
+    profile_id: Rc<i64>, // multi-profile relationship
+}
+
+impl Database {
+    // Constructors
+
+    /// Create new `Self`
+    pub fn init(connection: &Rc<RwLock<Connection>>, profile_id: &Rc<i64>) -> Self {
+        Self {
+            connection: connection.clone(),
+            profile_id: profile_id.clone(),
+        }
+    }
+
+    // Getters
+
+    /// Get records from database
+    pub fn records(&self) -> Result<Vec<Row>, Error> {
+        let readable = self.connection.read().unwrap(); // @TODO handle
+        let tx = readable.unchecked_transaction()?;
+        select(&tx, *self.profile_id)
+    }
+
+    // Setters
+
+    /// Create new record in database
+    /// * return last insert ID on success
+    pub fn add(&self, query: String, name: Option<String>, is_default: bool) -> Result<i64, Error> {
+        // Begin new transaction
+        let mut writable = self.connection.write().unwrap(); // @TODO handle
+        let tx = writable.transaction()?;
+
+        // Create new record
+        if is_default {
+            // make sure only one default provider in set
+            reset(&tx, *self.profile_id, !is_default)?;
+        }
+        insert(&tx, *self.profile_id, query, name, is_default)?;
+
+        // Hold insert ID for result
+        let id = last_insert_id(&tx);
+
+        // Done
+        match tx.commit() {
+            Ok(_) => Ok(id),
+            Err(e) => Err(e),
+        }
+    }
+
+    /// Delete record from database
+    pub fn delete(&self, id: i64) -> Result<(), Error> {
+        // Begin new transaction
+        let mut writable = self.connection.write().unwrap(); // @TODO
+        let tx = writable.transaction()?;
+
+        // Delete record by ID
+        match delete(&tx, id) {
+            Ok(_) => match tx.commit() {
+                Ok(_) => Ok(()),
+                Err(e) => Err(e),
+            },
+            Err(e) => Err(e),
+        }
+    }
+}
+
+// Low-level DB API
+
+pub fn init(tx: &Transaction) -> Result<usize, Error> {
+    tx.execute(
+        "CREATE TABLE IF NOT EXISTS `profile_search`
+        (
+            `id`         INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
+            `profile_id` INTEGER NOT NULL,
+            `is_default` INTEGER NOT NULL,
+            `query`      TEXT NOT NULL,
+            `name`       VARCHAR(64),
+
+            FOREIGN KEY (`profile_id`) REFERENCES `profile` (`id`)
+        )",
+        [],
+    )
+}
+
+fn insert(
+    tx: &Transaction,
+    profile_id: i64,
+    query: String,
+    name: Option<String>,
+    is_default: bool,
+) -> Result<usize, Error> {
+    tx.execute(
+        "INSERT INTO `profile_search` (
+            `profile_id`,
+            `is_default`,
+            `query`,
+            `name`
+        ) VALUES (?, ?, ?, ?)",
+        (profile_id, is_default, query, name),
+    )
+}
+
+fn select(tx: &Transaction, profile_id: i64) -> Result<Vec<Row>, Error> {
+    let mut stmt = tx.prepare(
+        "SELECT `id`, `profile_id`, `is_default`, `query`, `name`
+            FROM `profile_search`
+            WHERE `profile_id` = ?",
+    )?;
+
+    let result = stmt.query_map([profile_id], |row| {
+        Ok(Row {
+            id: row.get(0)?,
+            //profile_id: row.get(1)?,
+            is_default: row.get(2)?,
+            name: row.get(3)?,
+            query: row.get(4)?,
+        })
+    })?;
+
+    let mut records = Vec::new();
+
+    for record in result {
+        let table = record?;
+        records.push(table);
+    }
+
+    Ok(records)
+}
+
+fn delete(tx: &Transaction, id: i64) -> Result<usize, Error> {
+    tx.execute("DELETE FROM `profile_search` WHERE `id` = ?", [id])
+}
+
+fn reset(tx: &Transaction, profile_id: i64, is_default: bool) -> Result<usize, Error> {
+    tx.execute(
+        "UPDATE `profile_search` SET `is_default` = ? WHERE `profile_id` = ?",
+        (is_default, profile_id),
+    )
+}
+
+fn last_insert_id(tx: &Transaction) -> i64 {
+    tx.last_insert_rowid()
+}