webpack-5-chain/src/createClass/createSet.js

[![NPM version][npm-image]][npm-url] [![NPM downloads][npm-downloads]][npm-url] [![CI Status][ci-image]][ci-url]

module.exports = function createSet(superClass) {
  return class extends superClass {
    constructor(...args) {
      super(...args);
      this.store = new Set();
    }

    add(value) {
      this.store.add(value);
      return this;
    }

    prepend(value) {
      this.store = new Set([value, ...this.store]);
      return this;
    }

    clear() {
      this.store.clear();
      return this;
    }

    delete(value) {
      this.store.delete(value);
      return this;
    }

    values() {
      return [...this.store];
    }

    has(value) {
      return this.store.has(value);
    }

    merge(arr) {
      this.store = new Set([...this.store, ...arr]);
      return this;
    }

    when(
      condition,
      whenTruthy = Function.prototype,
      whenFalsy = Function.prototype,
    ) {
      if (condition) {
        whenTruthy(this);
      } else {
        whenFalsy(this);
      }

      return this;
    }
  };
};