useObjectState hook in react with type definitions

useObjectState.d.ts

import { Dispatch, SetStateAction } from 'react';

/**
 * This method needs to provide the type of the initialState like
 * `useObjectState<initialStateInterface>(initialState)`
 * @param {*} initialState The initial state the user wants to set
 * @returns                Array containing `state` object, `setState` method and `updateState` method
 */
function useObjectState<S>(initialState: S): [
  S,
  Dispatch<SetStateAction<S>>,
  Dispatch<Partial<S>>
];

useObjectState.js

import { useState } from 'react';
/**
 * This method needs to provide the type of the initialState like
 * `useObjectState<initialStateInterface>(initialState)`
 * @param {object} initialState The initial state the user wants to set
 * @returns                Array containing `state` object, `setState` method and `updateState` method
 */
export const useObjectState = (initialState) => {
  const [state, setState] = useState(initialState);

  const updateState = (newState) => {
    setState((prevState) => {
      return {
        ...prevState,
        ...newState
      }
    });
  }

  return [state, setState, updateState];
};