import { notFound, redirect } from "next/navigation"
import { and, asc, eq } from "drizzle-orm"
import { db } from "@/lib/db"
import { tenants as tenantsTable, properties, leases as leasesTable } from "@/lib/db/schema"
import { getSessionUser } from "@/lib/session"
import { getEffectiveOwnerId } from "@/lib/account"
import { LeaseForm } from "@/components/forms/lease-form"
import { BackButton } from "@/components/ui/back-button"

export const metadata = { title: "Edit Lease" }

export default async function EditLeasePage({ params }: { params: Promise<{ leaseId: string }> }) {
  const user = await getSessionUser()
  if (!user) redirect("/login")

  const ownerId = await getEffectiveOwnerId(user.id)

  const { leaseId } = await params

  const [lease, tenants, properties_] = await Promise.all([
    db.query.leases.findFirst({
      where: and(eq(leasesTable.id, leaseId), eq(leasesTable.user_id, ownerId)),
      with: {
        tenant: { columns: { id: true, first_name: true, last_name: true } },
      },
    }),
    db
      .select({
        id: tenantsTable.id,
        first_name: tenantsTable.first_name,
        last_name: tenantsTable.last_name,
        unit_id: tenantsTable.unit_id,
        property_id: tenantsTable.property_id,
      })
      .from(tenantsTable)
      .where(and(eq(tenantsTable.user_id, ownerId), eq(tenantsTable.status, "active")))
      .orderBy(asc(tenantsTable.first_name)),
    db.query.properties.findMany({
      where: eq(properties.user_id, ownerId),
      columns: { id: true, name: true },
      with: {
        units: { columns: { id: true, unit_number: true } },
      },
      orderBy: asc(properties.name),
    }),
  ])

  if (!lease) notFound()

  // Ensure the lease's own tenant is selectable even if now inactive.
  const tenantList = tenants.some((t) => t.id === lease.tenant_id)
    ? tenants
    : [
        {
          id: lease.tenant_id,
          first_name: lease.tenant?.first_name ?? "",
          last_name: lease.tenant?.last_name ?? "",
          unit_id: lease.unit_id,
          property_id: lease.property_id,
        },
        ...tenants,
      ]

  return (
    <div className="mx-auto max-w-2xl space-y-6">
      <BackButton href={`/leases/${leaseId}`} />
      <div>
        <h2 className="text-lg font-semibold text-white">Edit Lease</h2>
        <p className="text-sm text-white/40">
          {lease.tenant?.first_name} {lease.tenant?.last_name}
        </p>
      </div>
      <LeaseForm tenants={tenantList} properties={properties_ ?? []} lease={lease} />
    </div>
  )
}
