xref: /MusicFree/src/utils/stateMapper.ts (revision 233cafa6d759d9d8f74658f7f6ae6d72949a2bc1)

import {useEffect, useState} from 'react';

export default class StateMapper<T> {
  private getFun: () => T;
  private cbs: Set<Function> = new Set([]);
  constructor(getFun: () => T) {
    this.getFun = getFun;
  }

  notify = () => {
    this.cbs.forEach(_ => _?.());
  }

  useMappedState = () => {
    const [_state, _setState] = useState<T>(this.getFun);
    const updateState = () => {
      _setState(this.getFun());
    };
    useEffect(() => {
      this.cbs.add(updateState);
      return () => {
        this.cbs.delete(updateState);
      };
    }, []);
    return _state;
  }
}