Files: 4f7d162bf1625d6dc4e8e518f55ee5d842d12bc2 / index.js
11414 bytesRaw
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 | defaultOpts.marked = { |
51 | gfm: true, |
52 | mentions: true, |
53 | tables: true, |
54 | breaks: true, |
55 | pedantic: false, |
56 | sanitize: true, |
57 | smartLists: true, |
58 | smartypants: false, |
59 | emoji: renderEmoji, |
60 | renderer: new MdRenderer(defaultOpts) |
61 | } |
62 | |
63 | var getMsg = memo({cache: lru(100)}, getMsgWithValue, sbot) |
64 | var getAbout = memo({cache: lru(100)}, require('./lib/about'), sbot) |
65 | |
66 | http.createServer(serve).listen(port, host, function () { |
67 | console.log('[viewer] Listening on http://' + host + ':' + port) |
68 | }) |
69 | |
70 | function serve(req, res) { |
71 | if (req.method !== 'GET' && req.method !== 'HEAD') { |
72 | return respond(res, 405, 'Method must be GET or HEAD') |
73 | } |
74 | |
75 | var m = urlIdRegex.exec(req.url) |
76 | |
77 | if (req.url.startsWith('/user-feed/')) return serveUserFeed(req, res, m[4]) |
78 | else if (req.url.startsWith('/channel/')) return serveChannel(req, res, m[4]) |
79 | |
80 | if (m[2] && m[2].length === 3) { |
81 | m[1] = decodeURIComponent(m[1]) |
82 | m[2] = m[1][0] |
83 | } |
84 | switch (m[2]) { |
85 | case '%': return serveId(req, res, m[1], m[3], m[5]) |
86 | case '@': return serveFeed(req, res, m[1], m[3], m[5]) |
87 | case '&': return serveBlob(req, res, sbot, m[1]) |
88 | default: return servePath(req, res, m[4]) |
89 | } |
90 | } |
91 | |
92 | function serveFeed(req, res, feedId) { |
93 | console.log("serving feed: " + feedId) |
94 | |
95 | var opts = defaultOpts |
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 | pull( |
166 | sbot.createLogStream({ reverse: true, limit: 2500 }), |
167 | pull.filter((msg) => { |
168 | return !msg.value || |
169 | (msg.value.author in following || |
170 | msg.value.content.channel in channelSubscriptions) |
171 | }), |
172 | pull.take(100), |
173 | pull.collect(function (err, logs) { |
174 | if (err) return respond(res, 500, err.stack || err) |
175 | res.writeHead(200, { |
176 | 'Content-Type': ctype("html") |
177 | }) |
178 | pull( |
179 | pull.values(logs), |
180 | paramap(addAuthorAbout, 8), |
181 | paramap(addFollowAbout, 8), |
182 | paramap(addVoteMessage, 8), |
183 | pull(renderThread(opts), wrapPage(name)), |
184 | toPull(res, function (err) { |
185 | if (err) console.error('[viewer]', err) |
186 | }) |
187 | ) |
188 | }) |
189 | ) |
190 | } |
191 | |
192 | function serveChannel(req, res, url) { |
193 | var channelId = url.substring(url.lastIndexOf('channel/')+8, 100) |
194 | console.log("serving channel: " + channelId) |
195 | |
196 | var opts = defaultOpts |
197 | |
198 | pull( |
199 | sbot.query.read({ limit: 500, reverse: true, query: [{$filter: { value: { content: { channel: channelId }}}}]}), |
200 | pull.collect(function (err, logs) { |
201 | if (err) return respond(res, 500, err.stack || err) |
202 | res.writeHead(200, { |
203 | 'Content-Type': ctype("html") |
204 | }) |
205 | pull( |
206 | pull.values(logs), |
207 | paramap(addAuthorAbout, 8), |
208 | paramap(addVoteMessage, 8), |
209 | pull(renderThread(opts), wrapPage('#' + channelId)), |
210 | toPull(res, function (err) { |
211 | if (err) console.error('[viewer]', err) |
212 | }) |
213 | ) |
214 | }) |
215 | ) |
216 | } |
217 | |
218 | function addFollowAbout(msg, cb) { |
219 | if (msg.value.content.contact) |
220 | getAbout(msg.value.content.contact, function (err, about) { |
221 | if (err) return cb(err) |
222 | msg.value.content.contactAbout = about |
223 | cb(null, msg) |
224 | }) |
225 | else |
226 | cb(null, msg) |
227 | } |
228 | |
229 | function addVoteMessage(msg, cb) { |
230 | if (msg.value.content.type == 'vote' && msg.value.content.vote.link[0] == '%') |
231 | getMsg(msg.value.content.vote.link, function (err, linkedMsg) { |
232 | if (linkedMsg) |
233 | msg.value.content.vote.linkedText = linkedMsg.value.content.text |
234 | cb(null, msg) |
235 | }) |
236 | else |
237 | cb(null, msg) |
238 | } |
239 | |
240 | function serveId(req, res, id, ext, query) { |
241 | var q = query ? qs.parse(query) : {} |
242 | var includeRoot = !('noroot' in q) |
243 | var base = q.base || conf.base |
244 | var baseToken |
245 | if (!base) { |
246 | if (ext === 'js') base = baseToken = '__BASE_' + Math.random() + '_' |
247 | else base = '/' |
248 | } |
249 | var opts = { |
250 | base: base, |
251 | base_token: baseToken, |
252 | msg_base: q.msg_base || conf.msg_base || base, |
253 | feed_base: q.feed_base || conf.feed_base || base, |
254 | blob_base: q.blob_base || conf.blob_base || base, |
255 | img_base: q.img_base || conf.img_base || base, |
256 | emoji_base: q.emoji_base || conf.emoji_base || (base + 'emoji/'), |
257 | } |
258 | opts.marked = { |
259 | gfm: true, |
260 | mentions: true, |
261 | tables: true, |
262 | breaks: true, |
263 | pedantic: false, |
264 | sanitize: true, |
265 | smartLists: true, |
266 | smartypants: false, |
267 | emoji: renderEmoji, |
268 | renderer: new MdRenderer(opts) |
269 | } |
270 | |
271 | var format = formatMsgs(id, ext, opts) |
272 | if (format === null) return respond(res, 415, 'Invalid format') |
273 | |
274 | pull( |
275 | sbot.links({dest: id, values: true, rel: 'root'}), |
276 | includeRoot && prepend(getMsg, id), |
277 | pull.unique('key'), |
278 | pull.collect(function (err, links) { |
279 | if (err) return respond(res, 500, err.stack || err) |
280 | var etag = hash(sort.heads(links).concat(appHash, ext, qs)) |
281 | if (req.headers['if-none-match'] === etag) return respond(res, 304) |
282 | res.writeHead(200, { |
283 | 'Content-Type': ctype(ext), |
284 | 'etag': etag |
285 | }) |
286 | pull( |
287 | pull.values(sort(links)), |
288 | paramap(addAuthorAbout, 8), |
289 | format, |
290 | toPull(res, function (err) { |
291 | if (err) console.error('[viewer]', err) |
292 | }) |
293 | ) |
294 | }) |
295 | ) |
296 | } |
297 | |
298 | function addAuthorAbout(msg, cb) { |
299 | getAbout(msg.value.author, function (err, about) { |
300 | if (err) return cb(err) |
301 | msg.author = about |
302 | cb(null, msg) |
303 | }) |
304 | } |
305 | } |
306 | |
307 | function serveBlob(req, res, sbot, id) { |
308 | if (req.headers['if-none-match'] === id) return respond(res, 304) |
309 | sbot.blobs.has(id, function (err, has) { |
310 | if (err) { |
311 | if (/^invalid/.test(err.message)) return respond(res, 400, err.message) |
312 | else return respond(res, 500, err.message || err) |
313 | } |
314 | if (!has) return respond(res, 404, 'Not found') |
315 | res.writeHead(200, { |
316 | 'Cache-Control': 'public, max-age=315360000', |
317 | 'etag': id |
318 | }) |
319 | pull( |
320 | sbot.blobs.get(id), |
321 | toPull(res, function (err) { |
322 | if (err) console.error('[viewer]', err) |
323 | }) |
324 | ) |
325 | }) |
326 | } |
327 | |
328 | function getMsgWithValue(sbot, id, cb) { |
329 | sbot.get(id, function (err, value) { |
330 | if (err) return cb(err) |
331 | cb(null, {key: id, value: value}) |
332 | }) |
333 | } |
334 | |
335 | function respond(res, status, message) { |
336 | res.writeHead(status) |
337 | res.end(message) |
338 | } |
339 | |
340 | function ctype(name) { |
341 | switch (name && /[^.\/]*$/.exec(name)[0] || 'html') { |
342 | case 'html': return 'text/html' |
343 | case 'js': return 'text/javascript' |
344 | case 'css': return 'text/css' |
345 | case 'json': return 'application/json' |
346 | } |
347 | } |
348 | |
349 | function servePath(req, res, url) { |
350 | switch (url) { |
351 | case '/robots.txt': return res.end('User-agent: *') |
352 | } |
353 | var m = /^(\/?[^\/]*)(\/.*)?$/.exec(url) |
354 | switch (m[1]) { |
355 | case '/static': return serveStatic(req, res, m[2]) |
356 | case '/emoji': return serveEmoji(req, res, m[2]) |
357 | } |
358 | return respond(res, 404, 'Not found') |
359 | } |
360 | |
361 | function ifModified(req, lastMod) { |
362 | var ifModSince = req.headers['if-modified-since'] |
363 | if (!ifModSince) return false |
364 | var d = new Date(ifModSince) |
365 | return d && Math.floor(d/1000) >= Math.floor(lastMod/1000) |
366 | } |
367 | |
368 | function serveStatic(req, res, file) { |
369 | serveFile(req, res, path.join(__dirname, 'static', file)) |
370 | } |
371 | |
372 | function serveFile(req, res, file) { |
373 | fs.stat(file, function (err, stat) { |
374 | if (err && err.code === 'ENOENT') return respond(res, 404, 'Not found') |
375 | if (err) return respond(res, 500, err.stack || err) |
376 | if (!stat.isFile()) return respond(res, 403, 'May only load files') |
377 | if (ifModified(req, stat.mtime)) return respond(res, 304, 'Not modified') |
378 | res.writeHead(200, { |
379 | 'Content-Type': ctype(file), |
380 | 'Content-Length': stat.size, |
381 | 'Last-Modified': stat.mtime.toGMTString() |
382 | }) |
383 | fs.createReadStream(file).pipe(res) |
384 | }) |
385 | } |
386 | |
387 | function prepend(fn, arg) { |
388 | return function (read) { |
389 | return function (abort, cb) { |
390 | if (fn && !abort) { |
391 | var _fn = fn |
392 | fn = null |
393 | return _fn(arg, function (err, value) { |
394 | if (err) return read(err, function (err) { |
395 | cb(err || true) |
396 | }) |
397 | cb(null, value) |
398 | }) |
399 | } |
400 | read(abort, cb) |
401 | } |
402 | } |
403 | } |
404 |
Built with git-ssb-web