appdb 0.2.14

Lightweight SurrealDB helper library for Tauri embedded database apps
Documentation 
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
61

use anyhow::{Result, anyhow};
use serde::Serialize;
use surrealdb::opt::auth::Record;
use surrealdb::types::SurrealValue;

use crate::connection::get_db;

#[derive(Serialize, SurrealValue)]
/// Credentials used for the built-in root account bootstrap flow.
struct RootCredentials {
    user: String,
    pass: String,
}

fn root_user(pass: &str) -> Record<RootCredentials> {
    Record {
        namespace: "app".to_owned(),
        database: "app".to_owned(),
        access: "account".to_owned(),
        params: RootCredentials {
            user: "root".to_owned(),
            pass: pass.to_owned(),
        },
    }
}

/// Ensures the default root record-access account exists and is usable.
pub async fn ensure_root_user(pass: &str) -> Result<()> {
    let db = get_db()?;
    ensure_root_user_with(
        pass,
        |creds| async { db.signin(creds).await.map(|_| ()).map_err(Into::into) },
        |creds| async { db.signup(creds).await.map(|_| ()).map_err(Into::into) },
    )
    .await
}

async fn ensure_root_user_with<Signin, SigninFuture, Signup, SignupFuture>(
    pass: &str,
    signin: Signin,
    signup: Signup,
) -> Result<()>
where
    Signin: FnOnce(Record<RootCredentials>) -> SigninFuture,
    SigninFuture: Future<Output = Result<()>>,
    Signup: FnOnce(Record<RootCredentials>) -> SignupFuture,
    SignupFuture: Future<Output = Result<()>>,
{
    match signin(root_user(pass)).await {
        Ok(_) => Ok(()),
        Err(signin_err) => signup(root_user(pass)).await.map_err(|signup_err| {
            anyhow!(
                "root record access signin failed: {signin_err}; signup fallback failed: {signup_err}"
            )
        }),
    }
}

#[cfg(test)]
#[path = "tests.rs"]
mod tests;