gearbox-rs-macros 1.3.0

use proc_macro2::TokenStream as TokenStream2;
use quote::{format_ident, quote};

use crate::crud::parse::CrudEntityInfo;

/// Generate the get handler (GET /{path}/{id}).
pub(super) fn generate_get_handler(entity: &CrudEntityInfo, path: &str, core: &TokenStream2, pg: &TokenStream2) -> (TokenStream2, TokenStream2) {
    let name = &entity.name;
    let snake_name = entity.snake_case_name();
    let handler_name = format_ident!("__crud_{}_get", snake_name);
    let response_name = format_ident!("{}Response", name);

    let pk_fields = entity.pk_fields();
    let id_path = format!("{}/{{id}}", path);

    let id_type = super::pk_id_type(&pk_fields);

    let not_found = super::not_found_response(core);
    let err_response = super::internal_error_response(core, &snake_name, "get");

    let handler = quote! {
        async fn #handler_name(
            #core::Inject(db): #core::Inject<#pg::PgClient>,
            #core::Path(id): #core::Path<#id_type>,
        ) -> axum::response::Response {
            use axum::response::IntoResponse;

            let result: Result<Option<#name>, #pg::PgError> =
                <#pg::PgClient as #pg::PgRepository<#name>>::find_by_id(&*db, &id).await;

            match result {
                Ok(Some(entity)) => {
                    let response = #response_name::from(entity);
                    (axum::http::StatusCode::OK, #core::Json(response)).into_response()
                }
                Ok(None) => {
                    #not_found
                }
                Err(e) => {
                    #err_response
                }
            }
        }
    };

    let registration = quote! {
        #core::inventory::submit!(#core::RouteRegistration {
            path: #id_path,
            method: "GET",
            handler: || axum::routing::get(#handler_name),
        });
    };

    (handler, registration)
}