import { buildBurialSiteName } from '../helpers/burialSites.helpers.js'

import addOrUpdateBurialSiteField from './addOrUpdateBurialSiteField.js'
import deleteBurialSiteField from './deleteBurialSiteField.js'
import getCemetery from './getCemetery.js'
import { acquireConnection } from './pool.js'

export interface UpdateBurialSiteForm {
  burialSiteId: string | number

  burialSiteNameSegment1?: string
  burialSiteNameSegment2?: string
  burialSiteNameSegment3?: string
  burialSiteNameSegment4?: string
  burialSiteNameSegment5?: string

  burialSiteTypeId: string | number
  burialSiteStatusId: string | number

  cemeteryId: string | number
  cemeterySvgId: string

  burialSiteLatitude: string
  burialSiteLongitude: string

  burialSiteTypeFieldIds?: string
  [fieldValue_burialSiteTypeFieldId: string]: unknown
}

/**
 * Updates a burial site.
 * @param updateForm - The burial site's updated information
 * @param user - The user making the request
 * @returns True if the burial site was updated.
 * @throws If an active burial site with the same name already exists.
 */
export default async function updateBurialSite(
  updateForm: UpdateBurialSiteForm,
  user: User
): Promise<boolean> {
  const database = await acquireConnection()

  const cemetery =
    updateForm.cemeteryId === ''
      ? undefined
      : await getCemetery(updateForm.cemeteryId, database)

  const burialSiteName = buildBurialSiteName(cemetery?.cemeteryKey, updateForm)

  // Ensure no active burial sites share the same name

  const existingBurialSite = database
    .prepare(
      `select burialSiteId
        from BurialSites
        where burialSiteName = ?
        and burialSiteId <> ?
        and recordDelete_timeMillis is null`
    )
    .pluck()
    .get(burialSiteName, updateForm.burialSiteId) as number | undefined

  if (existingBurialSite !== undefined) {
    database.release()
    throw new Error('An active burial site with that name already exists.')
  }

  const result = database
    .prepare(
      `update BurialSites
        set burialSiteNameSegment1 = ?,
        burialSiteNameSegment2 = ?,
        burialSiteNameSegment3 = ?,
        burialSiteNameSegment4 = ?,
        burialSiteNameSegment5 = ?,
        burialSiteName = ?,
        burialSiteTypeId = ?,
        burialSiteStatusId = ?,
        cemeteryId = ?,
        cemeterySvgId = ?,
        burialSiteLatitude = ?,
        burialSiteLongitude = ?,
        recordUpdate_userName = ?,
        recordUpdate_timeMillis = ?
        where burialSiteId = ?
        and recordDelete_timeMillis is null`
    )
    .run(
      updateForm.burialSiteNameSegment1 ?? '',
      updateForm.burialSiteNameSegment2 ?? '',
      updateForm.burialSiteNameSegment3 ?? '',
      updateForm.burialSiteNameSegment4 ?? '',
      updateForm.burialSiteNameSegment5 ?? '',
      burialSiteName,
      updateForm.burialSiteTypeId,
      updateForm.burialSiteStatusId === ''
        ? undefined
        : updateForm.burialSiteStatusId,
      updateForm.cemeteryId === '' ? undefined : updateForm.cemeteryId,
      updateForm.cemeterySvgId,
      updateForm.burialSiteLatitude === ''
        ? undefined
        : updateForm.burialSiteLatitude,
      updateForm.burialSiteLongitude === ''
        ? undefined
        : updateForm.burialSiteLongitude,
      user.userName,
      Date.now(),
      updateForm.burialSiteId
    )

  if (result.changes > 0) {
    const burialSiteTypeFieldIds = (
      updateForm.burialSiteTypeFieldIds ?? ''
    ).split(',')

    for (const burialSiteTypeFieldId of burialSiteTypeFieldIds) {
      const fieldValue = updateForm[`fieldValue_${burialSiteTypeFieldId}`] as
        | string
        | undefined

      await ((fieldValue ?? '') === ''
        ? deleteBurialSiteField(
            updateForm.burialSiteId,
            burialSiteTypeFieldId,
            user,
            database
          )
        : addOrUpdateBurialSiteField(
            {
              burialSiteId: updateForm.burialSiteId,
              burialSiteTypeFieldId,
              fieldValue: fieldValue ?? ''
            },
            user,
            database
          ))
    }
  }

  database.release()

  return result.changes > 0
}

export async function updateBurialSiteStatus(
  burialSiteId: number | string,
  burialSiteStatusId: number | string,
  user: User
): Promise<boolean> {
  const database = await acquireConnection()

  const rightNowMillis = Date.now()

  const result = database
    .prepare(
      `update BurialSites
        set burialSiteStatusId = ?,
        recordUpdate_userName = ?,
        recordUpdate_timeMillis = ?
        where burialSiteId = ?
        and recordDelete_timeMillis is null`
    )
    .run(
      burialSiteStatusId === '' ? undefined : burialSiteStatusId,
      user.userName,
      rightNowMillis,
      burialSiteId
    )

  database.release()

  return result.changes > 0
}