git ssb

1
const FlumeView = require('flumeview-level')
2
const get = require('lodash/get')
3
const pull = require('pull-stream')
4
const defer = require('pull-defer')
5
const isBlog = require('scuttle-blog/isBlog')
6
const { isMsg: isMsgRef } = require('ssb-ref')
7
8
const getType = (msg) => get(msg, 'value.content.type')
9
const getAuthor = (msg) => get(msg, 'value.author')
10
const getCommentRoot = (msg) => get(msg, 'value.content.root')
11
const getLikeRoot = (msg) => get(msg, 'value.content.vote.link')
12
const getTimestamp = (msg) => get(msg, 'value.timestamp')
13
14
const FLUME_VIEW_VERSION = 1
15
16
module.exports = {
17
  name: 'blogStats',
18
  version: 1,
19
  manifest: {
20
    get: 'async',
21
    read: 'source',
22
    readBlogs: 'source',
23
    getBlogs: 'async',
24
    readComments: 'source',
25
    readAllComments: 'source', // TEMP
26
    readLikes: 'source'
27
  },
28
  init: (server, config) => {
29
    console.log('> initialising blog-stats plugin')
30
    const myKey = server.keys.id
31
32
    const view = server._flumeUse(
33
      'internalblogStats',
34
      FlumeView(FLUME_VIEW_VERSION, map)
35
    )
36
37
    return {
38
      get: view.get,
39
      read: view.read,
40
      readBlogs,
41
      getBlogs,
42
      readComments,
43
      readAllComments,
44
      readLikes
45
      // readShares
46
    }
47
48
    function map (msg, seq) {
49
      var root
50
51
      switch (getType(msg)) {
52
        case 'blog':
53
          if (isBlog(msg) && isMyMsg(msg)) return [['B', msg.key, getTimestamp(msg)]]
54
          else return []
55
56
        case 'vote':
57
          root = getLikeRoot(msg)
58
          // TODO figure out how to only store likes I care about
59
          if (root) return [['L', root, getTimestamp(msg)]]
60
          else return []
61
62
          // Note this catches:
63
          //   - all likes, on all things D:
64
          //   - likes AND unlikes
65
66
        case 'post':
67
          root = getCommentRoot(msg)
68
          // TODO figure out how to only store comments I care about
69
          if (!root && isMyMsg(msg) && isPlog(msg)) return [['B', msg.key, getTimestamp(msg)]]
70
          else if (root) return [['C', root, getTimestamp(msg)]]
71
          else return []
72
73
          // Note this catches:
74
          //   - all comments, on all things D:
75
76
        default:
77
          return []
78
      }
79
    }
80
81
    // a Plog is a Blog shaped Post!
82
    function isPlog (msg) {
83
      // if (get(msg, 'value.content.text', '').length >= 2500) console.log(get(msg, 'value.content.text', '').length)
84
      return get(msg, 'value.content.text', '').length >= 2500
85
    }
86
87
    function readBlogs (options = {}) {
88
      const query = Object.assign({}, {
89
        gte: ['B', null, null],
90
        // null is the 'minimum' structure in bytewise ordering
91
        lte: ['B', undefined, undefined],
92
        reverse: true,
93
        values: true,
94
        keys: false,
95
        seqs: false
96
      }, options)
97
98
      return view.read(query)
99
    }
100
101
    function getBlogs (options, cb) {
102
      if (typeof options === 'function') {
103
        cb = options
104
        options = {}
105
      }
106
107
      pull(
108
        readBlogs(options),
109
        pull.collect(cb)
110
      )
111
    }
112
113
    function readComments (blog, options = {}) {
114
      var key = getBlogKey(blog)
115
116
      const query = Object.assign({}, {
117
        gt: ['C', key, null],
118
        lt: ['C', key, undefined],
119
        // undefined is the 'maximum' structure in bytewise ordering https://www.npmjs.com/package/bytewise#order-of-supported-structures
120
        reverse: true,
121
        values: true,
122
        keys: false,
123
        seqs: false
124
      }, options)
125
126
      return view.read(query)
127
    }
128
129
    function readAllComments () {
130
      var source = defer.source()
131
132
      getBlogs({ keys: true, values: false }, (err, data) => {
133
        if (err) throw err
134
135
        const blogIds = data.map(d => d[1])
136
137
        const _source = pull(
138
          view.read({
139
            // live: true,
140
            gt: [ 'C', null, null ],
141
            lt: [ 'C', undefined, undefined ],
142
            reverse: true,
143
            values: true,
144
            keys: true,
145
            seqs: false
146
          }),
147
          pull.filter(result => {
148
            return blogIds.includes(result.key[1])
149
          }),
150
          pull.map(result => result.value)
151
        )
152
153
        source.resolve(_source)
154
      })
155
156
      return pull(
157
        source
158
      )
159
    }
160
161
    function readLikes (blog, options = {}) {
162
      var key = getBlogKey(blog)
163
164
      const query = Object.assign({}, {
165
        gt: ['L', key, null],
166
        lt: ['L', key, undefined],
167
        reverse: true,
168
        values: true,
169
        keys: false,
170
        seqs: false
171
      }, options)
172
173
      return view.read(query)
174
    }
175
176
    function getBlogKey (blog) {
177
      if (isMsgRef(blog)) return blog
178
      // else if (isMsgRef(blog.key) && isBlog(blog)) return blog.key
179
      else if (isMsgRef(blog.key) && (isBlog(blog) || isPlog(blog))) return blog.key
180
    }
181
182
    function isMyMsg (msg) {
183
      return getAuthor(msg) === myKey
184
    }
185
  }
186
}
187