minor improvements

2 files changed, 26 insertions, 18 deletions

diff --git a/src/lib/contexts/Auth.context.tsx b/src/lib/contexts/Auth.context.tsx
index 29e2005..fbf2e24 100644
--- a/src/lib/contexts/Auth.context.tsx
+++ b/src/lib/contexts/Auth.context.tsx
@@ -1,39 +1,49 @@
 'use client';
 import React, { createContext, useState, useContext, useEffect } from 'react';
 
-type User = {
+export type User = {
     id: number;
     username: string;
+    avatar?: string;
+    banner_file?: string;
+    premium?: boolean;
+    is_banned?: boolean;
+    is_moderator?: boolean;
 };
-
 interface AuthContextType {
     user: User | null;
     setUser: (user: User | null) => void;
     logout: () => void;
 }
-
 const AuthContext = createContext<AuthContextType | null>(null);
+const API_URL = process.env.NEXT_PUBLIC_API_URL;
 
 export const AuthContextProvider = ({ children }: React.PropsWithChildren) => {
     const [user, setUser] = useState<User | null>(null);
 
     useEffect(() => {
-        const token = localStorage.getItem('token');
-        if (!token) return;
+        const fetchUser = async () => {
+            const hasCookies = document.cookie.includes('access_token');
+            if (!hasCookies) return;
+
+            try {
+                const res = await fetch(`${API_URL}/api/me`, {
+                    credentials: 'include',
+                });
+                if (!res.ok) return;
+                const userData = await res.json();
+                setUser(userData);
+            } catch {}
+        };
 
-        fetch('http://localhost:8000/api/me', {
-            headers: { Authorization: `Bearer ${token}` },
-        })
-            .then((res) => {
-                if (!res.ok) throw new Error('Not authenticated');
-                return res.json();
-            })
-            .then((userData) => setUser(userData))
-            .catch(() => setUser(null));
+        fetchUser();
     }, []);
 
-    const logout = () => {
-        localStorage.removeItem('token');
+    const logout = async () => {
+        await fetch(`${API_URL}/api/auth/logout`, {
+            method: 'POST',
+            credentials: 'include',
+        });
         setUser(null);
     };
 
diff --git a/src/lib/contexts/Global.context.tsx b/src/lib/contexts/Global.context.tsx
index c903456..3f8b61b 100644
--- a/src/lib/contexts/Global.context.tsx
+++ b/src/lib/contexts/Global.context.tsx
@@ -1,5 +1,3 @@
-import react from 'react';
-import { authcontextprovider } from './auth.context';
 import React from 'react';
 import { AuthContextProvider } from './Auth.context';