1 files changed, 71 insertions, 0 deletions
diff --git a/lib/app.js b/lib/app.js
index d86b4b9..d1f76f3 100644
--- a/lib/app.js
+++ b/lib/app.js
@@ -172,6 +172,17 @@ App.prototype.publish = function (content, cb) {
   tryPublish(2)
 }
 
+App.prototype.wantSizeBlob = function (id, cb) {
+  var blobs = this.sbot.blobs
+  blobs.size(id, function (err, size) {
+    if (size != null) return cb(null, size)
+    blobs.want(id, function (err) {
+      if (err) return cb(err)
+      blobs.size(id, cb)
+    })
+  })
+}
+
 App.prototype.addBlob = function (cb) {
   var done = multicb({pluck: 1, spread: true})
   var hashCb = done()
@@ -378,6 +389,66 @@ App.prototype.createContactStreams = function (id) {
   return new Contacts(this.sbot).createContactStreams(id)
 }
 
+function compareVoted(a, b) {
+  return b.value - a.value
+}
+
+App.prototype.getVoted = function (_opts, cb) {
+  if (isNaN(_opts.limit)) return pull.error(new Error('missing limit'))
+  var self = this
+  var opts = {
+    type: 'vote',
+    limit: _opts.limit * 100,
+    reverse: !!_opts.reverse,
+    gt: _opts.gt || undefined,
+    lt: _opts.lt || undefined,
+  }
+
+  var votedObj = {}
+  var votedArray = []
+  var numItems = 0
+  var firstTimestamp, lastTimestamp
+  pull(
+    self.sbot.messagesByType(opts),
+    self.unboxMessages(),
+    pull.take(function () {
+      return numItems < _opts.limit
+    }),
+    pull.drain(function (msg) {
+      if (!firstTimestamp) firstTimestamp = msg.timestamp
+      lastTimestamp = msg.timestamp
+      var vote = msg.value.content.vote
+      if (!vote) return
+      var target = u.linkDest(vote)
+      var votes = votedObj[target]
+      if (!votes) {
+        numItems++
+        votes = {id: target, value: 0, feedsObj: {}, feeds: []}
+        votedObj[target] = votes
+        votedArray.push(votes)
+      }
+      if (msg.value.author in votes.feedsObj) {
+        if (!opts.reverse) return // leave latest vote value as-is
+        // remove old vote value
+        votes.value -= votes.feedsObj[msg.value.author]
+      } else {
+        votes.feeds.push(msg.value.author)
+      }
+      var value = vote.value > 0 ? 1 : vote.value < 0 ? -1 : 0
+      votes.feedsObj[msg.value.author] = value
+      votes.value += value
+    }, function (err) {
+      if (err) return cb(err)
+      var items = votedArray
+      if (opts.reverse) items.reverse()
+      items.sort(compareVoted)
+      cb(null, {items: items,
+        firstTimestamp: firstTimestamp,
+        lastTimestamp: lastTimestamp})
+    })
+  )
+}
+
 App.prototype.createAboutStreams = function (id) {
   return this.about.createAboutStreams(id)
 }