/**
 * Use Case: GetRacePenaltiesUseCase
 *
 * Returns all penalties applied for a specific race, with driver details.
 * Orchestrates domain logic and delegates presentation to the presenter.
 */

import type { IPenaltyRepository } from '../../domain/repositories/IPenaltyRepository';
import type { IDriverRepository } from '../../domain/repositories/IDriverRepository';
import type {
  IRacePenaltiesPresenter,
  RacePenaltiesResultDTO,
  RacePenaltiesViewModel,
} from '../presenters/IRacePenaltiesPresenter';
import type { UseCase } from '@gridpilot/shared/application/UseCase';

export interface GetRacePenaltiesInput {
  raceId: string;
}

export class GetRacePenaltiesUseCase
  implements
    UseCase<GetRacePenaltiesInput, RacePenaltiesResultDTO, RacePenaltiesViewModel, IRacePenaltiesPresenter>
{
  constructor(
    private readonly penaltyRepository: IPenaltyRepository,
    private readonly driverRepository: IDriverRepository,
  ) {}

  async execute(input: GetRacePenaltiesInput, presenter: IRacePenaltiesPresenter): Promise<void> {
    const penalties = await this.penaltyRepository.findByRaceId(input.raceId);

    const driverIds = new Set<string>();
    penalties.forEach((penalty) => {
      driverIds.add(penalty.driverId);
      driverIds.add(penalty.issuedBy);
    });

    const drivers = await Promise.all(
      Array.from(driverIds).map((id) => this.driverRepository.findById(id)),
    );

    const driverMap = new Map<string, string>();
    drivers.forEach((driver) => {
      if (driver) {
        driverMap.set(driver.id, driver.name);
      }
    });

    presenter.reset();
    const dto: RacePenaltiesResultDTO = {
      penalties,
      driverMap,
    };
    presenter.present(dto);
  }
}