sdl/sdl-mgr/app/modules/sdl-mgr/index.js

233 lines
6.2 KiB
JavaScript

const module = 'sdl-mgr'; // Module Name
import { load_config, log, ipToInt, intToIp, computeBroadcast, getUptimeDHMS } from '../nwa-lib/index.js';
import mqtt from 'mqtt';
import os from 'os';
import dgram from 'dgram';
const config = load_config();
log(`Loaded module: ${module}`);
// log(`${module}: Cluster Conf: ${JSON.stringify(config.cluster, null, 2)}`, true);
function startSDLStatusPub() {
const sdlCfg = config.modules[module];
const mqttCfg = config.modules.mqtt;
if (!sdlCfg?.enabled) {
log(`${module}: disabled, not publishing status`);
return;
}
if (!mqttCfg?.enabled) {
log(`${module}: MQTT disabled, cannot publish status`);
return;
}
const statusInterval =
Number.isInteger(sdlCfg.update_interval.cluster_status) &&
sdlCfg.update_interval.cluster_status > 0
? sdlCfg.update_interval.cluster_status
: 10000;
// config.modules.mqtt.topics.sdl-mgr.pub.sdl_cluster-status
const statusTopic = mqttCfg.topics?.[module]?.pub?.['sdl_cluster-status'];
if (!statusTopic) {
log(`${module}: ${statusTopic} not configured`);
return;
}
// This is ok because we are running on localhost
const mqttUrl = `mqtt://127.0.0.1:${mqttCfg.mqtt_port}`;
const client = mqtt.connect(mqttUrl);
client.on('connect', () => {
log(`${module}: connected to MQTT at ${mqttUrl}`);
publishStatus(client, statusTopic);
setInterval(() => publishStatus(client, statusTopic), statusInterval);
});
client.on('error', err => {
log(`${module}: MQTT error: ${err}`);
});
}
function publishStatus(client, topic) {
const status = {
ts: new Date().toISOString(),
sdl_id: config.identity.sdl_id,
role: 'sdl-mgr',
host: config.identity.hostname,
type: 'cluster-status',
msg: {
sdl: {
version: config.package.version,
uptime: getUptimeDHMS()
},
cluster: config.cluster,
modules: Object.fromEntries(
Object.entries(config.modules).map(([name, mod]) => [
name,
{ enabled: mod?.enabled === true }
])
)
}
};
client.publish(
topic,
JSON.stringify(status),
{ qos: 1, retain: true },
err => {
if (err) {
log(`${module}: failed to publish status: ${err}`);
} else {
log(`${module}: published status to ${topic}`);
}
}
);
}
function startUdpBeacon() {
let sdlCfg = config.modules['sdl-mgr'];
// log(`${module}: DEBUG: SDL config: ${JSON.stringify(sdlCfg, null, 2)}`, true);
// log(`${module}: DEBUG: Cluster Conf: ${JSON.stringify(config.cluster, null, 2)}`, true);
sdlCfg.cluster = config.cluster;
const mqttCfg = config.modules.mqtt;
if (!sdlCfg?.enabled) {
log(`${module}: sdl disabled, not starting UDP beacon`);
return;
}
if (!mqttCfg?.enabled) {
log(`${module}: mqtt disabled, cannot advertise broker`);
return;
}
const beaconPort =
Number.isInteger(sdlCfg.beacon_udp_port)
? sdlCfg.beacon_udp_port
: 10101;
const beaconInterval =
Number.isInteger(sdlCfg.update_interval.udp_beacon)
? sdlCfg.update_interval.udp_beacon
: 2000;
const cluster = sdlCfg.cluster || {
id: 'default',
name: 'default',
desc: ''
};
log(`${module}: DEBUG: cluster: ${JSON.stringify(cluster)}`);
if (!config.host?.network) {
log(`${module}: no host network metadata available for UDP beacon`);
return;
}
const socket = dgram.createSocket('udp4');
socket.on('error', err => {
log(`${module}: UDP beacon socket error: ${err}`);
});
socket.bind(() => {
socket.setBroadcast(true);
log(`${module}: UDP beacon active on port ${beaconPort}`);
});
setInterval(() => {
const interfaces = os.networkInterfaces();
// for (const [iface, addrs] of Object.entries(config.host.network)) {
for (const [iface, addrs] of Object.entries(interfaces)) {
for (const addr of addrs) {
// --- FILTERS ---
// IPv4 only
if (addr.family !== 'IPv4') continue;
// Skip loopback
if (addr.internal === true) continue;
// Skip /32 networks (no broadcast: e.g. Tailscale)
const cidr = Number(addr.cidr?.split('/')[1]);
if (!Number.isInteger(cidr) || cidr >= 32) continue;
// Compute broadcast address
const broadcastAddr = computeBroadcast(addr.address, addr.netmask);
if (!broadcastAddr) continue;
const beacon = {
ts: new Date().toISOString(),
sdl_id: config.identity.sdl_id,
role: 'sdl-mgr',
host: config.identity.hostname,
type: 'udp-beacon',
msg: {
info: {
desc: config.package.description,
version: config.package.version,
copyright: config.package.copyright
},
cluster: {
id: cluster.id,
name: cluster.name,
desc: cluster.desc
},
mqtt: {
host: addr.address,
port: mqttCfg.mqtt_port
},
web: {
proto: 'http',
host: addr.address,
port: config.modules.web.port,
api_config: '/api/config',
config_url: `http://${addr.address}:${config.modules.web.port}/api/config`
},
sdl_wkr_install_cmd: {
curl: `curl -s http://${addr.address}:${config.modules.web.port}/dist/install-sdl-wkr.sh | bash -s ${addr.address}`,
wget: `wget -O - http://${addr.address}:${config.modules.web.port}/dist/install-sdl-wkr.sh | bash -s ${addr.address}`
}
}
};
const payload = Buffer.from(JSON.stringify(beacon)+"\n");
socket.send(
payload,
0,
payload.length,
beaconPort,
broadcastAddr,
err => {
if (err) {
log(`${module}: UDP beacon send error (${broadcastAddr}): ${err}`);
}
}
);
}
}
}, beaconInterval);
log(
`${module}: broadcasting UDP beacon every ${beaconInterval}ms on port ${beaconPort}`
);
}
// -------------------------------
// Entry point (ESM-safe)
// -------------------------------
startSDLStatusPub();
startUdpBeacon();