如何键入自定义挂钩useStateWithCallback React TypeScript

import { useState, useEffect, useRef } from 'react';

function useStateWithCallback(initialState: any) {
  const [state, setState] = useState(initialState);
  const callbackRef = useRef<Function | undefined>();

  useEffect(() => {
    if (callbackRef.current) {
      callbackRef.current(state);
      callbackRef.current = undefined;
    }
  }, [state]);

  const setStateWithCallback = (newState: any, callback: Function) => {
    callbackRef.current = callback;
    setState(newState);
  };

  return [state, setStateWithCallback];
}

export default useStateWithCallback;

import React from 'react';
import useStateWithCallback from './useStateWithCallback';

const MyComponent: React.FC = () => {
  const [count, setCount] = useStateWithCallback(0);

  const handleClick = () => {
    setCount(count + 1, (newCount: number) => {
      console.log('Count updated:', newCount);
    });
  };

  return (
    <div>
      <p>Count: {count}</p>
      <button onClick={handleClick}>Increase Count</button>
    </div>
  );
};

export default MyComponent;