ahooks/lib/useSet/index.js

react hooks library

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = void 0;
var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
var _react = require("react");
var _useMemoizedFn = _interopRequireDefault(require("../useMemoizedFn"));
function useSet(initialValue) {
  var getInitValue = function getInitValue() {
    return new Set(initialValue);
  };
  var _useState = (0, _react.useState)(getInitValue),
    _useState2 = (0, _slicedToArray2["default"])(_useState, 2),
    set = _useState2[0],
    setSet = _useState2[1];
  var updateSet = function updateSet(updater) {
    setSet(function (prevSet) {
      return updater(new Set(prevSet));
    });
  };
  var add = function add(key) {
    if (set.has(key)) {
      return;
    }
    updateSet(function (newSet) {
      newSet.add(key);
      return newSet;
    });
  };
  var remove = function remove(key) {
    if (!set.has(key)) {
      return;
    }
    updateSet(function (newSet) {
      newSet["delete"](key);
      return newSet;
    });
  };
  var reset = function reset() {
    return setSet(getInitValue());
  };
  return [set, {
    add: (0, _useMemoizedFn["default"])(add),
    remove: (0, _useMemoizedFn["default"])(remove),
    reset: (0, _useMemoizedFn["default"])(reset)
  }];
}
var _default = exports["default"] = useSet;