update to support websockets, update depends and gitignore

This commit is contained in:
creations 2025-02-27 11:57:09 -05:00
parent 5687fb4bec
commit 62dcb0f297
Signed by: creations
GPG key ID: 8F553AA4320FC711
4 changed files with 70 additions and 31 deletions

View file

@ -8,6 +8,8 @@ import {
} from "bun";
import { resolve } from "path";
import { webSocketHandler } from "@/WebSocketHandler";
class ServerHandler {
private router: FileSystemRouter;
@ -27,6 +29,11 @@ class ServerHandler {
port: this.port,
hostname: this.host,
fetch: this.handleRequest.bind(this),
websocket: {
open: webSocketHandler.handleOpen.bind(webSocketHandler),
message: webSocketHandler.handleMessage.bind(webSocketHandler),
close: webSocketHandler.handleClose.bind(webSocketHandler),
},
});
logger.info(
@ -195,29 +202,27 @@ class ServerHandler {
);
}
queueMicrotask(() => {
const headers: Headers = response.headers;
let ip: string | null = server.requestIP(request)?.address || null;
const headers: Headers = response.headers;
let ip: string | null = server.requestIP(request)?.address || null;
if (!ip) {
ip =
headers.get("CF-Connecting-IP") ||
headers.get("X-Real-IP") ||
headers.get("X-Forwarded-For") ||
null;
}
if (!ip) {
ip =
headers.get("CF-Connecting-IP") ||
headers.get("X-Real-IP") ||
headers.get("X-Forwarded-For") ||
null;
}
logger.custom(
`[${request.method}]`,
`(${response.status})`,
[
request.url,
`${(performance.now() - request.startPerf).toFixed(2)}ms`,
ip || "unknown",
],
"90",
);
});
logger.custom(
`[${request.method}]`,
`(${response.status})`,
[
request.url,
`${(performance.now() - request.startPerf).toFixed(2)}ms`,
ip || "unknown",
],
"90",
);
return response;
}

34
src/webSocketHandler.ts Normal file
View file

@ -0,0 +1,34 @@
import { logger } from "@helpers/logger";
import { type ServerWebSocket } from "bun";
class WebSocketHandler {
public handleMessage(ws: ServerWebSocket, message: string): void {
logger.info(`WebSocket received: ${message}`);
try {
ws.send(`You said: ${message}`);
} catch (error) {
logger.error(`WebSocket send error: ${(error as Error).message}`);
}
}
public handleOpen(ws: ServerWebSocket): void {
logger.info("WebSocket connection opened.");
try {
ws.send("Welcome to the WebSocket server!");
} catch (error) {
logger.error(`WebSocket send error: ${(error as Error).message}`);
}
}
public handleClose(
ws: ServerWebSocket,
code: number,
reason: string,
): void {
logger.warn(`WebSocket closed with code ${code}, reason: ${reason}`);
}
}
const webSocketHandler: WebSocketHandler = new WebSocketHandler();
export { webSocketHandler };