immer + useState = useImmerState

useImmerState.ts

import { useCallback, useState } from "react";
import produce, { Draft } from "immer";

function useImmerState<S = undefined>(initialState: S | (() => S)) {
  const [state, _setState] = useState(initialState);

  const setState = useCallback((mutateFn: (draft: Draft<S>) => S | void) => {
    _setState(produce(mutateFn) as (oldState: S) => S);
  }, []);

  return [state, setState] as const;
}

export default useImmerState;