import { throwIfAborted } from '../aborterror.js';
import { wrapWithAbort } from './operators/withabort.js';

export async function extremaBy<TSource, TKey>(
  source: AsyncIterable<TSource>,
  selector: (item: TSource, signal?: AbortSignal) => TKey | Promise<TKey>,
  comparer: (left: TKey, right: TKey, signal?: AbortSignal) => number | Promise<number>,
  signal?: AbortSignal
): Promise<TSource[]> {
  throwIfAborted(signal);

  let result = [];
  const it = wrapWithAbort(source, signal)[Symbol.asyncIterator]();
  const { value, done } = await it.next();
  if (done) {
    throw new Error('Sequence contains no elements');
  }

  let resKey = await selector(value, signal);
  result.push(value);

  let next: IteratorResult<TSource>;
  while (!(next = await it.next()).done) {
    const current = next.value;
    const key = await selector(current, signal);
    const cmp = await comparer(key, resKey, signal);

    if (cmp === 0) {
      result.push(current);
    } else if (cmp > 0) {
      result = [current];
      resKey = key;
    }
  }

  return result;
}