let intervals: NodeJS.Timeout[] = []

export function set(callback: () => any, period: number) {
  const interval = setInterval(callback, period)
  intervals.push(interval)
  return interval
}

export function clear(interval: NodeJS.Timeout) {
  const idx = intervals.indexOf(interval)
  if (idx !== -1) {
    intervals.splice(idx, 1)
  }
  clearInterval(interval)
}

export function cleanup() {
  for (let interval of intervals) {
    clearInterval(interval)
  }
  intervals = []
}

export class ExecutionTimeoutError extends Error {
  public readonly name = "ExecutionTimeoutError"
}

export class ExecutionTimeTracker {
  static withLimit(limitMs: number) {
    return new ExecutionTimeTracker(limitMs)
  }

  constructor(readonly limitMs: number) {}

  private totalTimeMs = 0

  track<T>(f: () => T): T {
    this.checkLimit()
    const start = process.hrtime.bigint()
    try {
      return f()
    } finally {
      const end = process.hrtime.bigint()
      this.totalTimeMs += Number(end - start) / 1e6
      this.checkLimit()
    }
  }

  get elapsedMS() {
    return this.totalTimeMs
  }

  checkLimit() {
    if (this.totalTimeMs > this.limitMs) {
      throw new ExecutionTimeoutError(
        `Execution time limit of ${this.limitMs}ms exceeded: ${this.totalTimeMs}ms`
      )
    }
  }
}