bitburner-src/src/ui/React/hooks.ts

import { useCallback, useEffect, useState } from "react";
import { GameCycleEvents } from "../../engine";

/** Hook that returns a function for the component. Optionally set an interval to rerender the component.
 * @param autoRerenderTime: Optional. If provided and nonzero, used as the ms interval to automatically call the rerender function.
 */
export function useRerender(autoRerenderTime?: number) {
  const [__, setRerender] = useState(0);

  const rerender = useCallback(() => setRerender((currentValue) => currentValue + 1), []);

  useEffect(() => {
    if (!autoRerenderTime) return;
    const intervalID = setInterval(rerender, autoRerenderTime);
    return () => clearInterval(intervalID);
  }, [rerender, autoRerenderTime]);

  return rerender;
}

/** Hook that rerenders the component shortly after the game engine processes a cycle.
 * @returns a function that will trigger a rerender.
 */
export function useCycleRerender(): () => void {
  const rerender = useRerender();

  useEffect(() => {
    const unsubscribe = GameCycleEvents.subscribe(rerender);
    return unsubscribe;
  }, [rerender]);
  return rerender;
}

export function useBoolean(initialValue = false) {
  const [value, setValue] = useState(initialValue);

  const toggle = useCallback(() => {
    setValue((old) => !old);
  }, []);

  const on = useCallback(() => {
    setValue(true);
  }, []);

  const off = useCallback(() => {
    setValue(false);
  }, []);

  return [value, { toggle, on, off }] as const;
}