xesite/src/handlers/feeds.rs

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)
}