All files / src/queues/jobs autoImport.ts

79.62% Statements 43/54
73.33% Branches 11/15
18.18% Functions 2/11
95.55% Lines 43/45

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111                  9x                 7x   7x   7x   7x     10x 10x 10x   10x         10x 10x 10x 10x       6x 1x 1x     5x 5x       5x 5x         5x 1x   1x 1x   4x     5x 5x 5x   5x 5x 5x 5x     5x   5x   5x 5x 5x 5x 5x 5x 5x               5x   5x                        
import { AboutService, AccountsService, TransactionRead, TransactionTypeProperty } from "@billos/firefly-iii-sdk"
import pino from "pino"
 
import { client } from "../../client"
import { env } from "../../config"
import DynamicConfig, { VConfig } from "../../modules/config/dynamic"
import { TemplateName } from "../../utils/renderTemplate"
import { SimpleJob } from "./BaseJob"
 
const logger = pino()
 
interface AccountTransactions {
  expenses: TransactionRead[]
  deposits: TransactionRead[]
  transfers: TransactionRead[]
}
 
export class AutoImportJob extends SimpleJob {
  readonly id = "auto-import"
 
  override readonly retryable = false // auto-import is triggered externally and should not be retried on failure
 
  override readonly uniqueNotificationKey = "sparkleft:notification:autoimport:id"
 
  override readonly cronConfigKey = VConfig.AutoImportCron
 
  private async getExpensesAndIncome(currentAccountId: string): Promise<AccountTransactions> {
    const now = new Date()
    const [start] = new Date(now.getFullYear(), now.getMonth(), now.getDate() - 7).toISOString().split("T")
    const [end] = now.toISOString().split("T")
 
    const { data: transactions } = await AccountsService.listTransactionByAccount({
      client,
      path: { id: currentAccountId },
      query: { start, end },
    })
    const expenses = transactions.filter(({ attributes: { transactions } }) => transactions[0].type === TransactionTypeProperty.WITHDRAWAL)
    const deposits = transactions.filter(({ attributes: { transactions } }) => transactions[0].type === TransactionTypeProperty.DEPOSIT)
    const transfers = transactions.filter(({ attributes: { transactions } }) => transactions[0].type === TransactionTypeProperty.TRANSFER)
    return { expenses, deposits, transfers }
  }
 
  async run(): Promise<void> {
    if (!env.importerUrl || !env.importDirectory || !env.autoImportSecret) {
      logger.warn("Missing required configuration for auto-import job (importerUrl, importDirectory, autoImportSecret), skipping")
      return
    }
 
    const currentAccountId = await DynamicConfig.get(VConfig.CurrentAccountId)
    Iif (!currentAccountId) {
      throw new Error("Current account ID is not set in the configuration")
    }
 
    const previousTransactions = await this.getExpensesAndIncome(currentAccountId)
    logger.info(
      "Found %d transactions in the last 7 days for asset account",
      previousTransactions.expenses.length + previousTransactions.deposits.length + previousTransactions.transfers.length,
    )
 
    if (env.fireflyCliToken) {
      logger.info("Triggering Firefly III cron job before auto-import")
      // await AboutService.getCron(env.fireflyCliToken)
      await AboutService.getCron({ client, path: { cliToken: env.fireflyCliToken } })
      logger.info("Firefly III cron job triggered successfully")
    } else {
      logger.warn("FIREFLY_III_CLI_TOKEN is not set, skipping Firefly III cron job trigger")
    }
 
    const params = new URLSearchParams()
    params.set("directory", env.importDirectory)
    params.set("secret", env.autoImportSecret)
 
    const url = `${env.importerUrl}/autoimport?${params.toString()}`
    logger.info("Triggering auto-import at %s/autoimport with directory: %s", env.importerUrl, env.importDirectory)
    const result = await fetch(url, { method: "POST" })
    Iif (!result.ok) {
      throw new Error(`Auto-import request failed with status ${result.status}: ${await result.text()}`)
    }
    logger.info("Auto-import triggered successfully")
 
    const newTransactions = await this.getExpensesAndIncome(currentAccountId)
 
    const expenses = newTransactions.expenses.filter((newTx) => !previousTransactions.expenses.some((prevTx) => prevTx.id === newTx.id))
    const deposits = newTransactions.deposits.filter((newTx) => !previousTransactions.deposits.some((prevTx) => prevTx.id === newTx.id))
    const transfers = newTransactions.transfers.filter((newTx) => !previousTransactions.transfers.some((prevTx) => prevTx.id === newTx.id))
    const diffExpenses = newTransactions.expenses.length - previousTransactions.expenses.length
    const diffDeposits = newTransactions.deposits.length - previousTransactions.deposits.length
    const diffTransfers = newTransactions.transfers.length - previousTransactions.transfers.length
    logger.info(
      "Found %d new transactions after auto-import (expenses: %d, income: %d, transfers: %d)",
      diffExpenses + diffDeposits + diffTransfers,
      diffExpenses,
      diffDeposits,
      diffTransfers,
    )
 
    const assetAccount = await AccountsService.getAccount({ client, path: { id: currentAccountId } })
 
    await this.sendUniqueNotification(TemplateName.AutoImport, {
      diffExpenses,
      diffDeposits,
      diffTransfers,
      expenses,
      deposits,
      transfers,
      accountBalance: assetAccount.data.attributes.current_balance || "0",
      accountCurrency: assetAccount.data.attributes.currency_symbol || "€",
    })
  }
}