2026-07-20 09:04:57 +07:00
|
|
|
//! Authentication middleware — session-lock based auth for Axum.
|
2026-07-20 12:26:08 +07:00
|
|
|
//!
|
|
|
|
|
//! Validates `X-Session-Id` header against the `SessionRepository` before
|
|
|
|
|
//! forwarding the request to the inner service.
|
2026-07-20 09:04:57 +07:00
|
|
|
|
|
|
|
|
use std::future::Future;
|
2026-07-20 12:26:08 +07:00
|
|
|
use std::path::PathBuf;
|
2026-07-20 09:04:57 +07:00
|
|
|
use std::pin::Pin;
|
2026-07-20 12:26:08 +07:00
|
|
|
use std::sync::Arc;
|
2026-07-20 09:04:57 +07:00
|
|
|
use std::task::{Context, Poll};
|
|
|
|
|
|
|
|
|
|
use axum::body::Body;
|
|
|
|
|
use axum::http::{Request, Response, StatusCode};
|
|
|
|
|
use axum::response::IntoResponse;
|
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
|
use tower::{Layer, Service};
|
2026-07-20 12:26:08 +07:00
|
|
|
use zesdex_domain::auth::{SessionId, SessionRepository};
|
2026-07-20 09:04:57 +07:00
|
|
|
|
|
|
|
|
/// Identity extracted from a validated session.
|
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
|
|
|
pub struct SessionIdentity {
|
|
|
|
|
pub session_id: String,
|
|
|
|
|
pub user_agent: String,
|
|
|
|
|
pub connected_at: i64,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl SessionIdentity {
|
|
|
|
|
pub fn new(session_id: String, user_agent: String) -> Self {
|
|
|
|
|
let connected_at = chrono::Utc::now().timestamp();
|
|
|
|
|
Self {
|
|
|
|
|
session_id,
|
|
|
|
|
user_agent,
|
|
|
|
|
connected_at,
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Tower Layer that produces SessionAuthMiddleware services.
|
2026-07-20 12:26:08 +07:00
|
|
|
///
|
|
|
|
|
/// Holds a reference to the `SessionRepository` and the base directory
|
|
|
|
|
/// needed to validate session IDs.
|
2026-07-20 09:04:57 +07:00
|
|
|
#[derive(Debug, Clone)]
|
2026-07-20 12:26:08 +07:00
|
|
|
pub struct SessionAuthLayer<R: SessionRepository + Send + Sync + 'static> {
|
|
|
|
|
base_dir: PathBuf,
|
|
|
|
|
repo: Arc<R>,
|
|
|
|
|
}
|
2026-07-20 09:04:57 +07:00
|
|
|
|
2026-07-20 12:26:08 +07:00
|
|
|
impl<R: SessionRepository + Send + Sync + 'static> SessionAuthLayer<R> {
|
|
|
|
|
pub fn new(base_dir: PathBuf, repo: Arc<R>) -> Self {
|
|
|
|
|
Self { base_dir, repo }
|
2026-07-20 09:04:57 +07:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2026-07-20 12:26:08 +07:00
|
|
|
impl<S, R> Layer<S> for SessionAuthLayer<R>
|
|
|
|
|
where
|
|
|
|
|
R: SessionRepository + Send + Sync + 'static,
|
|
|
|
|
{
|
|
|
|
|
type Service = SessionAuthMiddleware<S, R>;
|
2026-07-20 09:04:57 +07:00
|
|
|
|
|
|
|
|
fn layer(&self, inner: S) -> Self::Service {
|
2026-07-20 12:26:08 +07:00
|
|
|
SessionAuthMiddleware {
|
|
|
|
|
inner,
|
|
|
|
|
base_dir: self.base_dir.clone(),
|
|
|
|
|
repo: self.repo.clone(),
|
|
|
|
|
}
|
2026-07-20 09:04:57 +07:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Tower Service that validates X-Session-Id before forwarding.
|
|
|
|
|
#[derive(Debug, Clone)]
|
2026-07-20 12:26:08 +07:00
|
|
|
pub struct SessionAuthMiddleware<S, R: SessionRepository + Send + Sync + 'static> {
|
2026-07-20 09:04:57 +07:00
|
|
|
inner: S,
|
2026-07-20 12:26:08 +07:00
|
|
|
base_dir: PathBuf,
|
|
|
|
|
repo: Arc<R>,
|
2026-07-20 09:04:57 +07:00
|
|
|
}
|
|
|
|
|
|
2026-07-20 12:26:08 +07:00
|
|
|
impl<S, ReqBody, R> Service<Request<ReqBody>> for SessionAuthMiddleware<S, R>
|
2026-07-20 09:04:57 +07:00
|
|
|
where
|
|
|
|
|
S: Service<Request<ReqBody>, Response = Response<Body>> + Send + 'static,
|
|
|
|
|
S::Future: Send + 'static,
|
|
|
|
|
ReqBody: Send + 'static,
|
2026-07-20 12:26:08 +07:00
|
|
|
R: SessionRepository + Send + Sync + 'static,
|
2026-07-20 09:04:57 +07:00
|
|
|
{
|
|
|
|
|
type Response = S::Response;
|
|
|
|
|
type Error = S::Error;
|
|
|
|
|
type Future =
|
|
|
|
|
Pin<Box<dyn Future<Output = Result<Self::Response, Self::Error>> + Send + 'static>>;
|
|
|
|
|
|
|
|
|
|
fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
|
|
|
|
|
self.inner.poll_ready(cx)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn call(&mut self, req: Request<ReqBody>) -> Self::Future {
|
|
|
|
|
let session_id = req
|
|
|
|
|
.headers()
|
|
|
|
|
.get("X-Session-Id")
|
|
|
|
|
.and_then(|v| v.to_str().ok())
|
|
|
|
|
.map(|s| s.to_string());
|
|
|
|
|
|
2026-07-20 12:26:08 +07:00
|
|
|
// Validate the session against the repository.
|
|
|
|
|
match session_id {
|
|
|
|
|
Some(sid) => match SessionId::new(&sid) {
|
|
|
|
|
Ok(id) => match self.repo.load_session(&self.base_dir, &id) {
|
|
|
|
|
Ok(_session) => {
|
|
|
|
|
// Session is valid — forward the request.
|
|
|
|
|
let fut = self.inner.call(req);
|
|
|
|
|
return Box::pin(fut);
|
|
|
|
|
}
|
|
|
|
|
Err(_) => { /* fall through to 401 */ }
|
|
|
|
|
},
|
|
|
|
|
Err(_) => { /* fall through to 401 */ }
|
|
|
|
|
},
|
|
|
|
|
None => { /* fall through to 401 */ }
|
2026-07-20 09:04:57 +07:00
|
|
|
}
|
|
|
|
|
|
2026-07-20 12:26:08 +07:00
|
|
|
Box::pin(async move {
|
|
|
|
|
Ok((StatusCode::UNAUTHORIZED, "missing or invalid X-Session-Id header").into_response())
|
|
|
|
|
})
|
2026-07-20 09:04:57 +07:00
|
|
|
}
|
|
|
|
|
}
|
