use crate::Api; use futures::future::BoxFuture; use hyper::header::HeaderName; use hyper::{service::Service, Error, Request, Response, StatusCode}; use std::default::Default; use std::io; use std::marker::PhantomData; use std::task::{Context, Poll}; use swagger::auth::{AuthData, Authorization, Bearer, Scopes}; use swagger::{EmptyContext, Has, Pop, Push, XSpanIdString}; use url::form_urlencoded; pub struct MakeAddContext { inner: T, marker: PhantomData, } impl MakeAddContext where A: Default + Push, B: Push, Result = C>, C: Push, Result = D>, { pub fn new(inner: T) -> MakeAddContext { MakeAddContext { inner, marker: PhantomData, } } } // Make a service that adds context. impl Service for MakeAddContext where Target: Send, A: Default + Push + Send, B: Push, Result = C>, C: Push, Result = D>, D: Send + 'static, T: Service + Send, T::Future: Send + 'static, { type Error = T::Error; type Response = AddContext; type Future = BoxFuture<'static, Result>; fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll> { self.inner.poll_ready(cx) } fn call(&mut self, target: Target) -> Self::Future { let service = self.inner.call(target); Box::pin(async move { Ok(AddContext::new(service.await?)) }) } } /// Middleware to add context data from the request pub struct AddContext where A: Default + Push, B: Push, Result = C>, C: Push, Result = D>, { inner: T, marker: PhantomData, } impl AddContext where A: Default + Push, B: Push, Result = C>, C: Push, Result = D>, { pub fn new(inner: T) -> Self { AddContext { inner, marker: PhantomData, } } } impl Service> for AddContext where A: Default + Push, B: Push, Result = C>, C: Push, Result = D>, D: Send + 'static, T: Service<(Request, D)>, { type Error = T::Error; type Future = T::Future; type Response = T::Response; fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll> { self.inner.poll_ready(cx) } fn call(&mut self, request: Request) -> Self::Future { let context = A::default().push(XSpanIdString::get_or_generate(&request)); let headers = request.headers(); { use std::ops::Deref; use swagger::auth::Basic; if let Some(basic) = swagger::auth::from_headers::(&headers) { let auth_data = AuthData::Basic(basic); let context = context.push(Some(auth_data)); let context = context.push(None::); return self.inner.call((request, context)); } } let context = context.push(None::); let context = context.push(None::); self.inner.call((request, context)) } }