@react-hookz/web/dist/useControlledRerenderState/index.js

React hooks done right, for browser and SSR.

import { useCallback, useRef } from 'react';
import { useFirstMountState } from '../useFirstMountState/index.js';
import { useRerender } from '../useRerender/index.js';
import { resolveHookState } from '../util/resolve-hook-state.js';
/**
 * Like `React.useState`, but its state setter accepts extra argument, that allows to cancel
 * rerender.
 */
export function useControlledRerenderState(initialState) {
    const state = useRef(useFirstMountState() ? resolveHookState(initialState) : undefined);
    const rr = useRerender();
    return [
        state.current,
        useCallback((value, rerender) => {
            const newState = resolveHookState(value, state.current);
            if (newState !== state.current) {
                state.current = newState;
                if (rerender === undefined || rerender) {
                    rr();
                }
            }
            // eslint-disable-next-line react-hooks/exhaustive-deps
        }, []),
    ];
}