invoices.class.php

<?php
/* Copyright (C) 2026 Laurent Destailleur   <eldy@users.sourceforge.net>
 * Copyright (C) 2026 Nick Fragoulis
 * Copyright (C) 2026 Jose Martinez     <jose.martinez@pichinov.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY, without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <https://www.gnu.org/licenses/>.
 */
 
require_once DOL_DOCUMENT_ROOT . '/compta/facture/class/facture.class.php';
require_once DOL_DOCUMENT_ROOT . '/societe/class/societe.class.php';
require_once DOL_DOCUMENT_ROOT . '/product/class/product.class.php';
require_once DOL_DOCUMENT_ROOT . '/compta/paiement/class/paiement.class.php';
require_once DOL_DOCUMENT_ROOT . '/compta/bank/class/account.class.php';
require_once DOL_DOCUMENT_ROOT . '/core/lib/date.lib.php';
 
class ToolInvoices extends McpTool
{
 
  public function __construct(DoliDB $db, $user = null, $conf = null)
  {
    $this->db = $db;
    $this->user = $user;
    if ($conf !== null) {
      $this->conf = $conf;
    }
  }
 
  private function getUser()
  {
    // is_object() instead of empty() to avoid PHPStan flagging empty() on a
    // non-nullable typed parent property as unreachable.
    if (is_object($this->user) && !empty($this->user->id)) {
      return $this->user;
    }
    global $user;
    return is_object($user) && !empty($user->id) ? $user : null;
  }
 
  public function getDefinitions(): array
  {
    return [
      [
        "name" => "search_invoice",
        "description" => "Search for invoices. By default, lists UNPAID invoices. Excludes drafts.",
        "inputSchema" => [
          "type" => "object",
          "properties" => [
            "customer" => [
              "type" => "string",
              "description" => "Optional: Customer name."
            ],
            "status" => [
              "type" => "string",
              "enum" => ["unpaid", "paid", "draft", "all"],
              "description" => "Filter by status. Default is 'unpaid'. 'all' shows history but excludes drafts.",
              "default" => "unpaid"
            ],
            "limit" => [
              "type" => "integer",
              "default" => 10
            ],
            "ids" => [
              "type" => "array",
              "items" => ["type" => "integer"],
              "description" => "Restrict to these invoice ids (e.g. the rows the user selected on a list). The result then includes count and total sums."
            ]
          ]
        ]
      ],
      [
        "name" => "get_invoice",
        "description" => "Get details of a specific invoice by ID or Reference.",
        "inputSchema" => [
          "type" => "object",
          "properties" => [
            "ref" => ["type" => "string", "description" => "Invoice Ref (e.g. FA2401-001)"],
            "id" => ["type" => "integer", "description" => "Invoice ID"]
          ],
          "oneOf" => [
            ["required" => ["ref"]],
            ["required" => ["id"]]
          ]
        ]
      ],
      [
        "name" => "search_supplier_invoice",
        "description" => "Search SUPPLIER (vendor/purchase) invoices by supplier name, ref, or payment status. Use for supplier invoices only; for customer invoices use search_invoice.",
        "inputSchema" => [
          "type" => "object",
          "properties" => [
            "query" => ["type" => "string", "description" => "Supplier name, invoice ref or supplier's own ref (optional)"],
            "status" => ["type" => "string", "enum" => ["draft", "unpaid", "paid"], "description" => "Filter by status (optional)"],
            "limit" => ["type" => "integer", "default" => 10],
            "ids" => [
              "type" => "array",
              "items" => ["type" => "integer"],
              "description" => "Restrict to these supplier invoice ids (e.g. the rows the user selected on a list). The result then includes count and total sums."
            ]
          ]
        ]
      ],
      [
        "name" => "get_supplier_invoice",
        "description" => "Get details of a specific SUPPLIER (vendor/purchase) invoice by ID or Reference. For customer invoices use get_invoice.",
        "inputSchema" => [
          "type" => "object",
          "properties" => [
            "ref" => ["type" => "string", "description" => "Supplier invoice Ref"],
            "id" => ["type" => "integer", "description" => "Supplier invoice ID"]
          ],
          "oneOf" => [
            ["required" => ["ref"]],
            ["required" => ["id"]]
          ]
        ]
      ],
      [
        "name" => "validate_invoice",
        "description" => "Validate a draft invoice.",
        "inputSchema" => [
          "type" => "object",
          "properties" => [
            "invoice" => ["type" => "string", "description" => "Invoice ID or Ref."]
          ],
          "required" => ["invoice"]
        ]
      ],
      [
        "name" => "pay_invoice",
        "description" => "Register a payment for an invoice.",
        "inputSchema" => [
          "type" => "object",
          "properties" => [
            "invoice" => ["type" => "string", "description" => "Invoice ID or Reference."],
            "amount" => ["type" => "number", "description" => "Amount to pay. Defaults to full remaining."],
            "payment_mode" => ["type" => "string", "description" => "Code (VIR, CB, LIQ)."],
            "bank_account" => ["type" => "string", "description" => "Bank Account Name/Ref."]
          ],
          "required" => ["invoice"]
        ]
      ]
    ];
  }
 
