fix: Add token expiration check and remove unused utility

Author: tusharshah21

frontend/src/hooks/profiles/use-create-profile.ts

@@ -10,7 +10,6 @@ import type {
 } from "@/lib/types/api";
 import { API_BASE_URL } from "@/lib/constants/apiConstants";
 import { getToken } from "@/lib/utils/jwt";
-import { useAuthHeader } from "@/hooks/use-auth-header";
 
 async function postCreateProfile(
   input: CreateProfileInput,

frontend/src/hooks/use-auth-header.ts

@@ -1,6 +1,6 @@
 import { useCallback, useState } from "react";
 import { useAccount, useSignMessage } from "wagmi";
-import { getToken, saveToken, clearToken } from "@/lib/utils/jwt";
+import { getToken, saveToken, clearToken, isTokenValid } from "@/lib/utils/jwt";
 import { useGetNonce } from "./profiles/use-get-nonce";
 import { generateSiweMessage } from "@/lib/utils/siwe";
 import { API_BASE_URL } from "@/lib/constants/apiConstants";
@@ -16,13 +16,14 @@ export function useAuthHeader() {
     const [isRefreshing, setIsRefreshing] = useState(false);
 
     const getAuthHeaders = useCallback((): AuthHeaders => {
-        const token = getToken();
-
-        // If JWT token exists, use it
-        if (token) {
-            return {
-                Authorization: `Bearer ${token}`,
-            };
+        // Check if token is valid and not expired
+        if (isTokenValid()) {
+            const token = getToken();
+            if (token) {
+                return {
+                    Authorization: `Bearer ${token}`,
+                };
+            }
         }
 
         // Fall back to empty headers (caller will add SIWE headers)
@@ -74,6 +75,6 @@ export function useAuthHeader() {
         getAuthHeaders,
         refreshToken,
         isRefreshing,
-        hasToken: () => getToken() !== null,
+        hasToken: isTokenValid,
     };
 }

frontend/src/lib/utils/fetch-with-jwt.ts

@@ -1,17 +1,24 @@
 const JWT_TOKEN_KEY = "jwt_token";
 const JWT_ADDRESS_KEY = "jwt_address";
+const JWT_EXPIRY_KEY = "jwt_expiry";
 
 export interface JwtToken {
     token: string;
     address: string;
 }
 
-export function saveToken(token: JwtToken): void {
+export function saveToken(token: JwtToken, expiresInSeconds = 86400): void {
+    const expiryTime = Date.now() + expiresInSeconds * 1000;
     localStorage.setItem(JWT_TOKEN_KEY, token.token);
     localStorage.setItem(JWT_ADDRESS_KEY, token.address);
+    localStorage.setItem(JWT_EXPIRY_KEY, expiryTime.toString());
 }
 
 export function getToken(): string | null {
+    if (!isTokenValid()) {
+        clearToken();
+        return null;
+    }
     return localStorage.getItem(JWT_TOKEN_KEY);
 }
 
@@ -22,8 +29,24 @@ export function getAddress(): string | null {
 export function clearToken(): void {
     localStorage.removeItem(JWT_TOKEN_KEY);
     localStorage.removeItem(JWT_ADDRESS_KEY);
+    localStorage.removeItem(JWT_EXPIRY_KEY);
 }
 
 export function isTokenValid(): boolean {
-    return getToken() !== null;
-}
+    const token = localStorage.getItem(JWT_TOKEN_KEY);
+    const expiryStr = localStorage.getItem(JWT_EXPIRY_KEY);
+
+    if (!token || !expiryStr) {
+        return false;
+    }
+
+    const expiryTime = parseInt(expiryStr, 10);
+    const isExpired = Date.now() >= expiryTime;
+
+    if (isExpired) {
+        clearToken();
+        return false;
+    }
+
+    return true;
+}