bad-example.com
demos for spacedust
share a spacedust websocket
+122
-32
+122
-32
server/index.js
+122
-32
server/index.js
···
1
1
#!/usr/bin/env node
2
+
"use strict";
2
3
3
4
const webpush = require('web-push');
4
5
const fs = require('node:fs');
5
6
const http = require('http');
6
7
8
+
const DUMMY_DID = 'did:plc:zzzzzzzzzzzzzzzzzzzzzzzz';
9
+
10
+
let spacedust;
11
+
let spacedustEverStarted = false;
12
+
const subs = new Map();
13
+
14
+
const addSub = (did, sub) => {
15
+
if (!subs.has(did)) {
16
+
subs.set(did, []);
17
+
}
18
+
subs.get(did).push(sub);
19
+
updateSubs();
20
+
};
21
+
22
+
const updateSubs = () => {
23
+
if (!spacedust) {
24
+
console.warn('not updating subscription, no spacedust (reconnecting?)');
25
+
return;
26
+
}
27
+
const wantedSubjectDids = Array.from(subs.keys());
28
+
if (wantedSubjectDids.length === 0) {
29
+
wantedSubjectDids.push(DUMMY_DID);
30
+
}
31
+
console.log('updating for wantedSubjectDids', wantedSubjectDids);
32
+
spacedust.send(JSON.stringify({
33
+
type: 'options_update',
34
+
payload: {
35
+
wantedSubjectDids,
36
+
},
37
+
}));
38
+
};
39
+
40
+
const handleDust = async event => {
41
+
console.log('got', event.data);
42
+
let data;
43
+
try {
44
+
data = JSON.parse(event.data);
45
+
} catch (err) {
46
+
console.error(err);
47
+
return;
48
+
}
49
+
const { link: { subject, source, source_record } } = data;
50
+
51
+
let did;
52
+
if (subject.startsWith('did:')) did = subject;
53
+
else if (subject.startsWith('at://')) {
54
+
const [id, ..._] = subject.slice('at://'.length).split('/');
55
+
if (id.startsWith('did:')) did = id;
56
+
}
57
+
if (!did) {
58
+
console.warn(`ignoring link with non-DID subject: ${subject}`)
59
+
return;
60
+
}
61
+
62
+
const expiredSubs = [];
63
+
for (const sub of subs.get(did) ?? []) {
64
+
const title = `new ${source}`;
65
+
const body = `from ${source_record}`;
66
+
try {
67
+
await webpush.sendNotification(sub, JSON.stringify({ title, body }));
68
+
} catch (err) {
69
+
if (400 <= err.statusCode && err.statusCode < 500) {
70
+
expiredSubs.push(sub);
71
+
console.info(`removing sub for ${err.statusCode}`);
72
+
}
73
+
}
74
+
}
75
+
if (expiredSubs.length > 0) {
76
+
const activeSubs = subs.get(did)?.filter(s => !expiredSubs.includes(s));
77
+
if (!activeSubs) { // concurrently removed already
78
+
return;
79
+
}
80
+
if (activeSubs.length === 0) {
81
+
console.info('removed last subscriber for', did);
82
+
subs.delete(did);
83
+
updateSubs();
84
+
} else {
85
+
subs.set(did, activeSubs);
86
+
}
87
+
}
88
+
};
89
+
90
+
const connectSpacedust = host => {
91
+
spacedust = new WebSocket(`${host}/subscribe?instant=true&wantedSubjectDids=${DUMMY_DID}`);
92
+
let restarting = false;
93
+
94
+
const restart = () => {
95
+
if (restarting) return;
96
+
restarting = true;
97
+
let wait = Math.round(500 + (Math.random() * 1000));
98
+
console.info(`restarting spacedust connection in ${wait}ms...`);
99
+
setTimeout(() => connectSpacedust(host), wait);
100
+
spacedust = null;
101
+
}
102
+
103
+
spacedust.onopen = updateSubs
104
+
spacedust.onmessage = handleDust;
105
+
106
+
spacedust.onerror = e => {
107
+
console.error('spacedust errored:', e);
108
+
restart();
109
+
};
110
+
111
+
spacedust.onclose = () => {
112
+
console.log('spacedust closed');
113
+
restart();
114
+
};
115
+
}
116
+
117
+
const subscribeSpacedust = (did, sub) => {
118
+
if (!subs.has(did)) {
119
+
subs.set(did, []);
120
+
}
121
+
}
122
+
7
123
const getOrCreateKeys = filename => {
8
124
let keys;
9
125
try {
···
50
166
const handleSubscribe = async (req, res) => {
51
167
const body = await getRequesBody(req);
52
168
const { did, sub } = JSON.parse(body);
53
-
doStuff(did, sub);
169
+
addSub(did, sub);
54
170
res.setHeader('Content-Type', 'application/json');
55
171
res.writeHead(201);
56
172
res.end('{"oh": "hi"}');
57
173
}
58
174
59
-
const doStuff = (did, sub) => {
60
-
console.log('subscribing for', did);
61
-
const ws = new WebSocket(`wss://spacedust.microcosm.blue/subscribe?instant=true&wantedSubjectDids=${did}`);
62
-
63
-
ws.addEventListener('message', event => {
64
-
console.log('got', event.data);
65
-
let data;
66
-
try {
67
-
data = JSON.parse(event.data);
68
-
} catch (err) {
69
-
console.error(err);
70
-
return;
71
-
}
72
-
const { link: { source, source_record } } = data;
73
-
const title = `new ${source}`;
74
-
const body = `from ${source_record}`;
75
-
webpush.sendNotification(sub, JSON.stringify({ title, body }));
76
-
});
77
-
78
-
ws.addEventListener('error', err => {
79
-
console.log('uh oh', err);
80
-
});
81
-
82
-
ws.addEventListener('close', () => {
83
-
console.log('closed. bye!');
84
-
});
85
-
86
-
}
87
-
88
175
const requestListener = pubkey => (req, res) => {
89
176
if (req.method === 'GET' && req.url === '/')
90
177
return handleIndex(req, res, { PUBKEY: pubkey });
···
108
195
keys.privateKey,
109
196
);
110
197
111
-
const host = env.HOST || 'localhost';
112
-
const port = parseInt(env.PORT || 8000, 10);
198
+
const spacedustHost = env.SPACEDUST_HOST ?? 'wss://spacedust.microcosm.blue';
199
+
connectSpacedust(spacedustHost);
200
+
201
+
const host = env.HOST ?? 'localhost';
202
+
const port = parseInt(env.PORT ?? 8000, 10);
113
203
114
204
http
115
205
.createServer(requestListener(keys.publicKey))