150 lines
3.6 KiB
JavaScript
150 lines
3.6 KiB
JavaScript
require("dotenv").config();
|
|
const ws = require("ws")
|
|
|
|
/* Old NTFY Body from old system
|
|
ntfyBody = {
|
|
"topic": `${config.ntfy.prefix}${fromChannel}`,
|
|
"message": bodyData.string,
|
|
"tags": [`Timestamp: ${product_id.timestamp}`, `Station: ${product_id.station}`, `WMO: ${product_id.wmo}`, `PIL: ${product_id.pil}`, `Channel: ${fromChannel}`],
|
|
"priority": evt.priority,
|
|
"actions": [{ "action": "view", "label": "Product", "url": bodyData.url }, { "action": "view", "label": "Product Text", "url": `https://mesonet.agron.iastate.edu/api/1/nwstext/${product_id_raw}` }]
|
|
}
|
|
|
|
// Message body from websocket
|
|
{
|
|
"type": "iem-message",
|
|
"data": {
|
|
"channel": {
|
|
"location": "<location-name>",
|
|
"room": "<room-name>"
|
|
},
|
|
"event": {
|
|
"name": "<event-name>",
|
|
"priority": <priority-level>,
|
|
"code": "<event-code>"
|
|
},
|
|
"body": "<message-body>",
|
|
"timestamp": "<timestamp>",
|
|
"wmo": "<wmo-code>",
|
|
"pil": "<pil-code>",
|
|
"station": "<station-name>",
|
|
"raw": "<raw-product-id>",
|
|
"rawBody": "<raw-message-body>",
|
|
"image": "<image-url>", // Optional
|
|
"productText": "<product-text>" // Optional
|
|
}
|
|
}
|
|
|
|
*/
|
|
|
|
let retryQueue = [];
|
|
|
|
function handleIEMMessage(msg) {
|
|
if (!msg.data) return;
|
|
const data = msg.data;
|
|
ntfyBody = {
|
|
topic: `${data.channel.room}`,
|
|
message: data.body.string,
|
|
tags: [
|
|
`Timestamp: ${data.timestamp}`,
|
|
`Station: ${data.station}`,
|
|
`WMO: ${data.wmo}`,
|
|
`PIL: ${data.pil}`,
|
|
`Channel: ${data.channel.room}`,
|
|
],
|
|
priority: data.event.priority,
|
|
actions: [
|
|
{
|
|
action: "view",
|
|
label: "Product",
|
|
url: data.body.url,
|
|
},
|
|
{
|
|
action: "view",
|
|
label: "Product Text",
|
|
url: `https://mesonet.agron.iastate.edu/api/1/nwstext/${data.raw}`,
|
|
},
|
|
],
|
|
};
|
|
|
|
if (data.image) {
|
|
ntfyBody.attach = data.image;
|
|
}
|
|
|
|
// Send to NTFY
|
|
fetch(process.env.NTFY_HOST, {
|
|
method: "POST",
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
"Authorization": `Bearer ${process.env.NTFY_TOKEN}`,
|
|
},
|
|
body: JSON.stringify(ntfyBody),
|
|
}).then((res) => {
|
|
if (!res.ok) {
|
|
retryQueue.push(ntfyBody);
|
|
console.error(`Error sending to NTFY: ${res.statusText}. Added to retry queue.`);
|
|
} else {
|
|
console.log(`Sent to NTFY topic ${ntfyBody.topic}`);
|
|
}
|
|
}).catch((error) => {
|
|
retryQueue.push(ntfyBody);
|
|
console.error("Error sending to NTFY:", error);
|
|
});
|
|
};
|
|
|
|
function processRetryQueue() {
|
|
if (retryQueue.length === 0) return;
|
|
|
|
const body = retryQueue.shift();
|
|
fetch(process.env.NTFY_HOST, {
|
|
method: "POST",
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
"Authorization": `Bearer ${process.env.NTFY_TOKEN}`,
|
|
},
|
|
body: JSON.stringify(body),
|
|
}).then((res) => {
|
|
if (!res.ok) {
|
|
retryQueue.push(body);
|
|
console.error(`Error retrying to NTFY: ${res.statusText}. Re-added to retry queue.`);
|
|
} else {
|
|
console.log(`Retried and sent to NTFY topic ${body.topic}`);
|
|
}
|
|
}).catch((error) => {
|
|
retryQueue.push(body);
|
|
console.error("Error retrying to NTFY:", error);
|
|
});
|
|
};
|
|
|
|
// Process retry queue every second
|
|
setInterval(processRetryQueue, 1000);
|
|
|
|
|
|
// Connect to process.env.IEM_WS_SERVER
|
|
const wsClient = new ws(process.env.IEM_WS_SERVER, {
|
|
headers: {
|
|
"User-Agent": "ntfy-pusher/1.0",
|
|
},
|
|
});
|
|
|
|
wsClient.on("open", function open() {
|
|
console.log("Connected to IEM WebSocket server");
|
|
wsClient.send(JSON.stringify({ type: "subscribe", channel: "*" }));
|
|
});
|
|
|
|
wsClient.on("message", function incoming(data) {
|
|
let msg;
|
|
try {
|
|
msg = JSON.parse(data);
|
|
} catch (error) {
|
|
console.error("Error parsing message:", error);
|
|
}
|
|
|
|
if (!msg) return; // Ignore invalid messages
|
|
|
|
switch (msg.type) {
|
|
case "iem-message":
|
|
handleIEMMessage(msg);
|
|
break;
|
|
}
|
|
}); |