Add flash message

Author: Gabatxo1312

Cargo.lock

@@ -16,6 +16,7 @@ askama = "0.14.0"
 # askama_web::WebTemplate implements axum::IntoResponse
 askama_web = { version = "0.14.6", features = ["axum-0.8"] }
 axum = { version = "0.8.4", features = ["macros"] }
+axum-extra = { version = "0.12.1", features = ["cookie"] }
 # UTF-8 paths for easier String/PathBuf interop
 camino = { version = "1.1.12", features = ["serde1"] }
 # Date/time management
@@ -70,4 +71,4 @@ uucore = { version = "0.1.0", features = ["fsext"] }
 xdg = "3.0.0"
 
 [dev-dependencies]
-async-tempfile = "0.7"
+async-tempfile = "0.7"

Cargo.toml

@@ -2,13 +2,15 @@ use askama::Template;
 use askama_web::WebTemplate;
 use axum::Form;
 use axum::extract::{Path, State};
-use axum::response::IntoResponse;
+use axum::response::{IntoResponse, Redirect};
+use axum_extra::extract::CookieJar;
 use serde::{Deserialize, Serialize};
 use snafu::prelude::*;
 
 use crate::database::category::CategoryError;
 use crate::database::{category, category::CategoryOperator};
 use crate::extractors::user::User;
+use crate::state::flash_message::{OperationStatus, get_cookie};
 use crate::state::{AppState, AppStateContext, error::*};
 
 #[derive(Clone, Debug, Deserialize, Serialize)]
@@ -17,13 +19,6 @@ pub struct CategoryForm {
     pub path: String,
 }
 
