117 lines
3.1 KiB
JavaScript
117 lines
3.1 KiB
JavaScript
"use strict";
|
|
|
|
/**
|
|
* Represents a WebSocket connection and its state.
|
|
* @typedef {Object} WebSocketConnection
|
|
* @property {WebSocket} socket - The WebSocket instance.
|
|
* @property {number} state - The state of the WebSocket connection. Possible values:
|
|
* - 0: Connection is in the CONNECTING state.
|
|
* - 1: Connection is in the OPEN state.
|
|
* - 2: Connection is in the CLOSING state.
|
|
* - 3: Connection is in the CLOSED state.
|
|
* @property {Array} received - An array to store received data or messages.
|
|
*/
|
|
|
|
/**
|
|
* An instance of WebSocketConnection.
|
|
* @type {{[key: number]: WebSocketConnection}}
|
|
*/
|
|
|
|
let ws = {};
|
|
let i = 0;
|
|
|
|
const WS_NOT_EXISTING = -1;
|
|
const WS_DISCONNECTED = 0;
|
|
const WS_CONNECTED = 1;
|
|
const WS_CLOSED = 2;
|
|
|
|
function ws_open_js(url, id) {
|
|
let conn;
|
|
try {
|
|
conn = {
|
|
socket: new WebSocket(url),
|
|
url,
|
|
state: WS_CONNECTED,
|
|
received: []
|
|
};
|
|
} catch (error) {
|
|
return -1;
|
|
}
|
|
|
|
conn.socket.onopen = (s, e) => {
|
|
conn.state = WS_CONNECTED;
|
|
};
|
|
conn.socket.onmessage = (e) => {
|
|
conn.received.push(new Uint8Array(e.data.arrayBuffer()));
|
|
};
|
|
conn.socket.onerror = (s, e) => {
|
|
conn.state = WS_DISCONNECTED;
|
|
};
|
|
conn.socket.onclose = (s, e) => {
|
|
conn.state = WS_DISCONNECTED;
|
|
//delete ws[id];
|
|
//ws[id] = undefined;
|
|
};
|
|
ws[id] = conn;
|
|
return id;
|
|
}
|
|
|
|
function ws_open(ptr, len) {
|
|
let url = UTF8ToString(ptr, len);
|
|
return ws_open_js(url, i++);
|
|
}
|
|
|
|
function ws_revive(id) {
|
|
if (ws[id] == null) return false;
|
|
if (ws[id].state == WS_NOT_EXISTING || ws[id].state == WS_CLOSED) return false;
|
|
if (ws[id].state == WS_CONNECTED) return true;
|
|
return ws_open_js(ws[id].url, id) == id;
|
|
}
|
|
|
|
function ws_write(id, ptr, len) {
|
|
let data = new Uint8Array(wasm_memory.buffer, ptr, len);
|
|
/*let data = new Uint8Array(len);
|
|
for (let i = 0; i < len; i++) {
|
|
data[i] = data_in[i];
|
|
}*/
|
|
if (ws[id] != null && ws[id].socket.readyState == WebSocket.OPEN) {
|
|
ws[id].socket.send(data);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
function ws_available(id) {
|
|
if (ws[id] == null || ws[id].received == null || ws[id].received.length == 0) return -1;
|
|
return ws[id].received[0].length;
|
|
}
|
|
|
|
function ws_read(id, ptr, max_length) {
|
|
let file = ws[id].received.pop();
|
|
var dest = new Uint8Array(wasm_memory.buffer, ptr, max_length);
|
|
for (let i = 0; i < file.length && i < max_length; i++) {
|
|
dest[i] = file[i];
|
|
}
|
|
}
|
|
|
|
function ws_state(id) {
|
|
return ws[id]?.state ?? WS_NOT_EXISTING;
|
|
}
|
|
|
|
function ws_close(id) {
|
|
if (ws[id] == null) return;
|
|
ws[id].socket.close();
|
|
ws[id].state = WS_CLOSED;
|
|
ws[id] = { state: WS_CLOSED };
|
|
}
|
|
|
|
function register_plugin(importObject) {
|
|
importObject.env.ws_open = ws_open;
|
|
importObject.env.ws_revive = ws_revive;
|
|
importObject.env.ws_write = ws_write;
|
|
importObject.env.ws_read = ws_read;
|
|
importObject.env.ws_available = ws_available;
|
|
importObject.env.ws_state = ws_state;
|
|
importObject.env.ws_close = ws_close;
|
|
}
|
|
|
|
miniquad_add_plugin({ register_plugin, version: "0.0.1", name: "miniquad_websocket" });
|