Skip to content

Commit

Permalink
remove hanging if no connection
Browse files Browse the repository at this point in the history
  • Loading branch information
rsbondi authored and fiatjaf committed Jan 19, 2021
1 parent 63259eb commit 6586a1b
Showing 1 changed file with 74 additions and 56 deletions.
130 changes: 74 additions & 56 deletions relay.js
Expand Up @@ -11,88 +11,106 @@ export function normalizeRelayURL(url) {
export function relayConnect(url, onEvent, onNotice) {
url = normalizeRelayURL(url)

const ws = new WebSocket(
url + (url.indexOf('?') !== -1 ? '&' : '?') + `session=${Math.random()}`,
{
pingTimeout: 180 * 1000
}
)
let ws, resolveOpen, untilOpen, rejectOpen

var isOpen
let untilOpen = new Promise(resolve => {
isOpen = resolve
})
function connect() {

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)
untilOpen = new Promise((resolve, reject) => {
resolveOpen = resolve
rejectOpen = reject
})

ws = new WebSocket(
url + (url.indexOf('?') !== -1 ? '&' : '?') + `session=${Math.random()}`
)

ws.onmessage = async e => {
var data
try {
data = JSON.parse(e.data)
} catch (err) {
data = e.data
ws.onopen = () => {
console.log('connected to', url)
resolveOpen()
}

if (data.length > 1) {
if (data === 'PING') {
ws.send('PONG')
return
ws.onerror = err => {
console.log('error connecting to relay', url, err)
rejectOpen()
}
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
}

if (data[0] === 'notice') {
console.log('message from relay ' + url + ': ' + data[1])
onNotice(data[1])
return
}
if (data.length > 1) {
if (data === 'PING') {
ws.send('PONG')
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 (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
)
if (await verifySignature(event)) {
onEvent(event, context)
} else {
console.warn(
'got event with invalid signature from ' + url,
event,
context
)
}
return
}
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}`)
}
}
}

return {
url,
async subKey(key) {
await untilOpen
ws.send('sub-key:' + key)
trySend('sub-key:' + key)
},
async unsubKey(key) {
await untilOpen
ws.send('unsub-key:' + key)
trySend('unsub-key:' + key)
},
async reqFeed(params = {}) {
await untilOpen
ws.send('req-feed:' + JSON.stringify(params))
trySend('req-feed:' + JSON.stringify(params))
},
async reqEvent(params) {
await untilOpen
ws.send('req-event:' + JSON.stringify(params))
trySend('req-event:' + JSON.stringify(params))
},
async reqKey(params) {
await untilOpen
ws.send('req-key:' + JSON.stringify(params))
trySend('req-key:' + JSON.stringify(params))
},
async publish(event) {
await untilOpen
ws.send(JSON.stringify(event))
trySend(JSON.stringify(event))
},
close() {
ws.close()
Expand Down

0 comments on commit 6586a1b

Please sign in to comment.