import { NextResponse } from "next/server"
import { and, desc, eq, sql } from "drizzle-orm"
import { db } from "@/lib/db"
import { rent_payments } from "@/lib/db/schema"
import { getSessionUser } from "@/lib/session"
import { rentPaymentSchema } 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 page = Math.max(1, parseInt(searchParams.get("page") ?? "1", 10))
  const limit = Math.min(100, Math.max(1, parseInt(searchParams.get("limit") ?? "50", 10)))
  const offset = (page - 1) * limit

  const where = and(
    eq(rent_payments.user_id, user.id),
    status ? eq(rent_payments.status, status as typeof rent_payments.$inferSelect.status) : undefined,
    tenantId ? eq(rent_payments.tenant_id, tenantId) : undefined
  )

  const [data, [{ count }]] = await Promise.all([
    db.query.rent_payments.findMany({
      where,
      with: {
        tenant: { columns: { first_name: true, last_name: true, email: true } },
        property: { columns: { name: true } },
        unit: { columns: { unit_number: true } },
      },
      orderBy: desc(rent_payments.due_date),
      limit,
      offset,
    }),
    db.select({ count: sql<number>`count(*)::int` }).from(rent_payments).where(where),
  ])

  return NextResponse.json({ data, total: count ?? 0, page, limit })
}

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 = rentPaymentSchema.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(rent_payments)
    .values({ ...parsed.data, user_id: user.id })
    .returning()

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