-pub struct OperationStatus {
-    /// Status of operation
-    pub success: bool,
-    /// Message for confirmation alert
-    pub message: String,
-}
-
 #[derive(Template, WebTemplate)]
 #[template(path = "categories/index.html")]
 pub struct CategoriesTemplate {
@@ -34,7 +29,7 @@ pub struct CategoriesTemplate {
     /// Logged-in user.
     pub user: Option<User>,
     /// Operation status for UI confirmation
-    pub operation_status: Option<OperationStatus>,
+    pub flash: Option<OperationStatus>,
 }
 
 #[derive(Template, WebTemplate)]
@@ -68,37 +63,33 @@ pub async fn delete(
     State(app_state): State<AppState>,
     user: Option<User>,
     Path(id): Path<i32>,
+    jar: CookieJar,
 ) -> Result<impl axum::response::IntoResponse, AppStateError> {
-    let app_state_context = app_state.context().await?;
+    // let app_state_context = app_state.context().await?;
     let categories = CategoryOperator::new(app_state.clone(), user.clone());
 
     let deleted = categories.delete(id, user.clone()).await;
 
-    match deleted {
-        Ok(name) => Ok(CategoriesTemplate {
-            categories: categories.list().await.context(CategorySnafu)?,
-            operation_status: Some(OperationStatus {
-                success: true,
-                message: format!("The category {} has been successfully deleted", name),
-            }),
-            state: app_state_context,
-            user,
-        }),
-        Err(error) => Ok(CategoriesTemplate {
-            categories: categories.list().await.context(CategorySnafu)?,
-            operation_status: Some(OperationStatus {
-                success: false,
-                message: format!("{}", error),
-            }),
-            state: app_state_context,
-            user,
-        }),
-    }
+    let operation_status = match deleted {
+        Ok(name) => OperationStatus {
+            success: true,
+            message: format!("The category {} has been successfully deleted", name),
+        },
+        Err(error) => OperationStatus {
+            success: false,
+            message: format!("{}", error),
+        },
+    };
+
+    let jar = operation_status.set_cookie(jar);
+
+    Ok((jar, Redirect::to("/categories")))
 }
 
 pub async fn create(
     State(app_state): State<AppState>,
     user: Option<User>,
+    jar: CookieJar,
     Form(form): Form<CategoryForm>,
 ) -> Result<impl axum::response::IntoResponse, AppStateError> {
     let app_state_context = app_state.context().await?;
@@ -107,40 +98,49 @@ pub async fn create(
     let created = categories.create(&form, user.clone()).await;
 
     match created {
-        Ok(created) => Ok(CategoriesTemplate {
-            categories: categories.list().await.context(CategorySnafu)?,
-            state: app_state_context,
-            user,
-            operation_status: Some(OperationStatus {
+        Ok(created) => {
+            let operation_status = OperationStatus {
                 success: true,
                 message: format!(
                     "The category {} has been successfully created (ID {})",
                     created.name, created.id
                 ),
-            }),
-        }
-        .into_response()),
-        Err(error) => Ok(NewCategoryTemplate {
-            state: app_state_context,
-            user,
-            category_form: Some(form),
-            error: Some(error),
+            };
+
+            let jar = operation_status.set_cookie(jar);
+
+            Ok((jar, Redirect::to("/categories").into_response()))
         }
-        .into_response()),
+        Err(error) => Ok((
+            jar,
+            NewCategoryTemplate {
+                state: app_state_context,
+                user,
+                category_form: Some(form),
+                error: Some(error),
+            }
+            .into_response(),
+        )),
     }
 }
 
 pub async fn index(
     State(app_state): State<AppState>,
     user: Option<User>,
-) -> Result<CategoriesTemplate, AppStateError> {
+    jar: CookieJar,
+) -> Result<(CookieJar, CategoriesTemplate), AppStateError> {
     let app_state_context = app_state.context().await?;
     let categories = CategoryOperator::new(app_state.clone(), user.clone());
 
-    Ok(CategoriesTemplate {
-        categories: categories.list().await.context(CategorySnafu)?,
-        state: app_state_context,
-        user,
-        operation_status: None,
-    })
+    let (jar, operation_status) = get_cookie(jar);
+
+    Ok((
+        jar,
+        CategoriesTemplate {
+            categories: categories.list().await.context(CategorySnafu)?,
+            state: app_state_context,
+            user,
+            flash: operation_status,
+        },
+    ))
 }

src/routes/category.rs

@@ -0,0 +1,46 @@
+use axum_extra::extract::{CookieJar, cookie::Cookie};
+
+pub struct OperationStatus {
+    /// Status of operation
+    pub success: bool,
+    /// Message for confirmation alert
+    pub message: String,
+}
+
+impl OperationStatus {
+    pub fn set_cookie(&self, jar: CookieJar) -> CookieJar {
+        let mut cookie_operation_status_success =
+            Cookie::new("operation_status_success", self.success.to_string());
+
+        let mut cookie_operation_status_message =
+            Cookie::new("operation_status_message", self.message.clone());
+        cookie_operation_status_success.set_path("/");
+        cookie_operation_status_message.set_path("/");
+
+        jar.add(cookie_operation_status_success)
+            .add(cookie_operation_status_message)
+    }
+}
+
+pub fn get_cookie(jar: CookieJar) -> (CookieJar, Option<OperationStatus>) {
+    let operation_status = match (
+        jar.get("operation_status_success"),
+        jar.get("operation_status_message"),
+    ) {
+        (Some(success), Some(message)) => Some(OperationStatus {
+            success: if let Ok(success) = success.value().parse() {
+                success
+            } else {
+                return (jar, None);
+            },
+            message: message.value().to_string(),
+        }),
+        _ => None,
+    };
+
+    let jar = jar
+        .remove(Cookie::from("operation_status_success"))
+        .remove(Cookie::from("operation_status_message"));
+
+    (jar, operation_status)
+}

src/state/flash_message.rs

@@ -7,6 +7,7 @@ use snafu::prelude::*;
 use crate::config::AppConfig;
 
 pub mod error;
+pub mod flash_message;
 pub mod free_space;
 pub mod logger;
 

src/state/mod.rs

@@ -1,6 +1,6 @@
-{% if let Some(operation_status) = operation_status %}
-  <div class="alert {% if operation_status.success %}alert-success{% else %}alert-danger{% endif %} mt-4">
-    <p class="mb-0">{{ operation_status.message }}</p>
+{% if let Some(flash) = flash %}
+  <div class="alert {% if flash.success %}alert-success{% else %}alert-danger{% endif %} mt-4">
+    <p class="mb-0">{{ flash.message }}</p>
   </div>
 {% endif %}