Uecko_ERP/modules/customer-invoices/src/api/domain/entities/customer-invoice-items/customer-invoice-items.ts

import type { Tax } from "@erp/core/api";
import type { CurrencyCode, LanguageCode, Percentage } from "@repo/rdx-ddd";
import { Collection } from "@repo/rdx-utils";

import { ItemAmount, ItemDiscount } from "../../value-objects";

import type { CustomerInvoiceItem } from "./customer-invoice-item";

export interface CustomerInvoiceItemsProps {
  items?: CustomerInvoiceItem[];
  languageCode: LanguageCode;
  currencyCode: CurrencyCode;
  globalDiscountPercentage: Percentage;
}

export class CustomerInvoiceItems extends Collection<CustomerInvoiceItem> {
  private _languageCode!: LanguageCode;
  private _currencyCode!: CurrencyCode;
  private _globalDiscountPercentage!: Percentage;

  constructor(props: CustomerInvoiceItemsProps) {
    super(props.items ?? []);
    this._languageCode = props.languageCode;
    this._currencyCode = props.currencyCode;
    this._globalDiscountPercentage = props.globalDiscountPercentage;
  }

  public static create(props: CustomerInvoiceItemsProps): CustomerInvoiceItems {
    return new CustomerInvoiceItems(props);
  }

  // Helpers

  private _sumAmounts(selector: (item: CustomerInvoiceItem) => ItemAmount): ItemAmount {
    return this.getAll().reduce(
      (acc, item) => acc.add(selector(item)),
      ItemAmount.zero(this._currencyCode.code)
    );
  }

  /**
   * @summary Helper puro para sumar impuestos individuales por tipo.
   */
  private _calculateIndividualTaxes() {
    let iva = ItemAmount.zero(this._currencyCode.code);
    let rec = ItemAmount.zero(this._currencyCode.code);
    let retention = ItemAmount.zero(this._currencyCode.code);

    for (const item of this.getAll()) {
      const { ivaAmount, recAmount, retentionAmount } = item.getIndividualTaxAmounts();

      iva = iva.add(ivaAmount);
      rec = rec.add(recAmount);
      retention = retention.add(retentionAmount);
    }

    return { iva, rec, retention };
  }

  //

  /**
   * @summary Añade un nuevo ítem a la colección.
   * @param item - El ítem de factura a añadir.
   * @returns `true` si el ítem fue añadido correctamente; `false` si fue rechazado.
   * @remarks
   * Sólo se aceptan ítems cuyo `LanguageCode` y `CurrencyCode` coincidan con
   * los de la colección. Si no coinciden, el método devuelve `false` sin modificar
   * la colección.
   */
  add(item: CustomerInvoiceItem): boolean {
    // Antes de añadir un nuevo item, debo comprobar que el item a añadir
    // tiene el mismo "currencyCode" y "languageCode" que la colección de items.
    if (
      !(
        this._languageCode.equals(item.languageCode) &&
        this._currencyCode.equals(item.currencyCode) &&
        this._globalDiscountPercentage.equals(
          item.globalDiscountPercentage.match(
            (v) => v,
            () => ItemDiscount.zero()
          )
        )
      )
    ) {
      return false;
    }
    return super.add(item);
  }

  // Cálculos

