import type { z } from 'zod';
import { init } from '@accounter/modern-poalim-scraper';
import type { ServerMessage } from '../../shared/ws-protocol.js';
import type { MaxPayload } from '../payload-schemas/max.schema.js';
import { validatePayload } from '../validate-payload.js';
import type { MaxAccountSchema } from '../vault.js';

export type MaxCreds = z.infer<typeof MaxAccountSchema>;

export type Emitter = (msg: ServerMessage) => void;

export async function scrapeMax(
  creds: MaxCreds,
  dateFrom: Date,
  _dateTo: Date,
  emit: Emitter,
  headless = true,
): Promise<MaxPayload> {
  const { max: maxFn, close } = await init({ headless });

  try {
    const scraper = await maxFn(
      { username: creds.username, password: creds.password },
      { startDate: dateFrom },
    );

    const accounts = await scraper.getTransactions();
    const validated = validatePayload('max', accounts);

    for (const account of validated) {
      emit({
        type: 'task-month-fetched',
        sourceId: creds.id,
        month: account.accountNumber,
        transactionCount: account.txns.length,
      });
    }

    return validated;
  } finally {
    await close();
  }
}