1use async_trait::async_trait;
2use tide::{Middleware, Next, Request};
3
4mod basic;
5mod bearer;
6mod error;
7mod traits;
8
9pub use basic::Basic;
10pub use bearer::Bearer;
11pub use error::{Error, Result};
12pub use traits::{AuthValue, Scheme};
13
14pub struct Auth<S: Scheme + Send + Sync>(S);
15
16#[async_trait]
17impl<S: Scheme + Send + Sync> Scheme for Auth<S> {
18 fn header_name() -> http_types::headers::HeaderName {
19 S::header_name()
20 }
21
22 async fn parse(&self, header_value: &http_types::headers::HeaderValue) -> Result<AuthValue> {
23 self.0.parse(header_value).await
24 }
25}
26
27impl<S: Scheme + Send + Sync> From<S> for Auth<S> {
28 fn from(scheme: S) -> Self {
29 Auth(scheme)
30 }
31}
32
33#[async_trait]
34impl<State, S> Middleware<State> for Auth<S>
35where
36 State: Clone + Send + Sync + 'static,
37 S: Scheme + Send + Sync + 'static,
38{
39 async fn handle(&self, mut request: Request<State>, next: Next<'_, State>) -> tide::Result {
40 let res = if let None = request.ext::<AuthValue>() {
41 let header_value = match request.header(&S::header_name()) {
42 Some(header_value) => header_value,
43 _ => return Err(Error::Invalid.into()),
44 };
45
46 let value = self.parse(header_value).await?;
47
48 request.set_ext(value);
49
50 next.run(request).await
51 } else {
52 next.run(request).await
53 };
54
55 Ok(res)
56 }
57}