101 lines
2.2 KiB
Rust
101 lines
2.2 KiB
Rust
use std::net::SocketAddr;
|
|
use askama::Template;
|
|
use axum::{
|
|
async_trait,
|
|
Extension,
|
|
extract::{self, RequestParts, FromRequest},
|
|
http::StatusCode,
|
|
response::IntoResponse,
|
|
routing::get,
|
|
Router,
|
|
};
|
|
use crate::{
|
|
html_template::HtmlTemplate,
|
|
RequestFactory,
|
|
request_mux::RequestMux,
|
|
trends::TrendsResults,
|
|
};
|
|
|
|
type Mux = RequestMux<RequestFactory>;
|
|
|
|
#[async_trait]
|
|
impl<B> FromRequest<B> for Mux
|
|
where
|
|
B: Send,
|
|
{
|
|
type Rejection = (StatusCode, String);
|
|
|
|
async fn from_request(req: &mut RequestParts<B>) -> Result<Self, Self::Rejection> {
|
|
let Extension(mux) = Extension::<Mux>::from_request(req)
|
|
.await
|
|
.map_err(internal_error)?;
|
|
|
|
Ok(mux)
|
|
}
|
|
}
|
|
|
|
/// Utility function for mapping any error into a `500 Internal Server Error`
|
|
/// response.
|
|
fn internal_error<E>(err: E) -> (StatusCode, String)
|
|
where
|
|
E: std::error::Error,
|
|
{
|
|
(StatusCode::INTERNAL_SERVER_ERROR, err.to_string())
|
|
}
|
|
|
|
|
|
#[derive(Template)]
|
|
#[template(path = "trends.html")]
|
|
struct TrendsPage {
|
|
language: Option<String>,
|
|
results: TrendsResults,
|
|
}
|
|
|
|
impl TrendsPage {
|
|
async fn generate(language: Option<String>, mux: Mux) -> Self {
|
|
let results = mux.request(language.clone()).await;
|
|
|
|
TrendsPage {
|
|
language,
|
|
results,
|
|
}
|
|
}
|
|
|
|
fn template(self) -> HtmlTemplate<Self> {
|
|
HtmlTemplate(self)
|
|
}
|
|
}
|
|
|
|
#[axum_macros::debug_handler]
|
|
async fn home(Extension(mux): Extension<Mux>) -> impl IntoResponse {
|
|
TrendsPage::generate(None, mux)
|
|
.await
|
|
.template()
|
|
}
|
|
|
|
async fn in_language(
|
|
Extension(mux): Extension<Mux>,
|
|
extract::Path(language): extract::Path<String>,
|
|
) -> impl IntoResponse {
|
|
TrendsPage::generate(Some(language), mux)
|
|
.await
|
|
.template()
|
|
}
|
|
|
|
pub async fn start(listen_port: u16, mux: Mux) {
|
|
cave::systemd::status("Starting HTTP server");
|
|
|
|
// build our application with some routes
|
|
let app = Router::new()
|
|
.route("/", get(home))
|
|
.route("/in/:language", get(in_language))
|
|
.layer(Extension(mux));
|
|
|
|
// run it
|
|
let addr = SocketAddr::from(([127, 0, 0, 1], listen_port));
|
|
axum::Server::bind(&addr)
|
|
.serve(app.into_make_service())
|
|
.await
|
|
.unwrap();
|
|
}
|