refactor : Conditionally return login route for other users and admin

ad956 · Jul 27, 2024 · 1d20dc1 · 1d20dc1
1 parent ee5aa0e
commit 1d20dc1
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/middlewares/handlePrivateRoute.ts b/middlewares/handlePrivateRoute.ts
@@ -5,13 +5,16 @@ export default async function handlePrivateRoute(
   request: NextRequest,
   token: string | undefined
 ) {
+  const currentRole = request.nextUrl.pathname.split("/")[1];
+
   if (!token) {
-    return NextResponse.redirect(new URL("/login", request.url));
+    let loginUrl = currentRole === "admin" ? "/admin-login" : "/login";
+
+    return NextResponse.redirect(new URL(loginUrl, request.url));
   }
 
   const decryptedToken = await decrypt(token);
   const userRole = decryptedToken.user.role;
-  const currentRole = request.nextUrl.pathname.split("/")[1];
 
   if (currentRole !== userRole) {
     return NextResponse.redirect(new URL(`/${userRole}`, request.url));