src/state.ts

import { StageDefinition } from "./stage";
import { PopoverDOM } from "./popover";

export type State = {
  // Whether driver is initialized or not
  isInitialized?: boolean;

  // Used to bounce the resize event
  resizeTimeout?: number;

  // Used while transitioning between stages
  previousHighlight?: Element;
  activeHighlight?: Element;
  transitionCallback?: () => void;

  activeStagePosition?: StageDefinition;
  stageSvg?: SVGSVGElement;

  popover?: PopoverDOM;
};

let currentState: State = {};

export function setState<K extends keyof State>(key: K, value: State[K]) {
  currentState[key] = value;
}

export function getState(): State;
export function getState<K extends keyof State>(key: K): State[K];
export function getState<K extends keyof State>(key?: K) {
  return key ? currentState[key] : currentState;
}