From 09083b2bc095b2e33e2a2b3a7f2d9f2ab87d161a Mon Sep 17 00:00:00 2001
From: EuAndreh <eu@euandre.org>
Date: Fri, 15 Mar 2024 13:03:04 -0300
Subject: src/hero.mjs: Add handlerForConnection()

---
 src/hero.mjs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'src')

diff --git a/src/hero.mjs b/src/hero.mjs
index 10fa06e..61621af 100644
--- a/src/hero.mjs
+++ b/src/hero.mjs
@@ -309,6 +309,22 @@ export const buildHttpPayload = (code, {
 	"\r\n\r\n" +
 	message;
 
+export const fallback404Handler = (_req, socket) =>
+	socket.end(buildHttpPayload(404)).destroySoon();
+
+export const fallback405Handler = (_req, socket) =>
+	socket.end(buildHttpPayload(405)).destroySoon();
+
+export const handlerForConnection = (table, method, path) => {
+	if (method !== WEBSOCKET_METHOD) {
+		return fallback405Handler;
+	}
+
+	const segments = pathToSegments(path);
+	const handlerFn = u.getIn(table, [WEBSOCKET_KEYWORD, WEBSOCKET_METHOD].concat(segments));
+	return handlerFn || fallback404Handler;
+};
+
 export const makeRequestListener = table => async (req, res) => {
 	const { status, headers, body } = await handleRequest(table, req);
 	res.writeHead(status, headers);
-- 
cgit v1.2.3