mirror of https://github.com/YGGverse/Yoda.git
yggverse
2 months ago
4 changed files with 212 additions and 25 deletions
@ -0,0 +1,88 @@
@@ -0,0 +1,88 @@
|
||||
use sqlite::{Error, Transaction}; |
||||
|
||||
pub struct Table { |
||||
pub id: i64, |
||||
// pub app_browser_window_tab_id: i64, not in use
|
||||
pub is_pinned: bool, |
||||
} |
||||
|
||||
pub struct Database { |
||||
// nothing yet..
|
||||
} |
||||
|
||||
impl Database { |
||||
pub fn new() -> Self { |
||||
Self {} |
||||
} |
||||
|
||||
pub fn init(tx: &Transaction) -> Result<Database, Error> { |
||||
tx.execute( |
||||
"CREATE TABLE IF NOT EXISTS `app_browser_window_tab_label` |
||||
( |
||||
`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, |
||||
`app_browser_window_tab_id` INTEGER NOT NULL, |
||||
`is_pinned` INTEGER NOT NULL |
||||
)", |
||||
[], |
||||
)?; |
||||
|
||||
Ok(Self {}) |
||||
} |
||||
|
||||
pub fn add( |
||||
&self, |
||||
tx: &Transaction, |
||||
app_browser_window_tab_id: &i64, |
||||
is_pinned: &bool, |
||||
) -> Result<usize, Error> { |
||||
tx.execute( |
||||
"INSERT INTO `app_browser_window_tab_label` ( |
||||
`app_browser_window_tab_id`, |
||||
`is_pinned` |
||||
) VALUES (?,?)", |
||||
[app_browser_window_tab_id, &(*is_pinned as i64)], |
||||
) |
||||
} |
||||
|
||||
pub fn records( |
||||
&self, |
||||
tx: &Transaction, |
||||
app_browser_window_tab_id: &i64, |
||||
) -> Result<Vec<Table>, Error> { |
||||
let mut stmt = tx.prepare( |
||||
"SELECT `id`, |
||||
`app_browser_window_tab_id`, |
||||
`is_pinned` FROM `app_browser_window_tab_label` |
||||
WHERE `app_browser_window_tab_id` = ?", |
||||
)?; |
||||
|
||||
let result = stmt.query_map([app_browser_window_tab_id], |row| { |
||||
Ok(Table { |
||||
id: row.get(0)?, |
||||
// app_browser_window_tab_id: row.get(1)?, not in use
|
||||
is_pinned: row.get(2)?, |
||||
}) |
||||
})?; |
||||
|
||||
let mut records = Vec::new(); |
||||
|
||||
for record in result { |
||||
let table = record?; |
||||
records.push(table); |
||||
} |
||||
|
||||
Ok(records) |
||||
} |
||||
|
||||
pub fn delete(&self, tx: &Transaction, id: &i64) -> Result<usize, Error> { |
||||
tx.execute( |
||||
"DELETE FROM `app_browser_window_tab_label` WHERE `id` = ?", |
||||
[id], |
||||
) |
||||
} |
||||
|
||||
/* not in use
|
||||
pub fn last_insert_id(&self, tx: &Transaction) -> i64 { |
||||
tx.last_insert_rowid() |
||||
} */ |
||||
} |
Loading…
Reference in new issue