victory-core/src/victory-util/timer.ts

Victory Core

import { timer, now, Timer as D3Timer } from "victory-vendor/d3-timer";

type TimerCallback = (elapsed: number, duration: number) => void;

export default class Timer {
  private shouldAnimate: boolean;
  private readonly subscribers: Array<{
    callback: TimerCallback;
    startTime: number;
    duration: number;
  }>;
  private activeSubscriptions: number;
  private timer: D3Timer | null;

  constructor() {
    this.shouldAnimate = true;
    this.subscribers = [];
    this.timer = null;
    this.activeSubscriptions = 0;
  }

  bypassAnimation() {
    this.shouldAnimate = false;
  }

  resumeAnimation() {
    this.shouldAnimate = true;
  }

  loop = () => {
    this.subscribers.forEach((s) => {
      s.callback(now() - s.startTime, s.duration);
    });
  };

  start() {
    if (!this.timer) {
      this.timer = timer(this.loop);
    }
  }

  stop() {
    if (this.timer) {
      this.timer.stop();
      this.timer = null;
    }
  }

  subscribe(callback: TimerCallback, duration: number) {
    const subscriptionID = this.subscribers.push({
      startTime: now(),
      callback,
      duration: this.shouldAnimate ? duration : 0,
    });
    this.activeSubscriptions++;
    this.start();
    return subscriptionID;
  }

  unsubscribe(id) {
    if (id !== null && this.subscribers[id - 1]) {
      delete this.subscribers[id - 1];
      this.activeSubscriptions--;
    }
    if (this.activeSubscriptions === 0) {
      this.stop();
    }
  }
}