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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
|
use std::{str::FromStr, time::Duration};
use anyhow::Context;
use axum::{
extract::{Query, State},
http::HeaderMap,
response::{IntoResponse, Redirect},
};
use axum_extra::{TypedHeader, headers};
use oauth2::{AuthorizationCode, TokenResponse};
use reqwest::{StatusCode, header::SET_COOKIE};
use sellershut_core::{
auth::{RegisterUserRequest, auth_server::Auth, register_user_request::AccountDetails},
users::CreateUserRequest,
};
use serde::{Deserialize, Serialize};
use sqlx::types::uuid;
use time::OffsetDateTime;
use tonic::IntoRequest;
use tower_sessions::{
SessionStore,
session::{Id, Record},
};
use uuid::Uuid;
use crate::{
auth::Claims,
error::AppError,
server::{
OAUTH_CSRF_COOKIE, csrf_token_validation::csrf_token_validation_workflow,
grpc::auth::DbUser, keys::generate_actor_keypair, routes::Provider,
},
state::AppHandle,
};
#[derive(Debug, Deserialize)]
pub struct AuthRequest {
provider: Provider,
code: String,
pub state: String,
}
#[derive(Debug, Deserialize, Serialize, Clone)]
struct User {
id: String,
avatar: Option<String>,
username: String,
discriminator: String,
verified: bool,
email: String,
}
/// The cookie to store the session id for user information.
const SESSION_COOKIE: &str = "info";
const SESSION_DATA_KEY: &str = "data";
pub async fn login_authorised(
Query(query): Query<AuthRequest>,
State(state): State<AppHandle>,
TypedHeader(cookies): TypedHeader<headers::Cookie>,
) -> Result<impl IntoResponse, AppError> {
let provider = query.provider.to_string();
let oauth_session_id = Id::from_str(
cookies
.get(OAUTH_CSRF_COOKIE)
.context("missing session cookie")?,
)?;
csrf_token_validation_workflow(&query, &state.session_store, oauth_session_id).await?;
let client = state.http_client.clone();
let store = state.session_store.clone();
// Get an auth token
let token = state
.discord_client
.exchange_code(AuthorizationCode::new(query.code.clone()))
.request_async(&client)
.await
.context("failed in sending request request to authorisation server")?;
let user_data = client
// https://discord.com/developers/docs/resources/user#get-current-user
.get("https://discordapp.com/api/users/@me")
.bearer_auth(token.access_token().secret())
.send()
.await
.context("failed in sending request to target Url")?
.json::<serde_json::Value>()
.await
.context("failed to deserialise response as JSON")?;
let user_data: User = serde_json::from_value(user_data)?;
if !user_data.verified {
return Ok((StatusCode::UNAUTHORIZED, "email is not verified").into_response());
}
// Create a new session filled with user data
let session_id = Id(i128::from_le_bytes(uuid::Uuid::new_v4().to_bytes_le()));
let user = sqlx::query_as!(
DbUser,
"
select
p.*
from
auth_user p
inner join
oauth_account a
on
p.id=a.user_id
where a.provider_id = $1 and a.provider_user_id = $2
",
provider,
user_data.id
)
.fetch_optional(&state.services.postgres)
.await?;
let user = if let Some(user) = user {
user.id
} else {
let data = user_data.clone();
let request = RegisterUserRequest {
email: data.email,
account: Some(AccountDetails {
provider_id: provider,
provider_user_id: data.id,
})
};
let resp = state.register_user(request.into_request()).await.unwrap().into_inner();
Uuid::parse_str(&resp.auth_id).unwrap()
};
let exp = OffsetDateTime::now_utc() + Duration::from_secs(15 * 60);
let claims = Claims {
sub: user,
exp: exp.unix_timestamp(),
iss: "sellershut".to_owned(),
sid: session_id.to_string(),
aud: "sellershut".to_owned(),
iat: OffsetDateTime::now_utc().unix_timestamp(),
};
let token = jsonwebtoken::encode(
&jsonwebtoken::Header::default(),
&claims,
&jsonwebtoken::EncodingKey::from_secret(
state.local_config.oauth.jwt_encoding_key.as_bytes(),
),
)?;
store
.create(&mut Record {
id: session_id,
data: [(
SESSION_DATA_KEY.to_string(),
serde_json::to_value(user_data).unwrap(),
)]
.into(),
expiry_date: time::OffsetDateTime::now_utc()
+ Duration::from_secs(state.local_config.oauth.session_lifespan),
})
.await
.context("failed in inserting serialised value into session")?;
sqlx::query!(
"insert into token (user_id, token, session_id) values ($1, $2, $3)",
user,
token,
session_id.to_string()
)
.execute(&state.services.postgres)
.await?;
let cookie = format!("{SESSION_COOKIE}={session_id}; SameSite=Lax; HttpOnly; Secure; Path=/");
let mut headers = HeaderMap::new();
headers.insert(
SET_COOKIE,
cookie.parse().context("failed to parse cookie")?,
);
Ok((headers, Redirect::to(&format!("/?token={token}"))).into_response())
}
|
