import 'package:nc_photos/iterator_extension.dart';

/// Contain results from the diff functions
///
/// [onlyInA] contains items exist in a but not b, [onlyInB] contains items
/// exist in b but not a
class DiffResult<T> {
  const DiffResult({
    required this.onlyInA,
    required this.onlyInB,
  });

  final List<T> onlyInB;
  final List<T> onlyInA;
}

/// Return the difference between two sorted lists, [a] and [b]
///
/// [a] and [b] MUST be sorted in ascending order, otherwise the result is
/// undefined
DiffResult<T> diffWith<T>(
    Iterable<T> a, Iterable<T> b, int Function(T a, T b) comparator) {
  final aIt = a.iterator, bIt = b.iterator;
  final aMissing = <T>[], bMissing = <T>[];
  while (true) {
    if (!aIt.moveNext()) {
      // no more elements in a
      bIt.iterate((obj) => aMissing.add(obj));
      return DiffResult(onlyInB: aMissing, onlyInA: bMissing);
    }
    if (!bIt.moveNext()) {
      // no more elements in b
      // needed because aIt has already advanced
      bMissing.add(aIt.current);
      aIt.iterate((obj) => bMissing.add(obj));
      return DiffResult(onlyInB: aMissing, onlyInA: bMissing);
    }
    final result = _diffUntilEqual(aIt, bIt, comparator);
    aMissing.addAll(result.onlyInB);
    bMissing.addAll(result.onlyInA);
  }
}

DiffResult<T> diff<T extends Comparable>(Iterable<T> a, Iterable<T> b) =>
    diffWith(a, b, Comparable.compare);

DiffResult<T> _diffUntilEqual<T>(
    Iterator<T> aIt, Iterator<T> bIt, int Function(T a, T b) comparator) {
  final a = aIt.current, b = bIt.current;
  final diff = comparator(a, b);
  if (diff < 0) {
    // a < b
    if (!aIt.moveNext()) {
      return DiffResult(onlyInB: [b] + bIt.toList(), onlyInA: [a]);
    } else {
      final result = _diffUntilEqual(aIt, bIt, comparator);
      return DiffResult(onlyInB: result.onlyInB, onlyInA: [a] + result.onlyInA);
    }
  } else if (diff > 0) {
    // a > b
    if (!bIt.moveNext()) {
      return DiffResult(onlyInB: [b], onlyInA: [a] + aIt.toList());
    } else {
      final result = _diffUntilEqual(aIt, bIt, comparator);
      return DiffResult(onlyInB: [b] + result.onlyInB, onlyInA: result.onlyInA);
    }
  } else {
    // a == b
    return const DiffResult(onlyInB: [], onlyInA: []);
  }
}