mirror of
https://github.com/nbd-wtf/nostr-tools.git
synced 2025-12-09 00:28:51 +00:00
remove hanging if no connection
This commit is contained in:
138
relay.js
138
relay.js
@@ -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()
|
||||||
|
|||||||
Reference in New Issue
Block a user