package model.result;

import java.util.*;
import model.result.comparators.IdNumberComparator;
import model.result.comparators.ResultComparator;

public class RaceResultList {
  private SortedSet<RaceResult> results;

  public RaceResultList() {
    results = new TreeSet<>(new IdNumberComparator());
  }

  public RaceResultList(ResultComparator comp) {
    results = new TreeSet<>(comp);
  }

  public void setComp(ResultComparator comp) {
    SortedSet<RaceResult> newResults = new TreeSet<>(comp);
    newResults.addAll(results);
    results = newResults;
  }

  /**
   * Updates the RaceResultList with a new RaceResult.
   *
   * @param rr the new RaceResult to be added.
   */
  public void update(RaceResult rr) {
    results.add(rr);
  }

  /**
   * Locates the RaceResult in the results list.
   *
   * @param startNbr the index for the RaceResult.
   * @return An optional RaceResult, depending on if the startNbr exists or not.
   */
  public Optional<RaceResult> getResultByStartNbr(int startNbr) {
    return results.stream().filter(r -> r.getCompetitorNumber() == startNbr).findAny();
  }

  /** Gets a list of all results, in the order specified by the given RaceComparator. */
  public List<RaceResult> getList() {
    return new ArrayList<>(results);
  }
}