58 lines
1.5 KiB
Rust
58 lines
1.5 KiB
Rust
use rusqlite::Connection;
|
|
|
|
pub fn get_setting(key: String, connection: &Connection) -> Option<String> {
|
|
let mut stmt = connection
|
|
.prepare("SELECT value FROM settings WHERE id = ?1")
|
|
.unwrap();
|
|
let mut iter = stmt.query_map((key,), |row| row.get(0)).unwrap();
|
|
|
|
if let Some(row) = iter.next()
|
|
&& let Ok(value) = row
|
|
{
|
|
return Some(value);
|
|
}
|
|
|
|
None
|
|
}
|
|
|
|
pub fn set_setting(key: String, value: String, connection: &Connection) {
|
|
let current = get_settings_list(connection);
|
|
if current.contains(&key) {
|
|
update_settings(key, value, connection);
|
|
} else {
|
|
create_settings(key, value, connection);
|
|
}
|
|
}
|
|
|
|
pub fn delete_settings(key: String, connection: &Connection) {
|
|
connection
|
|
.execute("DELETE FROM settings WHERE id = ?1", (&key,))
|
|
.unwrap();
|
|
}
|
|
|
|
fn create_settings(key: String, value: String, connection: &Connection) {
|
|
connection
|
|
.execute(
|
|
"INSERT into settings (id, value) VALUES (?1, ?2)",
|
|
(key, value),
|
|
)
|
|
.unwrap();
|
|
}
|
|
|
|
fn update_settings(key: String, value: String, connection: &Connection) {
|
|
connection
|
|
.execute(
|
|
"update settings
|
|
set value = ?2
|
|
where id = ?1",
|
|
(key, value),
|
|
)
|
|
.unwrap();
|
|
}
|
|
|
|
fn get_settings_list(connection: &Connection) -> Vec<String> {
|
|
let mut stmt = connection.prepare("SELECT id FROM settings").unwrap();
|
|
let iter = stmt.query_map((), |row| row.get(0)).unwrap();
|
|
iter.map(|row| row.unwrap()).collect()
|
|
}
|