middleware.ts

import { auth } from "@/lib/auth";
import { routes } from "@/lib/routes";
import { NextResponse } from "next/server";

const PREFIX_ROLE = [
  { prefix: "/owner", role: "owner" },
  { prefix: "/agent", role: "agent" },
  { prefix: "/customer", role: "customer" },
] as const;

function requiredRole(pathname: string) {
  return PREFIX_ROLE.find(
    (r) => pathname === r.prefix || pathname.startsWith(`${r.prefix}/`)
  )?.role;
}

export default auth((req) => {
  const role = requiredRole(req.nextUrl.pathname);
  if (!role) return NextResponse.next();

  if (!req.auth?.user) {
    const login = new URL(routes.login, req.nextUrl.origin);
    login.searchParams.set("callbackUrl", req.nextUrl.pathname + req.nextUrl.search);
    return NextResponse.redirect(login);
  }

  if (req.auth.user.role !== role) {
    return NextResponse.redirect(new URL(routes.home, req.nextUrl.origin));
  }

  return NextResponse.next();
});

export const config = {
  matcher: ["/owner/:path*", "/agent/:path*", "/customer/:path*"],
};