yjs/src/Types/Map.js

/* global EventHandler, Y, CustomType, copyObject, compareIds */

;(function () {
  class YMap {
    constructor (os, model) {
      this._model = model.id
      this.os = os
      this.map = copyObject(model.map)
      this.contents = {}
      this.opContents = {}
      this.eventHandler = new EventHandler(ops => {
        var userEvents = []
        for (var i in ops) {
          var op = ops[i]
          var oldValue
          // key is the name to use to access (op)content
          var key = op.struct === 'Delete' ? op.key : op.parentSub

          // compute oldValue
          if (this.opContents[key] != null) {
            let prevType = this.opContents[key]
            oldValue = () => {// eslint-disable-line
              let def = Promise.defer()
              this.os.requestTransaction(function *() {// eslint-disable-line
                def.resolve(yield* this.getType(prevType))
              })
              return def.promise
            }
          } else {
            oldValue = this.contents[key]
          }
          // compute op event
          if (op.struct === 'Insert') {
            if (op.left === null) {
              if (op.opContent != null) {
                delete this.contents[key]
                if (op.deleted) {
                  delete this.opContents[key]
                } else {
                  this.opContents[key] = op.opContent
                }
              } else {
                delete this.opContents[key]
                if (op.deleted) {
                  delete this.contents[key]
                } else {
                  this.contents[key] = op.content
                }
              }
              this.map[key] = op.id
              var insertEvent = {
                name: key,
                object: this
              }
              if (oldValue === undefined) {
                insertEvent.type = 'add'
              } else {
                insertEvent.type = 'update'
                insertEvent.oldValue = oldValue
              }
              userEvents.push(insertEvent)
            }
          } else if (op.struct === 'Delete') {
            if (compareIds(this.map[key], op.target)) {
              delete this.opContents[key]
              delete this.contents[key]
              var deleteEvent = {
                name: key,
                object: this,
                oldValue: oldValue,
                type: 'delete'
              }
              userEvents.push(deleteEvent)
            }
          } else {
            throw new Error('Unexpected Operation!')
          }
        }
        this.eventHandler.callUserEventListeners(userEvents)
      })
    }
    get (key) {
      // return property.
      // if property does not exist, return null
      // if property is a type, return a promise
      if (this.opContents[key] == null) {
        if (key == null) {
          return copyObject(this.contents)
        } else {
          return this.contents[key]
        }
      } else {
        let def = Promise.defer()
        var oid = this.opContents[key]
        this.os.requestTransaction(function *() {
          def.resolve(yield* this.getType(oid))
        })
        return def.promise
      }
    }
    delete (key) {
      var right = this.map[key]
      if (right != null) {
        var del = {
          target: right,
          struct: 'Delete'
        }
        var eventHandler = this.eventHandler
        var modDel = copyObject(del)
        modDel.key = key
        eventHandler.awaitAndPrematurelyCall([modDel])
        this.os.requestTransaction(function *() {
          yield* this.applyCreatedOperations([del])
          eventHandler.awaitedLastDeletes(1)
        })
      }
    }
    set (key, value) {
      // set property.
      // if property is a type, return a promise
      // if not, apply immediately on this type an call event

      var right = this.map[key] || null
      var insert = {
        left: null,
        right: right,
        origin: null,
        parent: this._model,
        parentSub: key,
        struct: 'Insert'
      }
      var def = Promise.defer()
      if (value instanceof CustomType) {
        // construct a new type
        this.os.requestTransaction(function *() {
          var type = yield* value.createType.call(this)
          insert.opContent = type._model
          insert.id = this.store.getNextOpId()
          yield* this.applyCreatedOperations([insert])
          def.resolve(type)
        })
      } else {
        insert.content = value
        insert.id = this.os.getNextOpId()
        var eventHandler = this.eventHandler
        eventHandler.awaitAndPrematurelyCall([insert])

        this.os.requestTransaction(function *() {
          yield* this.applyCreatedOperations([insert])
          eventHandler.awaitedLastInserts(1)
        })
        def.resolve(value)
      }
      return def.promise
    }
    observe (f) {
      this.eventHandler.addUserEventListener(f)
    }
    unobserve (f) {
      this.eventHandler.removeUserEventListener(f)
    }
    observePath (path, f) {
      var self = this
      if (path.length === 0) {
        this.observe(f)
        return Promise.resolve(function () {
          self.unobserve(f)
        })
      } else {
        var deleteChildObservers
        var resetObserverPath = function () {
          var promise = self.get(path[0])
          if (!promise instanceof Promise) {
            // its either not defined or a premitive value
            promise = self.set(path[0], Y.Map)
          }
          return promise.then(function (map) {
            return map.observePath(path.slice(1), f)
          }).then(function (_deleteChildObservers) {
            deleteChildObservers = _deleteChildObservers
            return Promise.resolve()
          })
        }
        var observer = function (events) {
          for (var e in events) {
            var event = events[e]
            if (event.name === path[0]) {
              deleteChildObservers()
              if (event.type === 'add' || event.type === 'update') {
                resetObserverPath()
              }
            }
          }
        }
        self.observe(observer)
        return resetObserverPath().then(
          Promise.resolve(function () {
            deleteChildObservers()
            self.unobserve(observer)
          })
        )
      }
    }
    * _changed (transaction, op) {
      if (op.struct === 'Delete') {
        op.key = (yield* transaction.getOperation(op.target)).parentSub
      }
      this.eventHandler.receivedOp(op)
    }
  }
  Y.Map = new CustomType({
    class: YMap,
    createType: function * YMapCreator () {
      var model = {
        map: {},
        struct: 'Map',
        type: 'Map',
        id: this.store.getNextOpId()
      }
      yield* this.applyCreatedOperations([model])
      return yield* this.createType(model)
    },
    initType: function * YMapInitializer (os, model) { // eslint-disable-line
      return new YMap(os, model)
    }
  })
})()