Boostnote/browser/lib/Mutable.js

class MutableMap {
  constructor(iterable) {
    this._map = new Map(iterable)
    Object.defineProperty(this, 'size', {
      get: () => this._map.size,
      set: function(value) {
        this['size'] = value
      }
    })
  }

  get(...args) {
    return this._map.get(...args)
  }

  set(...args) {
    return this._map.set(...args)
  }

  delete(...args) {
    return this._map.delete(...args)
  }

  has(...args) {
    return this._map.has(...args)
  }

  clear(...args) {
    return this._map.clear(...args)
  }

  forEach(...args) {
    return this._map.forEach(...args)
  }

  [Symbol.iterator]() {
    return this._map[Symbol.iterator]()
  }

  map(cb) {
    const result = []
    for (const [key, value] of this._map) {
      result.push(cb(value, key))
    }
    return result
  }

  toJS() {
    const result = {}
    for (let [key, value] of this._map) {
      if (value instanceof MutableSet || value instanceof MutableMap) {
        value = value.toJS()
      }
      result[key] = value
    }
    return result
  }
}

class MutableSet {
  constructor(iterable) {
    this._set = new Set(iterable)
    Object.defineProperty(this, 'size', {
      get: () => this._set.size,
      set: function(value) {
        this['size'] = value
      }
    })
  }

  add(...args) {
    return this._set.add(...args)
  }

  delete(...args) {
    return this._set.delete(...args)
  }

  forEach(...args) {
    return this._set.forEach(...args)
  }

  [Symbol.iterator]() {
    return this._set[Symbol.iterator]()
  }

  map(cb) {
    const result = []
    this._set.forEach(function(value, key) {
      result.push(cb(value, key))
    })

    return result
  }

  toJS() {
    return Array.from(this._set)
  }
}

const Mutable = {
  Map: MutableMap,
  Set: MutableSet
}

module.exports = Mutable