useResetState

use-reset-State.js

const NO_VALUE_SYMBOL = Symbol("USE_RESET_STATE_NO_VALUE");
const useResetState = (createValue, deps = []) => {
  const [, triggerRerender] = useState(createValue);
  const stateRef = useRef(NO_VALUE_SYMBOL);
  const depsRef = useRef(deps);

  if (stateRef.current === NO_VALUE_SYMBOL) {
    stateRef.current = createValue();
  }
  if (depsRef.current.some((value, index) => value !== deps[index])) {
    depsRef.current = deps;
    stateRef.current = createValue();
  }

  const setState = useCallback(
    newState => {
      if (typeof newState === "function") {
        stateRef.current = newState(stateRef.current);
      } else {
        stateRef.current = newState;
      }
      triggerRerender(i => i + 1);
    },
    [triggerRerender]
  );

  return [stateRef.current, setState];
};