UseSet

1'use client';
2 
3import { useCallback, useMemo, useState } from 'react';
4 
5export type UseSetResult<T> = {
6  set: Set<T>;
7  add: (item: T) => void;
8  delete: (item: T) => void;
9  has: (item: T) => boolean;
10  clear: () => void;
11  reset: () => void;
12  setValue: (newSet: Set<T>) => void;
13  getAll: () => T[];
14};
15 
16export function useSet<T>(initialSet: Iterable<T> = []): UseSetResult<T> {
17  const [set, setSet] = useState<Set<T>>(() => new Set(initialSet));
18  const initial = useMemo(() => new Set(initialSet), [initialSet]);
19 
20  const add = useCallback((item: T) => {
21    setSet((prev) => {
22      if (prev.has(item)) return prev;
23      const next = new Set(prev);
24      next.add(item);
25      return next;
26    });
27  }, []);
28 
29  const deleteItem = useCallback((item: T) => {
30    setSet((prev) => {
31      if (!prev.has(item)) return prev;
32      const next = new Set(prev);
33      next.delete(item);
34      return next;
35    });
36  }, []);
37 
38  const has = useCallback((item: T) => set.has(item), [set]);
39 
40  const clear = useCallback(() => {
41    setSet(new Set());
42  }, []);
43 
44  const reset = useCallback(() => {
45    setSet(new Set(initial));
46  }, [initial]);
47 
48  const setValue = useCallback((newSet: Set<T>) => {
49    setSet(new Set(newSet));
50  }, []);
51 
52  const getAll = useCallback(() => Array.from(set), [set]);
53 
54  return {
55    set,
56    add,
57    delete: deleteItem,
58    has,
59    clear,
60    reset,
61    setValue,
62    getAll,
63  };
64}