  public function getRequiredRights(string $toolName)
  {
    $map = array(
      'search_invoice' => array(array('facture', 'lire')),
      'search_invoices' => array(array('facture', 'lire')),
      'get_invoice' => array(array('facture', 'lire')),
      'validate_invoice' => array(array('facture', 'creer')),
      'pay_invoice' => array(array('facture', 'paiement')),
      'search_supplier_invoice' => array(array('fournisseur', 'facture', 'lire')),
      'get_supplier_invoice' => array(array('fournisseur', 'facture', 'lire'))
    );
 
    return isset($map[$toolName]) ? $map[$toolName] : self::RIGHTS_UNDECLARED;
  }
 
  public function writeConfirmationPreview(string $toolName, array $args)
  {
    global $langs;
 
    $langs->load("other");
 
    switch ($toolName) {
      case 'validate_invoice':
        return $langs->trans("AIPreviewValidateInvoice", (string) ($args['ref'] ?? $args['id'] ?? ''));
      case 'pay_invoice':
        return $langs->trans("AIPreviewPayInvoice", price((float) ($args['amount'] ?? 0)), (string) ($args['ref'] ?? $args['id'] ?? ''));
      default:
        return McpTool::NO_WRITE;
    }
  }
 
  public function getCategories(): array
  {
    return ['billing'];
  }
 
  public function execute(string $name, array $args)
  {
    switch ($name) {
      case 'search_invoice':
      case 'search_invoices':
        return $this->searchInvoices($args);
 
      case 'search_supplier_invoice':
        return $this->searchSupplierInvoices($args);
      case 'get_supplier_invoice':
        return $this->getSupplierInvoice($args);
      case 'get_invoice':
        return $this->getInvoice($args);
 
      case 'validate_invoice':
        return $this->validateInvoice($args);
 
      case 'pay_invoice':
        return $this->payInvoice($args);
 
      default:
        return ["error" => "Tool function '$name' not found."];
    }
  }
 
