ahooks/es/useSet/index.js

react hooks library

import { __read } from "tslib";
import { useState } from 'react';
import useMemoizedFn from '../useMemoizedFn';
function useSet(initialValue) {
    var getInitValue = function () { return new Set(initialValue); };
    var _a = __read(useState(getInitValue), 2), set = _a[0], setSet = _a[1];
    var updateSet = function (updater) {
        setSet(function (prevSet) { return updater(new Set(prevSet)); });
    };
    var add = function (key) {
        if (set.has(key)) {
            return;
        }
        updateSet(function (newSet) {
            newSet.add(key);
            return newSet;
        });
    };
    var remove = function (key) {
        if (!set.has(key)) {
            return;
        }
        updateSet(function (newSet) {
            newSet.delete(key);
            return newSet;
        });
    };
    var reset = function () { return setSet(getInitValue()); };
    return [
        set,
        {
            add: useMemoizedFn(add),
            remove: useMemoizedFn(remove),
            reset: useMemoizedFn(reset),
        },
    ];
}
export default useSet;