diff --git a/src/auth.rs b/src/auth.rs index f5aeaa1..3842530 100644 --- a/src/auth.rs +++ b/src/auth.rs @@ -307,6 +307,25 @@ pub struct OrgHeaders { pub org_user_type: UserOrgType, } +// org_id is usually the second param ("/organizations/") +// But there are cases where it is located in a query value. +// First check the param, if this is not a valid uuid, we will try the query value. +fn get_org_id(request: &Request) -> Option { + if let Some(Ok(org_id)) = request.get_param::(1) { + if uuid::Uuid::parse_str(&org_id).is_ok() { + return Some(org_id); + } + } + + if let Some(Ok(org_id)) = request.get_query_value::("organizationId") { + if uuid::Uuid::parse_str(&org_id).is_ok() { + return Some(org_id); + } + } + + None +} + impl<'a, 'r> FromRequest<'a, 'r> for OrgHeaders { type Error = &'static str; @@ -315,9 +334,8 @@ impl<'a, 'r> FromRequest<'a, 'r> for OrgHeaders { Outcome::Forward(_) => Outcome::Forward(()), Outcome::Failure(f) => Outcome::Failure(f), Outcome::Success(headers) => { - // org_id is expected to be the second param ("/organizations/") - match request.get_param::(1) { - Some(Ok(org_id)) => { + match get_org_id(request) { + Some(org_id) => { let conn = match request.guard::() { Outcome::Success(conn) => conn, _ => err_handler!("Error getting DB"), @@ -348,7 +366,7 @@ impl<'a, 'r> FromRequest<'a, 'r> for OrgHeaders { } }, }) - } + }, _ => err_handler!("Error getting the organization id"), } }