  private function searchInvoices($args)
  {
    $limit = isset($args['limit']) ? (int) $args['limit'] : 10;
    $status = isset($args['status']) ? $args['status'] : 'unpaid';
 
    // Safety fallback
    if ($limit <= 0) {
      $limit = 5;
    }
    if ($limit > 1000) {
      dol_syslog("Search DB Error: Too many record requested", LOG_ERR);
      return ["error" => "DB Error"];
    }
 
    $sql = "SELECT f.rowid, f.ref, f.total_ht, f.total_ttc, f.fk_statut, f.paye, f.datef, s.nom
        FROM " . MAIN_DB_PREFIX . "facture as f
        LEFT JOIN " . MAIN_DB_PREFIX . "societe as s ON f.fk_soc = s.rowid
        WHERE f.entity IN (" . getEntity('facture') . ")";
 
    // Selection mode: restrict to explicit ids (e.g. the user's checked
    // rows) and let the server do the arithmetic the single-shot LLM
    // architecture cannot: the model never sees tool results, so sums
    // must be computed here. Status filtering is bypassed - a selection
    // means these exact records, drafts included.
    $onlyIds = array();
    if (!empty($args['ids']) && is_array($args['ids'])) {
      foreach ($args['ids'] as $selId) {
        if ((int) $selId > 0 && count($onlyIds) < 100) {
          $onlyIds[] = (int) $selId;
        }
      }
      if (!empty($onlyIds)) {
        $sql .= " AND f.rowid IN (".$this->db->sanitize(implode(',', $onlyIds)).")";
        $status = 'ids';
      }
    }
 
    // Status filtering (bypassed entirely in selection mode: the ids ARE the filter)
    if ($status === 'ids') {
      // no status clause
    } elseif ($status === 'draft') {
      // Explicitly asking for drafts
      $sql .= " AND f.fk_statut = 0";
    } elseif ($status === 'paid') {
      // Fully paid
      $sql .= " AND f.fk_statut = 2";
    } elseif ($status === 'all') {
      // Valid invoices (Unpaid + Paid). EXCLUDES Drafts (0) and Abandoned (3)
      $sql .= " AND f.fk_statut IN (1, 2)";
    } else {
      // Default: 'unpaid'
      // In Dolibarr: fk_statut=1 means Validated but not fully paid.
      $sql .= " AND f.fk_statut = 1 AND f.paye = 0";
    }
 
    // Customer Filter
    if (!empty($args['customer'])) {
      $cust = $this->findCustomer($args['customer']);
      if (!is_array($cust)) {
        $sql .= " AND f.fk_soc = " . ((int) $cust->id);
      } else {
        $sql .= " AND s.nom LIKE '%" . $this->db->escape($args['customer']) . "%'";
      }
    }
 
    $sql .= " ORDER BY f.datef DESC LIMIT " . ((int) $limit);
 
    $resql = $this->db->query($sql);
    $list = [];
    $sumHt = 0.0;
    $sumTtc = 0.0;
 
    if ($resql) {
      while ($r = $this->db->fetch_object($resql)) {
        // Double check to ensure no PROV/Drafts slip through unless asked
        if (!in_array($status, array('draft', 'ids')) && $r->fk_statut == 0) {
          continue;
        }
 
        $ref = ($r->fk_statut == 0 ? "(PROV" . $r->rowid . ")" : $r->ref);
 
        // Calculate Status Label
        $statusLabel = "Unknown";
        if ($r->fk_statut == 0) {
          $statusLabel = "Draft";
        } elseif ($r->fk_statut == 1) {
          $statusLabel = "Unpaid";
        } elseif ($r->fk_statut == 2) {
          $statusLabel = "Paid";
        } elseif ($r->fk_statut == 3) {
          $statusLabel = "Abandoned";
        }
 
        $sumHt += (float) $r->total_ht;
        $sumTtc += (float) $r->total_ttc;
        $list[] = [
          "ref" => $ref,
          "date" => dol_print_date($this->db->jdate($r->datef), 'day'),
          "customer" => $r->nom,
          "amount" => price($r->total_ttc),
          "status" => $statusLabel,
          "url" => DOL_URL_ROOT . "/compta/facture/card.php?id=" . $r->rowid
        ];
      }
      $this->db->free($resql);
    }
 
    if (empty($list)) {
      return ["info" => "No " . $status . " invoices found matching your criteria."];
    }
 
    if ($status === 'ids') {
      // Selection mode: hand back the aggregate the user actually asked
      // for, next to the rows themselves.
      return [
        "count" => count($list),
        "total_ht" => price($sumHt),
        "total_ttc" => price($sumTtc),
        "invoices" => $list
      ];
    }
 
    return $list;
  }
 
  private function getInvoice($args)
  {
    $id = isset($args['ref']) ? $args['ref'] : (isset($args['id']) ? $args['id'] : null);
 
    $invoice = $this->findInvoice($id);
    if (is_array($invoice)) {
      return $invoice;
    }
 
    $invoice->fetch_thirdparty();
    $invoice->fetch_lines();
 
    $lines = [];
    foreach ($invoice->lines as $l) {
      $prodRef = !empty($l->product_ref) ? $l->product_ref : (!empty($l->product_label) ? $l->product_label : '');
 
      $lines[] = [
        "product" => $prodRef,
        "desc" => dol_html_entity_decode(strip_tags($l->desc), ENT_QUOTES),
        "qty" => (float) $l->qty,
        "price" => price($l->subprice),
        "total_line" => price($l->total_ht),
        "vat" => $l->tva_tx . "%"
      ];
    }
 
    return [
      "id" => $invoice->id,
      "ref" => $invoice->ref,
      "date" => dol_print_date($invoice->date, 'day'),
      "status" => $invoice->getLibStatut(1),
      "customer" => $invoice->thirdparty->name,
      "total_ht" => price($invoice->total_ht),
      "total_ttc" => price($invoice->total_ttc),
      "lines" => $lines,
      "url" => DOL_URL_ROOT . "/compta/facture/card.php?id=" . $invoice->id
    ];
  }
 
