56 lines
1.7 KiB
Rust
56 lines
1.7 KiB
Rust
use crate::State;
|
|
use actix_web::{
|
|
error::ErrorInternalServerError, web::Data, Error as WebError, HttpResponse,
|
|
Result as WebResult,
|
|
};
|
|
use tera::{Context, Tera};
|
|
#[cfg(feature = "rss_feed")]
|
|
use rss::{ChannelBuilder, ItemBuilder};
|
|
|
|
pub fn index(tmpl: Data<Tera>, state: Data<State>) -> WebResult<HttpResponse, WebError> {
|
|
let state = state.read().unwrap();
|
|
let mut ctx = Context::new();
|
|
ctx.insert("results", &*state);
|
|
let s = tmpl.render("index.html", &ctx).map_err(|e| {
|
|
println!("{:?}", e);
|
|
ErrorInternalServerError("Template error")
|
|
})?;
|
|
Ok(HttpResponse::Ok().content_type("text/html").body(s))
|
|
}
|
|
|
|
#[cfg(feature = "json")]
|
|
pub fn json_endpoint(state: Data<State>) -> HttpResponse {
|
|
let state = state.read().unwrap();
|
|
HttpResponse::Ok().json(&*state.groups)
|
|
}
|
|
|
|
#[cfg(feature = "rss_feed")]
|
|
pub fn rss_endpoint(state: Data<State>) -> HttpResponse {
|
|
let state = state.read().unwrap();
|
|
let mut chan = ChannelBuilder::default()
|
|
.title("endstat")
|
|
.description("Endpoint statuses")
|
|
.link("https://git.eddie.sh/edward/endstat")
|
|
.build()
|
|
.unwrap();
|
|
|
|
let mut endpoints = vec![];
|
|
|
|
for group in &state.groups {
|
|
for endpoint in &group.endpoints {
|
|
endpoints.push(
|
|
ItemBuilder::default()
|
|
.title(format!("{}: {}", group.label, endpoint.label))
|
|
.link(endpoint.location.clone())
|
|
.description(Some(endpoint.status.clone().into()))
|
|
.build()
|
|
.unwrap()
|
|
);
|
|
}
|
|
}
|
|
|
|
chan.set_items(endpoints);
|
|
|
|
HttpResponse::Ok().content_type("application/rss+xml").body(chan.to_string())
|
|
}
|