From 306ae13467d848af2a8422c4fa0b662043bfb296 Mon Sep 17 00:00:00 2001
From: philipp <philipp@hofer.link>
Date: Thu, 16 Feb 2023 11:59:01 +0100
Subject: [PATCH] add flash

---
 db.sqlite                | Bin 57344 -> 57344 bytes
 src/rest/mod.rs          |  63 +++++++++++++++++++++++++--------------
 templates/base.html.tera |  19 ++++++++++++
 3 files changed, 59 insertions(+), 23 deletions(-)

diff --git a/db.sqlite b/db.sqlite
index 04ccdc829c4db38d8928187a6db0ce867da5191b..3f773596d694802be950458dbc00cf49cbbb968e 100644
GIT binary patch
delta 92
zcmV-i0HgnazypB51CSd5tdSf;0j#lLpI;LN4FCWD1rIy`53mnl4^<CF4?MFG5GoG}
y2n7ZJ5da7XP-v4zZ@vx%56%D&)eq?p&JXnu;j<A?yblKk0}%mGXa%#?e=A^Z7a1r3

delta 68
zcmV-K0K5NyzypB51CSd5sgWE*0jaTIpI;FK4FCWD1P?|453mnl4^<CFvk?$B50gf3
az6b;l)c_CG59zZGaKI0dPzkfue=A_Rp%qR5

diff --git a/src/rest/mod.rs b/src/rest/mod.rs
index 671be62..770a77c 100644
--- a/src/rest/mod.rs
+++ b/src/rest/mod.rs
@@ -9,10 +9,11 @@ use rocket::{
     form::{self, Form, ValueField},
     fs::FileServer,
     http::{Cookie, CookieJar},
-    response::Redirect,
+    request::FlashMessage,
+    response::{Flash, Redirect},
     Build, Rocket, State,
 };
-use rocket_dyn_templates::{context, Template};
+use rocket_dyn_templates::{tera, Template};
 use sea_orm::{Database, DatabaseConnection};
 use sha3::{Digest, Sha3_256};
 
@@ -37,7 +38,12 @@ impl Deref for NaiveDateForm {
 }
 
 #[get("/?<all>")]
-async fn index(db: &State<DatabaseConnection>, user: user::Model, all: Option<String>) -> Template {
+async fn index(
+    db: &State<DatabaseConnection>,
+    user: user::Model,
+    all: Option<String>,
+    flash: Option<FlashMessage<'_>>,
+) -> Template {
     let mut data = Vec::new();
 
     let mut show_next_n_days = 6;
@@ -56,12 +62,24 @@ async fn index(db: &State<DatabaseConnection>, user: user::Model, all: Option<St
         data.push(DayWithTrips::new(day, db.inner()).await);
     }
 
-    Template::render("index", context! { data, user })
+    let mut context = tera::Context::new();
+
+    if let Some(msg) = flash {
+        context.insert("flash", &msg.into_inner());
+    }
+    context.insert("data", &data);
+    context.insert("user", &user);
+    Template::render("index", &context.into_json())
 }
 
 #[get("/name")]
-fn name() -> Template {
-    Template::render("name", context! {})
+fn name(flash: Option<FlashMessage<'_>>) -> Template {
+    let mut context = tera::Context::new();
+
+    if let Some(msg) = flash {
+        context.insert("flash", &msg.into_inner());
+    }
+    Template::render("name", &context.into_json())
 }
 
 #[derive(FromForm)]
@@ -75,32 +93,31 @@ async fn savename(
     name: Form<NameForm>,
     cookies: &CookieJar<'_>,
     db: &State<DatabaseConnection>,
-) -> Redirect {
+) -> Flash<Redirect> {
     let user = user::Model::find_or_create_user(&name.name, db.inner()).await;
     match user.pw {
-        Some(pw) => {
-            match &name.pw {
-                Some(entered_pw) => {
-                    let mut hasher = Sha3_256::new();
-                    hasher.update(entered_pw);
-                    let entered_pw = hasher.finalize();
+        Some(pw) => match &name.pw {
+            Some(entered_pw) => {
+                let mut hasher = Sha3_256::new();
+                hasher.update(entered_pw);
+                let entered_pw = hasher.finalize();
 
-                    if hex::encode(entered_pw) == pw {
-                        cookies.add_private(Cookie::new("name", name.name.clone()));
-                    } else {
-                        Redirect::to("/"); // Wrong PW
-                    }
-                }
-                None => {
-                    Redirect::to("/"); // No PW supplied
+                if hex::encode(entered_pw) == pw {
+                    cookies.add_private(Cookie::new("name", name.name.clone()));
+                    return Flash::success(Redirect::to("/"), "Erfolgreich eingeloggt");
+                } else {
+                    return Flash::error(Redirect::to("/name"), "Falsches Passwort");
                 }
             }
-        }
+            None => {
+                return Flash::error(Redirect::to("/name"), "Benutzer besitzt hat Passwort, du hast jedoch keines eingegeben. Bitte nochmal probieren");
+            }
+        },
         None => {
             cookies.add_private(Cookie::new("name", name.name.clone()));
+            return Flash::success(Redirect::to("/"), "Name erfolgreich ausgewählt");
         }
     }
-    Redirect::to("/")
 }
 
 #[get("/logout")]
diff --git a/templates/base.html.tera b/templates/base.html.tera
index bedfac8..4a4a080 100644
--- a/templates/base.html.tera
+++ b/templates/base.html.tera
@@ -43,6 +43,25 @@
 	{% endif %}
 	<a href="/logout">LOGOUT</a>
     {% endif %}
+
+    {% if flash %}
+      {% if flash.0 == "success" %}
+        <div class="row">
+          <div class="one-column" style="padding: 15px; background-color: #2b8c68; margin-bottom: 1.6em;">
+            {{ flash.1 }}
+          </div>
+        </div>
+      {% endif %}
+      {% if flash.0 == "error" %}
+        <div class="row">
+          <div class="one-column" style="padding: 15px; background-color: #ff6961; margin-bottom: 1.6em;">
+            {{ flash.1 }}
+          </div>
+        </div>
+      {% endif %}
+    {% endif %}
+
+
     <div class="row">
       <div class="column">
 	{% block content %}