import type { PoolConnection } from 'better-sqlite-pool'

import {
  calculateFeeAmount,
  calculateTaxAmount
} from '../helpers/functions.fee.js'
import type { Contract, Fee } from '../types/recordTypes.js'

import getContract from './getContract.js'
import getFee from './getFee.js'
import { acquireConnection } from './pool.js'

export interface AddContractFeeForm {
  contractId: number | string
  feeId: number | string
  quantity: number | string
  feeAmount?: number | string
  taxAmount?: number | string
}

export default async function addContractFee(
  addFeeForm: AddContractFeeForm,
  user: User,
  connectedDatabase?: PoolConnection
): Promise<boolean> {
  const database = connectedDatabase ?? (await acquireConnection())

  const rightNowMillis = Date.now()

  // Calculate fee and tax (if not set)
  let feeAmount: number
  let taxAmount: number

  if ((addFeeForm.feeAmount ?? '') === '') {
    const contract = (await getContract(
      addFeeForm.contractId
    )) as Contract

    const fee = (await getFee(addFeeForm.feeId)) as Fee

    feeAmount = calculateFeeAmount(fee, contract)
    taxAmount = calculateTaxAmount(fee, feeAmount)
  } else {
    feeAmount =
      typeof addFeeForm.feeAmount === 'string'
        ? Number.parseFloat(addFeeForm.feeAmount)
        : 0
    taxAmount =
      typeof addFeeForm.taxAmount === 'string'
        ? Number.parseFloat(addFeeForm.taxAmount)
        : 0
  }

  try {
    // Check if record already exists
    const record = database
      .prepare(
        `select feeAmount, taxAmount, recordDelete_timeMillis
          from ContractFees
          where contractId = ?
          and feeId = ?`
      )
      .get(addFeeForm.contractId, addFeeForm.feeId) as
      | {
          feeAmount: number | null
          taxAmount: number | null
          recordDelete_timeMillis: number | null
        }
      | undefined

    if (record !== undefined) {
      if (record.recordDelete_timeMillis !== null) {
        database
          .prepare(
            `delete from ContractFees
              where recordDelete_timeMillis is not null
              and contractId = ?
              and feeId = ?`
          )
          .run(addFeeForm.contractId, addFeeForm.feeId)
      } else if (
        record.feeAmount === feeAmount &&
        record.taxAmount === taxAmount
      ) {
        database
          .prepare(
            `update ContractFees
              set quantity = quantity + ?,
              recordUpdate_userName = ?,
              recordUpdate_timeMillis = ?
              where contractId = ?
              and feeId = ?`
          )
          .run(
            addFeeForm.quantity,
            user.userName,
            rightNowMillis,
            addFeeForm.contractId,
            addFeeForm.feeId
          )

        return true
      } else {
        const quantity =
          typeof addFeeForm.quantity === 'string'
            ? Number.parseFloat(addFeeForm.quantity)
            : addFeeForm.quantity

        database
          .prepare(
            `update ContractFees
              set feeAmount = (feeAmount * quantity) + ?,
              taxAmount = (taxAmount * quantity) + ?,
              quantity = 1,
              recordUpdate_userName = ?,
              recordUpdate_timeMillis = ?
              where contractId = ?
              and feeId = ?`
          )
          .run(
            feeAmount * quantity,
            taxAmount * quantity,
            user.userName,
            rightNowMillis,
            addFeeForm.contractId,
            addFeeForm.feeId
          )

        return true
      }
    }

    // Create new record
    const result = database
      .prepare(
        `insert into ContractFees (
          contractId, feeId,
          quantity, feeAmount, taxAmount,
          recordCreate_userName, recordCreate_timeMillis,
          recordUpdate_userName, recordUpdate_timeMillis)
          values (?, ?, ?, ?, ?, ?, ?, ?, ?)`
      )
      .run(
        addFeeForm.contractId,
        addFeeForm.feeId,
        addFeeForm.quantity,
        feeAmount,
        taxAmount,
        user.userName,
        rightNowMillis,
        user.userName,
        rightNowMillis
      )

    return result.changes > 0
  } finally {
    if (connectedDatabase === undefined) {
      database.release()
    }
  }
}