import { NextResponse } from "next/server"
import { and, eq } from "drizzle-orm"
import { db } from "@/lib/db"
import { maintenance_requests } from "@/lib/db/schema"
import { getSessionUser } from "@/lib/session"
import { maintenanceSchema } from "@/lib/validations"
import { ownsProperty, ownsUnit, ownsTenant } from "@/lib/db/ownership"
import { z } from "zod"

const maintenancePatchSchema = maintenanceSchema.partial().extend({
  status: z.enum(["open", "in_progress", "resolved", "closed"]).optional(),
  resolution_notes: z.string().optional(),
  actual_cost: z.number().positive().optional(),
  resolved_at: z.string().optional(),
})

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

  const { id } = await params
  const data = await db.query.maintenance_requests.findFirst({
    where: and(eq(maintenance_requests.id, id), eq(maintenance_requests.user_id, user.id)),
    with: {
      property: { columns: { name: true, address_line1: true, city: true } },
      unit: { columns: { unit_number: true } },
      tenant: { columns: { first_name: true, last_name: true, email: true, phone: true } },
    },
  })

  if (!data) return NextResponse.json({ error: "Not found" }, { status: 404 })
  return NextResponse.json(data)
}

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 = maintenancePatchSchema.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 updateData: Record<string, unknown> = { ...parsed.data }

  // Auto-set resolved_at when status → resolved
  if (parsed.data.status === "resolved" && !parsed.data.resolved_at) {
    updateData.resolved_at = new Date().toISOString()
  }

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

  if (!data) return NextResponse.json({ error: "Not found" }, { status: 404 })
  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(maintenance_requests)
    .where(and(eq(maintenance_requests.id, id), eq(maintenance_requests.user_id, user.id)))

  return NextResponse.json({ success: true })
}