import { NextResponse } from "next/server"
import { and, eq } 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"
import { logActivity } from "@/lib/activity"

export async function PATCH(request: Request, { params }: { params: Promise<{ id: string }> }) {
  const user = await getSessionUser()
  if (!user) return NextResponse.json({ error: "Unauthorized" }, { status: 401 })

  const { id } = await params
  const body = await request.json()
  const parsed = rentPaymentSchema.partial().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 })
  }

  // Auto-set paid_date when status → paid
  const updateData = { ...parsed.data }
  if (parsed.data.status === "paid" && !parsed.data.paid_date) {
    updateData.paid_date = new Date().toISOString().slice(0, 10)
  }

  const [data] = await db
    .update(rent_payments)
    .set(updateData)
    .where(and(eq(rent_payments.id, id), eq(rent_payments.user_id, user.id)))
    .returning()

  if (parsed.data.status === "paid") {
    await logActivity({
      userId: user.id,
      type: "rent_paid",
      title: "Rent payment marked as paid",
      entityType: "rent_payment",
      entityId: id,
    })
  }

  return NextResponse.json(data)
}

export async function DELETE(_: Request, { params }: { params: Promise<{ id: string }> }) {
  const user = await getSessionUser()
  if (!user) return NextResponse.json({ error: "Unauthorized" }, { status: 401 })

  const { id } = await params
  await db.delete(rent_payments).where(and(eq(rent_payments.id, id), eq(rent_payments.user_id, user.id)))
  return NextResponse.json({ success: true })
}