update how to get user

app/api/auth/route.ts

@@ -1,5 +1,4 @@
 import { NextRequest, NextResponse } from "next/server";
-import MY_TOKEN_KEY from "@/lib/get-cookie-name";
 
 export async function POST(req: NextRequest) {
   const body = await req.json();
@@ -71,10 +70,7 @@ export async function POST(req: NextRequest) {
   }
   const user = await userResponse.json();
 
-  const cookieName = MY_TOKEN_KEY();
-  const isProduction = process.env.NODE_ENV === "production";
-  
-  const nextResponse = NextResponse.json(
+  return NextResponse.json(
     {
       access_token: response.access_token,
       expires_in: response.expires_in,
@@ -87,17 +83,4 @@ export async function POST(req: NextRequest) {
       },
     }
   );
-  
-  // Set HTTP-only cookie
-  const cookieOptions = [
-    `${cookieName}=${response.access_token}`,
-    `Max-Age=${response.expires_in || 3600}`, // Default 1 hour if not provided
-    "Path=/deepsite",
-    "HttpOnly",
-    ...(isProduction ? ["Secure", "SameSite=None"] : ["SameSite=Lax"])
-  ].join("; ");
-  
-  nextResponse.headers.set("Set-Cookie", cookieOptions);
-  
-  return nextResponse;
 }

hooks/useUser.ts

@@ -17,6 +17,7 @@ export const useUser = (initialData?: {
   const client = useQueryClient();
   const router = useRouter();
   const [currentRoute, setCurrentRoute, removeCurrentRoute] = useCookie("deepsite-currentRoute");
+  const [token, setToken, removeToken] = useCookie("deepsite-auth-token");
 
   const { data: { user, errCode } = { user: null, errCode: null }, isLoading } =
     useQuery({
@@ -68,11 +69,31 @@ export const useUser = (initialData?: {
     await api
       .post("/auth", { code })
       .then(async (res: any) => {
-        if (res.data) {
-          client.setQueryData(["user.me"], {
-            user: res.data.user,
-            errCode: null,
+        if (res.data && res.data.access_token) {
+          // Set cookie using useCookie hook
+          const expiresIn = res.data.expires_in || 3600;
+          const expiresDate = new Date();
+          expiresDate.setTime(expiresDate.getTime() + expiresIn * 1000);
+          
+          setToken(res.data.access_token, {
+            expires: expiresDate,
+            path: '/deepsite',
+            sameSite: 'lax',
+            secure: window.location.protocol === 'https:',
           });
+          
+          // Refetch user data from /api/me
+          const meResponse = await api.get("/me");
+          if (meResponse.data) {
+            client.setQueryData(["user.me"], {
+              user: meResponse.data.user,
+              errCode: null,
+            });
+            if (meResponse.data.projects) {
+              setProjects(meResponse.data.projects);
+            }
+          }
+          
           // if (currentRoute) {
           //   router.push(currentRoute);
           //   removeCurrentRoute();
@@ -93,12 +114,14 @@ export const useUser = (initialData?: {
   const logout = async () => {
     try {
       await api.post("/auth/logout");
+      removeToken();
       removeCurrentRoute();
       client.clear();
       toast.success("Logout successful");
       window.location.reload();
     } catch (error) {
       console.error("Logout error:", error);
+      removeToken();
       removeCurrentRoute();
       client.clear()
       toast.success("Logout successful");