From c5ba268e0aa9bef54b0f73e42689a9e2fbd81724 Mon Sep 17 00:00:00 2001
From: marzban-dev <marzban.dev@gmail.com>
Date: Thu, 16 Jan 2025 16:02:54 +0330
Subject: [PATCH] Implement token validation cycle

---
 frontend/layouts/Default.vue | 67 ++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)

diff --git a/frontend/layouts/Default.vue b/frontend/layouts/Default.vue
index cb94ffa..3d5c39c 100644
--- a/frontend/layouts/Default.vue
+++ b/frontend/layouts/Default.vue
@@ -1,3 +1,70 @@
+<script lang="ts" setup>
+
+// import
+
+import { useAuth } from "~/composables/api/auth/useAuth";
+import useRefreshAuth from "~/composables/api/auth/useRefreshAuth";
+import useVerify from "~/composables/api/auth/useVerify";
+
+// state
+
+const { mutateAsync: refreshAuth } = useRefreshAuth();
+const { token, refreshToken, updateToken, updateRefreshToken, logout } = useAuth();
+const { mutateAsync: verify } = useVerify();
+
+// lifecycle
+
+onServerPrefetch(async () => {
+    if (!!token.value) {
+
+        // 1.1 - token is there
+
+        try {
+
+            await verify({
+                token: token.value
+            });
+
+            // 2.1 - token is valid, finish
+
+        } catch (e) {
+
+            // 2.2 - token is there, but not valid, try to refresh token
+
+            if (!!refreshToken.value) {
+
+                // 3.1 - refresh token is there, try to refresh
+
+                try {
+                    const refreshResponse = await refreshAuth({ refresh: refreshToken.value });
+
+                    // 4.1 - token is refreshed successfully, finish
+
+                    updateToken(refreshResponse.access);
+                    updateRefreshToken(refreshResponse.refresh);
+                } catch (e) {
+
+                    // 4.2 - cant refreshing token, logout
+
+                    logout();
+                }
+            } else {
+
+                // 3.2 - refresh token is not exist, logout
+
+                logout();
+            }
+        }
+    } else {
+
+        // 1.2 - token is not exist, logout
+
+        logout();
+    }
+});
+
+</script>
+
 <template>
     <div
         class="w-full flex flex-col-center persian-number font-iran-yekan-x"