git ssb

1
var pull = require('pull-stream')
2
var paramap = require('pull-paramap')
3
var asyncMemo = require('asyncmemo')
4
var Issues = require('ssb-issues')
5
6
function readNext(fn) {
7
  var next
8
  return function (end, cb) {
9
    if (next) return next(end, cb)
10
    fn(function (err, _next) {
11
      if (err) return cb(err)
12
      next = _next
13
      next(null, cb)
14
    })
15
  }
16
}
17
18
module.exports = {
19
  name: 'pull-requests',
20
  version: '1.0.0',
21
  manifest: {
22
    get: 'async',
23
    getRevs: 'async',
24
    list: 'source'
25
  },
26
  schemas: require('./lib/schemas'),
27
  init: function (sbot) {
28
29
    var issues = Issues.init(sbot)
30
    var getMsg = asyncMemo(sbot.get)
31
32
    var getPullReq = asyncMemo(function (id, cb) {
33
      issues.get(id, function (err, issue) {
34
        if (err) return cb(err)
35
        var c = issue.msg.value.content
36
        issue.baseRepo = c.repo
37
        issue.baseBranch = c.branch
38
        issue.headRepo = c.head_repo
39
        issue.headBranch = c.head_branch
40
        cb(null, issue)
41
      })
42
    })
43
44
    function getBranchLastUpdate(repoId, branch, lte, cb) {
45
      // TODO: detect and skip updates from subsequent PR with same branch name
46
      getMsg(repoId, function (err, msg) {
47
        if (err) return cb(err)
48
        var repoAuthor = msg.author
49
        pull(
50
          sbot.links({
51
            dest: repoId,
52
            source: repoAuthor,
53
            rel: 'repo',
54
            values: true,
55
            reverse: true
56
          }),
57
          pull.filter(function (link) {
58
            return link.value.content.type == 'git-update'
59
          }),
60
          pull.map(function (link) {
61
            return {
62
              timestamp: link.value.timestamp,
63
              rev: (link.value.content.refs || {})['refs/heads/' + branch]
64
            }
65
          }),
66
          pull.filter(function (update) {
67
            return update.rev
68
              && (lte ? update.timestamp <= lte : true)
69
          }),
70
          pull.take(1)
71
        )(null, cb)
72
      })
73
    }
74
75
    function getRevs(prId, cb) {
76
      getPullReq(prId, function (err, pr) {
77
        if (err) return cb(err)
78
        // get the latest rev of the head branch before it was deleted or the
79
        // PR closed
80
        var lastTime = pr.open ? null : pr.updated_at
81
        getBranchLastUpdate(pr.headRepo, pr.headBranch, lastTime,
82
          function (err, headUpdate) {
83
            if (err) return cb(err)
84
            // get the rev of base when head was last updated
85
            getBranchLastUpdate(pr.baseRepo, pr.baseBranch,
86
              headUpdate.timestamp, function (err, baseUpdate) {
87
                if (err) return cb(err)
88
                cb(null, {base: baseUpdate.rev, head: headUpdate.rev})
89
              })
90
          })
91
      })
92
    }
93
94
    function listPullReqs(opts) {
95
      opts = opts || {}
96
      opts.type = 'pull-request'
97
      return pull(
98
        sbot.messagesByType(opts),
99
        pull.filter(function (msg) {
100
          return (!opts.repo || opts.repo == msg.value.content.repo)
101
            && (!opts.author || opts.author == msg.value.author)
102
        }),
103
        paramap(function (msg, cb) {
104
          getPullReq(msg.key, cb)
105
        }, 8),
106
        pull.filter(opts.open != null && function (pr) {
107
          return pr.open == opts.open
108
        })
109
      )
110
    }
111
112
    return {
113
      get: getPullReq,
114
      getRevs: getRevs,
115
      list: listPullReqs
116
    }
117
  }
118
}
119