1use serde::{Deserialize, Serialize};
2use url::Url;
3
4use crate::error::Error;
5use crate::pkce;
6use crate::types::{Ppnum, PpnumId};
7
8const DEFAULT_AUTH_URL: &str = "https://accounts.ppoppo.com/oauth/authorize";
9const DEFAULT_TOKEN_URL: &str = "https://accounts.ppoppo.com/oauth/token";
10const DEFAULT_USERINFO_URL: &str = "https://accounts.ppoppo.com/oauth/userinfo";
11
12#[derive(Debug, Clone)]
25#[non_exhaustive]
26pub struct OAuthConfig {
27 pub(crate) client_id: String,
28 pub(crate) auth_url: Url,
29 pub(crate) token_url: Url,
30 pub(crate) userinfo_url: Url,
31 pub(crate) redirect_uri: Url,
32 pub(crate) scopes: Vec<String>,
33}
34
35impl OAuthConfig {
36 #[must_use]
40 #[allow(clippy::expect_used)] pub fn new(client_id: impl Into<String>, redirect_uri: Url) -> Self {
42 Self {
43 client_id: client_id.into(),
44 redirect_uri,
45 auth_url: DEFAULT_AUTH_URL.parse().expect("valid default URL"),
46 token_url: DEFAULT_TOKEN_URL.parse().expect("valid default URL"),
47 userinfo_url: DEFAULT_USERINFO_URL.parse().expect("valid default URL"),
48 scopes: vec!["profile".into()],
49 }
50 }
51
52 #[must_use]
54 pub fn with_auth_url(mut self, url: Url) -> Self {
55 self.auth_url = url;
56 self
57 }
58
59 #[must_use]
61 pub fn with_token_url(mut self, url: Url) -> Self {
62 self.token_url = url;
63 self
64 }
65
66 #[must_use]
68 pub fn with_userinfo_url(mut self, url: Url) -> Self {
69 self.userinfo_url = url;
70 self
71 }
72
73 #[must_use]
75 pub fn with_scopes(mut self, scopes: Vec<String>) -> Self {
76 self.scopes = scopes;
77 self
78 }
79
80 #[must_use]
82 pub fn client_id(&self) -> &str {
83 &self.client_id
84 }
85
86 #[must_use]
88 pub fn auth_url(&self) -> &Url {
89 &self.auth_url
90 }
91
92 #[must_use]
94 pub fn token_url(&self) -> &Url {
95 &self.token_url
96 }
97
98 #[must_use]
100 pub fn userinfo_url(&self) -> &Url {
101 &self.userinfo_url
102 }
103
104 #[must_use]
106 pub fn redirect_uri(&self) -> &Url {
107 &self.redirect_uri
108 }
109
110 #[must_use]
112 pub fn scopes(&self) -> &[String] {
113 &self.scopes
114 }
115}
116
117pub struct AuthClient {
119 config: OAuthConfig,
120 http: reqwest::Client,
121}
122
123#[non_exhaustive]
125pub struct AuthorizationRequest {
126 pub url: String,
127 pub state: String,
128 pub code_verifier: String,
129}
130
131#[derive(Debug, Clone, Deserialize)]
133#[non_exhaustive]
134pub struct TokenResponse {
135 pub access_token: String,
136 pub token_type: String,
137 #[serde(default)]
138 pub expires_in: Option<u64>,
139 #[serde(default)]
140 pub refresh_token: Option<String>,
141}
142
143#[derive(Debug, Clone, Serialize, Deserialize)]
145#[non_exhaustive]
146pub struct UserInfo {
147 pub sub: PpnumId,
148 #[serde(default)]
149 pub email: Option<String>,
150 pub ppnum: Ppnum,
151 #[serde(default)]
152 pub email_verified: Option<bool>,
153 #[serde(default, with = "time::serde::rfc3339::option")]
154 pub created_at: Option<time::OffsetDateTime>,
155}
156
157impl UserInfo {
158 #[must_use]
160 pub fn new(sub: PpnumId, ppnum: Ppnum) -> Self {
161 Self {
162 sub,
163 ppnum,
164 email: None,
165 email_verified: None,
166 created_at: None,
167 }
168 }
169
170 #[must_use]
172 pub fn with_email(mut self, email: impl Into<String>) -> Self {
173 self.email = Some(email.into());
174 self
175 }
176
177 #[must_use]
179 pub fn with_email_verified(mut self, verified: bool) -> Self {
180 self.email_verified = Some(verified);
181 self
182 }
183}
184
185impl AuthClient {
186 pub fn try_new(config: OAuthConfig) -> Result<Self, Error> {
198 let builder = reqwest::Client::builder();
199 #[cfg(not(target_arch = "wasm32"))]
200 let builder = builder
201 .timeout(std::time::Duration::from_secs(10))
202 .connect_timeout(std::time::Duration::from_secs(5));
203 Ok(Self {
204 config,
205 http: builder.build()?,
206 })
207 }
208
209 #[must_use]
215 pub fn with_http_client(config: OAuthConfig, client: reqwest::Client) -> Self {
216 Self {
217 config,
218 http: client,
219 }
220 }
221
222 #[must_use]
224 pub fn authorization_url(&self) -> AuthorizationRequest {
225 let state = pkce::generate_state();
226 let code_verifier = pkce::generate_code_verifier();
227 let code_challenge = pkce::generate_code_challenge(&code_verifier);
228 let scope = self.config.scopes.join(" ");
229
230 let mut url = self.config.auth_url.clone();
231 url.query_pairs_mut()
232 .append_pair("response_type", "code")
233 .append_pair("client_id", &self.config.client_id)
234 .append_pair("redirect_uri", self.config.redirect_uri.as_str())
235 .append_pair("state", &state)
236 .append_pair("code_challenge", &code_challenge)
237 .append_pair("code_challenge_method", "S256")
238 .append_pair("scope", &scope);
239
240 AuthorizationRequest {
241 url: url.into(),
242 state,
243 code_verifier,
244 }
245 }
246
247 pub async fn exchange_code(
254 &self,
255 code: &str,
256 code_verifier: &str,
257 ) -> Result<TokenResponse, Error> {
258 let params = [
259 ("grant_type", "authorization_code"),
260 ("code", code),
261 ("redirect_uri", self.config.redirect_uri.as_str()),
262 ("client_id", self.config.client_id.as_str()),
263 ("code_verifier", code_verifier),
264 ];
265
266 self.send_and_deserialize(
267 self.http.post(self.config.token_url.clone()).form(¶ms),
268 "token exchange",
269 )
270 .await
271 }
272
273 pub async fn refresh_token(&self, refresh_token: &str) -> Result<TokenResponse, Error> {
280 let params = [
281 ("grant_type", "refresh_token"),
282 ("refresh_token", refresh_token),
283 ("client_id", self.config.client_id.as_str()),
284 ];
285
286 self.send_and_deserialize(
287 self.http.post(self.config.token_url.clone()).form(¶ms),
288 "token refresh",
289 )
290 .await
291 }
292
293 pub async fn get_user_info(&self, access_token: &str) -> Result<UserInfo, Error> {
300 self.send_and_deserialize(
301 self.http
302 .get(self.config.userinfo_url.clone())
303 .bearer_auth(access_token),
304 "userinfo request",
305 )
306 .await
307 }
308
309 async fn send_and_deserialize<T: serde::de::DeserializeOwned>(
310 &self,
311 request: reqwest::RequestBuilder,
312 operation: &'static str,
313 ) -> Result<T, Error> {
314 let response = request.send().await?;
315
316 if !response.status().is_success() {
317 let status = response.status().as_u16();
318 let body = response.text().await.unwrap_or_default();
319 return Err(Error::OAuth {
320 operation,
321 status: Some(status),
322 detail: body,
323 });
324 }
325
326 response.json::<T>().await.map_err(|e| Error::OAuth {
327 operation,
328 status: None,
329 detail: format!("response deserialization failed: {e}"),
330 })
331 }
332}
333
334#[cfg(test)]
335#[allow(clippy::unwrap_used)]
336mod tests {
337 use super::*;
338
339 fn test_config() -> OAuthConfig {
340 OAuthConfig::new(
341 "test-client",
342 "https://example.com/callback".parse().unwrap(),
343 )
344 }
345
346 #[test]
347 fn test_authorization_url_contains_pkce() {
348 let client = AuthClient::try_new(test_config()).unwrap();
349 let req = client.authorization_url();
350
351 assert!(req.url.contains("code_challenge="));
352 assert!(req.url.contains("code_challenge_method=S256"));
353 assert!(req.url.contains("state="));
354 assert!(req.url.contains("response_type=code"));
355 assert!(req.url.contains("client_id=test-client"));
356 assert!(!req.code_verifier.is_empty());
357 assert!(!req.state.is_empty());
358 }
359
360 #[test]
361 fn test_authorization_url_unique_per_call() {
362 let client = AuthClient::try_new(test_config()).unwrap();
363 let req1 = client.authorization_url();
364 let req2 = client.authorization_url();
365
366 assert_ne!(req1.state, req2.state);
367 assert_ne!(req1.code_verifier, req2.code_verifier);
368 }
369
370 #[test]
371 fn test_config_constructor() {
372 let config = OAuthConfig::new("my-app", "https://my-app.com/callback".parse().unwrap());
373
374 assert_eq!(config.client_id(), "my-app");
375 assert_eq!(
376 config.redirect_uri().as_str(),
377 "https://my-app.com/callback"
378 );
379 assert_eq!(
380 config.auth_url().as_str(),
381 "https://accounts.ppoppo.com/oauth/authorize"
382 );
383 }
384
385 #[test]
386 fn test_config_with_overrides() {
387 let config = OAuthConfig::new("my-app", "https://my-app.com/callback".parse().unwrap())
388 .with_auth_url("https://custom.example.com/authorize".parse().unwrap())
389 .with_scopes(vec!["profile".into(), "email".into()]);
390
391 assert_eq!(
392 config.auth_url().as_str(),
393 "https://custom.example.com/authorize"
394 );
395 assert_eq!(config.scopes(), &["profile", "email"]);
396 }
397}