From b43e8c4a61a485f558afe9c099a9d44eb7144b58 Mon Sep 17 00:00:00 2001
From: Zack Fu Zi Xiang <speed2exe@live.com.sg>
Date: Thu, 21 Mar 2024 15:25:40 +0800
Subject: [PATCH] feat: add fragment redirect for web home

---
 .../templates/components/sidebar.html         | 21 +++++++++++
 admin_frontend/templates/oc                   |  0
 admin_frontend/templates/pages/login.html     | 36 ++++++++++++++-----
 src/biz/workspace/ops.rs                      |  2 +-
 4 files changed, 50 insertions(+), 9 deletions(-)
 create mode 100644 admin_frontend/templates/oc

diff --git a/admin_frontend/templates/components/sidebar.html b/admin_frontend/templates/components/sidebar.html
index 2141bf71..2500d469 100644
--- a/admin_frontend/templates/components/sidebar.html
+++ b/admin_frontend/templates/components/sidebar.html
@@ -3,6 +3,7 @@
     class="sidebar-item"
     hx-target="#sidebar-content"
     hx-get="/web/components/user/navigate"
+    data-section="navigate"
   >
     Navigate
   </div>
@@ -10,6 +11,7 @@
     class="sidebar-item"
     hx-target="#sidebar-content"
     hx-get="/web/components/user/change-password"
+    data-section="change-password"
   >
     Change Password
   </div>
@@ -17,6 +19,7 @@
     class="sidebar-item"
     hx-target="#sidebar-content"
     hx-get="/web/components/user/invite"
+    data-section="invite"
   >
     Invite
   </div>
@@ -24,6 +27,7 @@
     class="sidebar-item"
     hx-target="#sidebar-content"
     hx-get="/web/components/user/user-usage"
+    data-section="user-usage"
   >
     User Usage
   </div>
@@ -31,7 +35,24 @@
     class="sidebar-item"
     hx-target="#sidebar-content"
     hx-get="/web/components/user/workspace-usage"
+    data-section="workspace-usage"
   >
     Workspace Usage
   </div>
 </div>
+
+<script>
+  document.addEventListener('DOMContentLoaded', (event) => {
+    const frag = window.location.href.split('#');
+    if (frag.length > 1) {
+      const section = frag[1];
+      const sidebarItems = document.querySelectorAll('.sidebar-item');
+
+      sidebarItems.forEach(item => {
+        if (item.getAttribute('data-section') === section) {
+          item.click();
+        }
+      });
+    }
+  });
+</script>
diff --git a/admin_frontend/templates/oc b/admin_frontend/templates/oc
new file mode 100644
index 00000000..e69de29b
diff --git a/admin_frontend/templates/pages/login.html b/admin_frontend/templates/pages/login.html
index b52e011d..ad33401a 100644
--- a/admin_frontend/templates/pages/login.html
+++ b/admin_frontend/templates/pages/login.html
@@ -109,24 +109,44 @@
   <script>
     // OAuthLogin (This is done because axum doesn't support fragment in the URL)
     // https://github.com/tokio-rs/axum/discussions/2147
+
+    // Parse the fragment and extract the parameters
+    const paramMap = {};
     if (window.location.hash) {
-      // Extract data from the URL fragment
-      const fragmentData = window.location.hash.substring(1);   // Remove the leading #
-      const fragmentParams = new URLSearchParams(fragmentData); // Parse the fragment data as a URLSearchParams object
-      const refreshToken = fragmentParams.get("refresh_token"); // Extract the refresh_token
+      const fragments = window.location.href.split('#').slice(1);
+      for (let i = 0; i < fragments.length; i++) {
+        const params = fragments[i].split('&');
+        for (let j = 0; j < params.length; j++) {
+          const keyValue = params[j].split('=');
+          if (keyValue.length === 2) {
+            const key = keyValue[0];
+            const value = keyValue[1];
+            paramMap[key] = value;
+          }
+        }
+      }
+
+      // Clear the fragment from the URL
+      history.replaceState("", document.title, window.location.pathname + window.location.search);
+
+      // Login in via refresh_token
+      const refreshToken = paramMap['refresh_token'];
       fetch(`/web-api/login-refresh/${refreshToken}`, {
-        // Login in via refresh_token
         method: "POST",
       }).then((response) => {
         if (!response.ok) {
           displayHttpStatusAndPayload(response);
         } else {
-          window.location.href = "/web/home";
+          const redirect_to = paramMap['redirect_to'];
+          if (redirect_to && redirect_to.trim() !== "") {
+            console.log("redirect_to is not empty");
+            window.location.href = "/web/home#" + redirect_to;
+          } else {
+            window.location.href = "/web/home";
+          }
         }
       });
 
-      // Remove # from url, prevent access_token from being reused event though refresh token is already invalid
-      history.replaceState("", document.title, window.location.pathname + window.location.search);
     }
   </script>
   {% endblock %}
diff --git a/src/biz/workspace/ops.rs b/src/biz/workspace/ops.rs
index cf83346c..92dcd4d0 100644
--- a/src/biz/workspace/ops.rs
+++ b/src/biz/workspace/ops.rs
@@ -181,7 +181,7 @@ pub async fn invite_workspace_members(
           email: invitation.email.clone(),
           ..Default::default()
         },
-        Some("/web/home#invite".to_owned()),
+        Some("/web/home#redirect_to=invite".to_owned()),
       )
       .await?;