property-management-network/app/api/units/[id]/route.ts

import { NextResponse } from "next/server"
import { and, eq } from "drizzle-orm"
import { db } from "@/lib/db"
import { units } from "@/lib/db/schema"
import { getSessionUser } from "@/lib/session"
import { unitSchema } from "@/lib/validations"
import { ownsProperty } from "@/lib/db/ownership"
import { getAccountContext } from "@/lib/account"

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 ctx = await getAccountContext(user.id)
  const ownerId = ctx.ownerId
  if (!ctx.canWrite) return NextResponse.json({ error: "Forbidden" }, { status: 403 })

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

  if (!(await ownsProperty(ownerId, parsed.data.property_id))) {
    return NextResponse.json({ error: "Invalid reference" }, { status: 403 })
  }

  try {
    const [data] = await db
      .update(units)
      .set(parsed.data)
      .where(and(eq(units.id, id), eq(units.user_id, ownerId)))
      .returning()

    return NextResponse.json(data)
  } catch (e) {
    return NextResponse.json({ error: (e as Error).message }, { status: 500 })
  }
}

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

  const ctx = await getAccountContext(user.id)
  const ownerId = ctx.ownerId
  if (!ctx.canWrite) return NextResponse.json({ error: "Forbidden" }, { status: 403 })

  const { id } = await params
  try {
    await db.delete(units).where(and(eq(units.id, id), eq(units.user_id, ownerId)))
    return NextResponse.json({ success: true })
  } catch (e) {
    return NextResponse.json({ error: (e as Error).message }, { status: 500 })
  }
}