web permission system updated

web/src/pages/admin/permissions/logic.rs

@@ -36,15 +36,17 @@ pub(crate) async fn add_role(
     headers: HeaderMap,
     Form(form): Form<AddRoleForm>,
 ) -> Response {
+    let role = form.name.trim().to_string();
+    let destination = format!("/admin/permissions?role={role}&updated=true");
     let request_headers = headers.clone();
-    mutate(&headers, async move {
+    mutate(&headers, &destination, async move {
         let mut auth = state.auth;
         auth.add_role(authenticated_request(&request_headers, AddRoleRequest {
-            name: form.name.trim().to_string(),
+            name: role,
             parent: form.parent.trim().to_string(),
         }).map_err(|_| "Missing session".to_string())?)
             .await.map_err(|error| error.message().to_string())?;
-        Ok("Role created. Sign in again to continue.")
+        Ok(())
     }).await
 }
 
@@ -53,14 +55,15 @@ pub(crate) async fn remove_role(
     headers: HeaderMap,
     Form(form): Form<RoleForm>,
 ) -> Response {
+    let destination = "/admin/permissions?updated=true";
     let request_headers = headers.clone();
-    mutate(&headers, async move {
+    mutate(&headers, destination, async move {
         let mut auth = state.auth;
         auth.remove_role(authenticated_request(&request_headers, RemoveRoleRequest {
             name: form.role,
         }).map_err(|_| "Missing session".to_string())?)
             .await.map_err(|error| error.message().to_string())?;
-        Ok("Role removed. Sign in again to continue.")
+        Ok(())
     }).await
 }
 
@@ -69,8 +72,9 @@ pub(crate) async fn grant(
     headers: HeaderMap,
     Form(form): Form<PermissionForm>,
 ) -> Response {
+    let destination = format!("/admin/permissions?role={}&updated=true", form.role);
     let request_headers = headers.clone();
-    mutate(&headers, async move {
+    mutate(&headers, &destination, async move {
         let mut auth = state.auth;
         auth.grant_permission(authenticated_request(&request_headers, GrantPermissionRequest {
             role: form.role,
@@ -78,7 +82,7 @@ pub(crate) async fn grant(
             action: form.action,
         }).map_err(|_| "Missing session".to_string())?)
             .await.map_err(|error| error.message().to_string())?;
-        Ok("Permission granted. Sign in again to continue.")
+        Ok(())
     }).await
 }
 
@@ -87,8 +91,9 @@ pub(crate) async fn revoke(
     headers: HeaderMap,
     Form(form): Form<PermissionForm>,
 ) -> Response {
+    let destination = format!("/admin/permissions?role={}&updated=true", form.role);
     let request_headers = headers.clone();
-    mutate(&headers, async move {
+    mutate(&headers, &destination, async move {
         let mut auth = state.auth;
         auth.revoke_permission(authenticated_request(&request_headers, RevokePermissionRequest {
             role: form.role,
@@ -96,7 +101,7 @@ pub(crate) async fn revoke(
             action: form.action,
         }).map_err(|_| "Missing session".to_string())?)
             .await.map_err(|error| error.message().to_string())?;
-        Ok("Permission revoked. Sign in again to continue.")
+        Ok(())
     }).await
 }
 
@@ -105,27 +110,28 @@ pub(crate) async fn assign_user_role(
     headers: HeaderMap,
     Form(form): Form<AssignRoleForm>,
 ) -> Response {
+    let destination = "/admin/permissions?updated=true";
     let request_headers = headers.clone();
-    mutate(&headers, async move {
+    mutate(&headers, destination, async move {
         let mut auth = state.auth;
         auth.assign_user_role(authenticated_request(&request_headers, AssignUserRoleRequest {
             username: form.username,
             role: form.role,
         }).map_err(|_| "Missing session".to_string())?)
             .await.map_err(|error| error.message().to_string())?;
-        Ok("User role changed. Sign in again to continue.")
+        Ok(())
     }).await
 }
 
-async fn mutate<F>(headers: &HeaderMap, operation: F) -> Response
+async fn mutate<F>(headers: &HeaderMap, destination: &str, operation: F) -> Response
 where
-    F: std::future::Future<Output = Result<&'static str, String>>,
+    F: std::future::Future<Output = Result<(), String>>,
 {
     if let Some(rejection) = reject_cross_site(headers) {
         return rejection;
     }
     match operation.await {
-        Ok(_) => stale_session_response(),
+        Ok(()) => success_redirect(destination),
         Err(message) => (
             StatusCode::UNPROCESSABLE_ENTITY,
             Html(ui::render_mutation_error(&message)),
@@ -133,19 +139,18 @@ where
     }
 }
 
-fn stale_session_response() -> Response {
+fn success_redirect(destination: &str) -> Response {
     let mut response = StatusCode::SEE_OTHER.into_response();
-    response.headers_mut().insert(
-        header::SET_COOKIE,
-        HeaderValue::from_static("analytics_token=; Path=/; HttpOnly; SameSite=Strict; Max-Age=0"),
-    );
+    let Ok(destination) = HeaderValue::try_from(destination) else {
+        return (StatusCode::INTERNAL_SERVER_ERROR, "Invalid redirect").into_response();
+    };
     response.headers_mut().insert(
         header::LOCATION,
-        HeaderValue::from_static("/login?permissions_changed=1"),
+        destination.clone(),
     );
     response.headers_mut().insert(
         "hx-redirect",
-        HeaderValue::from_static("/login?permissions_changed=1"),
+        destination,
     );
     response
 }