src/state.ts

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

export type State = {
  isInitialized?: boolean;
  resizeTimeout?: number;

  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;
}