12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
import { proxy } from '../vanilla'
// properties that we don't want to expose to the end-usertype InternalProxySet<T> = Set<T> & {  data: T[]  toJSON: object}
/** * proxySet * * This is to create a proxy which mimic the native Set behavior. * The API is the same as Set API * * @example * import { proxySet } from 'valtio/utils' * const state = proxySet([1,2,3]) * //can be used inside a proxy as well * const state = proxy({ *   count: 1, *   set: proxySet() * }) */export function proxySet<T>(initialValues?: Iterable<T> | null): Set<T> {  const set: InternalProxySet<T> = proxy({    data: Array.from(new Set(initialValues)),    has(value) {      return this.data.indexOf(value) !== -1    },    add(value) {      let hasProxy = false      if (typeof value === 'object' && value !== null) {        hasProxy = this.data.indexOf(proxy(value as T & object)) !== -1      }      if (this.data.indexOf(value) === -1 && !hasProxy) {        this.data.push(value)      }      return this    },    delete(value) {      const index = this.data.indexOf(value)      if (index === -1) {        return false      }      this.data.splice(index, 1)      return true    },    clear() {      this.data.splice(0)    },    get size() {      return this.data.length    },    forEach(cb) {      this.data.forEach((value) => {        cb(value, value, this)      })    },    get [Symbol.toStringTag]() {      return 'Set'    },    toJSON() {      return {}    },    [Symbol.iterator]() {      return this.data[Symbol.iterator]()    },    values() {      return this.data.values()    },    keys() {      // for Set.keys is an alias for Set.values()      return this.data.values()    },    entries() {      // array.entries returns [index, value] while Set [value, value]      return new Set(this.data).entries()    },  })
  Object.defineProperties(set, {    data: {      enumerable: false,    },    size: {      enumerable: false,    },    toJSON: {      enumerable: false,    },  })
  Object.seal(set)
  return set as Set<T>}