Ben / RubHub

use axum::{
    extract::{FromRef, FromRequestParts, Path},
    http::{StatusCode, request::Parts},
};

use crate::{GlobalState, Project, User};

pub struct PathUserProjectBranch(pub User, pub Project, pub String);

impl<S> FromRequestParts<S> for PathUserProjectBranch
where
    S: Send + Sync,
    GlobalState: FromRef<S>,
{
    type Rejection = (StatusCode, &'static str);

    async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
        let Path((username, slug, branch)): Path<(String, String, String)> =
            Path::from_request_parts(parts, state)
                .await
                .map_err(|_| (StatusCode::BAD_REQUEST, "Invalid path parameter"))?;

        let state = GlobalState::from_ref(state);

        if let Some(user_slug) = username.strip_prefix("~") {
            if let Ok(user) = User::load(&state, user_slug).await {
                if let Ok(project) = Project::load(&state, user_slug, &slug).await {
                    return Ok(PathUserProjectBranch(user, project, branch));
                };
            }
        };
        Err((StatusCode::NOT_FOUND, "PathUserProject not found"))
    }
}