109 lines
2.6 KiB
Rust
109 lines
2.6 KiB
Rust
use std::sync::Arc;
|
|
|
|
use axum::{extract::Path, response::Html, routing::get, Extension, Router};
|
|
use serde_derive::Serialize;
|
|
use tracing::{instrument, log::*};
|
|
|
|
use crate::{
|
|
post::{render_post, Post},
|
|
State, WebsiteError,
|
|
};
|
|
|
|
use super::HIT_COUNTER;
|
|
|
|
pub fn router() -> Router {
|
|
Router::new()
|
|
.route("/", get(index))
|
|
.route("/:slug/", get(view))
|
|
.route("/:slug/index.md", get(super::not_found))
|
|
.fallback_service(tower_http::services::ServeDir::new("./posts"))
|
|
}
|
|
|
|
#[derive(Serialize, Debug)]
|
|
struct PageContext<'a> {
|
|
title: &'a str,
|
|
}
|
|
|
|
#[instrument(skip(state))]
|
|
pub async fn index(Extension(state): Extension<Arc<State>>) -> Result<Html<String>, WebsiteError> {
|
|
let mut posts: Vec<&Post> = state
|
|
.posts
|
|
.values()
|
|
.filter(|p| p.is_published())
|
|
.collect();
|
|
|
|
posts.sort_by_key(|p| &p.date);
|
|
posts.reverse();
|
|
|
|
let ctx = PageContext {
|
|
title: "Posts",
|
|
};
|
|
|
|
let mut c = tera::Context::new();
|
|
c.insert("page", &ctx);
|
|
c.insert("posts", &posts);
|
|
|
|
let res = state.tera
|
|
.render("posts_index.html", &c)?;
|
|
|
|
HIT_COUNTER.with_label_values(&["/posts/"]).inc();
|
|
Ok(Html(res))
|
|
}
|
|
|
|
#[instrument(skip(state))]
|
|
pub async fn view(
|
|
Path(slug): Path<String>,
|
|
Extension(state): Extension<Arc<State>>,
|
|
) -> Result<Html<String>, WebsiteError> {
|
|
let post = state.posts.get(&slug).ok_or(WebsiteError::NotFound)?;
|
|
if !post.is_published() {
|
|
warn!("attempted to view post before it has been published!");
|
|
return Err(WebsiteError::NotFound);
|
|
}
|
|
|
|
let res = render_post(&state.tera, post).await?;
|
|
|
|
HIT_COUNTER
|
|
.with_label_values(&[&format!("/posts/{slug}/")])
|
|
.inc();
|
|
Ok(Html(res))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use chrono::DateTime;
|
|
|
|
use crate::post::Post;
|
|
|
|
use super::PageContext;
|
|
|
|
#[test]
|
|
fn render_index() {
|
|
let posts = vec![Post {
|
|
title: "test".into(),
|
|
slug: "test".into(),
|
|
date: Some(DateTime::parse_from_rfc3339("2023-03-26T13:04:01+02:00").unwrap()),
|
|
..Default::default()
|
|
},
|
|
Post {
|
|
title: "test2".into(),
|
|
slug: "test2".into(),
|
|
date: None,
|
|
..Default::default()
|
|
}];
|
|
let page = PageContext {
|
|
title: "Posts",
|
|
};
|
|
let mut ctx = tera::Context::new();
|
|
ctx.insert("page", &page);
|
|
ctx.insert("posts", &posts);
|
|
|
|
let tera = tera::Tera::new("templates/**/*").unwrap();
|
|
let _res = tera
|
|
.render(
|
|
"posts_index.html",
|
|
&ctx,
|
|
)
|
|
.unwrap();
|
|
}
|
|
}
|