feat: add from redirect for auth (#18)

* feat: add from redirect for auth

* fix: add missing import

components/user-auth-form.tsx

@@ -1,6 +1,7 @@
 "use client"
 
 import * as React from "react"
+import { useSearchParams } from "next/navigation"
 import { signIn } from "next-auth/react"
 import * as z from "zod"
 import { useForm } from "react-hook-form"
@@ -24,14 +25,15 @@ export function UserAuthForm({ className, ...props }: UserAuthFormProps) {
     resolver: zodResolver(userAuthSchema),
   })
   const [isLoading, setIsLoading] = React.useState<boolean>(false)
+  const searchParams = useSearchParams();
 
   async function onSubmit(data: FormData) {
     setIsLoading(true)
 
     const signInResult = await signIn("email", {
       email: data.email.toLowerCase(),
       redirect: false,
-      callbackUrl: `${window.location.origin}/dashboard`,
+      callbackUrl: searchParams.get("from") || "/dashboard",
     })
 
     setIsLoading(false)

middleware.ts

@@ -19,7 +19,14 @@ export default withAuth(
     }
 
     if (!isAuth) {
-      return NextResponse.redirect(new URL("/login", req.url))
+      let from = req.nextUrl.pathname;
+      if (req.nextUrl.search) {
+        from += req.nextUrl.search;
+      }
+
+      return NextResponse.redirect(
+        new URL(`/login?from=${encodeURIComponent(from)}`, req.url)
+      );
     }
   },
   {