129 lines
3.4 KiB
Rust
129 lines
3.4 KiB
Rust
use serde_json::{Value, json};
|
|
use serde::Serialize;
|
|
use rocket::http::{Status};
|
|
use rocket::http::uri::Origin;
|
|
use rocket::form::Form;
|
|
|
|
use crate::template::Template;
|
|
use crate::models;
|
|
use crate::controllers;
|
|
use crate::DbConn;
|
|
use crate::dns::ZoneConnector;
|
|
|
|
|
|
#[derive(Serialize)]
|
|
pub struct RecordsPage {
|
|
zone: String
|
|
}
|
|
|
|
|
|
// TODO: Check if origin changes if application mounted on different path
|
|
#[get("/zone/<zone>/records")]
|
|
pub async fn get_zone_records_page(user_info: models::UserInfo, zone: models::AbsoluteName, conn: DbConn, origin: &Origin<'_>) -> Result<Template<'static, Value>, Status> {
|
|
let zone_name = zone.to_utf8();
|
|
|
|
let zones = conn.run(move |c| {
|
|
if user_info.is_admin() {
|
|
models::Zone::get_by_name(c, &zone_name)?;
|
|
models::Zone::get_all(c)
|
|
} else {
|
|
user_info.get_zone(c, &zone_name)?;
|
|
user_info.get_zones(c)
|
|
|
|
}
|
|
}).await.map_err(|e| models::ErrorResponse::from(e).status)?;
|
|
|
|
|
|
Ok(Template::new(
|
|
"pages/zone/records.html",
|
|
json!({
|
|
"current_zone": zone.to_utf8(),
|
|
"zones": zones,
|
|
"nav_page": origin.clone().into_normalized().path().as_str(),
|
|
"nav_sections": vec!["zones", zone.to_utf8().as_str()],
|
|
})
|
|
))
|
|
}
|
|
|
|
#[get("/zones")]
|
|
pub async fn get_zones_page(user_info: models::UserInfo, conn: DbConn, origin: &Origin<'_>) -> Result<Template<'static, Value>, Status> {
|
|
let zones = controllers::get_zones(
|
|
&conn,
|
|
user_info
|
|
).await.map_err(|e| e.status)?;
|
|
|
|
Ok(Template::new(
|
|
"pages/zones.html",
|
|
json!({
|
|
"zones": zones,
|
|
"nav_page": origin.clone().into_normalized().path().as_str(),
|
|
"nav_sections": vec!["zones"],
|
|
})
|
|
))
|
|
}
|
|
|
|
|
|
#[get("/zones/new")]
|
|
pub async fn get_create_zone_page(
|
|
conn: DbConn,
|
|
user_info: models::UserInfo,
|
|
origin: &Origin<'_>
|
|
) -> Result<Template<'static, Value>, Status> {
|
|
|
|
user_info
|
|
.check_admin()
|
|
.map_err(|e| models::ErrorResponse::from(e).status)?;
|
|
|
|
let zones = controllers::get_zones(
|
|
&conn,
|
|
user_info
|
|
).await.map_err(|e| e.status)?;
|
|
|
|
Ok(Template::new(
|
|
"pages/zones/new.html",
|
|
json!({
|
|
"zone": None::<models::Zone>,
|
|
"zones": zones,
|
|
"error": None::<String>,
|
|
"nav_page": origin.clone().into_normalized().path().as_str(),
|
|
"nav_sections": vec!["zones", "_new-zone"],
|
|
})
|
|
))
|
|
}
|
|
|
|
#[post("/zones/new", data = "<zone_request>")]
|
|
pub async fn post_create_zone_page(
|
|
conn: DbConn,
|
|
dns_api: Box<dyn ZoneConnector>,
|
|
user_info: models::UserInfo,
|
|
zone_request: Form<models::CreateZoneRequest>,
|
|
origin: &Origin<'_>
|
|
) -> Result<Template<'static, Value>, Status> {
|
|
user_info
|
|
.check_admin()
|
|
.map_err(|e| models::ErrorResponse::from(e).status)?;
|
|
|
|
let zone = controllers::create_zone(
|
|
&conn,
|
|
dns_api,
|
|
user_info.clone(),
|
|
zone_request.into_inner()
|
|
).await.map_err(|e| e.status)?;
|
|
|
|
let zones = controllers::get_zones(
|
|
&conn,
|
|
user_info
|
|
).await.map_err(|e| e.status)?;
|
|
|
|
Ok(Template::new(
|
|
"pages/zones/new.html",
|
|
json!({
|
|
"zone": Some(zone),
|
|
"zones": zones,
|
|
"error": None::<String>,
|
|
"nav_page": origin.clone().into_normalized().path().as_str(),
|
|
"nav_sections": vec!["zones", "_new-zone"],
|
|
})
|
|
))
|
|
}
|