alibi-sqlx 0.10.0

SQLx integration for Alibi
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
62
63

use super::SqlxStore;
use super::entities::jwk::Model;
use crate::model::{self, ActiveRow};
use crate::pool::Exec;
use crate::schema::AuthSchema;
use alibi_core::error::AuthResult;
use alibi_core::store::JwkStore;
use alibi_core::types::{CreateJwk, Jwk};
use async_trait::async_trait;

impl<S: AuthSchema> SqlxStore<S> {
    pub(super) async fn list_jwks_with(&self, exec: Exec<'_>) -> AuthResult<Vec<Jwk>> {
        let mut sql = model::select_model::<Model>(exec);
        sql.push(" LIMIT ");
        sql.bind(self.find_many_limit());
        Ok(exec
            .fetch_all::<Model>(sql)
            .await?
            .into_iter()
            .map(Into::into)
            .collect())
    }

    pub(super) async fn get_jwk_with(&self, exec: Exec<'_>, id: &str) -> AuthResult<Option<Jwk>> {
        let mut sql = model::by_id::<Model>(exec, id);
        model::limit_one(&mut sql);
        Ok(exec.fetch_optional::<Model>(sql).await?.map(Into::into))
    }

    pub(super) async fn create_jwk_with(&self, exec: Exec<'_>, data: CreateJwk) -> AuthResult<Jwk> {
        let mut active = ActiveRow::new();
        active.set(
            "id",
            match data.id {
                Some(id) => id,
                None => self
                    .generated_id(exec, "jwks", "jwks", "id")
                    .await?
                    .unwrap_or_else(|| uuid::Uuid::new_v4().to_string()),
            },
        );
        active.set("publicKey", data.public_key);
        active.set("privateKey", data.private_key);
        active.set("createdAt", data.created_at);
        active.set("expiresAt", data.expires_at);
        active.set("alg", data.alg);
        active.set("crv", data.crv);
        model::insert::<Model>(exec, &active).await.map(Into::into)
    }
}

#[async_trait]
impl<S: AuthSchema> JwkStore for SqlxStore<S> {
    async fn list_jwks(&self) -> AuthResult<Vec<Jwk>> {
        self.list_jwks_with(self.exec()).await
    }
    async fn get_jwk_by_id(&self, id: &str) -> AuthResult<Option<Jwk>> {
        self.get_jwk_with(self.exec(), id).await
    }
    async fn create_jwk(&self, data: CreateJwk) -> AuthResult<Jwk> {
        self.create_jwk_with(self.exec(), data).await
    }
}