use crate::endpoint_lib::diagnostic::DiagnosticCollector;
use std::borrow::Cow;
use std::error::Error;
use std::fmt::{Display, Formatter};
#[derive(Debug, Eq, PartialEq)]
pub(crate) struct Arn<'a> {
partition: &'a str,
service: &'a str,
region: &'a str,
account_id: &'a str,
resource_id: Vec<&'a str>,
}
#[allow(unused)]
impl<'a> Arn<'a> {
pub(crate) fn partition(&self) -> &'a str {
self.partition
}
pub(crate) fn service(&self) -> &'a str {
self.service
}
pub(crate) fn region(&self) -> &'a str {
self.region
}
pub(crate) fn account_id(&self) -> &'a str {
self.account_id
}
pub(crate) fn resource_id(&self) -> &Vec<&'a str> {
&self.resource_id
}
}
#[derive(Debug, PartialEq)]
pub(crate) struct InvalidArn {
message: Cow<'static, str>,
}
impl InvalidArn {
fn from_static(message: &'static str) -> InvalidArn {
Self {
message: Cow::Borrowed(message),
}
}
}
impl Display for InvalidArn {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", self.message)
}
}
impl Error for InvalidArn {}
impl<'a> Arn<'a> {
pub(crate) fn parse(arn: &'a str) -> Result<Self, InvalidArn> {
let mut split = arn.splitn(6, ':');
let invalid_format =
|| InvalidArn::from_static("ARN must have 6 components delimited by `:`");
let arn = split.next().ok_or_else(invalid_format)?;
let partition = split.next().ok_or_else(invalid_format)?;
let service = split.next().ok_or_else(invalid_format)?;
let region = split.next().ok_or_else(invalid_format)?;
let account_id = split.next().ok_or_else(invalid_format)?;
let resource_id = split.next().ok_or_else(invalid_format)?;
if arn != "arn" {
return Err(InvalidArn::from_static(
"first component of the ARN must be `arn`",
));
}
if partition.is_empty() || service.is_empty() || resource_id.is_empty() {
return Err(InvalidArn::from_static(
"partition, service, and resource id must all be non-empty",
));
}
let resource_id = resource_id.split([':', '/']).collect::<Vec<_>>();
Ok(Self {
partition,
service,
region,
account_id,
resource_id,
})
}
}
pub(crate) fn parse_arn<'a, 'b>(input: &'a str, e: &'b mut DiagnosticCollector) -> Option<Arn<'a>> {
e.capture(Arn::parse(input))
}
#[cfg(test)]
mod test {
use super::Arn;
#[test]
fn arn_parser() {
let arn = "arn:aws:s3:us-east-2:012345678:outpost:op-1234";
let parsed = Arn::parse(arn).expect("valid ARN");
assert_eq!(
parsed,
Arn {
partition: "aws",
service: "s3",
region: "us-east-2",
account_id: "012345678",
resource_id: vec!["outpost", "op-1234"]
}
);
}
#[test]
fn allow_slash_arns() {
let arn = "arn:aws:s3:us-east-2:012345678:outpost/op-1234";
let parsed = Arn::parse(arn).expect("valid ARN");
assert_eq!(
parsed,
Arn {
partition: "aws",
service: "s3",
region: "us-east-2",
account_id: "012345678",
resource_id: vec!["outpost", "op-1234"]
}
);
}
#[test]
fn resource_id_must_be_nonempty() {
let arn = "arn:aws:s3:us-east-2:012345678:";
Arn::parse(arn).expect_err("empty resource");
}
#[test]
fn arns_with_empty_parts() {
let arn = "arn:aws:s3:::my_corporate_bucket/Development/*";
assert_eq!(
Arn::parse(arn).expect("valid arn"),
Arn {
partition: "aws",
service: "s3",
region: "",
account_id: "",
resource_id: vec!["my_corporate_bucket", "Development", "*"]
}
);
}
}