Compare commits
5
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
403a941ece
|
||
|
|
904e50f981
|
||
|
|
6b84cc42d6
|
||
|
|
b25b651241
|
||
|
|
cc6e6286c7
|
+3
-41
@@ -2,52 +2,14 @@
|
||||
/// extract it into some strings
|
||||
pub fn delimitize(var: &crate::envy::EnvVar) -> Result<Vec<String>, std::io::Error> {
|
||||
if var.has_delimiter {
|
||||
let values: Vec<String> = match var
|
||||
Ok(var
|
||||
.value
|
||||
.split(var.delimiter)
|
||||
.map(|c| c.parse::<String>())
|
||||
.collect()
|
||||
{
|
||||
Ok(v) => v,
|
||||
Err(err) => {
|
||||
return Err(std::io::Error::other(err));
|
||||
}
|
||||
};
|
||||
|
||||
Ok(values)
|
||||
.map(|c| c.parse::<String>().unwrap())
|
||||
.collect())
|
||||
} else {
|
||||
Err(std::io::Error::other(
|
||||
"Environment variable does not have a delimiter",
|
||||
))
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_delimitize() {
|
||||
let envvar = crate::envy::EnvVar {
|
||||
key: "OCEANS".to_string(),
|
||||
value: "Atlantic|Pacific|Indian|Arctic".to_string(),
|
||||
has_delimiter: true,
|
||||
delimiter: '|',
|
||||
};
|
||||
let all_values = vec![
|
||||
"Atlantic".to_string(),
|
||||
"Pacific".to_string(),
|
||||
"Indian".to_string(),
|
||||
"Arctic".to_string(),
|
||||
];
|
||||
|
||||
match delimitize(&envvar) {
|
||||
Ok(values) => {
|
||||
assert_eq!(values, all_values, "Failure in delimitizing EnvVar values");
|
||||
}
|
||||
Err(err) => {
|
||||
assert!(false, "Error delimitizing: {err:?}");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
+13
-67
@@ -85,50 +85,10 @@ pub fn create_token(
|
||||
let mut header = josekit::jws::JwsHeader::new();
|
||||
header.set_token_type(TOKEN_TYPE);
|
||||
|
||||
match init_payload(token_resource, duration) {
|
||||
Ok((payload, issued, expire)) => {
|
||||
let signer = match josekit::jws::alg::hmac::HmacJwsAlgorithm::Hs256
|
||||
.signer_from_bytes(key.as_bytes())
|
||||
{
|
||||
Ok(signer) => signer,
|
||||
Err(err) => {
|
||||
return Err(err);
|
||||
}
|
||||
};
|
||||
let access_token = match josekit::jwt::encode_with_signer(&payload, &header, &signer) {
|
||||
Ok(access_token) => access_token,
|
||||
Err(err) => {
|
||||
return Err(err);
|
||||
}
|
||||
};
|
||||
|
||||
Ok(CreateTokenResult {
|
||||
access_token: access_token,
|
||||
issued: issued.unix_timestamp(),
|
||||
expires_in: expire.unix_timestamp(),
|
||||
token_issued: issued,
|
||||
})
|
||||
}
|
||||
Err(e) => Err(josekit::JoseError::InvalidClaim(e.into())),
|
||||
}
|
||||
}
|
||||
|
||||
fn init_payload(
|
||||
token_resource: &TokenResource,
|
||||
duration: time::Duration,
|
||||
) -> Result<
|
||||
(
|
||||
josekit::jwt::JwtPayload,
|
||||
time::OffsetDateTime,
|
||||
time::OffsetDateTime,
|
||||
),
|
||||
josekit::JoseError,
|
||||
> {
|
||||
let mut payload = josekit::jwt::JwtPayload::new();
|
||||
payload.set_subject(&token_resource.message);
|
||||
payload.set_issuer(&token_resource.issuer);
|
||||
payload.set_audience(token_resource.audiences.clone());
|
||||
|
||||
if !token_resource.user_id.is_nil() {
|
||||
match payload.set_claim("user_id", Some(serde_json::json!(token_resource.user_id))) {
|
||||
Ok(_) => {}
|
||||
@@ -137,38 +97,24 @@ fn init_payload(
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
match init_payload_issued(&mut payload, duration) {
|
||||
Ok((issued, expire)) => Ok((payload, issued, expire)),
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
|
||||
fn init_payload_issued(
|
||||
payload: &mut josekit::jwt::JwtPayload,
|
||||
duration: time::Duration,
|
||||
) -> Result<(time::OffsetDateTime, time::OffsetDateTime), josekit::JoseError> {
|
||||
match get_issued() {
|
||||
Ok(issued) => {
|
||||
let expire = issued + duration;
|
||||
let issued_at = match util::time_to_std_time(&issued) {
|
||||
Ok(issued_at) => issued_at,
|
||||
Err(err) => {
|
||||
return Err(josekit::JoseError::InvalidClaim(err.into()));
|
||||
}
|
||||
};
|
||||
let expires_at = match util::time_to_std_time(&expire) {
|
||||
Ok(expires_at) => expires_at,
|
||||
Err(err) => {
|
||||
return Err(josekit::JoseError::InvalidClaim(err.into()));
|
||||
}
|
||||
};
|
||||
payload.set_issued_at(&util::time_to_std_time(&issued).unwrap());
|
||||
payload.set_expires_at(&util::time_to_std_time(&expire).unwrap());
|
||||
|
||||
payload.set_issued_at(&issued_at);
|
||||
payload.set_expires_at(&expires_at);
|
||||
Ok((issued, expire))
|
||||
let signer = josekit::jws::alg::hmac::HmacJwsAlgorithm::Hs256
|
||||
.signer_from_bytes(key.as_bytes())
|
||||
.unwrap();
|
||||
|
||||
Ok(CreateTokenResult {
|
||||
access_token: josekit::jwt::encode_with_signer(&payload, &header, &signer).unwrap(),
|
||||
issued: issued.unix_timestamp(),
|
||||
expires_in: expire.unix_timestamp(),
|
||||
token_issued: issued,
|
||||
})
|
||||
}
|
||||
Err(err) => Err(josekit::JoseError::InvalidClaim(err.into())),
|
||||
Err(e) => Err(josekit::JoseError::InvalidClaim(e.into())),
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user