refactor: split database into multiple files, more error handling, cleaner code

src/database/globals.rs

@@ -0,0 +1,61 @@
+use crate::{utils, Result};
+
+pub const COUNTER: &str = "c";
+
+pub struct Globals {
+    pub(super) globals: sled::Tree,
+    hostname: String,
+    keypair: ruma_signatures::Ed25519KeyPair,
+    reqwest_client: reqwest::Client,
+}
+
+impl Globals {
+    pub fn load(globals: sled::Tree, hostname: String) -> Self {
+        let keypair = ruma_signatures::Ed25519KeyPair::new(
+            &*globals
+                .update_and_fetch("keypair", utils::generate_keypair)
+                .unwrap()
+                .unwrap(),
+            "key1".to_owned(),
+        )
+        .unwrap();
+
+        Self {
+            globals,
+            hostname,
+            keypair,
+            reqwest_client: reqwest::Client::new(),
+        }
+    }
+
+    /// Returns the hostname of the server.
+    pub fn hostname(&self) -> &str {
+        &self.hostname
+    }
+
+    /// Returns this server's keypair.
+    pub fn keypair(&self) -> &ruma_signatures::Ed25519KeyPair {
+        &self.keypair
+    }
+
+    /// Returns a reqwest client which can be used to send requests.
+    pub fn reqwest_client(&self) -> &reqwest::Client {
+        &self.reqwest_client
+    }
+
+    pub fn next_count(&self) -> Result<u64> {
+        Ok(utils::u64_from_bytes(
+            &self
+                .globals
+                .update_and_fetch(COUNTER, utils::increment)?
+                .expect("utils::increment will always put in a value"),
+        ))
+    }
+
+    pub fn current_count(&self) -> Result<u64> {
+        Ok(self
+            .globals
+            .get(COUNTER)?
+            .map_or(0_u64, |bytes| utils::u64_from_bytes(&bytes)))
+    }
+}