Initial commit

src/main.rs

@@ -0,0 +1,50 @@
+/* hrtime - transgender survey website
+ * Copyright (C) 2025 Olive <hello@grasswren.net>
+ * see LICENCE file for licensing information */
+
+pub mod schema;
+pub mod models;
+pub mod form;
+
+use std::path::{Path, PathBuf};
+
+use rocket::Request;
+use rocket::fs::NamedFile;
+use rocket_sync_db_pools::{database, diesel};
+
+#[database("db")]
+pub struct Database(diesel::SqliteConnection);
+
+#[rocket::get("/<file..>")]
+async fn files(file: PathBuf) -> Option<NamedFile> {
+    let pages = vec!["about", "future", "contact", "form", "thanks"];
+    let other = vec!["hamburger.svg", "logo.webp", "script.js", "style.css"];
+
+    let mut path = match file.to_str() {
+        Some(path) => path.to_owned(),
+        None => return NamedFile::open(Path::new("/var/www/404.html")).await.ok(),
+    };
+
+    if path == "" {
+        path = "index.html".to_string();
+    } else if pages.contains(&path.as_str()) {
+        path += ".html";
+    } else if !other.contains(&path.as_str()) {
+        path = "404.html".to_string();
+    }
+    NamedFile::open(Path::new("/var/www/").join(path)).await.ok()
+}
+
+#[rocket::catch(404)]
+async fn not_found(_: &Request<'_>) -> Option<NamedFile> {
+    NamedFile::open(Path::new("/var/www/404.html")).await.ok()
+}
+
+#[rocket::launch]
+fn rocket() -> _ {
+    rocket::build()
+        .attach(Database::fairing())
+        .mount("/form", rocket::routes![form::form])
+        .mount("/", rocket::routes![files])
+        .register("/", rocket::catchers![not_found])
+}