import addContractFee from './addContractFee.js'
import { getFeeCategory } from './getFeeCategories.js'
import { acquireConnection } from './pool.js'

export interface AddContractCategoryForm {
  contractId: number | string
  feeCategoryId: number | string
}

export default async function addContractFeeCategory(
  addFeeCategoryForm: AddContractCategoryForm,
  user: User
): Promise<number> {
  const database = await acquireConnection()

  const feeCategory = await getFeeCategory(
    addFeeCategoryForm.feeCategoryId,
    database
  )

  let addedFeeCount = 0

  for (const fee of feeCategory?.fees ?? []) {
    const success = await addContractFee(
      {
        contractId: addFeeCategoryForm.contractId,
        feeId: fee.feeId,
        quantity: 1
      },
      user,
      database
    )

    if (success) {
      addedFeeCount += 1
    }
  }

  database.release()

  return addedFeeCount
}