import { NextResponse } from "next/server"
import { and, asc, eq } from "drizzle-orm"
import { db } from "@/lib/db"
import { leases } from "@/lib/db/schema"
import { getSessionUser } from "@/lib/session"
import { leaseSchema } from "@/lib/validations"
import { ownsProperty, ownsUnit, ownsTenant } from "@/lib/db/ownership"

export async function GET(request: Request) {
  const user = await getSessionUser()
  if (!user) return NextResponse.json({ error: "Unauthorized" }, { status: 401 })

  const { searchParams } = new URL(request.url)
  const status = searchParams.get("status")
  const tenantId = searchParams.get("tenant_id")

  const data = await db.query.leases.findMany({
    where: and(
      eq(leases.user_id, user.id),
      status ? eq(leases.status, status as typeof leases.$inferSelect.status) : undefined,
      tenantId ? eq(leases.tenant_id, tenantId) : undefined
    ),
    with: {
      tenant: { columns: { first_name: true, last_name: true } },
      property: { columns: { name: true } },
      unit: { columns: { unit_number: true } },
    },
    orderBy: asc(leases.lease_end),
  })

  return NextResponse.json(data)
}

export async function POST(request: Request) {
  const user = await getSessionUser()
  if (!user) return NextResponse.json({ error: "Unauthorized" }, { status: 401 })

  const body = await request.json()
  const parsed = leaseSchema.safeParse(body)
  if (!parsed.success) return NextResponse.json({ error: parsed.error.flatten() }, { status: 400 })

  if (
    !(await ownsProperty(user.id, parsed.data.property_id)) ||
    !(await ownsUnit(user.id, parsed.data.unit_id)) ||
    !(await ownsTenant(user.id, parsed.data.tenant_id))
  ) {
    return NextResponse.json({ error: "Invalid reference" }, { status: 403 })
  }

  const [data] = await db
    .insert(leases)
    .values({ ...parsed.data, user_id: user.id })
    .returning()

  return NextResponse.json(data, { status: 201 })
}