From 493f1d2f37381bb922acf54544e3d8601a9b6be3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20Hampa=C3=AF?= <clement.hampai@cypressxt.net>
Date: Wed, 14 Jun 2023 15:56:37 +0200
Subject: [PATCH] Support HEAD query on fs endpoint

Some CPEs query the FS endpoint with a `HEAD` request first before downloading
the image using a `GET`.
---
 lib/fs.ts | 24 +++++++++++++++---------
 1 file changed, 15 insertions(+), 9 deletions(-)

diff --git a/lib/fs.ts b/lib/fs.ts
index 3d44868..6e58335 100644
--- a/lib/fs.ts
+++ b/lib/fs.ts
@@ -62,8 +62,8 @@ export async function listener(
   request: IncomingMessage,
   response: ServerResponse
 ): Promise<void> {
-  if (request.method !== "GET") {
-    response.writeHead(405, { Allow: "GET" });
+  if (request.method !== "GET" && request.method !== "HEAD") {
+    response.writeHead(405, { Allow: "GET, HEAD" });
     response.end("405 Method Not Allowed");
     return;
   }
@@ -75,6 +75,7 @@ export async function listener(
     message: "Fetch file",
     filename: filename,
     remoteAddress: getRequestOrigin(request).remoteAddress,
+    method: request.method,
   };
 
   const file = await collections.files.findOne({ _id: filename });
@@ -87,20 +88,25 @@ export async function listener(
     return;
   }
 
+  response.writeHead(200, {
+    "Content-Type": "application/octet-stream",
+    "Content-Length": file.length,
+  });
+
+  logger.accessInfo(log);
+
+  if (request.method === "HEAD") {
+    response.end();
+    return;
+  }
+
   const chunks = await getFile(
     file["uploadDate"].getTime(),
     file.length,
     filename
   );
 
-  response.writeHead(200, {
-    "Content-Type": "application/octet-stream",
-    "Content-Length": file.length,
-  });
-
   pipeline(Readable.from(chunks), response, () => {
     // Ignore errors resulting from client disconnecting
   });
-
-  logger.accessInfo(log);
 }