import type { Request, Response } from 'express'

import addWorkOrderBurialSite from '../../database/addWorkOrderBurialSite.js'
import getBurialSites from '../../database/getBurialSites.js'

export default async function handler(
  request: Request<
    unknown,
    unknown,
    { workOrderId: string; burialSiteId: string }
  >,
  response: Response
): Promise<void> {
  const success = await addWorkOrderBurialSite(
    {
      workOrderId: request.body.workOrderId,
      burialSiteId: request.body.burialSiteId
    },
    request.session.user as User
  )

  const workOrderLotsResults = await getBurialSites(
    {
      workOrderId: request.body.workOrderId
    },
    {
      limit: -1,
      offset: 0,
      includeBurialSiteContractCount: false
    }
  )

  response.json({
    success,
    workOrderBurialSites: workOrderLotsResults.burialSites
  })
}