use miette::{miette, Context, IntoDiagnostic};
use ockam_api::EnrollmentTicket;
use std::str::FromStr;
use url::Url;
pub fn parse_key_val<T, U>(s: &str) -> miette::Result<(T, U)>
where
T: FromStr,
T::Err: std::error::Error + Send + Sync + 'static,
U: FromStr,
U::Err: std::error::Error + Send + Sync + 'static,
{
let pos = s
.find('=')
.ok_or_else(|| miette!("invalid KEY=value: no `=` found in `{s}`"))?;
Ok((
s[..pos].parse().into_diagnostic()?,
s[pos + 1..].parse().into_diagnostic()?,
))
}
pub fn parse_enrollment_ticket(value: &str) -> miette::Result<EnrollmentTicket> {
let contents = parse_string_or_path_or_url(value)?;
if let Ok(enrollment_ticket) = serde_json::from_str(&contents) {
Ok(enrollment_ticket)
}
else if let Ok(hex_decoded) = hex::decode(contents.trim()) {
Ok(serde_json::from_slice(&hex_decoded)
.into_diagnostic()
.context("Failed to parse enrollment ticket from hex-encoded contents")?)
} else {
Err(miette!("Failed to parse enrollment ticket argument"))
}
}
pub async fn async_parse_path_or_url(value: &str) -> miette::Result<String> {
if Url::parse(value).is_ok() {
reqwest::get(value)
.await
.into_diagnostic()
.context(format!("Failed to download file from {value}"))?
.text()
.await
.into_diagnostic()
.context("Failed to read contents from downloaded file")
} else if tokio::fs::metadata(value).await.is_ok() {
std::fs::read_to_string(value)
.into_diagnostic()
.context("Failed to read contents from file")
} else {
Err(miette!("Failed to parse value {} as a path or URL", value))
}
}
pub fn parse_path_or_url(value: &str) -> miette::Result<String> {
if Url::parse(value).is_ok() {
reqwest::blocking::get(value)
.into_diagnostic()
.context(format!("Failed to download file from {value}"))?
.text()
.into_diagnostic()
.context("Failed to read contents from downloaded file")
} else if std::fs::metadata(value).is_ok() {
std::fs::read_to_string(value)
.into_diagnostic()
.context("Failed to read contents from file")
} else {
Err(miette!("Failed to parse value {} as a path or URL", value))
}
}
fn parse_string_or_path_or_url(value: &str) -> miette::Result<String> {
parse_path_or_url(value).or_else(|_| Ok(value.to_string()))
}