1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

use chrono::{Duration, NaiveDateTime};
use std::ops::Deref;
use uuid::Uuid;

#[derive(Clone, Debug, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct Session<T> {
    #[serde(skip)]
    pub session_id: Uuid,
    pub created_at: NaiveDateTime,
    pub value: T,
    #[serde(skip)]
    pub max_age: Option<Duration>,
    #[serde(skip)]
    pub expires: Option<NaiveDateTime>,
}

impl<T> Session<T> {
    pub fn map<U>(self, map_fn: impl FnOnce(T) -> U) -> Session<U> {
        Session {
            session_id: self.session_id,
            created_at: self.created_at,
            value: map_fn(self.value),
            max_age: self.max_age,
            expires: self.expires,
        }
    }

    pub fn try_map<U, E>(self, try_map_fn: impl FnOnce(T) -> Result<U, E>) -> Result<Session<U>, E> {
        Ok(Session {
            session_id: self.session_id,
            created_at: self.created_at,
            value: try_map_fn(self.value)?,
            max_age: self.max_age,
            expires: self.expires,
        })
    }
}

impl<T> Deref for Session<T> {
    type Target = T;
    fn deref(&self) -> &T {
        &self.value
    }
}

pub trait RawSession<ParsedSession>: Sized {
    type Key;
    type Validation;
    fn try_decode(self, key: &Self::Key, validation: &Self::Validation) -> Result<ParsedSession, anyhow::Error>;
    fn add_extensions(
        session: Result<Option<Self>, anyhow::Error>,
        key: &Self::Key,
        validation: &Self::Validation,
        extensions: &mut http::Extensions,
    );
}

impl<T: Send + Sync + 'static> RawSession<T> for T {
    type Key = ();
    type Validation = ();
    fn try_decode(self, _: &Self::Key, _: &Self::Validation) -> Result<T, anyhow::Error> {
        Ok(self)
    }
    fn add_extensions(
        session: Result<Option<Self>, anyhow::Error>,
        _: &Self::Key,
        _: &Self::Validation,
        extensions: &mut http::Extensions,
    ) {
        match session {
            Ok(Some(session)) => extensions.insert(Some(session)),
            _ => extensions.insert(None::<T>),
        };
    }
}

#[derive(Clone, Debug)]
pub enum RequestSession<T> {
    None,
    SessionId(Uuid),
    Session(Session<T>),
}

#[cfg(feature = "axum-core-02")]
#[async_trait]
impl<B, T> axum_core_02::extract::FromRequest<B> for Session<T>
where
    B: Send,
    T: serde::de::DeserializeOwned + Send + Sync + 'static,
{
    type Rejection = http::StatusCode;

    async fn from_request(req: &mut axum_core_02::extract::RequestParts<B>) -> Result<Self, Self::Rejection> {
        req.extensions_mut()
            .remove::<Option<Session<T>>>()
            .flatten()
            .ok_or_else(|| {
                log::error!("tried to extract session::Session from request when there was none, use axum::extract::Extract<Option<session::Session<T>>> instead of axum::extract::Extract<session::Session<T>> for correct extraction session from requests");
                http::StatusCode::INTERNAL_SERVER_ERROR
            })
    }
}

#[cfg(feature = "axum-core-03")]
#[async_trait]
impl<S, B, T> axum_core_03::extract::FromRequest<S, B> for Session<T>
where
    S: Send + Sync,
    B: Send + 'static,
    T: serde::de::DeserializeOwned + Send + Sync + 'static,
{
    type Rejection = http::StatusCode;

    async fn from_request(mut req: http::Request<B>, _: &S) -> Result<Self, Self::Rejection> {
        req.extensions_mut()
            .remove::<Option<Session<T>>>()
            .flatten()
            .ok_or_else(|| {
                log::error!("tried to extract session::Session from request when there was none, use axum::extract::Extract<Option<session::Session<T>>> instead of axum::extract::Extract<session::Session<T>> for correct extraction session from requests");
                http::StatusCode::INTERNAL_SERVER_ERROR
            })
    }
}