summaryrefslogtreecommitdiff
path: root/src/token/repo/auth.rs
blob: a42fa1a6697206fff3004d232515685bbb076c26 (plain)
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
use sqlx::{SqliteConnection, Transaction, sqlite::Sqlite};

use crate::{
    clock::DateTime,
    db::NotFound,
    event::{Instant, Sequence},
    name::{self, Name},
    password::StoredHash,
    user::{self, History, User},
};

pub trait Provider {
    fn auth(&mut self) -> Auth<'_>;
}

impl Provider for Transaction<'_, Sqlite> {
    fn auth(&mut self) -> Auth<'_> {
        Auth(self)
    }
}

pub struct Auth<'t>(&'t mut SqliteConnection);

impl Auth<'_> {
    pub async fn for_name(&mut self, name: &Name) -> Result<(History, StoredHash), LoadError> {
        let name = name.canonical();
        let row = sqlx::query!(
            r#"
                select
                    id as "id: user::Id",
                    login.display_name as "display_name: String",
                    login.canonical_name as "canonical_name: String",
                    user.created_sequence as "created_sequence: Sequence",
                    user.created_at as "created_at: DateTime",
                    login.password as "password: StoredHash"
                from user
                join login using (id)
                where login.canonical_name = $1
            "#,
            name,
        )
        .fetch_one(&mut *self.0)
        .await?;

        let login = History {
            user: User {
                id: row.id,
                name: Name::new(row.display_name, row.canonical_name)?,
            },
            created: Instant::new(row.created_at, row.created_sequence),
        };

        Ok((login, row.password))
    }

    pub async fn for_user(&mut self, user: &User) -> Result<(History, StoredHash), LoadError> {
        let row = sqlx::query!(
            r#"
                select
                    id as "id: user::Id",
                    login.display_name as "display_name: String",
                    login.canonical_name as "canonical_name: String",
                    user.created_sequence as "created_sequence: Sequence",
                    user.created_at as "created_at: DateTime",
                    login.password as "password: StoredHash"
                from user
                join login using (id)
                where id = $1
            "#,
            user.id,
        )
        .fetch_one(&mut *self.0)
        .await?;

        let user = History {
            user: User {
                id: row.id,
                name: Name::new(row.display_name, row.canonical_name)?,
            },
            created: Instant::new(row.created_at, row.created_sequence),
        };

        Ok((user, row.password))
    }
}

#[derive(Debug, thiserror::Error)]
#[error(transparent)]
pub enum LoadError {
    Database(#[from] sqlx::Error),
    Name(#[from] name::Error),
}

impl<T> NotFound for Result<T, LoadError> {
    type Ok = T;
    type Error = LoadError;

    fn optional(self) -> Result<Option<T>, LoadError> {
        match self {
            Ok(value) => Ok(Some(value)),
            Err(LoadError::Database(sqlx::Error::RowNotFound)) => Ok(None),
            Err(other) => Err(other),
        }
    }
}