Split db into youmubot-db

youmubot-db/src/lib.rs

@@ -0,0 +1,73 @@
+use rustbreak::{deser::Yaml as Ron, FileDatabase};
+use serde::{de::DeserializeOwned, Deserialize, Serialize};
+use serenity::{framework::standard::CommandError as Error, model::id::GuildId, prelude::*};
+use std::collections::HashMap;
+use std::path::Path;
+
+/// GuildMap defines the guild-map type.
+/// It is basically a HashMap from a GuildId to a data structure.
+pub type GuildMap<V> = HashMap<GuildId, V>;
+/// The generic DB type we will be using.
+pub struct DB<T>(std::marker::PhantomData<T>);
+impl<T: std::any::Any> serenity::prelude::TypeMapKey for DB<T> {
+    type Value = FileDatabase<T, Ron>;
+}
+
+impl<T: std::any::Any + Default + Send + Sync + Clone + Serialize + std::fmt::Debug> DB<T>
+where
+    for<'de> T: Deserialize<'de>,
+{
+    /// Insert into a ShareMap.
+    pub fn insert_into(data: &mut ShareMap, path: impl AsRef<Path>) -> Result<(), Error> {
+        let db = FileDatabase::<T, Ron>::from_path(path, T::default())?;
+        db.load().or_else(|e| {
+            dbg!(e);
+            db.save()
+        })?;
+        data.insert::<DB<T>>(db);
+        Ok(())
+    }
+
+    /// Open a previously inserted DB.
+    pub fn open(data: &ShareMap) -> DBWriteGuard<'_, T> {
+        data.get::<Self>().expect("DB initialized").into()
+    }
+}
+
+/// The write guard for our FileDatabase.
+/// It wraps the FileDatabase in a write-on-drop lock.
+pub struct DBWriteGuard<'a, T>(&'a FileDatabase<T, Ron>)
+where
+    T: Send + Sync + Clone + std::fmt::Debug + Serialize + DeserializeOwned;
+
+impl<'a, T> From<&'a FileDatabase<T, Ron>> for DBWriteGuard<'a, T>
+where
+    T: Send + Sync + Clone + std::fmt::Debug + Serialize + DeserializeOwned,
+{
+    fn from(v: &'a FileDatabase<T, Ron>) -> Self {
+        DBWriteGuard(v)
+    }
+}
+
+impl<'a, T> DBWriteGuard<'a, T>
+where
+    T: Send + Sync + Clone + std::fmt::Debug + Serialize + DeserializeOwned,
+{
+    /// Borrows the FileDatabase.
+    pub fn borrow(&self) -> Result<std::sync::RwLockReadGuard<T>, rustbreak::RustbreakError> {
+        (*self).0.borrow_data()
+    }
+    /// Borrows the FileDatabase for writing.
+    pub fn borrow_mut(&self) -> Result<std::sync::RwLockWriteGuard<T>, rustbreak::RustbreakError> {
+        (*self).0.borrow_data_mut()
+    }
+}
+
+impl<'a, T> Drop for DBWriteGuard<'a, T>
+where
+    T: Send + Sync + Clone + std::fmt::Debug + Serialize + DeserializeOwned,
+{
+    fn drop(&mut self) {
+        self.0.save().expect("Save succeed")
+    }
+}