remove hanging if no connection

This commit is contained in:
Richard Bondi
2021-01-18 22:01:30 -06:00
committed by fiatjaf
parent 63259eb7c3
commit 6586a1b85d

138
relay.js
View File

@@ -11,59 +11,83 @@ export function normalizeRelayURL(url) {
export function relayConnect(url, onEvent, onNotice) { export function relayConnect(url, onEvent, onNotice) {
url = normalizeRelayURL(url) url = normalizeRelayURL(url)
const ws = new WebSocket( let ws, resolveOpen, untilOpen, rejectOpen
url + (url.indexOf('?') !== -1 ? '&' : '?') + `session=${Math.random()}`,
{ function connect() {
pingTimeout: 180 * 1000
untilOpen = new Promise((resolve, reject) => {
resolveOpen = resolve
rejectOpen = reject
})
ws = new WebSocket(
url + (url.indexOf('?') !== -1 ? '&' : '?') + `session=${Math.random()}`
)
ws.onopen = () => {
console.log('connected to', url)
resolveOpen()
} }
) ws.onerror = err => {
console.log('error connecting to relay', url, err)
var isOpen rejectOpen()
let untilOpen = new Promise(resolve => {
isOpen = resolve
})
ws.onopen = () => {
console.log('connected to', url)
isOpen()
}
ws.onerror = err => console.log('error connecting to relay', url, err)
ws.onclose = () => console.log('relay connection closed', url)
ws.onmessage = async e => {
var data
try {
data = JSON.parse(e.data)
} catch (err) {
data = e.data
} }
ws.onclose = () => console.log('relay connection closed', url)
if (data.length > 1) { ws.onmessage = async e => {
if (data === 'PING') { var data
ws.send('PONG') try {
return data = JSON.parse(e.data)
} catch (err) {
data = e.data
} }
if (data[0] === 'notice') { if (data.length > 1) {
console.log('message from relay ' + url + ': ' + data[1]) if (data === 'PING') {
onNotice(data[1]) ws.send('PONG')
return return
}
if (typeof data[0] === 'object') {
let event = data[0]
let context = data[1]
if (await verifySignature(event)) {
onEvent(event, context)
} else {
console.warn(
'got event with invalid signature from ' + url,
event,
context
)
} }
return
if (data[0] === 'notice') {
console.log('message from relay ' + url + ': ' + data[1])
onNotice(data[1])
return
}
if (typeof data[0] === 'object') {
let event = data[0]
let context = data[1]
if (await verifySignature(event)) {
onEvent(event, context)
} else {
console.warn(
'got event with invalid signature from ' + url,
event,
context
)
}
return
}
}
}
}
setInterval(() => {
if (!ws || ws.readyState !== WebSocket.OPEN)
connect()
}, 180 * 1000)
connect()
async function trySend(msg) {
if (ws && ws.readyState === WebSocket.OPEN) {
ws.send(msg)
} else {
try {
await untilOpen
ws.send(msg)
} catch(e) {
console.log(`waiting to connect to ${url}`)
} }
} }
} }
@@ -71,28 +95,22 @@ export function relayConnect(url, onEvent, onNotice) {
return { return {
url, url,
async subKey(key) { async subKey(key) {
await untilOpen trySend('sub-key:' + key)
ws.send('sub-key:' + key)
}, },
async unsubKey(key) { async unsubKey(key) {
await untilOpen trySend('unsub-key:' + key)
ws.send('unsub-key:' + key)
}, },
async reqFeed(params = {}) { async reqFeed(params = {}) {
await untilOpen trySend('req-feed:' + JSON.stringify(params))
ws.send('req-feed:' + JSON.stringify(params))
}, },
async reqEvent(params) { async reqEvent(params) {
await untilOpen trySend('req-event:' + JSON.stringify(params))
ws.send('req-event:' + JSON.stringify(params))
}, },
async reqKey(params) { async reqKey(params) {
await untilOpen trySend('req-key:' + JSON.stringify(params))
ws.send('req-key:' + JSON.stringify(params))
}, },
async publish(event) { async publish(event) {
await untilOpen trySend(JSON.stringify(event))
ws.send(JSON.stringify(event))
}, },
close() { close() {
ws.close() ws.close()