quick rust website

src/handlers.rs

@@ -0,0 +1,19 @@
+use askama::Template;
+use axum::{
+    http::StatusCode,
+    response::{Html, IntoResponse},
+};
+
+pub async fn render_homepage() -> impl IntoResponse {
+    #[derive(Template)]
+    #[template(path = "homepage.html")]
+    struct HomePageTemplate<'a> {
+        lang: &'a str,
+    }
+
+    let tmpl = HomePageTemplate { lang: "en" };
+    match tmpl.render() {
+        Ok(html) => Html(html).into_response(),
+        Err(_) => (StatusCode::INTERNAL_SERVER_ERROR, "Template render error").into_response(),
+    }
+}

src/lib.rs

@@ -0,0 +1,59 @@
+use std::{net::SocketAddr, time::Duration};
+
+use axum::{
+    Router,
+    http::{Method, header},
+    routing::get,
+};
+use tower_governor::{GovernorLayer, governor::GovernorConfigBuilder};
+use tower_http::cors::{Any, CorsLayer};
+
+use crate::handlers::render_homepage;
+
+pub mod handlers;
+
+/// Start the server with the given configuration and output file.
+/// `addr` should be something like "127.0.0.1:8081"
+pub async fn run_server(addr: &str, verbose: bool) -> anyhow::Result<()> {
+    // CORS
+    let cors = CorsLayer::new()
+        .allow_origin(Any)
+        .allow_methods([Method::GET, Method::POST])
+        .allow_headers([header::AUTHORIZATION, header::CONTENT_TYPE]);
+
+    // rate limiter
+    let governor_conf = GovernorConfigBuilder::default()
+        .per_second(3)
+        .burst_size(10)
+        .finish()
+        .unwrap();
+
+    // a separate background task to clean up
+    let governor_limiter = governor_conf.limiter().clone();
+    let interval = Duration::from_secs(60);
+    std::thread::spawn(move || {
+        loop {
+            std::thread::sleep(interval);
+            if verbose {
+                tracing::info!("rate limiting storage size: {}", governor_limiter.len());
+            }
+            governor_limiter.retain_recent();
+        }
+    });
+
+    let app = Router::new()
+        .route("/", get(render_homepage))
+        .layer(cors)
+        .layer(GovernorLayer::new(governor_conf));
+
+    let listener = tokio::net::TcpListener::bind(addr).await?;
+    tracing::info!("Listening on {}", addr);
+
+    axum::serve(
+        listener,
+        app.into_make_service_with_connect_info::<SocketAddr>(),
+    )
+    .await?;
+
+    Ok(())
+}

src/main.rs

@@ -0,0 +1,26 @@
+use clap::Parser;
+
+use mycoolwebsite::run_server;
+
+#[derive(Parser, Debug)]
+#[command(author, version, about, long_about = None)]
+struct Cli {
+    #[arg(short, long)]
+    verbose: bool,
+}
+
+/// Cli handler
+#[tokio::main]
+async fn main() -> anyhow::Result<()> {
+    let cli = Cli::parse();
+
+    let subscriber = tracing_subscriber::FmtSubscriber::new();
+    tracing::subscriber::set_global_default(subscriber).ok();
+
+    let port = std::env::var("SERVER_PORT").unwrap_or("8081".to_string());
+    let addr = format!("127.0.0.1:{port}");
+
+    run_server(&addr, cli.verbose).await?;
+
+    Ok(())
+}