meteor-ddp.js

/* MeteorDdp - a client for DDP version pre1 */
/* Lightly modified for Drupal */
var MeteorDdp = function(wsUri) {
  this.VERSIONS = ["pre1"];

  this.wsUri = wsUri;
  this.sock;
  this.defs = {};         // { deferred_id => deferred_object }
  this.subs = {};         // { pub_name => deferred_id }
  this.watchers = {};     // { coll_name => [cb1, cb2, ...] }
  this.collections = {};  // { coll_name => {docId => {doc}, docId => {doc}, ...} }
};

(function ($) {
  MeteorDdp.prototype._Ids = function() {
    var count = 0;
    return {
      next: function() {
        return ++count + '';
      }
    }
  }();
  
  MeteorDdp.prototype.connect = function() {
    var self = this;
    var conn = new $.Deferred();
    
    self.sock = new SockJS(self.wsUri);
    self.sock.onopen = function() {
      self.send({
        msg: 'connect',
        version: self.VERSIONS[0],
        support: self.VERSIONS
      });
    };

    self.sock.onerror = function(err) {
      conn.reject(err);
    };
  
    self.sock.onmessage = function(msg) {
      var data = JSON.parse(msg.data);
  
      //console.log(msg);
  
      switch (data.msg) {
        case 'connected':
          conn.resolve(data);
          break;
        case 'result':
          self._resolveCall(data);
          break;
        case 'updated':
          // TODO method call was acked
          break;
        case 'changed':
          self._changeDoc(data);
          break;
        case 'added':
          self._addDoc(data);
          break;
        case 'removed':
          self._removeDoc(data);
          break;
        case 'ready':
          self._resolveSubs(data);
          break;
        case 'nosub':
          self._resolveNoSub(data);
          break;
        case 'addedBefore':
          self._addDoc(data);
          break;
        case 'movedBefore':
          // TODO
          break;
      }
    };
    return conn.promise();
  };
  
  MeteorDdp.prototype._resolveNoSub = function(data) {
    if (data.error) {
      var error = data.error;
      this.defs[data.id].reject(error.reason || 'Subscription not found');
    } else {
      this.defs[data.id].resolve();
    }
  };
  
  MeteorDdp.prototype._resolveCall = function(data) {
    if (data.error) {
      this.defs[data.id].reject(data.error.reason);
    } else if (data.result) {
      this.defs[data.id].resolve(data.result);
    }
  };
  
  MeteorDdp.prototype._resolveSubs = function(data) {
    var subIds = data.subs;
    for (var i = 0; i < subIds.length; i++) {
      this.defs[subIds[i]].resolve();
    }
  };
  
  MeteorDdp.prototype._changeDoc = function(msg) {
    var collName = msg.collection;
    var id = msg.id;
    var fields = msg.fields;
    var cleared = msg.cleared;
    var coll = this.collections[collName];
  
    if (fields) {
      for (var k in fields) {
        coll[id][k] = fields[k];
      }
    } else if (cleared) {
      for (var i = 0; i < cleared.length; i++) {
        var fieldName = cleared[i];
        delete coll[id][fieldName];
      }
    }
  
    var changedDoc = coll[id];
    this._notifyWatchers(collName, changedDoc, id);
  };
  
  MeteorDdp.prototype._addDoc = function(msg) {
    var collName = msg.collection;
    var id = msg.id;
    if (!this.collections[collName]) {
      this.collections[collName] = {};
    }
    /* NOTE: Ordered docs will have a 'before' field containing the id of
     * the doc after it. If it is the last doc, it will be null.
     */
    this.collections[collName][id] = msg.fields;
  
    var changedDoc = this.collections[collName][id];
    this._notifyWatchers(collName, changedDoc, id);
  };
  
  MeteorDdp.prototype._removeDoc = function(msg) {
    var collName = msg.collection;
    var id = msg.id;
    var doc = this.collections[collName][id];
  
    var docCopy = JSON.parse(JSON.stringify(doc));
    docCopy.__wasDeleted = true;
    delete this.collections[collName][id];
    this._notifyWatchers(collName, docCopy, id);
  };
  
  MeteorDdp.prototype._notifyWatchers = function(collName, changedDoc, docId) {
    changedDoc = JSON.parse(JSON.stringify(changedDoc)); // make a copy      
    changedDoc._id = docId; // id might be useful to watchers, attach it.
  
    if (!this.watchers[collName]) {
      this.watchers[collName] = [];
    } else {
      for (var i = 0; i < this.watchers[collName].length; i++) {
        this.watchers[collName][i](changedDoc);
      }
    }
  };
  
  MeteorDdp.prototype._deferredSend = function(actionType, name, params) {
    var id = this._Ids.next();
    this.defs[id] = new $.Deferred();
  
    var args = params || [];
  
    var o = {
      msg: actionType,
      params: args,
      id: id
    };
  
    if (actionType === 'method') {
      o.method = name;
    } else if (actionType === 'sub') {
      o.name = name;
      this.subs[name] = id;
    }
  
    this.send(o);
    return this.defs[id].promise();
  };
  
  MeteorDdp.prototype.call = function(methodName, params) {
    return this._deferredSend('method', methodName, params);
  };
  
  MeteorDdp.prototype.subscribe = function(pubName, params) {
    return this._deferredSend('sub', pubName, params);
  };
  
  MeteorDdp.prototype.unsubscribe = function(pubName) {
    this.defs[id] = new $.Deferred();
    if (!this.subs[pubName]) {
      this.defs[id].reject(pubName + " was never subscribed");
    } else {
      var id = this.subs[pubName];
      var o = {
        msg: 'unsub',
        id: id
      };
      this.send(o);
    }
    return this.defs[id].promise();
  };
  
  MeteorDdp.prototype.watch = function(collectionName, cb) {
    if (!this.watchers[collectionName]) {
      this.watchers[collectionName] = [];
    }
    this.watchers[collectionName].push(cb);
  };
  
  MeteorDdp.prototype.getCollection = function(collectionName) {
    return this.collections[collectionName] || null;
  }
  
  MeteorDdp.prototype.getDocument = function(collectionName, docId) {
    return this.collections[collectionName][docId] || null;
  }
  
  MeteorDdp.prototype.send = function(msg) {
    this.sock.send(JSON.stringify(msg));
  };
  
  MeteorDdp.prototype.close = function() {
    this.sock.close();
  };

})(jQuery);