119 lines
3.9 KiB
Rust
119 lines
3.9 KiB
Rust
use super::LAST_MODIFIED;
|
|
use crate::{app::State, post::Post, templates};
|
|
use lazy_static::lazy_static;
|
|
use prometheus::{opts, register_int_counter_vec, IntCounterVec};
|
|
use std::{io, sync::Arc};
|
|
use tracing::instrument;
|
|
use warp::{http::Response, Rejection, Reply};
|
|
|
|
lazy_static! {
|
|
pub static ref HIT_COUNTER: IntCounterVec = register_int_counter_vec!(
|
|
opts!("feed_hits", "Number of hits to various feeds"),
|
|
&["kind"]
|
|
)
|
|
.unwrap();
|
|
pub static ref ETAG: String = format!(r#"W/"{}""#, uuid::Uuid::new_v4().to_simple());
|
|
}
|
|
|
|
#[instrument(skip(state))]
|
|
pub async fn jsonfeed(state: Arc<State>, since: Option<String>) -> Result<impl Reply, Rejection> {
|
|
HIT_COUNTER.with_label_values(&["json"]).inc();
|
|
let state = state.clone();
|
|
Ok(warp::reply::json(&state.jf))
|
|
}
|
|
|
|
#[instrument(skip(state))]
|
|
pub async fn new_post(state: Arc<State>) -> Result<impl Reply, Rejection> {
|
|
let state = state.clone();
|
|
let everything = state.everything.clone();
|
|
let p: &Post = everything.iter().next().unwrap();
|
|
Ok(warp::reply::json(&p.new_post))
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub enum RenderError {
|
|
Build(warp::http::Error),
|
|
IO(io::Error),
|
|
}
|
|
|
|
impl warp::reject::Reject for RenderError {}
|
|
|
|
#[instrument(skip(state))]
|
|
pub async fn atom(state: Arc<State>, since: Option<String>) -> Result<impl Reply, Rejection> {
|
|
if let Some(etag) = since {
|
|
if etag == ETAG.clone() {
|
|
return Response::builder()
|
|
.status(304)
|
|
.header("Content-Type", "text/plain")
|
|
.body(
|
|
"You already have the newest version of this feed."
|
|
.to_string()
|
|
.into_bytes(),
|
|
)
|
|
.map_err(RenderError::Build)
|
|
.map_err(warp::reject::custom);
|
|
}
|
|
}
|
|
|
|
HIT_COUNTER.with_label_values(&["atom"]).inc();
|
|
let state = state.clone();
|
|
let mut buf = Vec::new();
|
|
templates::blog_atom_xml(&mut buf, state.everything.clone())
|
|
.map_err(RenderError::IO)
|
|
.map_err(warp::reject::custom)?;
|
|
Response::builder()
|
|
.status(200)
|
|
.header("Content-Type", "application/atom+xml")
|
|
.header("ETag", ETAG.clone())
|
|
.header("Last-Modified", &*LAST_MODIFIED)
|
|
.body(buf)
|
|
.map_err(RenderError::Build)
|
|
.map_err(warp::reject::custom)
|
|
}
|
|
|
|
#[instrument(skip(state))]
|
|
pub async fn rss(state: Arc<State>, since: Option<String>) -> Result<impl Reply, Rejection> {
|
|
if let Some(etag) = since {
|
|
if etag == ETAG.clone() {
|
|
return Response::builder()
|
|
.status(304)
|
|
.header("Content-Type", "text/plain")
|
|
.body(
|
|
"You already have the newest version of this feed."
|
|
.to_string()
|
|
.into_bytes(),
|
|
)
|
|
.map_err(RenderError::Build)
|
|
.map_err(warp::reject::custom);
|
|
}
|
|
}
|
|
|
|
HIT_COUNTER.with_label_values(&["rss"]).inc();
|
|
let state = state.clone();
|
|
let mut buf = Vec::new();
|
|
templates::blog_rss_xml(&mut buf, state.everything.clone())
|
|
.map_err(RenderError::IO)
|
|
.map_err(warp::reject::custom)?;
|
|
Response::builder()
|
|
.status(200)
|
|
.header("Content-Type", "application/rss+xml")
|
|
.header("ETag", ETAG.clone())
|
|
.header("Last-Modified", &*LAST_MODIFIED)
|
|
.body(buf)
|
|
.map_err(RenderError::Build)
|
|
.map_err(warp::reject::custom)
|
|
}
|
|
|
|
#[instrument(skip(state))]
|
|
pub async fn sitemap(state: Arc<State>) -> Result<impl Reply, Rejection> {
|
|
HIT_COUNTER.with_label_values(&["sitemap"]).inc();
|
|
let state = state.clone();
|
|
Response::builder()
|
|
.status(200)
|
|
.header("Content-Type", "application/xml")
|
|
.header("Last-Modified", &*LAST_MODIFIED)
|
|
.body(state.sitemap.clone())
|
|
.map_err(RenderError::Build)
|
|
.map_err(warp::reject::custom)
|
|
}
|