8x48/homepage
1
0
Fork 0
mirror of https://github.com/gethomepage/homepage.git synced 2026-01-15 05:42:08 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

UNify this

Browse Source
This commit is contained in:
shamoon
2026-01-03 23:52:05 -08:00
parent e4b189557e
commit 9f234d5d3b
1 changed files with 11 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
src/widgets/truenas/proxy.js
View File

@@ -19,17 +19,17 @@ function buildWebsocketUrl(baseUrl) {
return url.toString(); return url.toString();
} }
function waitForMessage(ws, matcher) { function waitForEvent(ws, matcher, { event = "message", parseJson = true, timeoutMs = 10000 } = {}) {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
const timeout = setTimeout(() => { const timeout = setTimeout(() => {
cleanup(); cleanup();
reject(new Error("TrueNAS websocket request timed out")); reject(new Error("TrueNAS websocket wait timed out"));
}, 10000); }, timeoutMs);
const handleMessage = (data) => { const handleEvent = (payload) => {
try { try {
const parsed = JSON.parse(data.toString()); const parsed = parseJson ? JSON.parse(payload.toString()) : payload;
logger.info("Received TrueNAS websocket message: %o", parsed); if (parseJson) logger.info("Received TrueNAS websocket message: %o", parsed);
const matchResult = matcher(parsed); const matchResult = matcher(parsed);
if (matchResult !== undefined) { if (matchResult !== undefined) {
cleanup(); cleanup();
@@ -59,12 +59,12 @@ function waitForMessage(ws, matcher) {
function cleanup() { function cleanup() {
clearTimeout(timeout); clearTimeout(timeout);
ws.off("message", handleMessage); ws.off(event, handleEvent);
ws.off("error", handleError); ws.off("error", handleError);
ws.off("close", handleClose); ws.off("close", handleClose);
} }
ws.on("message", handleMessage); ws.on(event, handleEvent);
ws.on("error", handleError); ws.on("error", handleError);
ws.on("close", handleClose); ws.on("close", handleClose);
}); });
@@ -72,7 +72,7 @@ function waitForMessage(ws, matcher) {
async function ensureConnected(ws) { async function ensureConnected(ws) {
ws.send(JSON.stringify({ msg: "connect", version: "1", support: ["1"] })); ws.send(JSON.stringify({ msg: "connect", version: "1", support: ["1"] }));
await waitForMessage(ws, (message) => (message?.msg === "connected" ? true : undefined)); await waitForEvent(ws, (message) => (message?.msg === "connected" ? true : undefined));
} }
let nextId = 1; let nextId = 1;
@@ -81,7 +81,7 @@ async function sendMethod(ws, method, params = []) {
logger.info("Sending TrueNAS websocket method %s with id %d", method, id); logger.info("Sending TrueNAS websocket method %s with id %d", method, id);
ws.send(JSON.stringify({ id, msg: "method", method, params })); ws.send(JSON.stringify({ id, msg: "method", method, params }));
return waitForMessage(ws, (message) => { return waitForEvent(ws, (message) => {
if (message?.msg === "result" && message.id === id) { if (message?.msg === "result" && message.id === id) {
if (message.error) { if (message.error) {
return new Error(message.error.reason || JSON.stringify(message.error)); return new Error(message.error.reason || JSON.stringify(message.error));
@@ -115,39 +115,12 @@ async function authenticate(ws, widget) {
throw new Error("TrueNAS authentication failed"); throw new Error("TrueNAS authentication failed");
} }
function waitForOpen(ws) {
return new Promise((resolve, reject) => {
const handleOpen = () => {
cleanup();
resolve();
};
const handleError = (err) => {
cleanup();
reject(err);
};
const handleClose = () => {
cleanup();
reject(new Error("TrueNAS websocket closed before the request was sent"));
};
function cleanup() {
ws.off("open", handleOpen);
ws.off("error", handleError);
ws.off("close", handleClose);
}
ws.once("open", handleOpen);
ws.once("error", handleError);
ws.once("close", handleClose);
});
}
async function callWebsocket(widget, method) { async function callWebsocket(widget, method) {
const wsUrl = buildWebsocketUrl(widget.url); const wsUrl = buildWebsocketUrl(widget.url);
logger.info("Connecting to TrueNAS websocket at %s", wsUrl); logger.info("Connecting to TrueNAS websocket at %s", wsUrl);
const ws = new WebSocket(wsUrl, { rejectUnauthorized: false }); const ws = new WebSocket(wsUrl, { rejectUnauthorized: false });
await waitForOpen(ws); await waitForEvent(ws, () => true, { event: "open", parseJson: false });
logger.info("Connected to TrueNAS websocket at %s", wsUrl); logger.info("Connected to TrueNAS websocket at %s", wsUrl);
try { try {
await ensureConnected(ws); await ensureConnected(ws);