74 lines
1.7 KiB
Rust
74 lines
1.7 KiB
Rust
|
use actix::{Actor, AsyncContext, Context, Handler};
|
||
|
|
use actix::Message;
|
||
|
|
|
||
|
|
use crate::constants::*;
|
||
|
|
use crate::data::client::ClientID;
|
||
|
|
use crate::data::user::UserID;
|
||
|
|
use crate::utils::time::time;
|
||
|
|
|
||
|
|
#[derive(Clone, Debug, serde::Serialize, serde::Deserialize, Eq, PartialEq)]
|
||
|
|
pub struct SessionID(pub String);
|
||
|
|
|
||
|
|
#[derive(Clone, Debug)]
|
||
|
|
pub struct Session {
|
||
|
|
pub session_id: SessionID,
|
||
|
|
pub client: ClientID,
|
||
|
|
pub user: UserID,
|
||
|
|
pub redirect_uri: String,
|
||
|
|
|
||
|
|
pub authorization_code: String,
|
||
|
|
pub code_expire_on: u64,
|
||
|
|
|
||
|
|
pub token: String,
|
||
|
|
pub token_expire_at: u64,
|
||
|
|
|
||
|
|
pub nonce: Option<String>,
|
||
|
|
pub code_challenge: Option<(String, String)>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Session {
|
||
|
|
pub fn is_expired(&self) -> bool {
|
||
|
|
self.code_expire_on < time() || self.token_expire_at < time()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Message)]
|
||
|
|
#[rtype(result = "()")]
|
||
|
|
pub struct PushNewSession(pub Session);
|
||
|
|
|
||
|
|
|
||
|
|
pub struct OpenIDSessionsActor {
|
||
|
|
session: Vec<Session>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl OpenIDSessionsActor {
|
||
|
|
pub fn clean_old_sessions(&mut self) {
|
||
|
|
self.session.retain(|s| !s.is_expired());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Default for OpenIDSessionsActor {
|
||
|
|
fn default() -> Self {
|
||
|
|
Self { session: vec![] }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Actor for OpenIDSessionsActor {
|
||
|
|
type Context = Context<Self>;
|
||
|
|
|
||
|
|
fn started(&mut self, ctx: &mut Self::Context) {
|
||
|
|
// Clean up at a regular interval failed attempts
|
||
|
|
ctx.run_interval(OPEN_ID_SESSION_CLEANUP_INTERVAL, |act, _ctx| {
|
||
|
|
log::trace!("Cleaning up old login sessions");
|
||
|
|
act.clean_old_sessions();
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Handler<PushNewSession> for OpenIDSessionsActor {
|
||
|
|
type Result = ();
|
||
|
|
|
||
|
|
fn handle(&mut self, msg: PushNewSession, _ctx: &mut Self::Context) -> Self::Result {
|
||
|
|
self.session.push(msg.0)
|
||
|
|
}
|
||
|
|
}
|