Agent Protocol

import { createSocket } from "node:dgram";

const SERVER_URL = "wss://netignite.example.com/api/agent-control-service";
const AGENT_ID = "3";
const AGENT_SECRET = "your-connection-secret";
const BROADCAST_ADDRESS = "192.168.1.255";

type Payload = Record<string, unknown>;

function sendMagicPacket(macAddress: string, port: number) {
    const mac = macAddress.split(/[:-]/).map((part) => Number.parseInt(part, 16));
    if (mac.length !== 6 || mac.some(Number.isNaN)) throw new Error("Invalid MAC address");

    // 6 × 0xFF, then the MAC address 16 times.
    const packet = Buffer.alloc(102, 0xff);
    for (let i = 0; i < 16; i++) Buffer.from(mac).copy(packet, 6 + i * 6);

    const socket = createSocket("udp4");
    socket.bind(() => {
        socket.setBroadcast(true);
        socket.send(packet, port, BROADCAST_ADDRESS, () => socket.close());
    });
}

function handleCommand(command: string, payload: Payload): Payload {
    switch (command) {
        case "WAKEUP":
            sendMagicPacket(String(payload.macAddress), Number(payload.port));
            return { status: "OK", message: "Wakeup successful" };
        case "GET_STATUS":
            // Look up and ping the device here, then report "online" or "offline".
            return { status: "OK", message: "Status check not implemented", online: "unknown" };
        case "HEARTBEAT":
            return {};
        default:
            return { status: "ERROR", message: "Unknown command" };
    }
}

function connect() {
    const ws = new WebSocket(SERVER_URL, {
        headers: { Authorization: `Basic ${btoa(`${AGENT_ID}:${AGENT_SECRET}`)}` },
    });

    ws.onopen = () => console.log("Connected");

    ws.onmessage = (event) => {
        const message = String(event.data);
        const first = message.indexOf(":");
        const second = message.indexOf(":", first + 1);
        const command = message.slice(0, first);
        const id = message.slice(first + 1, second);

        let response: Payload;
        try {
            response = handleCommand(command, JSON.parse(message.slice(second + 1)));
        } catch (error) {
            response = { status: "ERROR", message: String(error) };
        }
        ws.send(`${command}:${id}:${JSON.stringify(response)}`);
    };

    ws.onclose = (event) => {
        console.log(`Disconnected (${event.code} ${event.reason}), reconnecting in 10 seconds`);
        setTimeout(connect, 10_000);
    };
}

connect();

bun run agent.ts