  private function validateInvoice($args)
  {
    $user = $this->getUser();
    if ($user === null) {
      return ["error" => "User not authenticated."];
    }
 
    // Checked here as well as centrally: validating is irreversible, so it
    // must not depend on a single caller remembering to check.
    if (!$user->hasRight('facture', 'creer')) {
      return ["error" => "Permission denied: validating an invoice requires the right facture/creer."];
    }
    $invoice = $this->findInvoice($args['invoice']);
    if (is_array($invoice)) {
      return $invoice;
    }
 
    if ($invoice->statut != 0) {
      return ["error" => "Invoice is already validated."];
    }
 
    if ($invoice->validate($user) < 0) {
      $error = $invoice->error;
      if (!empty($invoice->errors)) {
        $error .= ' ' . implode(', ', $invoice->errors);
      }
      if (empty(trim($error))) {
        $error = 'Unknown error (validate returned < 0 with no message)';
      }
      return ["error" => "Validation failed: " . $error];
    }
 
    $invoice->fetch($invoice->id);
    return [
      "success" => true,
      "new_ref" => $invoice->ref,
      "status" => "Validated (Unpaid)",
      "url" => DOL_URL_ROOT . "/compta/facture/card.php?id=" . $invoice->id
    ];
  }
 
  private function payInvoice($args)
  {
    $user = $this->getUser();
    if ($user === null) {
      return ["error" => "User not authenticated."];
    }
 
    // Checked here as well as centrally: a payment cannot be taken back.
    if (!$user->hasRight('facture', 'paiement')) {
      return ["error" => "Permission denied: recording a payment requires the right facture/paiement."];
    }
    $invoice = $this->findInvoice($args['invoice']);
    if (is_array($invoice)) {
      return $invoice;
    }
 
    // Cannot pay drafts
    if ($invoice->statut == 0) {
      return ["error" => "Cannot pay a Draft invoice. Please validate it first."];
    }
 
    $bank = $this->findBankAccount(isset($args['bank_account']) ? $args['bank_account'] : '');
    if (!$bank) {
      return ["error" => "No active Bank account found to receive payment."];
    }
 
    $remaining = $invoice->total_ttc - $invoice->getSommePaiement();
    if ($remaining <= 0) {
      return ["error" => "Invoice is already fully paid."];
    }
 
    $amount = isset($args['amount']) ? (float) $args['amount'] : $remaining;
    if ($amount > $remaining) {
      $amount = $remaining;
    }
 
    $code = isset($args['payment_mode']) ? $args['payment_mode'] : 'VIR';
    $modeId = dol_getIdFromCode($this->db, $code, 'c_paiement', 'code', 'id');
 
    $this->db->begin();
    $payment = new Paiement($this->db);
    $payment->datepaye = dol_now();
    $payment->amounts = [$invoice->id => $amount];
    $payment->paiementid = $modeId;
    $payment->paiementcode = $code;
 
    $paymentId = $payment->create($user, 1);
    if ($paymentId < 0) {
      $this->db->rollback();
      return ["error" => "Payment creation failed: " . implode(', ', $payment->errors)];
    }
    $payment->fetch($paymentId);
    if ($payment->addPaymentToBank($user, 'payment', '(Payment via AI)', $bank->rowid, '', '') < 0) {
      $this->db->rollback();
      return ["error" => "Failed to add payment to bank ledger."];
    }
 
    $this->db->commit();
 
    return [
      "success" => true,
      "paid_amount" => price($amount),
      "remaining_due" => price($remaining - $amount),
      "status" => ($remaining - $amount <= 0) ? "Fully Paid" : "Partially Paid",
      "payment_url" => DOL_URL_ROOT . "/compta/paiement/card.php?id=" . $paymentId
    ];
  }
 
