use http::{uri, Request};
use tower::{Layer, Service};
#[derive(Debug, Clone)]
pub struct BaseUriLayer {
base_uri: http::Uri,
}
impl BaseUriLayer {
pub fn new(base_uri: http::Uri) -> Self {
Self { base_uri }
}
}
impl<S> Layer<S> for BaseUriLayer {
type Service = BaseUri<S>;
fn layer(&self, inner: S) -> Self::Service {
BaseUri {
base_uri: self.base_uri.clone(),
inner,
}
}
}
#[derive(Debug, Clone)]
pub struct BaseUri<S> {
base_uri: http::Uri,
inner: S,
}
impl<S, ReqBody> Service<Request<ReqBody>> for BaseUri<S>
where
S: Service<Request<ReqBody>>,
{
type Error = S::Error;
type Future = S::Future;
type Response = S::Response;
fn poll_ready(&mut self, cx: &mut std::task::Context<'_>) -> std::task::Poll<Result<(), Self::Error>> {
self.inner.poll_ready(cx)
}
fn call(&mut self, req: Request<ReqBody>) -> Self::Future {
let (mut parts, body) = req.into_parts();
let req_pandq = parts.uri.path_and_query();
parts.uri = set_base_uri(&self.base_uri, req_pandq);
self.inner.call(Request::from_parts(parts, body))
}
}
fn set_base_uri(base_uri: &http::Uri, req_pandq: Option<&uri::PathAndQuery>) -> http::Uri {
let mut builder = uri::Builder::new();
if let Some(scheme) = base_uri.scheme() {
builder = builder.scheme(scheme.as_str());
}
if let Some(authority) = base_uri.authority() {
builder = builder.authority(authority.as_str());
}
if let Some(pandq) = base_uri.path_and_query() {
builder = if let Some(req_pandq) = req_pandq {
let base_path = pandq.path().trim_end_matches('/');
builder.path_and_query(format!("{base_path}{req_pandq}"))
} else {
builder.path_and_query(pandq.as_str())
};
} else if let Some(req_pandq) = req_pandq {
builder = builder.path_and_query(req_pandq.as_str());
}
builder.build().expect("Valid Uri")
}
#[cfg(test)]
mod tests {
#[test]
fn normal_host() {
let base_uri = http::Uri::from_static("https://192.168.1.65:8443");
let apipath = http::Uri::from_static("/api/v1/nodes?hi=yes");
let pandq = apipath.path_and_query();
assert_eq!(
super::set_base_uri(&base_uri, pandq),
"https://192.168.1.65:8443/api/v1/nodes?hi=yes"
);
}
#[test]
fn rancher_host() {
let base_uri = http::Uri::from_static("https://example.com/foo/bar");
let api_path = http::Uri::from_static("/api/v1/nodes?hi=yes");
let pandq = api_path.path_and_query();
assert_eq!(
super::set_base_uri(&base_uri, pandq),
"https://example.com/foo/bar/api/v1/nodes?hi=yes"
);
}
}