mod database;
mod memory;
use anyhow::Result;
use database::Database;
use gtk::glib::Uri;
use memory::Memory;
use r2d2::Pool;
use r2d2_sqlite::SqliteConnectionManager;
use sqlite::Transaction;
pub struct Search {
database: Database, memory: Memory, }
impl Search {
pub fn build(database_pool: &Pool<SqliteConnectionManager>, profile_id: i64) -> Result<Self> {
let database = Database::init(database_pool, profile_id)?;
let memory = Memory::init();
index(&database, &memory)?;
Ok(Self { database, memory })
}
pub fn add(&self, query: &Uri, is_default: bool) -> Result<()> {
self.database.add(query.to_string(), is_default)?;
Ok(())
}
pub fn set_default(&self, profile_search_id: i64) -> Result<()> {
self.database.set_default(profile_search_id)?;
index(&self.database, &self.memory)
}
pub fn records(&self) -> Vec<database::Row> {
self.memory.records()
}
pub fn delete(&self, id: i64) -> Result<()> {
self.database.delete(id)?;
index(&self.database, &self.memory)
}
pub fn default(&self) -> Option<database::Row> {
self.memory.default()
}
}
pub fn migrate(tx: &Transaction) -> Result<()> {
database::init(tx)?;
Ok(())
}
fn index(database: &Database, memory: &Memory) -> Result<()> {
memory.clear();
for record in database.records()? {
memory.push(record.id, record.query, record.is_default)
}
Ok(())
}