eeg_certs/backend/src/controller/person/mod.rs

use reqwest::Client;
use rocket::http::Status;
use rocket::serde::json::Json;

use crate::model::reniec_person::ReniecPerson;
use crate::{db, model::person::Person};

#[get("/person/<dni>")]
pub async fn get_by_dni(dni: i32) -> (Status, Json<Person>) {
    let db = db();

    // TODO: move db logic to model
    /*
     * Search person in DB
     */
    let query = sqlx::query!("SELECT * FROM person WHERE person_dni = ?", dni)
        .fetch_one(db)
        .await;

    // Return if found
    if let Ok(row) = query {
        let person = Person {
            person_id: row.person_id,
            person_dni: row.person_dni,
            person_names: row.person_names,
            person_paternal_surname: row.person_paternal_surname,
            person_maternal_surname: row.person_maternal_surname,
        };

        return (Status::Ok, Json(person));
    }

    /*
     * Person not found in DB. Search in fake RENIEC API with reqwest
     */
    let reniec_api_key = std::env::var("RENIEC_API").expect("RENIEC_API env var is not set!");
    let client = Client::new();
    let reqwest_result = client
        .get(format!("https://api.apis.net.pe/v1/dni?numero={}", &dni))
        .bearer_auth(reniec_api_key)
        .header("Content-Type", "application/json")
        .header("charset", "utf-8")
        .send()
        .await;

    let reniec_person = match reqwest_result {
        Ok(data) => {
            let person = data.json::<ReniecPerson>().await;
            match person {
                Ok(p) => Some(p),
                Err(reason) => {
                    eprintln!(
                        "Error parsing data from fake RENIEC API into ReniecPerson: {:?}",
                        reason
                    );
                    None
                }
            }
        }
        Err(error) => {
            eprintln!("Error fetching person from fake RENIEC API: {:?}", error);
            None
        }
    };

    // If person is found in fake RENIEC API, insert to DB and return
    if let Some(p) = reniec_person {
        let query = sqlx::query!(
            "INSERT INTO person (person_dni, person_names, person_paternal_surname, person_maternal_surname) VALUES (?, ?, ?, ?)",
            dni,
            p.nombres,
            p.apellidoPaterno,
            p.apellidoMaterno,
        )
        .execute(db)
        .await;

        if let Err(reason) = query {
            eprintln!("Error inserting person into DB: {:?}", reason);
            return (Status::InternalServerError, Json(Person::default()));
        }

        let person = Person {
            person_id: 0,
            person_dni: format!("{}", dni),
            person_names: p.nombres,
            person_paternal_surname: p.apellidoPaterno,
            person_maternal_surname: p.apellidoMaterno,
        };

        return (Status::Ok, Json(person));
    }

    // Return error
    (Status::NotFound, Json(Person::default()))
}