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

import { Tax } from "@erp/core/api";
import { CurrencyCode, LanguageCode } from "@repo/rdx-ddd";
import { Collection } from "@repo/rdx-utils";
import { ItemAmount } from "../../value-objects";
import { CustomerInvoiceItem } from "./customer-invoice-item";

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

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

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

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

  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)
    ) {
      return false;
    }
    return super.add(item);
  }

  public getSubtotalAmount(): ItemAmount {
    return this.getAll().reduce(
      (total, tax) => total.add(tax.getSubtotalAmount()),
      ItemAmount.zero(this._currencyCode.code)
    );
  }

  public getDiscountAmount(): ItemAmount {
    return this.getAll().reduce(
      (total, item) => total.add(item.getDiscountAmount()),
      ItemAmount.zero(this._currencyCode.code)
    );
  }

  public getTaxableAmount(): ItemAmount {
    return this.getAll().reduce(
      (total, item) => total.add(item.getTaxableAmount()),
      ItemAmount.zero(this._currencyCode.code)
    );
  }

  public getTaxesAmount(): ItemAmount {
    return this.getAll().reduce(
      (total, item) => total.add(item.getTaxesAmount()),
      ItemAmount.zero(this._currencyCode.code)
    );
  }

  public getTotalAmount(): ItemAmount {
    return this.getAll().reduce(
      (total, item) => total.add(item.getTotalAmount()),
      ItemAmount.zero(this._currencyCode.code)
    );
  }

  public getTaxesAmountByTaxes(): Array<{
    tax: Tax;
    taxableAmount: ItemAmount;
    taxesAmount: ItemAmount;
  }> {
    const getTaxCode = (tax: Tax): string => tax.code; // clave estable para Map
    const currencyCode = this._currencyCode.code;

    // Mapeamos por clave (tax code), pero también guardamos el Tax original
    const resultMap = new Map<
      string,
      { tax: Tax; taxableAmount: ItemAmount; taxesAmount: ItemAmount }
    >();

    for (const item of this.getAll()) {
      for (const { taxableAmount, tax, taxesAmount } of item.getTaxesAmountByTaxes()) {
        const key = getTaxCode(tax);
        const current = resultMap.get(key) ?? {
          tax,
          taxableAmount: ItemAmount.zero(currencyCode),
          taxesAmount: ItemAmount.zero(currencyCode),
        };

        resultMap.set(key, {
          tax: current.tax,
          taxableAmount: current.taxableAmount.add(taxableAmount),
          taxesAmount: current.taxesAmount.add(taxesAmount),
        });
      }
    }

    return Array.from(resultMap.values());
  }
}