useStateWithDeps.ts

import { useState } from "react";

export const useStateWithDeps = <T>(
  initialState: T | (() => T),
  deps: unknown[],
) => {
  const [state, setState] = useState(initialState);
  const [currentDeps, setCurrentDeps] = useState(deps);

  if (!shallowEqual(deps, currentDeps)) {
    const newState = getState(initialState);
    setState(newState);
    setCurrentDeps(deps);
    return [newState, setState] as const;
  }

  return [state, setState] as const;
};

const getState = <T>(initialState: T | (() => T)) => {
  return typeof initialState === "function"
    ? (initialState as () => T)()
    : initialState;
};