37 lines
1.0 KiB
JavaScript
37 lines
1.0 KiB
JavaScript
import {encode} from '@/packets/index.js';
|
|
|
|
let connected = false;
|
|
let socket;
|
|
|
|
function onMessage(event) {
|
|
postMessage(event.data);
|
|
}
|
|
|
|
onmessage = async (event) => {
|
|
if (!connected) {
|
|
const url = new URL(`wss://${event.data.host}/ws`)
|
|
if ('production' === process.env.NODE_ENV) {
|
|
url.protocol = 'ws:';
|
|
}
|
|
socket = new WebSocket(url.href);
|
|
socket.binaryType = 'arraybuffer';
|
|
const {promise, resolve} = Promise.withResolvers();
|
|
socket.addEventListener('open', resolve);
|
|
socket.addEventListener('error', () => {
|
|
postMessage(encode({type: 'ConnectionStatus', payload: 'aborted'}));
|
|
close();
|
|
});
|
|
await promise;
|
|
socket.removeEventListener('open', resolve);
|
|
socket.addEventListener('message', onMessage);
|
|
socket.addEventListener('close', () => {
|
|
postMessage(encode({type: 'ConnectionStatus', payload: 'aborted'}));
|
|
close();
|
|
});
|
|
postMessage(encode({type: 'ConnectionStatus', payload: 'connected'}));
|
|
connected = true;
|
|
return;
|
|
}
|
|
socket.send(event.data);
|
|
};
|