From 054ea6aa077e011a01f4f8fe145f342d5119dcf4 Mon Sep 17 00:00:00 2001
From: mrjvs <jellevs@gmail.com>
Date: Thu, 4 Jan 2024 19:54:24 +0100
Subject: [PATCH] Support overwriting user agent

---
 src/utils/headers.ts | 21 ++++++++++++---------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/src/utils/headers.ts b/src/utils/headers.ts
index f87c2f6..3130378 100644
--- a/src/utils/headers.ts
+++ b/src/utils/headers.ts
@@ -26,20 +26,23 @@ function copyHeader(
 export function getProxyHeaders(headers: Headers): Headers {
   const output = new Headers();
 
-  const headerMap: Record<string, string> = {
-    'X-Cookie': 'Cookie',
-    'X-Referer': 'Referer',
-    'X-Origin': 'Origin',
-  };
-  Object.entries(headerMap).forEach((entry) => {
-    copyHeader(headers, output, entry[0], entry[1]);
-  });
-
+  // default user-agent
   output.set(
     'User-Agent',
     'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:93.0) Gecko/20100101 Firefox/93.0',
   );
 
+  const headerMap: Record<string, string> = {
+    'X-Cookie': 'Cookie',
+    'X-Referer': 'Referer',
+    'X-Origin': 'Origin',
+    'X-User-Agent': 'User-Agent',
+    'X-X-Real-Ip': 'X-Real-Ip',
+  };
+  Object.entries(headerMap).forEach((entry) => {
+    copyHeader(headers, output, entry[0], entry[1]);
+  });
+
   return output;
 }