use std::borrow::Cow;
use url::Url;
pub trait IntoUrl: IntoUrlSealed {}
impl IntoUrl for Url {}
impl IntoUrl for String {}
impl IntoUrl for &Url {}
impl IntoUrl for &str {}
impl IntoUrl for &String {}
impl IntoUrl for Cow<'_, str> {}
pub trait IntoUrlSealed {
fn into_url(self) -> crate::Result<Url>;
fn as_str(&self) -> &str;
}
impl IntoUrlSealed for Url {
fn into_url(self) -> crate::Result<Url> {
if self.has_host() {
Ok(self)
} else {
Err(crate::error::url_bad_scheme(self))
}
}
fn as_str(&self) -> &str {
self.as_ref()
}
}
impl IntoUrlSealed for &Url {
fn into_url(self) -> crate::Result<Url> {
if self.has_host() {
Ok(self.clone())
} else {
Err(crate::error::url_bad_scheme(self.clone()))
}
}
fn as_str(&self) -> &str {
self.as_ref()
}
}
impl IntoUrlSealed for &str {
fn into_url(self) -> crate::Result<Url> {
Url::parse(self).map_err(crate::error::builder)?.into_url()
}
fn as_str(&self) -> &str {
self
}
}
impl IntoUrlSealed for &String {
fn into_url(self) -> crate::Result<Url> {
(&**self).into_url()
}
fn as_str(&self) -> &str {
self.as_ref()
}
}
impl IntoUrlSealed for String {
fn into_url(self) -> crate::Result<Url> {
(&*self).into_url()
}
fn as_str(&self) -> &str {
self.as_ref()
}
}
impl IntoUrlSealed for Cow<'_, str> {
fn into_url(self) -> crate::Result<Url> {
(&*self).into_url()
}
fn as_str(&self) -> &str {
self
}
}
pub(crate) fn try_uri(url: &Url) -> Option<http::Uri> {
url.as_str().parse().ok()
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn into_url_file_scheme() {
let err = "file:///etc/hosts".into_url().unwrap_err();
assert_eq!(
err.to_string(),
"builder error for url (file:///etc/hosts): URL scheme is not allowed"
);
}
#[test]
fn into_url_blob_scheme() {
let err = "blob:https://example.com".into_url().unwrap_err();
assert_eq!(
err.to_string(),
"builder error for url (blob:https://example.com): URL scheme is not allowed"
);
}
#[tokio::test]
async fn execute_request_rejects_invalid_hostname() {
let url_str = "https://{{hostname}}/";
let url = url::Url::parse(url_str).unwrap();
let result = crate::Client::new().get(url).send().await;
assert!(result.is_err());
let err = result.err().unwrap();
assert!(err.is_builder());
assert_eq!(url_str, err.url().unwrap().as_str());
}
}