git ssb

1
var fs = require('fs')
2
var http = require('http')
3
var qs = require('querystring')
4
var path = require('path')
5
var crypto = require('crypto')
6
var pull = require('pull-stream')
7
var paramap = require('pull-paramap')
8
var sort = require('ssb-sort')
9
var toPull = require('stream-to-pull-stream')
10
var memo = require('asyncmemo')
11
var lru = require('lrucache')
12
var serveEmoji = require('emoji-server')()
13
var {
14
    MdRenderer,
15
    renderEmoji,
16
    formatMsgs,
17
    wrapPage,
18
    renderThread
19
} = require('./render');
20
21
var appHash = hash([fs.readFileSync(__filename)])
22
23
var urlIdRegex = /^(?:\/(([%&@]|%25|%26|%40)(?:[A-Za-z0-9\/+]|%2[Ff]|%2[Bb]){43}(?:=|%3[Dd])\.(?:sha256|ed25519))(?:\.([^?]*))?|(\/.*?))(?:\?(.*))?$/
24
25
function hash(arr) {
26
  return arr.reduce(function (hash, item) {
27
    return hash.update(String(item))
28
  }, crypto.createHash('sha256')).digest('base64')
29
}
30
31
exports.name = 'viewer'
32
exports.manifest = {}
33
exports.version = require('./package').version
34
35
exports.init = function (sbot, config) {
36
  var conf = config.viewer || {}
37
  var port = conf.port || 8807
38
  var host = conf.host || config.host || '::'
39
40
  var base = conf.base || '/'
41
  var defaultOpts = {
42
    base: base,
43
    msg_base: conf.msg_base || base,
44
    feed_base: conf.feed_base || base,
45
    blob_base: conf.blob_base || base,
46
    img_base: conf.img_base || base,
47
    emoji_base: conf.emoji_base || (base + 'emoji/'),
48
  }
49
50
  var getMsg = memo({cache: lru(100)}, getMsgWithValue, sbot)
51
  var getAbout = memo({cache: lru(100)}, require('./lib/about'), sbot)
52
53
  http.createServer(serve).listen(port, host, function () {
54
    console.log('[viewer] Listening on http://' + host + ':' + port)
55
  })
56
57
  function serve(req, res) {
58
    if (req.method !== 'GET' && req.method !== 'HEAD') {
59
      return respond(res, 405, 'Method must be GET or HEAD')
60
    }
61
62
    var m = urlIdRegex.exec(req.url)
63
64
    if (req.url.startsWith('/user-feed/')) return serveUserFeed(req, res, m[4])
65
    else if (req.url.startsWith('/channel/')) return serveChannel(req, res, m[4])
66
67
    if (m[2] && m[2].length === 3) {
68
      m[1] = decodeURIComponent(m[1])
69
      m[2] = m[1][0]
70
    }
71
    switch (m[2]) {
72
      case '%': return serveId(req, res, m[1], m[3], m[5])
73
      case '@': return serveFeed(req, res, m[1], m[3], m[5])
74
      case '&': return serveBlob(req, res, sbot, m[1])
75
      default: return servePath(req, res, m[4])
76
    }
77
  }
78
79
  function serveFeed(req, res, feedId) {
80
      console.log("serving feed: " + feedId)
81
82
      var opts = defaultOpts
83
84
      opts.marked = {
85
	  gfm: true,
86
	  mentions: true,
87
	  tables: true,
88
	  breaks: true,
89
	  pedantic: false,
90
	  sanitize: true,
91
	  smartLists: true,
92
	  smartypants: false,
93
	  emoji: renderEmoji,
94
	  renderer: new MdRenderer(opts)
95
      }
96
97
      getAbout(feedId, function (err, about) {
98
	  if (err) return cb(err)
99
100
	  pull(
101
	      sbot.createUserStream({ id: feedId, reverse: true }),
102
	      pull.collect(function (err, logs) {
103
		  if (err) return respond(res, 500, err.stack || err)
104
		  res.writeHead(200, {
105
		      'Content-Type': ctype("html")
106
		  })
107
		  pull(
108
		      pull.values(logs),
109
		      paramap(addAuthorAbout, 8),
110
		      paramap(addFollowAbout, 8),
111
		      paramap(addVoteMessage, 8),
112
		      pull(renderThread(opts), wrapPage(about.name)),
113
		      toPull(res, function (err) {
114
			  if (err) console.error('[viewer]', err)
115
		      })
116
		  )
117
	      })
118
	  )
119
      })
120
  }
121
122
  function serveUserFeed(req, res, url) {
123
      var feedId = url.substring(url.lastIndexOf('user-feed/')+10, 100)
124
      console.log("serving user feed: " + feedId)
125
126
      var following = []
127
      var channelSubscriptions = []
128
      
129
      getAbout(feedId, function (err, about) {
130
	  pull(
131
	      sbot.createUserStream({ id: feedId }),
132
	      pull.filter((msg) => {
133
		  return !msg.value ||
134
		      msg.value.content.type == 'contact' ||
135
		      (msg.value.content.type == 'channel' &&
136
		       typeof msg.value.content.subscribed != 'undefined')
137
	      }),
138
	      pull.collect(function (err, msgs) {
139
		  msgs.forEach((msg) => {
140
		      if (msg.value.content.type == 'contact')
141
		      {
142
			  if (msg.value.content.following)
143
			      following[msg.value.content.contact] = 1
144
			  else
145
			      delete following[msg.value.content.contact]
146
		      }
147
		      else // channel subscription
148
		      {
149
			  if (msg.value.content.subscribed)
150
			      channelSubscriptions[msg.value.content.channel] = 1
151
			  else
152
			      delete following[msg.value.content.channel]
153
		      }
154
		  })
155
		  
156
		  serveFeeds(req, res, following, channelSubscriptions, feedId, 'user feed ' + about.name)
157
	      })
158
	  )
159
      })
160
  }
161
162
  function serveFeeds(req, res, following, channelSubscriptions, feedId, name) {
163
      var opts = defaultOpts
164
      
165
      opts.marked = {
166
	  gfm: true,
167
	  mentions: true,
168
	  tables: true,
169
	  breaks: true,
170
	  pedantic: false,
171
	  sanitize: true,
172
	  smartLists: true,
173
	  smartypants: false,
174
	  emoji: renderEmoji,
175
	  renderer: new MdRenderer(opts)
176
      }
177
178
      pull(
179
	  sbot.createLogStream({ reverse: true, limit: 2500 }),
180
	  pull.filter((msg) => {
181
	      return !msg.value ||
182
		  (msg.value.author in following ||
183
		   msg.value.content.channel in channelSubscriptions)
184
	  }),
185
	  pull.take(100),
186
	  pull.collect(function (err, logs) {
187
	      if (err) return respond(res, 500, err.stack || err)
188
	      res.writeHead(200, {
189
		  'Content-Type': ctype("html")
190
	      })
191
	      pull(
192
		  pull.values(logs),
193
		  paramap(addAuthorAbout, 8),
194
		  paramap(addFollowAbout, 8),
195
		  paramap(addVoteMessage, 8),
196
		  pull(renderThread(opts), wrapPage(name)),
197
		  toPull(res, function (err) {
198
		      if (err) console.error('[viewer]', err)
199
		  })
200
	      )
201
	  })
202
      )
203
  }
204
205
  function serveChannel(req, res, url) {
206
      var channelId = url.substring(url.lastIndexOf('channel/')+8, 100)
207
      console.log("serving channel: " + channelId)
208
209
      var opts = defaultOpts
210
      
211
      opts.marked = {
212
	  gfm: true,
213
	  mentions: true,
214
	  tables: true,
215
	  breaks: true,
216
	  pedantic: false,
217
	  sanitize: true,
218
	  smartLists: true,
219
	  smartypants: false,
220
	  emoji: renderEmoji,
221
	  renderer: new MdRenderer(opts)
222
      }
223
224
      pull(
225
	  sbot.query.read({ limit: 500, reverse: true, query: [{$filter: { value: { content: { channel: channelId }}}}]}),
226
	  pull.collect(function (err, logs) {
227
	      if (err) return respond(res, 500, err.stack || err)
228
	      res.writeHead(200, {
229
		  'Content-Type': ctype("html")
230
	      })
231
	      pull(
232
		  pull.values(logs),
233
		  paramap(addAuthorAbout, 8),
234
		  paramap(addVoteMessage, 8),
235
		  pull(renderThread(opts), wrapPage('#' + channelId)),
236
		  toPull(res, function (err) {
237
		      if (err) console.error('[viewer]', err)
238
		  })
239
	      )
240
	  })
241
      )
242
  }
243
244
  function addFollowAbout(msg, cb) {
245
      if (msg.value.content.contact)
246
	  getAbout(msg.value.content.contact, function (err, about) {
247
	      if (err) return cb(err)
248
	      msg.value.content.contactAbout = about
249
	      cb(null, msg)
250
	  })
251
      else
252
	  cb(null, msg)
253
  }
254
255
  function addVoteMessage(msg, cb) {
256
      if (msg.value.content.type == 'vote' && msg.value.content.vote.link[0] == '%')
257
	  getMsg(msg.value.content.vote.link, function (err, linkedMsg) {
258
	      if (linkedMsg)
259
		  msg.value.content.vote.linkedText = linkedMsg.value.content.text
260
	      cb(null, msg)
261
	  })
262
      else
263
	  cb(null, msg)
264
  }
265
266
  function serveId(req, res, id, ext, query) {
267
    var q = query ? qs.parse(query) : {}
268
    var includeRoot = !('noroot' in q)
269
    var base = q.base || conf.base
270
    var baseToken
271
    if (!base) {
272
      if (ext === 'js') base = baseToken = '__BASE_' + Math.random() + '_'
273
      else base = '/'
274
    }
275
    var opts = {
276
      base: base,
277
      base_token: baseToken,
278
      msg_base: q.msg_base || conf.msg_base || base,
279
      feed_base: q.feed_base || conf.feed_base || base,
280
      blob_base: q.blob_base || conf.blob_base || base,
281
      img_base: q.img_base || conf.img_base || base,
282
      emoji_base: q.emoji_base || conf.emoji_base || (base + 'emoji/'),
283
    }
284
    opts.marked = {
285
      gfm: true,
286
      mentions: true,
287
      tables: true,
288
      breaks: true,
289
      pedantic: false,
290
      sanitize: true,
291
      smartLists: true,
292
      smartypants: false,
293
      emoji: renderEmoji,
294
      renderer: new MdRenderer(opts)
295
    }
296
297
    var format = formatMsgs(id, ext, opts)
298
    if (format === null) return respond(res, 415, 'Invalid format')
299
300
    pull(
301
      sbot.links({dest: id, values: true, rel: 'root'}),
302
      includeRoot && prepend(getMsg, id),
303
      pull.unique('key'),
304
      pull.collect(function (err, links) {
305
        if (err) return respond(res, 500, err.stack || err)
306
        var etag = hash(sort.heads(links).concat(appHash, ext, qs))
307
        if (req.headers['if-none-match'] === etag) return respond(res, 304)
308
        res.writeHead(200, {
309
          'Content-Type': ctype(ext),
310
          'etag': etag
311
        })
312
        pull(
313
          pull.values(sort(links)),
314
          paramap(addAuthorAbout, 8),
315
          format,
316
          toPull(res, function (err) {
317
            if (err) console.error('[viewer]', err)
318
          })
319
        )
320
      })
321
    )
322
  }
323
324
  function addAuthorAbout(msg, cb) {
325
    getAbout(msg.value.author, function (err, about) {
326
      if (err) return cb(err)
327
      msg.author = about
328
      cb(null, msg)
329
    })
330
  }
331
}
332
333
function serveBlob(req, res, sbot, id) {
334
  if (req.headers['if-none-match'] === id) return respond(res, 304)
335
  sbot.blobs.has(id, function (err, has) {
336
    if (err) {
337
      if (/^invalid/.test(err.message)) return respond(res, 400, err.message)
338
      else return respond(res, 500, err.message || err)
339
    }
340
    if (!has) return respond(res, 404, 'Not found')
341
    res.writeHead(200, {
342
      'Cache-Control': 'public, max-age=315360000',
343
      'etag': id
344
    })
345
    pull(
346
      sbot.blobs.get(id),
347
      toPull(res, function (err) {
348
        if (err) console.error('[viewer]', err)
349
      })
350
    )
351
  })
352
}
353
354
function getMsgWithValue(sbot, id, cb) {
355
  sbot.get(id, function (err, value) {
356
    if (err) return cb(err)
357
    cb(null, {key: id, value: value})
358
  })
359
}
360
361
function respond(res, status, message) {
362
  res.writeHead(status)
363
  res.end(message)
364
}
365
366
function ctype(name) {
367
  switch (name && /[^.\/]*$/.exec(name)[0] || 'html') {
368
    case 'html': return 'text/html'
369
    case 'js': return 'text/javascript'
370
    case 'css': return 'text/css'
371
    case 'json': return 'application/json'
372
  }
373
}
374
375
function servePath(req, res, url) {
376
  switch (url) {
377
    case '/robots.txt': return res.end('User-agent: *')
378
  }
379
  var m = /^(\/?[^\/]*)(\/.*)?$/.exec(url)
380
  switch (m[1]) {
381
    case '/static': return serveStatic(req, res, m[2])
382
    case '/emoji': return serveEmoji(req, res, m[2])
383
  }
384
  return respond(res, 404, 'Not found')
385
}
386
387
function ifModified(req, lastMod) {
388
  var ifModSince = req.headers['if-modified-since']
389
  if (!ifModSince) return false
390
  var d = new Date(ifModSince)
391
  return d && Math.floor(d/1000) >= Math.floor(lastMod/1000)
392
}
393
394
function serveStatic(req, res, file) {
395
  serveFile(req, res, path.join(__dirname, 'static', file))
396
}
397
398
function serveFile(req, res, file) {
399
  fs.stat(file, function (err, stat) {
400
    if (err && err.code === 'ENOENT') return respond(res, 404, 'Not found')
401
    if (err) return respond(res, 500, err.stack || err)
402
    if (!stat.isFile()) return respond(res, 403, 'May only load files')
403
    if (ifModified(req, stat.mtime)) return respond(res, 304, 'Not modified')
404
    res.writeHead(200, {
405
      'Content-Type': ctype(file),
406
      'Content-Length': stat.size,
407
      'Last-Modified': stat.mtime.toGMTString()
408
    })
409
    fs.createReadStream(file).pipe(res)
410
  })
411
}
412
413
function prepend(fn, arg) {
414
  return function (read) {
415
    return function (abort, cb) {
416
      if (fn && !abort) {
417
        var _fn = fn
418
        fn = null
419
        return _fn(arg, function (err, value) {
420
          if (err) return read(err, function (err) {
421
            cb(err || true)
422
          })
423
          cb(null, value)
424
        })
425
      }
426
      read(abort, cb)
427
    }
428
  }
429
}
430