69 lines
1.7 KiB
Rust
69 lines
1.7 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 iter = stmt.query_map((key,), |row| {
|
|
row.get(0)
|
|
}).unwrap();
|
|
|
|
for row in iter {
|
|
if let Ok(value) = row {
|
|
return Some(value);
|
|
}
|
|
return None;
|
|
}
|
|
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_or_else(|e| {
|
|
println!("{}", e);
|
|
0
|
|
});
|
|
}
|
|
|
|
fn create_settings(key: String, value: String, connection: &Connection) {
|
|
connection
|
|
.execute(
|
|
"INSERT into settings (id, value) VALUES (?1, ?2)",
|
|
(key, value),
|
|
)
|
|
.unwrap_or_else(|e| {
|
|
println!("{}", e);
|
|
0
|
|
});
|
|
}
|
|
|
|
fn update_settings(key: String, value: String, connection: &Connection) {
|
|
connection
|
|
.execute(
|
|
"update settings
|
|
set value = ?2
|
|
where id = ?1",
|
|
(key, value),
|
|
)
|
|
.unwrap_or_else(|e| {
|
|
println!("{}", e);
|
|
0
|
|
});}
|
|
|
|
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()
|
|
}
|
|
|