1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
/*
This file is part of jellything (https://codeberg.org/metamuffin/jellything)
which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
Copyright (C) 2023 metamuffin <metamuffin.org>
*/
use crate::routes::ui::account::hash_password;
use anyhow::Context;
use jellybase::CONF;
use jellycommon::{Node, SeekIndex};
use log::info;
use serde::{Deserialize, Serialize};
use std::path::Path;
use typed_sled::Tree;
pub struct Database {
pub db: sled::Db,
pub user: Tree<String, User>,
pub invite: Tree<String, ()>,
pub node: Tree<String, Node>,
pub seek_index: Tree<(String, usize), SeekIndex>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct User {
pub name: String,
pub display_name: String,
pub password: Vec<u8>,
pub admin: bool,
}
impl Database {
pub fn open(path: &Path) -> Result<Self, anyhow::Error> {
info!("opening database… (might take up to O(n) time)");
let db = sled::open(path).context("opening database")?;
info!("creating trees");
let r = Ok(Self {
user: Tree::open(&db, "user"),
invite: Tree::open(&db, "invite"),
node: Tree::open(&db, "node"),
seek_index: Tree::open(&db, "seek_index"),
db,
});
info!("ready");
r
}
pub fn create_admin(&self) {
self.user
.insert(
&CONF.admin_username,
&User {
admin: true,
display_name: "Admin".to_string(),
name: CONF.admin_username.clone(),
password: hash_password(&CONF.admin_username, &CONF.admin_password),
},
)
.unwrap();
}
}
|