  /**
   * @summary Orquestador central del cálculo agregado de la colección.
   * @remarks
   * Delega en los ítems individuales (DDD correcto) pero evita múltiples recorridos.
   */
  public calculateAllAmounts() {
    let subtotalAmount = ItemAmount.zero(this._currencyCode.code);

    let itemDiscountAmount = ItemAmount.zero(this._currencyCode.code);
    let globalDiscountAmount = ItemAmount.zero(this._currencyCode.code);
    let totalDiscountAmount = ItemAmount.zero(this._currencyCode.code);

    let taxableAmount = ItemAmount.zero(this._currencyCode.code);

    let ivaAmount = ItemAmount.zero(this._currencyCode.code);
    let recAmount = ItemAmount.zero(this._currencyCode.code);
    let retentionAmount = ItemAmount.zero(this._currencyCode.code);

    let taxesAmount = ItemAmount.zero(this._currencyCode.code);
    let totalAmount = ItemAmount.zero(this._currencyCode.code);

    for (const item of this.getAll()) {
      const amounts = item.calculateAllAmounts();

      // Subtotales
      subtotalAmount = subtotalAmount.add(amounts.subtotalAmount);

      // Descuentos
      itemDiscountAmount = itemDiscountAmount.add(amounts.itemDiscountAmount);
      globalDiscountAmount = globalDiscountAmount.add(amounts.globalDiscountAmount);
      totalDiscountAmount = totalDiscountAmount.add(amounts.totalDiscountAmount);

      // Base imponible
      taxableAmount = taxableAmount.add(amounts.taxableAmount);

      // Impuestos individuales
      ivaAmount = ivaAmount.add(amounts.ivaAmount);
      recAmount = recAmount.add(amounts.recAmount);
      retentionAmount = retentionAmount.add(amounts.retentionAmount);

      // Total impuestos del ítem
      taxesAmount = taxesAmount.add(amounts.taxesAmount);

      // Total final del ítem
      totalAmount = totalAmount.add(amounts.totalAmount);
    }

    return {
      subtotalAmount,

      itemDiscountAmount,
      globalDiscountAmount,
      totalDiscountAmount,

      taxableAmount,

      ivaAmount,
      recAmount,
      retentionAmount,

      taxesAmount,
      totalAmount,
    } as const;
  }

  public getSubtotalAmount(): ItemAmount {
    return this.calculateAllAmounts().subtotalAmount;
  }

  public getItemDiscountAmount(): ItemAmount {
    return this.calculateAllAmounts().itemDiscountAmount;
  }

  public getGlobalDiscountAmount(): ItemAmount {
    return this.calculateAllAmounts().globalDiscountAmount;
  }

  public getTotalDiscountAmount(): ItemAmount {
    return this.calculateAllAmounts().totalDiscountAmount;
  }

  public getTaxableAmount(): ItemAmount {
    return this.calculateAllAmounts().taxableAmount;
  }

  public getTaxesAmount(): ItemAmount {
    return this.calculateAllAmounts().taxesAmount;
  }

  public getTotalAmount(): ItemAmount {
    return this.calculateAllAmounts().totalAmount;
  }

  /**
   * @summary Agrupa bases e importes por código fiscal.
   * @remarks
   * Este método se usa para poblar la tabla `customer_invoice_taxes`.
   */
  public groupTaxesByCode() {
    const map = new Map<string, { tax: Tax; taxable: ItemAmount; total: ItemAmount }>();

    for (const item of this.getAll()) {
      const amounts = item.calculateAllAmounts();
      const taxable = amounts.taxableAmount;

      const { ivaAmount, recAmount, retentionAmount } = amounts;

      /* ----------------------------- IVA ----------------------------- */
      item.taxes.iva.match(
        (iva) => {
          const key = iva.code;
          const prev = map.get(key) ?? {
            tax: iva,
            taxable: ItemAmount.zero(taxable.currencyCode),
            total: ItemAmount.zero(taxable.currencyCode),
          };

          map.set(key, {
            tax: iva,
            taxable: prev.taxable.add(taxable),
            total: prev.total.add(ivaAmount),
          });
        },
        () => {
          //
        }
      );

      /* ----------------------------- REC ----------------------------- */
      item.taxes.rec.match(
        (rec) => {
          const key = rec.code;
          const prev = map.get(key) ?? {
            tax: rec,
            taxable: ItemAmount.zero(taxable.currencyCode),
            total: ItemAmount.zero(taxable.currencyCode),
          };

          map.set(key, {
            tax: rec,
            taxable: prev.taxable.add(taxable),
            total: prev.total.add(recAmount),
          });
        },
        () => {
          //
        }
      );

      /* -------------------------- RETENCIÓN -------------------------- */
      item.taxes.retention.match(
        (retention) => {
          const key = retention.code;
          const prev = map.get(key) ?? {
            tax: retention,
            taxable: ItemAmount.zero(taxable.currencyCode),
            total: ItemAmount.zero(taxable.currencyCode),
          };

          map.set(key, {
            tax: retention,
            taxable: prev.taxable.add(taxable),
            total: prev.total.add(retentionAmount),
          });
        },
        () => {
          //
        }
      );
    }

    return map;
  }
}