Add 'core/' from commit 'b0dfc4146ca0da535a87f8509aec68817fb2ab14'

git-subtree-dir: core
git-subtree-mainline: a07f3dcd23
git-subtree-split: b0dfc4146c

core/kordophone-db/src/settings.rs

@@ -0,0 +1,63 @@
+use anyhow::Result;
+use diesel::*;
+use serde::{de::DeserializeOwned, Serialize};
+
+#[derive(Insertable, Queryable, AsChangeset)]
+#[diesel(table_name = crate::schema::settings)]
+struct SettingsRow<'a> {
+    key: &'a str,
+    value: &'a [u8],
+}
+
+pub struct Settings<'a> {
+    connection: &'a mut SqliteConnection,
+}
+
+impl<'a> Settings<'a> {
+    pub fn new(connection: &'a mut SqliteConnection) -> Self {
+        Self { connection }
+    }
+
+    pub fn put<T: Serialize>(&mut self, k: &str, v: &T) -> Result<()> {
+        use crate::schema::settings::dsl::*;
+        let bytes = bincode::serialize(v)?;
+
+        diesel::insert_into(settings)
+            .values(SettingsRow {
+                key: k,
+                value: &bytes,
+            })
+            .on_conflict(key)
+            .do_update()
+            .set(value.eq(&bytes))
+            .execute(self.connection)?;
+
+        Ok(())
+    }
+
+    pub fn get<T: DeserializeOwned>(&mut self, k: &str) -> Result<Option<T>> {
+        use crate::schema::settings::dsl::*;
+        let blob: Option<Vec<u8>> = settings
+            .select(value)
+            .filter(key.eq(k))
+            .first(self.connection)
+            .optional()?;
+
+        Ok(match blob {
+            Some(b) => Some(bincode::deserialize(&b)?),
+            None => None,
+        })
+    }
+
+    pub fn del(&mut self, k: &str) -> Result<usize> {
+        use crate::schema::settings::dsl::*;
+        Ok(diesel::delete(settings.filter(key.eq(k))).execute(self.connection)?)
+    }
+
+    pub fn list_keys(&mut self) -> Result<Vec<String>> {
+        use crate::schema::settings::dsl::*;
+        let keys: Vec<String> = settings.select(key).load(self.connection)?;
+
+        Ok(keys)
+    }
+}