62 lines
1.5 KiB
Rust
62 lines
1.5 KiB
Rust
|
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)))
|
||
|
|
}
|
||
|
|
}
|