From 2dfed1b7fbf19e5e874340340f764a90252b5537 Mon Sep 17 00:00:00 2001
From: EuAndreh <eu@euandre.org>
Date: Sat, 16 Mar 2024 08:14:27 -0300
Subject: src/hero.mjs: Add makeUpgradeListener() (again, now rewritten)

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

(limited to 'src/hero.mjs')

diff --git a/src/hero.mjs b/src/hero.mjs
index 314360c..2168ff7 100644
--- a/src/hero.mjs
+++ b/src/hero.mjs
@@ -445,6 +445,19 @@ export const makeRequestListener = table => async (req, res) => {
 	res.end(body);
 };
 
+export const makeUpgradeListener = table => async (req, socket, _head) => {
+	assert.ok(req.upgrade);
+	const { status, headers, body } = await handleRequest(table, {
+		...req,
+		socket,
+	});
+	writeHead(socket, status, headers);
+	socket.write(body);
+	if (status !== 101) {
+		socket.end();
+	}
+};
+
 export const actionsFn = ({
 	logger = log,
 } = {}) => ({
-- 
cgit v1.2.3