inform people of participation; require updating personal data before joining

src/tera/ergo.rs

@@ -1,6 +1,6 @@
 use std::env;
 
-use chrono::Utc;
+use chrono::{Datelike, Utc};
 use rocket::{
     form::Form,
     fs::TempFile,
@@ -17,8 +17,7 @@ use sqlx::SqlitePool;
 use tera::Context;
 
 use crate::model::{
-    log::Log,
-    user::{AdminUser, User, UserWithDetails},
+    log::Log, notification::Notification, role::Role, user::{AdminUser, User, UserWithDetails}
 };
 
 #[derive(Serialize)]
@@ -50,7 +49,7 @@ async fn send(db: &State<SqlitePool>, _user: AdminUser) -> Template {
     .unwrap();
 
     Template::render(
-        "ergo.final",
+        "ergo/final",
         context!(loggedin_user: &UserWithDetails::from_user(_user.user, db).await, thirty, dozen),
     )
 }
@@ -98,6 +97,19 @@ async fn update(
 
 #[get("/")]
 async fn index(db: &State<SqlitePool>, user: User, flash: Option<FlashMessage<'_>>) -> Template {
+    let mut context = Context::new();
+    if let Some(msg) = flash {
+        context.insert("flash", &msg.into_inner());
+    }
+    context.insert("loggedin_user", &UserWithDetails::from_user(user.clone(), db).await);
+
+    if !user.has_role(db, "ergo").await {
+        return Template::render(
+            "ergo/missing-data",
+            context.into_json()
+        );
+    }
+
     let users = User::ergo(db).await;
 
     let thirty = sqlx::query_as!(
@@ -116,16 +128,56 @@ async fn index(db: &State<SqlitePool>, user: User, flash: Option<FlashMessage<'_
     .await
     .unwrap();
 
-    let mut context = Context::new();
-    if let Some(msg) = flash {
-        context.insert("flash", &msg.into_inner());
-    }
-    context.insert("loggedin_user", &UserWithDetails::from_user(user, db).await);
     context.insert("users", &users);
     context.insert("thirty", &thirty);
     context.insert("dozen", &dozen);
 
-    Template::render("ergo", context.into_json())
+    Template::render("ergo/index", context.into_json())
+}
+
+#[derive(FromForm, Debug)]
+pub struct UserAdd {
+    birthyear: i32,
+    weight: i64,
+    sex: String,
+}
+
+#[post("/set-data", data = "<data>")]
+async fn new_user(
+    db: &State<SqlitePool>,
+    data: Form<UserAdd>,
+    user: User,
+) -> Flash<Redirect> {
+    if user.has_role(db, "ergo").await {
+        return 
+    Flash::error(Redirect::to("/ergo"), "Du hast deine Daten schon eingegeben. Wenn du sie updaten willst, melde dich bitte bei it@rudernlinz.at");
+    }
+
+    // check data
+    if data.birthyear <1900 || data.birthyear > chrono::Utc::now().year() - 5 {
+        return 
+    Flash::error(Redirect::to("/ergo"), "Bitte überprüfe dein Geburtsjahr...");
+    }
+    if data.weight < 20 || data.weight > 200 {
+        return 
+    Flash::error(Redirect::to("/ergo"), "Bitte überprüfe dein Gewicht...");
+    }
+    if &data.sex != "f" && &data.sex != "m"{
+        return 
+    Flash::error(Redirect::to("/ergo"), "Bitte überprüfe dein Geschlecht...");
+    }
+
+    // set data
+    user.update_ergo(db, data.birthyear, data.weight, &data.sex).await;
+
+    // inform all other `ergo` users
+    let ergo = Role::find_by_name(db, "ergo").await.unwrap();
+    Notification::create_for_role(db, &ergo, &format!("{} nimmt heuer an der Ergochallenge teil 💪", user.name), "Ergo-Challenge", None, None).await;
+    
+    // add to `ergo`  group
+    user.add_role(db,&ergo).await.unwrap();
+    
+    Flash::success(Redirect::to("/ergo"), "Du hast deine Daten erfolgreich eingegeben. Viel Spaß beim Schwitzen :-)")
 }
 
 #[derive(FromForm, Debug)]
@@ -173,6 +225,9 @@ async fn new_thirty(
     )
     .await;
 
+    let ergo = Role::find_by_name(db, "ergo").await.unwrap();
+        Notification::create_for_role(db, &ergo, &format!("{} ist gerade die Dirty Thirty Challenge gefahren 🥵", user.name), "Ergo-Challenge",  Some("/ergo"), None).await;
+
     Flash::success(Redirect::to("/ergo"), "Erfolgreich eingetragen")
 }
 
@@ -214,11 +269,14 @@ async fn new_dozen(
     )
     .await;
 
+    let ergo = Role::find_by_name(db, "ergo").await.unwrap();
+        Notification::create_for_role(db, &ergo, &format!("{} ist gerade die Dirty Dozen Challenge gefahren 🥵", user.name), "Ergo-Challenge",  Some("/ergo"), None).await;
+
     Flash::success(Redirect::to("/ergo"), "Erfolgreich eingetragen")
 }
 
 pub fn routes() -> Vec<Route> {
-    routes![index, new_thirty, new_dozen, send, reset, update]
+    routes![index, new_thirty, new_dozen, send, reset, update, new_user]
 }
 
 #[cfg(test)]