import sqlite from 'better-sqlite3'

import { lotOccupancyDB as databasePath } from '../../data/databasePaths.js'

import { dateIntegerToString } from '@cityssm/expressjs-server-js/dateTimeFns.js'

import { getLots } from './getLots.js'

import { getLotOccupancies } from './getLotOccupancies.js'

import { getWorkOrderComments } from './getWorkOrderComments.js'

import { getWorkOrderMilestones } from './getWorkOrderMilestones.js'

import type * as recordTypes from '../../types/recordTypes'

interface WorkOrderOptions {
  includeLotsAndLotOccupancies: boolean
  includeComments: boolean
  includeMilestones: boolean
}

const baseSQL = `select w.workOrderId,
    w.workOrderTypeId, t.workOrderType,
    w.workOrderNumber, w.workOrderDescription,
    w.workOrderOpenDate, userFn_dateIntegerToString(w.workOrderOpenDate) as workOrderOpenDateString,
    w.workOrderCloseDate, userFn_dateIntegerToString(w.workOrderCloseDate) as workOrderCloseDateString,
    w.recordCreate_timeMillis, w.recordUpdate_timeMillis
    from WorkOrders w
    left join WorkOrderTypes t on w.workOrderTypeId = t.workOrderTypeId
    where w.recordDelete_timeMillis is null`

function _getWorkOrder(
  sql: string,
  workOrderIdOrWorkOrderNumber: number | string,
  options: WorkOrderOptions,
  connectedDatabase?: sqlite.Database
): recordTypes.WorkOrder {
  const database =
    connectedDatabase ??
    sqlite(databasePath, {
      readonly: true
    })

  database.function('userFn_dateIntegerToString', dateIntegerToString)

  const workOrder: recordTypes.WorkOrder = database
    .prepare(sql)
    .get(workOrderIdOrWorkOrderNumber)

  if (workOrder) {
    if (options.includeLotsAndLotOccupancies) {
      workOrder.workOrderLots = getLots(
        {
          workOrderId: workOrder.workOrderId
        },
        {
          limit: -1,
          offset: 0
        },
        database
      ).lots

      workOrder.workOrderLotOccupancies = getLotOccupancies(
        {
          workOrderId: workOrder.workOrderId
        },
        {
          limit: -1,
          offset: 0,
          includeOccupants: true
        },
        database
      ).lotOccupancies
    }

    if (options.includeComments) {
      workOrder.workOrderComments = getWorkOrderComments(
        workOrder.workOrderId as number,
        database
      )
    }

    if (options.includeMilestones) {
      workOrder.workOrderMilestones = getWorkOrderMilestones(
        {
          workOrderId: workOrder.workOrderId
        },
        {
          includeWorkOrders: false,
          orderBy: 'completion'
        },
        database
      )
    }
  }

  if (connectedDatabase === undefined) {
    database.close()
  }

  return workOrder
}

export function getWorkOrderByWorkOrderNumber(
  workOrderNumber: string
): recordTypes.WorkOrder {
  return _getWorkOrder(
    baseSQL + ' and w.workOrderNumber = ?',
    workOrderNumber,
    {
      includeLotsAndLotOccupancies: true,
      includeComments: true,
      includeMilestones: true
    }
  )
}

export function getWorkOrder(
  workOrderId: number | string,
  options: WorkOrderOptions,
  connectedDatabase?: sqlite.Database
): recordTypes.WorkOrder {
  return _getWorkOrder(
    baseSQL + ' and w.workOrderId = ?',
    workOrderId,
    options,
    connectedDatabase
  )
}

export default getWorkOrder