index.ts

import {useRef} from 'react';
import {useRerender} from '../useRerender/index.js';

const proto = Set.prototype;

/**
 * Tracks the state of a `Set`.
 *
 * @param values Initial values iterator for underlying `Set` constructor.
 */

export function useSet<T = any>(values?: readonly T[] | null): Set<T> {
	const setRef = useRef<Set<T>>(undefined);
	const rerender = useRerender();

	if (!setRef.current) {
		const set = new Set<T>(values);

		setRef.current = set;

		set.add = (...args) => {
			proto.add.apply(set, args);
			rerender();
			return set;
		};

		set.clear = (...args) => {
			proto.clear.apply(set, args);
			rerender();
		};

		set.delete = (...args) => {
			const result = proto.delete.apply(set, args);
			rerender();

			return result;
		};
	}

	return setRef.current;
}