youbase/lib/collection.js

var Collection, Definition, HARDENED_OFFSET, HDKey, bs, defer, deferObject, exports;

bs = require('bs58check');

defer = require('when');

deferObject = require('when/keys');

HDKey = require('hdkey');

Definition = require('./definition');

HARDENED_OFFSET = 0x80000000;

Collection = (function() {
  function Collection(custodian, model, key, hardened) {
    this.custodian = custodian;
    this.model = model;
    this.hardened = hardened != null ? hardened : false;
    if (!(this instanceof Collection)) {
      return new Collection(this.custodian, this.model, key, this.hardened);
    }
    if (typeof key === 'string') {
      key = bs.decode(key);
    }
    this.hdkey = HDKey.fromExtendedKey(bs.encode(key));
    this.privateKey = this.hdkey.privateKey;
    this.publicKey = this.hdkey.publicKey;
    this.readonly = this.privateKey == null;
    if (this.hardened) {
      this.offset = HARDENED_OFFSET;
    } else {
      this.offset = 0;
    }
    this._definitions = {};
    this._documents = [];
  }

  Collection.prototype.definition = function(key, definition) {
    if (definition != null) {
      return Definition(this.custodian, definition).save().then((function(_this) {
        return function(hash) {
          return _this._definitions[key] = bs.encode(hash);
        };
      })(this));
    } else if (key != null) {
      if (typeof key === 'string' && key.length <= 32) {
        key = this._definitions[key];
      }
      return Definition(this.custodian, key);
    } else {
      return deferObject.map(this._definitions, (function(_this) {
        return function(definition) {
          return Definition(_this.custodian, definition);
        };
      })(this));
    }
  };

  Collection.prototype.insert = function(definition, data, autosave) {
    if (autosave == null) {
      autosave = true;
    }
    if (this.readonly) {
      return false;
    }
    return this.sync().then((function(_this) {
      return function() {
        var document;
        document = _this.at();
        return document.definition(_this.definition(definition)).then(function() {
          return document.data(data);
        }).then(function() {
          if (autosave) {
            return document.save();
          }
        }).then(function() {
          return _this._documents[document.hdkey.index - _this.offset] = document;
        });
      };
    })(this));
  };

  Collection.prototype.at = function(index) {
    var document, hdkey, key, _ref;
    if (index == null) {
      index = this._documents.length + this.offset;
    }
    document = this._documents[index - this.offset];
    if (document != null) {
      return document;
    }
    hdkey = this.hdkey.deriveChild(index);
    key = (_ref = hdkey.privateExtendedKey) != null ? _ref : hdkey.publicExtendedKey;
    return new this.model(this.custodian, key);
  };

  Collection.prototype.all = function(refresh, pluck) {
    if (typeof refresh === 'string') {
      pluck = refresh;
      refresh = false;
    }
    return this.sync(refresh).then((function(_this) {
      return function() {
        console.log('all', _this._documents.length);
        if (pluck) {
          return defer.map(_this._documents, function(d) {
            return typeof d[pluck] === "function" ? d[pluck]() : void 0;
          });
        } else {
          return _this._documents;
        }
      };
    })(this));
  };

  Collection.prototype.sync = function(refresh) {
    var start;
    if (refresh == null) {
      refresh = false;
    }
    start = this.offset;
    if (!refresh) {
      start = start + this._documents.length;
    }
    return defer.iterate((function(_this) {
      return function(document) {
        return _this.at(document.hdkey.index + 1);
      };
    })(this), (function(_this) {
      return function(document) {
        return document.fetch()["yield"](false)["else"](true);
      };
    })(this), (function(_this) {
      return function(document) {
        return _this._documents[document.hdkey.index - _this.offset] = document;
      };
    })(this), this.at(start)).then((function(_this) {
      return function() {
        return _this;
      };
    })(this));
  };

  return Collection;

})();

exports = module.exports = Collection;