nomilo/src/routes/ui/zones.rs

use axum::extract::{Query, Path, State, OriginalUri};
use axum::Extension;
use serde::{Deserialize, Serialize};
use serde_json::{Value, json};
use unic_langid::LanguageIdentifier;

use crate::validation;
use crate::AppState;
use crate::errors::Error;
use crate::template::Template;
use crate::resources::dns::friendly;

pub async fn get_records_page(
    Path(zone_name): Path<String>,
    State(app): State<AppState>,
    OriginalUri(url): OriginalUri,
    Extension(lang): Extension<LanguageIdentifier>,
) -> Result<Template<'static, Value>, Error> {
    let zone = app.db.get_zone_by_name(&zone_name).await?;
    let records = zone.get_records(app.records).await?;
    let records = friendly::FriendlyRecords::from(records.clone());

    Ok(Template::new(
        "pages/records.html",
        app.template_engine,
        json!({
            "current_zone": zone.name,
            "records": records,
            "url": url.to_string(),
            "lang": lang.to_string(),
        })
    ))
}

#[derive(Deserialize)]
pub struct NewRecordQuery {
    subdomain: Option<String>,
    config: Option<ConfigurationType>,
    rtype: Option<friendly::FriendlyRType>,
}

#[derive(Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum ConfigurationType {
    Mail,
    Web,
}

pub async fn get_new_record_page(
    Path(zone_name): Path<String>,
    State(app): State<AppState>,
    Query(params): Query<NewRecordQuery>,
    OriginalUri(url): OriginalUri,
    Extension(lang): Extension<LanguageIdentifier>,
) -> Result<Template<'static, Value>, Error> {
    let zone = app.db.get_zone_by_name(&zone_name).await?;

    // Syntax validation of the subdomain
    let (name, domain_error) = if let Some(input_name) = params.subdomain {
        if input_name.is_empty() {
            (Some(zone.name.clone()), None)
        } else {
            let new_name = format!("{input_name}.{zone_name}");
            let res = validation::normalize_domain(&new_name);
            match res {
                Err(error) => (Some(input_name), Some(error)),
                Ok(new_name) => (Some(new_name), None),
            }
        }
    } else {
        (None, None)
    };

    println!("{}", lang);

    Ok(Template::new(
        "pages/new_record.html",
        app.template_engine,
        json!({
            "current_zone": zone.name,
            "new_record_name": name,
            "domain_error": domain_error,
            "config": params.config,
            "rtype": params.rtype,
            "url": url.to_string(),
            "lang": lang.to_string(),
        })
    ))
}