bigcapital/packages/server/src/modules/FinancialStatements/common/TableSheetPdf.ts

import * as R from 'ramda';
import { ITableColumn, ITableData, ITableRow } from '../types/Table.types';
import { FinancialTableStructure } from './FinancialTableStructure';
import { tableClassNames } from '../utils';
import { Injectable } from '@nestjs/common';
import { ChromiumlyTenancy } from '../../ChromiumlyTenancy/ChromiumlyTenancy.service';
import { renderFinancialSheetTemplateHtml } from '@bigcapital/pdf-templates';

@Injectable()
export class TableSheetPdf {
  /**
   * @param {ChromiumlyTenancy} chromiumlyTenancy - The chromiumly tenancy service.
   */
  constructor(
    private readonly chromiumlyTenancy: ChromiumlyTenancy,
  ) { }

  /**
   * Converts the table data into a PDF format.
   * @param {ITableData} table - The table data to be converted.
   * @param {string} organizationName - The organization name.
   * @param {string} sheetName - The name of the sheet.
   * @param {string} sheetDate - The date of the sheet.
   * @param {string} customCSS - Optional custom CSS to inject.
   * @returns A promise that resolves with the PDF conversion result.
   */
  public async convertToPdf(
    table: ITableData,
    organizationName: string,
    sheetName: string,
    sheetDate: string,
    customCSS?: string,
  ): Promise<Buffer> {
    // Prepare columns and rows for PDF conversion
    const columns = this.tablePdfColumns(table.columns);
    const rows = this.tablePdfRows(table.rows);
    const landscape = columns.length > 4;

    // Generate HTML content from the React template
    const htmlContent = renderFinancialSheetTemplateHtml({
      organizationName,
      sheetName,
      sheetDate,
      table: {
        columns: columns.map((col) => ({
          key: col.key,
          label: col.label,
          style: (col as any).style, // style may be added during transformation
        })),
        rows: rows.map((row) => ({
          cells: row.cells,
          classNames: (row as any).classNames,
        })),
      },
      customCSS,
    });
    // Convert the HTML content to PDF
    return this.chromiumlyTenancy.convertHtmlContent(htmlContent, {
      margins: { top: 0, bottom: 0, left: 0, right: 0 },
      landscape,
    });
  }

  /**
   * Converts the table columns to pdf columns.
   * @param {ITableColumn[]} columns
   * @returns {ITableColumn[]}
   */
  private tablePdfColumns = (columns: ITableColumn[]): ITableColumn[] => {
    return columns;
  };

  /**
   * Converts the table rows to pdf rows.
   * @param {ITableRow[]} rows - The table rows to be converted.
   * @returns {ITableRow[]} - The converted table rows.
   */
  private tablePdfRows = (rows: ITableRow[]): ITableRow[] => {
    const curriedFlatNestedTree = R.curry(
      FinancialTableStructure.flatNestedTree,
    );
    const flatNestedTree = curriedFlatNestedTree(R.__, {
      nestedPrefix: '<span style="padding-left: 15px;"></span>',
    });
    // @ts-ignore
    return R.compose(tableClassNames, flatNestedTree)(rows);
  };
}