  private function searchSupplierInvoices($args)
  {
    if (!$this->user->hasRight('fournisseur', 'facture', 'lire')) {
      return ["error" => "Permission denied: supplier invoice read right is missing."];
    }
 
    $limit = min(25, max(1, (int) ($args['limit'] ?? 10)));
    $query = trim((string) ($args['query'] ?? ''));
    $status = (string) ($args['status'] ?? '');
 
    $sql = "SELECT f.rowid FROM ".MAIN_DB_PREFIX."facture_fourn as f";
    $sql .= " LEFT JOIN ".MAIN_DB_PREFIX."societe as s ON s.rowid = f.fk_soc";
    $sql .= " WHERE f.entity IN (".getEntity('facture_fourn').")";
    $onlyIds = array();
    if (!empty($args['ids']) && is_array($args['ids'])) {
      foreach ($args['ids'] as $selId) {
        if ((int) $selId > 0 && count($onlyIds) < 100) {
          $onlyIds[] = (int) $selId;
        }
      }
      if (!empty($onlyIds)) {
        $sql .= " AND f.rowid IN (".$this->db->sanitize(implode(',', $onlyIds)).")";
      }
    }
    if ($query !== '') {
      $q = $this->db->escape($this->db->escapeforlike($query));
      $sql .= " AND (f.ref LIKE '%".$this->db->escape($q)."%' OR f.ref_supplier LIKE '%".$this->db->escape($q)."%' OR s.nom LIKE '%".$this->db->escape($q)."%')";
    }
    if ($status === 'draft') {
      $sql .= " AND f.fk_statut = 0";
    } elseif ($status === 'unpaid') {
      $sql .= " AND f.fk_statut = 1 AND f.paye = 0";
    } elseif ($status === 'paid') {
      $sql .= " AND f.paye = 1";
    }
    $sql .= " ORDER BY f.datef DESC";
    $sql .= " LIMIT ".((int) $limit);
 
    $resql = $this->db->query($sql);
    if (!$resql) {
      dol_syslog("AiToolsInvoices::searchSupplierInvoices sql error", LOG_ERR);
      return ["error" => "Search failed."];
    }
 
    require_once DOL_DOCUMENT_ROOT.'/fourn/class/fournisseur.facture.class.php';
    $out = [];
    while ($obj = $this->db->fetch_object($resql)) {
      $inv = new FactureFournisseur($this->db);
      if ($inv->fetch((int) $obj->rowid) > 0) {
        $inv->fetch_thirdparty();
        $out[] = [
          "id" => $inv->id,
          "ref" => $inv->ref,
          "ref_supplier" => $inv->ref_supplier,
          "supplier" => is_object($inv->thirdparty) ? $inv->thirdparty->name : '',
          "date" => dol_print_date($inv->date, 'day'),
          "status" => $inv->getLibStatut(1),
          "total_ttc" => price($inv->total_ttc),
          "_ht" => (float) $inv->total_ht,
          "_ttc" => (float) $inv->total_ttc,
          "url" => DOL_URL_ROOT."/fourn/facture/card.php?facid=".$inv->id
        ];
      }
    }
 
    if (!empty($out) && !empty($onlyIds)) {
      $sumHt = 0.0;
      $sumTtc = 0.0;
      foreach ($out as $row) {
        $sumHt += $row['_ht'];
        $sumTtc += $row['_ttc'];
      }
      foreach ($out as $k => $row) {
        unset($out[$k]['_ht'], $out[$k]['_ttc']);
      }
      return [
        "count" => count($out),
        "total_ht" => price($sumHt),
        "total_ttc" => price($sumTtc),
        "invoices" => $out
      ];
    }
    foreach ($out as $k => $row) {
      unset($out[$k]['_ht'], $out[$k]['_ttc']);
    }
 
    return empty($out) ? ["message" => "No supplier invoices found."] : $out;
  }
 
  private function getSupplierInvoice($args)
  {
    if (!$this->user->hasRight('fournisseur', 'facture', 'lire')) {
      return ["error" => "Permission denied: supplier invoice read right is missing."];
    }
 
    require_once DOL_DOCUMENT_ROOT.'/fourn/class/fournisseur.facture.class.php';
    $inv = new FactureFournisseur($this->db);
    $res = 0;
    if (!empty($args['id'])) {
      $res = $inv->fetch((int) $args['id']);
    } elseif (!empty($args['ref'])) {
      $res = $inv->fetch(0, (string) $args['ref']);
    }
    if ($res <= 0 || empty($inv->id)) {
      return ["error" => "Supplier invoice not found."];
    }
    if (!in_array((int) $inv->entity, explode(',', getEntity('facture_fourn')))) {
      return ["error" => "Supplier invoice not found."];
    }
 
    $inv->fetch_thirdparty();
    $inv->fetch_lines();
 
    $lines = [];
    foreach ($inv->lines as $l) {
      $lvars = get_object_vars($l);
      $lines[] = [
        "product" => !empty($l->product_ref) ? $l->product_ref : '',
        "desc" => dol_html_entity_decode(strip_tags((string) $l->description), ENT_QUOTES),
        "qty" => (float) $l->qty,
        "price" => price($lvars['pu_ht'] ?? $lvars['subprice'] ?? 0),
        "total_line" => price($l->total_ht),
        "vat" => $l->tva_tx."%"
      ];
    }
 
    return [
      "id" => $inv->id,
      "ref" => $inv->ref,
      "ref_supplier" => $inv->ref_supplier,
      "date" => dol_print_date($inv->date, 'day'),
      "status" => $inv->getLibStatut(1),
      "supplier" => is_object($inv->thirdparty) ? $inv->thirdparty->name : '',
      "total_ht" => price($inv->total_ht),
      "total_ttc" => price($inv->total_ttc),
      "lines" => $lines,
      "url" => DOL_URL_ROOT."/fourn/facture/card.php?facid=".$inv->id
    ];
  }
 
