Function useSet

src/array/useSet.ts:14–51

(initialState: Set<T> = new Set())

const clone = <T>(value: Set<T>) => new Set<T>(value);

export function useSet<T>(initialState: Set<T> = new Set()): UseSet<T> {
  const [value, setValue] = useState(initialState);
  const add = useCallback((item: T) => {
    setValue((prevValue) => {
      const copy = clone(prevValue);
      copy.add(item);
      return copy;
    });
  }, []);

  const remove = useCallback((item: T) => {
    setValue((prevValue) => {
      const copy = clone(prevValue);
      copy.delete(item);
      return copy;
    });
  }, []);

  const clear = useCallback(() => {
    setValue((prevValue) => {
      const copy = clone(prevValue);
      copy.clear();
      return copy;
    });
  }, []);

  const actions: UseSetActions<T> = useMemo(
    () => ({
      setValue,
      add,
      remove,
      clear,
    }),
    [add, clear, remove],
  );

  return useMemo(() => [value, actions], [value, actions]);
}