property-management-network/app/(dashboard)/layout.tsx

import { redirect } from "next/navigation"
import { eq } from "drizzle-orm"
import { db } from "@/lib/db"
import { profiles } from "@/lib/db/schema"
import { getSession, isAdminUser } from "@/lib/session"
import { getMaintenanceMode } from "@/lib/settings"
import { MaintenanceScreen } from "@/components/shared/maintenance-screen"
import { Sidebar } from "@/components/dashboard/sidebar"
import { Header } from "@/components/dashboard/header"
import { CommandPalette } from "@/components/dashboard/command-palette"
import { PageTransition } from "@/components/dashboard/page-transition"
import { Breadcrumbs } from "@/components/dashboard/breadcrumbs"
import { ScrollToTop } from "@/components/ui/scroll-to-top"
import { ImpersonationBanner } from "@/components/admin/impersonation-banner"

export default async function DashboardLayout({ children }: { children: React.ReactNode }) {
  const session = await getSession()
  const user = session?.user

  if (!user) {
    redirect("/login")
  }

  // Site maintenance mode: everyone except admins sees the maintenance screen.
  const maintenance = await getMaintenanceMode()
  if (maintenance.enabled && !isAdminUser(user)) {
    return <MaintenanceScreen message={maintenance.message} />
  }

  const profile = await db.query.profiles.findFirst({
    where: eq(profiles.id, user.id),
  })

  // Set by the Better Auth admin plugin while an admin is impersonating.
  const impersonating = Boolean((session?.session as { impersonatedBy?: string })?.impersonatedBy)

  return (
    <div className="flex h-screen flex-col bg-[#09090b] overflow-hidden">
      {impersonating && <ImpersonationBanner label={profile?.email ?? user.email} />}
      <div className="flex flex-1 overflow-hidden">
        <Sidebar profile={profile ?? null} isAdmin={isAdminUser(user)} />
        <div className="flex flex-1 flex-col overflow-hidden">
          <Header />
          <main id="main-scroll" className="flex-1 overflow-y-auto p-4 sm:p-6">
            <Breadcrumbs />
            <PageTransition>
              {children}
            </PageTransition>
          </main>
        </div>
        <CommandPalette />
        <ScrollToTop />
      </div>
    </div>
  )
}