  private function findCustomer($identifier)
  {
    global $conf;
 
    $customer = new Societe($this->db);
    $identifier = trim($identifier);
 
    if (preg_match('/^(?:socid|id)[:\s]+(\d+)$/i', $identifier, $m)) {
      $identifier = $m[1];
    } elseif (preg_match('/^(?:code|ref)[:\s]+(.+)$/i', $identifier, $m)) {
      $identifier = $m[1];
    }
 
    if (is_numeric($identifier)) {
      if ($customer->fetch((int) $identifier) > 0) {
        return $customer;
      }
    }
 
    // Exact
    $sql = "SELECT rowid FROM " . MAIN_DB_PREFIX . "societe
      WHERE (nom = '" . $this->db->escape($identifier) . "'
      OR code_client = '" . $this->db->escape($identifier) . "')
      AND entity IN (" . getEntity('societe') . ")";
 
    $resql = $this->db->query($sql);
 
    if ($resql && $this->db->num_rows($resql) > 0) {
      $obj = $this->db->fetch_object($resql);
      $customer->fetch($obj->rowid);
      $this->db->free($resql);
      return $customer;
    }
 
    $sql = "SELECT rowid, nom FROM " . MAIN_DB_PREFIX . "societe
      WHERE (nom LIKE '%" . $this->db->escape($identifier) . "%'
      OR code_client LIKE '%" . $this->db->escape($identifier) . "%')
      AND entity IN (" . getEntity('societe') . ")
      LIMIT 5";
 
    $resql = $this->db->query($sql);
 
    if ($resql) {
      $num = $this->db->num_rows($resql);
 
      if ($num == 1) {
        $obj = $this->db->fetch_object($resql);
        $customer->fetch($obj->rowid);
        $this->db->free($resql);
        return $customer;
      } elseif ($num > 1) {
        $matches = [];
        while ($obj = $this->db->fetch_object($resql)) {
          $matches[] = $obj->nom;
        }
        $this->db->free($resql);
        return ["error" => "Multiple customers found.", "matches" => $matches];
      }
    }
 
    return ["error" => "Customer not found."];
  }
 
  private function findInvoice($identifier)
  {
    $invoice = new Facture($this->db);
    $identifier = trim($identifier);
 
    if (preg_match('/^\‍(?prov[-_]?(\d+)\‍)?$/i', $identifier, $matches)) {
      if ($invoice->fetch((int) $matches[1]) > 0) {
        return $invoice;
      }
    }
    if (is_numeric($identifier)) {
      if ($invoice->fetch((int) $identifier) > 0) {
        return $invoice;
      }
    }
    if ($invoice->fetch(0, $identifier) > 0) {
      return $invoice;
    }
 
    return ["error" => "Invoice not found."];
  }
 
  private function findBankAccount($identifier)
  {
    global $conf;
 
    $identifier = trim($identifier);
    $params = [];
 
 
    $sql = "SELECT rowid, label FROM " . MAIN_DB_PREFIX . "bank_account
      WHERE entity IN (" . getEntity('bank_account') . ") AND clos = 0";
 
    if (is_numeric($identifier)) {
      $sql .= " AND rowid = " . ((int) $identifier);
    } elseif (!empty($identifier)) {
      $sql .= " AND (ref = '" . $this->db->escape($identifier) . "'
            OR label LIKE '%" . $this->db->escape($identifier) . "%')";
    }
 
    $resql = $this->db->query($sql);
 
    if ($resql && $this->db->num_rows($resql) > 0) {
      $obj = $this->db->fetch_object($resql);
      $this->db->free($resql);
      return $obj;
    }
 
    // fallback
    $sql = "SELECT rowid, label FROM " . MAIN_DB_PREFIX . "bank_account
      WHERE entity IN (" . getEntity('bank_account') . ") AND clos = 0
      LIMIT 1";
 
    $resql = $this->db->query($sql);
 
    if ($resql && $this->db->num_rows($resql) > 0) {
      $obj = $this->db->fetch_object($resql);
      $this->db->free($resql);
      return $obj;
    }
 
    return null;
  }
}