@accounter/israeli-vat-scraper/src/handlers/month-inputs-handler.ts

Scraper app for fetching Israeli VAT data

import type { Page } from 'puppeteer';
import { newPageByMonth } from '../utils/browser-util.js';
import { getReportExpansionInputs } from '../utils/evaluation-functions.js';
import { waitAndClick, waitForSelectorPlus } from '../utils/page-util.js';
import type { Config, Logger, ReportInputs } from '../utils/types.js';
import { UserPrompt } from '../utils/user-prompt.js';
import { monthExpansionRecordsHandler } from './month-expansion-records-handler.js';

const INPUTS_BUTTON_SELECTOR = '#LinkButton0';

/** תשומות */
export class MonthInputsHandler {
  private config: Config;
  private prompt: UserPrompt;
  private location: string[];
  private index: number;
  private page: Page | null = null;

  constructor(config: Config, prompt: UserPrompt, location: string[], index: number) {
    this.config = config;
    this.prompt = prompt;
    this.location = [...location, 'Inputs'];
    this.index = index;
  }

  public handle = async (logger: Logger): Promise<ReportInputs | undefined> => {
    this.prompt.update(this.location, 'Fetching...', logger);
    try {
      this.page = await newPageByMonth(
        this.config.visibleBrowser,
        this.location[0] as string,
        this.index,
        logger,
      );

      await waitAndClick(this.page, INPUTS_BUTTON_SELECTOR, logger);

      const inputsTable = await waitForSelectorPlus(this.page, '#tblSikum', logger);
      const inputsData = await inputsTable?.evaluate(getReportExpansionInputs);

      // gt income records
      for (const key in inputsData) {
        if (key === 'total') {
          continue;
        }
        if (inputsData[key as keyof ReportInputs].received.recordsCount > 0) {
          const index = ((): number | null => {
            switch (key) {
              case 'regularInput':
                return 3;
              case 'pettyCash':
                return 4;
              case 'selfInvoiceInput':
                return 5;
              case 'importList':
                return 6;
              case 'rashapSupplier':
                return 7;
              case 'otherDocument':
                return 8;
              default:
                return null;
            }
          })();
          const secondaryIndex = 2;

          if (index) {
            const recordsHandler = new monthExpansionRecordsHandler(
              this.config,
              this.prompt,
              this.location,
              INPUTS_BUTTON_SELECTOR,
              index,
              secondaryIndex,
            );
            inputsData[key as keyof ReportInputs].received.records =
              await recordsHandler.handle(logger);
          }
        }

        if (inputsData[key as keyof ReportInputs].incorrect.recordsCount > 0) {
          const index = ((): number | null => {
            switch (key) {
              case 'regularInput':
                return 3;
              case 'pettyCash':
                return 4;
              case 'selfInvoiceInput':
                return 5;
              case 'importList':
                return 6;
              case 'rashapSupplier':
                return 7;
              case 'otherDocument':
                return 8;
              default:
                return null;
            }
          })();
          const secondaryIndex = 5;

          if (index) {
            const recordsHandler = new monthExpansionRecordsHandler(
              this.config,
              this.prompt,
              this.location,
              INPUTS_BUTTON_SELECTOR,
              index,
              secondaryIndex,
            );
            inputsData[key as keyof ReportInputs].incorrect.records =
              await recordsHandler.handle(logger);
          }
        }
      }

      this.page.browser().close();
      this.prompt.update(this.location, 'Done', logger);
      return inputsData;
    } catch (e) {
      this.prompt.addError(this.location, (e as Error)?.message || e, logger);
      this.page?.browser().close();
      return undefined;
    }
  };
}