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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
use sqlx::{SqliteConnection, Transaction, sqlite::Sqlite};
use super::{Id, Invite, Summary};
use crate::{
clock::DateTime,
normalize::nfc,
user::{self, User},
};
pub trait Provider {
fn invites(&mut self) -> Invites<'_>;
}
impl Provider for Transaction<'_, Sqlite> {
fn invites(&mut self) -> Invites<'_> {
Invites(self)
}
}
pub struct Invites<'t>(&'t mut SqliteConnection);
impl Invites<'_> {
pub async fn create(
&mut self,
issuer: &User,
issued_at: &DateTime,
) -> Result<Invite, sqlx::Error> {
let id = Id::generate();
let invite = sqlx::query_as!(
Invite,
r#"
insert into invite (id, issuer, issued_at)
values ($1, $2, $3)
returning
id as "id: Id",
issuer as "issuer: user::Id",
issued_at as "issued_at: DateTime"
"#,
id,
issuer.id,
issued_at
)
.fetch_one(&mut *self.0)
.await?;
Ok(invite)
}
pub async fn by_id(&mut self, invite: &Id) -> Result<Invite, sqlx::Error> {
let invite = sqlx::query_as!(
Invite,
r#"
select
id as "id: Id",
issuer as "issuer: user::Id",
issued_at as "issued_at: DateTime"
from invite
where id = $1
"#,
invite,
)
.fetch_one(&mut *self.0)
.await?;
Ok(invite)
}
pub async fn summary(&mut self, invite: &Id) -> Result<Summary, sqlx::Error> {
let invite = sqlx::query!(
r#"
select
invite.id as "invite_id: Id",
issuer.id as "issuer_id: user::Id",
issuer_login.display_name as "issuer_name: nfc::String",
invite.issued_at as "invite_issued_at: DateTime"
from invite
join user as issuer on (invite.issuer = issuer.id)
join login as issuer_login on (issuer.id = issuer_login.id)
where invite.id = $1
"#,
invite,
)
.map(|row| Summary {
id: row.invite_id,
issuer: row.issuer_name,
issued_at: row.invite_issued_at,
})
.fetch_one(&mut *self.0)
.await?;
Ok(invite)
}
pub async fn accept(&mut self, invite: &Invite) -> Result<(), sqlx::Error> {
sqlx::query_scalar!(
r#"
delete from invite
where id = $1
returning 1 as "deleted: bool"
"#,
invite.id,
)
.fetch_one(&mut *self.0)
.await?;
Ok(())
}
pub async fn expire(&mut self, expire_at: &DateTime) -> Result<(), sqlx::Error> {
sqlx::query!(
r#"
delete from invite
where issued_at < $1
"#,
expire_at,
)
.execute(&mut *self.0)
.await?;
Ok(())
}
}
|
