use std::future::{Ready, ready};

use actix_web::dev::Payload;
use actix_web::{FromRequest, HttpMessage, HttpRequest};

use crate::error::SessionError;
use crate::state::Session;

pub struct AuthedUser(pub i64);

impl FromRequest for AuthedUser {
    type Error = actix_web::Error;
    type Future = Ready<Result<Self, Self::Error>>;

    fn from_request(req: &HttpRequest, _payload: &mut Payload) -> Self::Future {
        let Some(session) = req.extensions().get::<Session>().cloned() else {
            return ready(Err(actix_web::error::ErrorInternalServerError(
                "session is not initialized",
            )));
        };
        match session.uid() {
            Some(uid) => ready(Ok(Self(uid))),
            None => ready(Err(SessionError::Unauthorized.into())),
        }
    }
}

pub struct OptionalUser(pub Option<i64>);

impl FromRequest for OptionalUser {
    type Error = actix_web::Error;
    type Future = Ready<Result<Self, Self::Error>>;

    fn from_request(req: &HttpRequest, _payload: &mut Payload) -> Self::Future {
        match req.extensions().get::<Session>() {
            Some(session) => ready(Ok(Self(session.uid()))),
            None => ready(Err(actix_web::error::ErrorInternalServerError(
                "session is not initialized",
            ))),
        }
    }
}

impl FromRequest for Session {
    type Error = actix_web::Error;
    type Future = Ready<Result<Self, Self::Error>>;

    fn from_request(req: &HttpRequest, _payload: &mut Payload) -> Self::Future {
        match req.extensions().get::<Session>() {
            Some(session) => ready(Ok(session.clone())),
            None => ready(Err(actix_web::error::ErrorInternalServerError(
                "session is not initialized",
            ))),
        }
    }
}