import { redirect } from "next/navigation"
import { and, eq } from "drizzle-orm"
import { db } from "@/lib/db"
import { properties, units } from "@/lib/db/schema"
import { getSessionUser } from "@/lib/session"
import { getEffectiveOwnerId } from "@/lib/account"
import { BackButton } from "@/components/ui/back-button"
import { UnitForm } from "@/components/forms/unit-form"

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

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

  const ownerId = await getEffectiveOwnerId(user.id)

  const { propertyId, unitId } = await params

  const [property, unit] = await Promise.all([
    db.query.properties.findFirst({
      where: and(eq(properties.id, propertyId), eq(properties.user_id, ownerId)),
      columns: { id: true, name: true },
    }),
    db.query.units.findFirst({
      where: and(eq(units.id, unitId), eq(units.property_id, propertyId), eq(units.user_id, ownerId)),
    }),
  ])

  if (!property || !unit) redirect(`/properties/${propertyId}`)

  return (
    <div className="mx-auto max-w-2xl space-y-6">
      <BackButton href={`/properties/${propertyId}`} />
      <div>
        <h2 className="text-lg font-semibold text-white">Edit Unit {unit.unit_number}</h2>
        <p className="text-sm text-white/40">{property.name}</p>
      </div>
      <div className="rounded-xl border border-white/[0.06] bg-[#16161f] p-6">
        <UnitForm propertyId={propertyId} unit={unit} />
      </div>
    </div>
  )
}
