add HOSTNAME var

.env

@@ -1,4 +1,5 @@
 LOG_LEVEL=debug
+HOSTNAME=0.0.0.0
 PORT=9000
 HEARTBEAT_INTERVAL=30000
 PONG_TIMEOUT=60000

src/index.ts

@@ -7,13 +7,20 @@ import {
   DISCONNECT_TIMEOUT,
   REDIS_URL,
   ALLOWED_ORIGINS,
+  HOSTNAME,
 } from "./var";
-
+import os from "os";
 import Redis from "ioredis";
 
 import type { ServerWebSocket } from "bun";
 
-import type { ClientSignal, RawSignal, Room, ServerWebSocketData, TransferClient } from "./types";
+import type {
+  ClientSignal,
+  RawSignal,
+  Room,
+  ServerWebSocketData,
+  TransferClient,
+} from "./types";
 
 const logger = pino({
   level: LOG_LEVEL,
@@ -111,6 +118,7 @@ redisSub?.on("message", (channel, message) => {
 
 const server = Bun.serve<ServerWebSocketData>({
   port: PORT,
+  hostname: HOSTNAME,
   fetch(req, server) {
     const url = new URL(req.url);
     const roomId = url.searchParams.get("room") || "";
@@ -187,7 +195,10 @@ function handleWSOpen(ws: ServerWebSocket<ServerWebSocketData>) {
   );
 }
 
-function handleWSMessage(ws: ServerWebSocket<ServerWebSocketData>, message: string | Buffer) {
+function handleWSMessage(
+  ws: ServerWebSocket<ServerWebSocketData>,
+  message: string | Buffer
+) {
   try {
     const signal: RawSignal = JSON.parse(message.toString());
     const room: Room | undefined = rooms.get(ws.data.roomId);
@@ -264,7 +275,10 @@ function handleClientJoin(
       }
       existingClient.session = ws;
       existingClient.lastPongTime = Date.now();
-      logger.info({ clientId: client.clientId, name: client.name }, "Client reconnected");
+      logger.info(
+        { clientId: client.clientId, name: client.name },
+        "Client reconnected"
+      );
 
       // send cached messages
       existingClient.messageCache.forEach((message) => {
@@ -322,7 +336,10 @@ function handleClientJoin(
     if (session === ws) return;
     if (session.readyState === WebSocket.OPEN) {
       session.send(JSON.stringify(joinMessage));
-      logger.info({ clientId: client.clientId, name: client.name }, "send join message to client");
+      logger.info(
+        { clientId: client.clientId, name: client.name },
+        "send join message to client"
+      );
     } else {
       messageCache.push(joinMessage);
     }
@@ -443,7 +460,10 @@ function startHeartbeat() {
       room.clients.forEach((clientData, clientId) => {
         if (clientData.session.readyState === WebSocket.OPEN) {
           if (now - clientData.lastPongTime > PONG_TIMEOUT) {
-            logger.warn({ clientId, roomId }, "Client timed out, closing connection");
+            logger.warn(
+              { clientId, roomId },
+              "Client timed out, closing connection"
+            );
             clientData.session.close();
           } else {
             clientData.session.send(JSON.stringify({ type: "ping" }));
@@ -454,9 +474,23 @@ function startHeartbeat() {
   }, HEARTBEAT_INTERVAL);
 }
 
-logger.info({ port: server.port }, "WebSocket server started");
-startHeartbeat();
+const addresses: string[] = [HOSTNAME];
+
+// get all ip addresses
+if (HOSTNAME === "0.0.0.0") {
+  const interfaces = os.networkInterfaces();
+  const ips = Object.values(interfaces).flatMap((iface) =>
+    iface?.map((iface) => iface.address)
+  );
+  addresses.push(...(ips.filter((ip) => ip !== undefined) as string[]));
+}
+
+logger.info(
+  { port: server.port, hostname: HOSTNAME, addresses },
+  "WebSocket server started"
+);
 
+startHeartbeat();
 process.on("SIGINT", () => {
   logger.info("Shutting down server...");
   server.stop();

src/var.ts

@@ -1,4 +1,5 @@
 export const LOG_LEVEL = process.env["LOG_LEVEL"] || "info";
+export const HOSTNAME = process.env["HOSTNAME"] || "0.0.0.0";
 export const PORT = parseInt(process.env["PORT"] || "9000", 10);
 export const HEARTBEAT_INTERVAL = parseInt(process.env["HEARTBEAT_INTERVAL"] || "30000", 10);
 export const PONG_TIMEOUT = parseInt(process.env["PONG_TIMEOUT"] || "60000", 10);