import {
  constructZonesFromProfiles,
  modifyProfiles,
  profilesContainsZone,
} from "../../../helpers/recompute_zones";
import { UnboundedZone, Zone } from "../../../types/misc";

export class ZoneSet {
  private profilesStartingPosition: number[] = [0];
  private profiles = new Map<number, number[]>([[0, []]]);

  constructor(zones: Iterable<Zone | UnboundedZone> = []) {
    for (const zone of zones) {
      this.add(zone);
    }
  }

  isEmpty(): boolean {
    return this.profiles.size === 1 && this.profiles.get(0)?.length === 0;
  }

  add(zone: Zone | UnboundedZone) {
    modifyProfiles(this.profilesStartingPosition, this.profiles, [zone]);
  }

  delete(zone: Zone | UnboundedZone) {
    modifyProfiles(this.profilesStartingPosition, this.profiles, [zone], true);
  }

  has(zone: Zone | UnboundedZone): boolean {
    return profilesContainsZone(this.profilesStartingPosition, this.profiles, zone);
  }

  difference(other: ZoneSet): ZoneSet {
    const result = this.copy();
    for (const zone of other) {
      result.delete(zone);
    }
    return result;
  }

  copy(): ZoneSet {
    const result = new ZoneSet();
    result.profilesStartingPosition = [...this.profilesStartingPosition];
    result.profiles = new Map<number, number[]>();
    for (const [key, value] of this.profiles) {
      result.profiles.set(key, [...value]);
    }
    return result;
  }

  /**
   * iterator of all the zones in the ZoneSet
   */
  [Symbol.iterator](): IterableIterator<Zone> {
    return constructZonesFromProfiles<Zone>(this.profilesStartingPosition, this.profiles)[
      Symbol.iterator
    ]();
  }
}