useStateWithCallback like old setState callback behaviour

useStateWithCallback.ts

// https://betterprogramming.pub/synchronous-state-in-react-using-hooks-dc77f43d8521
import { useState } from 'react';

export function useStateWithCallback<T>(initialValue: T) {
  const [value, setValue] = useState(initialValue);

  const setValueAndCallback = (newValue: T, callback: (prevValue: T, newValue: T) => void) => {
    setValue((prevValue) => {
      if (callback) {
        callback(prevValue, newValue);
      }
      return newValue;
    });
  };

  return [value, setValueAndCallback];
}