Added auto get-user with jwt

2025-05-19 13:11:02 +01:00 · 2025-05-19 13:11:02 +01:00 · 31a0c622d5
commit 31a0c622d5
parent 158dbbf166
2 changed files with 76 additions and 10 deletions
--- a/src/app/api/get-user/route.ts
+++ b/src/app/api/get-user/route.ts
@ -0,0 +1,50 @@
+import { NextResponse } from "next/server";
+import { cookies } from "next/headers";
+import { env } from "@utils/env";
+import { PrismaClient } from "@prisma/client";
+import { verifyJwt } from "@utils/verifyJwt";
+
+const prisma = new PrismaClient();
+
+export async function POST(req: Request) {
+	let cookieStore;
+	try {
+		cookieStore = await cookies();
+		const token = cookieStore.get("jwt")?.value;
+
+		if (!token) {
+			return NextResponse.json({ error: "No JWT found" }, { status: 401 });
+		}
+
+		const payload = await verifyJwt({ token, secret: env.JWT_SECRET_KEY });
+
+		const user = await prisma.user.findUnique({
+			where: { id: payload.userId },
+			include: {
+				scientist: {
+					include: {
+						earthquakes: true,
+						observatories: true,
+						artefacts: true,
+						superior: true,
+						subordinates: true,
+					},
+				},
+				purchasedArtefacts: true,
+			},
+		});
+
+		if (user) {
+			return NextResponse.json({ message: "Got user successfully", user }, { status: 200 });
+		} else {
+			cookieStore.delete("jwt"); // Delete JWT cookie if user not found            }
+			return NextResponse.json({ message: "Failed to get user" }, { status: 401 });
+		}
+	} catch (error) {
+		console.error("Error in user endpoint:", error);
+		cookieStore?.delete("jwt"); // Delete JWT cookie on error
+		return NextResponse.json({ message: "Internal Server Error" }, { status: 500 });
+	} finally {
+		await prisma.$disconnect();
+	}
+}
--- a/src/app/layout.tsx
+++ b/src/app/layout.tsx
@ -1,8 +1,11 @@
 "use client";
+import axios from "axios";
+import { useEffect } from "react";
 import type { Metadata } from "next";
 import "./globals.css";

 import { action, createStore, StoreProvider } from "easy-peasy";
+import { useStoreActions } from "@hooks/store";
 import { Inter } from "next/font/google";

 import { StoreModel } from "@appTypes/StoreModel";
@ -24,21 +27,33 @@ const store = createStore<StoreModel>({
 		tickers: { GBP: "£", USD: "$", EUR: "€" },
 	},
 	user: null,
-	// user: {
-	// 	id: 123456,
-	// 	createdAt: new Date(8.64e15),
-	// 	email: "tim.howitz@dyson.com",
-	// 	passwordHash: "",
-	// 	name: "Tim Howitz",
-	// 	role: "ADMIN",
-	// 	scientist: undefined,
-	// 	purchasedArtefacts: [],
-	// },
 	setUser: action((state, payload) => {
 		state.user = payload;
 	}),
 });

+function UserFetcher() {
+	const setUser = useStoreActions((actions) => actions.setUser);
+
+	useEffect(() => {
+		async function fetchUser() {
+			try {
+				const response = await axios.post("/api/get-user");
+				if (response.status === 200 && response.data.user) {
+					setUser(response.data.user);
+				} else {
+					setUser(null); // Clear user if no user found
+				}
+			} catch (error) {
+				console.error("Error fetching user:", error);
+				setUser(null);
+			}
+		}
+		fetchUser();
+	}, [setUser]); // Run once on mount
+
+	return null; // No UI needed
+}
 export default function RootLayout({
 	children,
 }: Readonly<{
@ -48,6 +63,7 @@ export default function RootLayout({
 		<html lang="en">
 			<StoreProvider store={store}>
 				<body className={`${inter.variable} h-[calc(100vh-3.5rem)] flex flex-col min-h-screen antialiased`}>
+					<UserFetcher></UserFetcher>
 					<Navbar></Navbar>
 					<div className="flex-1 overflow-y-auto">{children}</div>
 				</body>