technofab/jwt-authorizer
1
0
Fork 0
mirror of https://github.com/TECHNOFAB11/jwt-authorizer.git synced 2025-12-11 23:50:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(discovery): replace join and add tests

Browse source
This commit is contained in:
cduvray 2023-03-15 08:04:25 +01:00
parent 2bb4b4ca34
commit 6dfc895876
1 changed files with 35 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

41
jwt-authorizer/src/oidc.rs
View file

@ -1,3 +1,4 @@
use reqwest::{Client, Url};
use serde::Deserialize; use serde::Deserialize;
use crate::error::InitError; use crate::error::InitError;
@ -8,13 +9,20 @@ pub struct OidcDiscovery {
pub jwks_uri: String, pub jwks_uri: String,
} }
fn discovery_url(issuer: &str) -> Result<Url, InitError> {
let mut url = Url::parse(issuer).map_err(|e| InitError::DiscoveryError(e.to_string()))?;
url.path_segments_mut()
.map_err(|_| InitError::DiscoveryError(format!("Issuer URL error! ('{issuer}' cannot be a base)")))?
.pop_if_empty()
.extend(&[".well-known", "openid-configuration"]);
Ok(url)
}
pub async fn discover_jwks(issuer: &str) -> Result<String, InitError> { pub async fn discover_jwks(issuer: &str) -> Result<String, InitError> {
let discovery_url = reqwest::Url::parse(issuer) Client::new()
.map_err(|e| InitError::DiscoveryError(e.to_string()))? .get(discovery_url(issuer)?)
.join(".well-known/openid-configuration")
.map_err(|e| InitError::DiscoveryError(e.to_string()))?;
reqwest::Client::new()
.get(discovery_url)
.send() .send()
.await .await
.map_err(|e| InitError::DiscoveryError(e.to_string()))? .map_err(|e| InitError::DiscoveryError(e.to_string()))?
@ -23,3 +31,24 @@ pub async fn discover_jwks(issuer: &str) -> Result<String, InitError> {
.map_err(|e| InitError::DiscoveryError(e.to_string())) .map_err(|e| InitError::DiscoveryError(e.to_string()))
.map(|d| d.jwks_uri) .map(|d| d.jwks_uri)
} }
#[test]
fn discovery() {
assert_eq!(
Url::parse("http://host.com:99/xx/.well-known/openid-configuration").unwrap(),
discovery_url("http://host.com:99/xx").unwrap()
);
assert_eq!(
Url::parse("http://host.com:99/xx/.well-known/openid-configuration").unwrap(),
discovery_url("http://host.com:99/xx/").unwrap()
);
assert_eq!(
Url::parse("http://host.com:99/xx/yy/.well-known/openid-configuration").unwrap(),
discovery_url("http://host.com:99/xx/yy").unwrap()
);
assert_eq!(
Url::parse("http://host.com:99/.well-known/openid-configuration").unwrap(),
discovery_url("http://host.com:99").unwrap()
);
assert!(discovery_url("xxx").is_err());
}