From 6a29c8ed40aacdcd37679e4b2f2ab551c5b419c0 Mon Sep 17 00:00:00 2001
From: Kuba Orlik <kontakt@kuba-orlik.name>
Date: Tue, 9 Nov 2021 21:57:19 +0100
Subject: [PATCH] Fix referer sometimes missing

---
 extended-request.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/extended-request.ts b/extended-request.ts
index 8ec7ca6..d0af544 100644
--- a/extended-request.ts
+++ b/extended-request.ts
@@ -117,7 +117,7 @@ export default class ExtendedRequest {
 
   getReferer() {
     return (
-      this.data.requestHeaders.filter((h) => h.name === "Referer")?.[0].value ||
+      this.data.requestHeaders.filter((h) => h.name === "Referer")[0]?.value ||
       "missing-referrer"
     );
   }
@@ -126,10 +126,14 @@ export default class ExtendedRequest {
     const url = new URL(this.origin);
     const host = url.host;
     const path = url.pathname;
+    const shorthost = getshorthost(host);
     return (
       this.getReferer().includes(host) ||
       this.stolenData.filter(
-        (entry) => entry.value.includes(host) || entry.value.includes(path)
+        (entry) =>
+          entry.value.includes(host) ||
+          entry.value.includes(path) ||
+          entry.value.includes(shorthost)
       ).length > 0
     );
   }