39 lines
1 KiB
Rust
39 lines
1 KiB
Rust
use directories::ProjectDirs;
|
|
use ruma_identifiers::UserId;
|
|
|
|
pub struct Data(sled::Db);
|
|
|
|
impl Data {
|
|
pub fn set_hostname(&self, hostname: &str) {
|
|
self.0.insert("hostname", hostname).unwrap();
|
|
}
|
|
pub fn hostname(&self) -> String {
|
|
String::from_utf8(self.0.get("hostname").unwrap().unwrap().to_vec()).unwrap()
|
|
}
|
|
pub fn load_or_create() -> Self {
|
|
Data(
|
|
sled::open(
|
|
ProjectDirs::from("xyz", "koesters", "matrixserver")
|
|
.unwrap()
|
|
.data_dir(),
|
|
)
|
|
.unwrap(),
|
|
)
|
|
}
|
|
|
|
pub fn user_exists(&self, user_id: &UserId) -> bool {
|
|
self.0
|
|
.open_tree("username_password")
|
|
.unwrap()
|
|
.contains_key(user_id.to_string())
|
|
.unwrap()
|
|
}
|
|
|
|
pub fn user_add(&self, user_id: UserId, password: Option<String>) {
|
|
self.0
|
|
.open_tree("username_password")
|
|
.unwrap()
|
|
.insert(user_id.to_string(), &*password.unwrap_or_default())
|
|
.unwrap();
|
|
}
|
|
}
|