use crate::{Error, Result};
use std::{
net::{IpAddr, SocketAddr},
str::FromStr,
};
use trillium_server_common::url::{ParseError, Url};
pub trait IntoUrl {
fn into_url(self, base: Option<&Url>) -> Result<Url>;
}
impl IntoUrl for Url {
fn into_url(self, base: Option<&Url>) -> Result<Url> {
if self.cannot_be_a_base() {
return Err(Error::UnexpectedUriFormat);
}
if base.is_some_and(|base| !self.as_str().starts_with(base.as_str())) {
Err(Error::UnexpectedUriFormat)
} else {
Ok(self)
}
}
}
impl IntoUrl for &str {
fn into_url(self, base: Option<&Url>) -> Result<Url> {
match (Url::from_str(self), base) {
(Ok(url), base) => url.into_url(base),
(Err(ParseError::RelativeUrlWithoutBase), Some(base)) => base
.join(self.trim_start_matches('/'))
.map_err(|_| Error::UnexpectedUriFormat),
_ => Err(Error::UnexpectedUriFormat),
}
}
}
impl IntoUrl for String {
#[inline(always)]
fn into_url(self, base: Option<&Url>) -> Result<Url> {
self.as_str().into_url(base)
}
}
impl<S: AsRef<str>> IntoUrl for &[S] {
fn into_url(self, base: Option<&Url>) -> Result<Url> {
let Some(mut url) = base.cloned() else {
return Err(Error::UnexpectedUriFormat);
};
url.path_segments_mut()
.map_err(|_| Error::UnexpectedUriFormat)?
.pop_if_empty()
.extend(self);
Ok(url)
}
}
impl<S: AsRef<str>, const N: usize> IntoUrl for [S; N] {
fn into_url(self, base: Option<&Url>) -> Result<Url> {
self.as_slice().into_url(base)
}
}
impl<S: AsRef<str>> IntoUrl for Vec<S> {
fn into_url(self, base: Option<&Url>) -> Result<Url> {
self.as_slice().into_url(base)
}
}
impl IntoUrl for SocketAddr {
fn into_url(self, base: Option<&Url>) -> Result<Url> {
let scheme = if self.port() == 443 { "https" } else { "http" };
format!("{scheme}://{self}").into_url(base)
}
}
impl IntoUrl for IpAddr {
fn into_url(self, base: Option<&Url>) -> Result<Url> {
match self {
IpAddr::V4(v4) => format!("http://{v4}"),
IpAddr::V6(v6) => format!("http://[{v6}]"),
}
.into_url(base)
}
}