1 files changed, 13 insertions, 11 deletions
diff --git a/code/api/wwwroot/scripts/components/profile-modal.js b/code/api/wwwroot/scripts/components/profile-modal.js
index 91d4ed3..f4e90d5 100644
--- a/code/api/wwwroot/scripts/components/profile-modal.js
+++ b/code/api/wwwroot/scripts/components/profile-modal.js
@@ -1,17 +1,19 @@
 class ProfileModal extends HTMLElement {
     constructor() {
         super();
-        const sessionData = session.get();
-        const root = create_element("div", {
-            style: {
-                padding: "5px",
-                display: ""
-            }
-        }, [
-            create_element("h4", {innerText: sessionData.username, style: {margin:0}}),
-            create_element("p", {innerText: sessionData.role})
-        ]);
-        this.innerHTML = root.innerHTML;
+        retry(session.get, res => (res?.username?.length > 0 ?? false), 0).then(sessionData => {
+            const root = create_element("div", {
+                style: {
+                    padding: "5px",
+                    display: ""
+                }
+            }, [
+                create_element("h4", {innerText: sessionData.username, style: {margin: 0}}),
+                create_element("p", {innerText: sessionData.role}),
+                create_element("button", {innerText: "Log out", classList: ["do-logout"]})
+            ]);
+            this.innerHTML = root.innerHTML;
+        });
     }
 }