From 22ec52561865f883ec08b57d71e1d1ddc34c4005 Mon Sep 17 00:00:00 2001
From: Junlai <1462011227@qq.com>
Date: Tue, 4 Apr 2023 03:34:33 +0000
Subject: [PATCH] update user host

---
 app/api/common.ts           |  5 +++--
 app/components/settings.tsx | 14 ++++++++++++++
 app/locales/cn.ts           |  5 +++++
 app/locales/en.ts           |  5 +++++
 app/locales/es.ts           |  5 +++++
 app/locales/it.ts           |  6 ++++++
 app/locales/tw.ts           |  5 +++++
 app/requests.ts             |  1 +
 app/store/access.ts         | 10 ++++++++--
 9 files changed, 52 insertions(+), 4 deletions(-)

diff --git a/app/api/common.ts b/app/api/common.ts
index 842eeacaf..d67d7cea8 100644
--- a/app/api/common.ts
+++ b/app/api/common.ts
@@ -3,12 +3,13 @@ import { NextRequest } from "next/server";
 const OPENAI_URL = "api.openai.com";
 const DEFAULT_PROTOCOL = "https";
 const PROTOCOL = process.env.PROTOCOL ?? DEFAULT_PROTOCOL;
-const BASE_URL = process.env.BASE_URL ?? OPENAI_URL;
 
 export async function requestOpenai(req: NextRequest) {
   const apiKey = req.headers.get("token");
   const openaiPath = req.headers.get("path");
-
+  const userHost = req.headers.get("userhost");
+  const BASE_URL = userHost ?? process.env.BASE_URL ?? OPENAI_URL;
+  console.log("[BASE_URL] ", openaiPath);
   console.log("[Proxy] ", openaiPath);
 
   return fetch(`${PROTOCOL}://${BASE_URL}/${openaiPath}`, {
diff --git a/app/components/settings.tsx b/app/components/settings.tsx
index 4645f3191..5f7ceb5ad 100644
--- a/app/components/settings.tsx
+++ b/app/components/settings.tsx
@@ -372,6 +372,20 @@ export function Settings(props: { closeSettings: () => void }) {
             <></>
           )}
 
+          <SettingItem
+            title={Locale.Settings.Host.Title}
+            subTitle={Locale.Settings.Host.SubTitle}
+          >
+            <input
+              value={accessStore.host}
+              type="text"
+              placeholder={Locale.Settings.Host.Placeholder}
+              onChange={(e) => {
+                accessStore.updateHost(e.currentTarget.value);
+              }}
+            ></input>
+          </SettingItem>
+
           <SettingItem
             title={Locale.Settings.Token.Title}
             subTitle={Locale.Settings.Token.SubTitle}
diff --git a/app/locales/cn.ts b/app/locales/cn.ts
index 49bcce235..ec9838ad3 100644
--- a/app/locales/cn.ts
+++ b/app/locales/cn.ts
@@ -97,6 +97,11 @@ const cn = {
       Title: "历史消息长度压缩阈值",
       SubTitle: "当未压缩的历史消息超过该值时，将进行压缩",
     },
+    Host: {
+      Title: "Personal Host",
+      SubTitle: "使用自己的 Host 可绕过受控访问限制",
+      Placeholder: "api.openai.com",
+    },
     Token: {
       Title: "API Key",
       SubTitle: "使用自己的 Key 可绕过受控访问限制",
diff --git a/app/locales/en.ts b/app/locales/en.ts
index ae8e88c2d..02e0949ba 100644
--- a/app/locales/en.ts
+++ b/app/locales/en.ts
@@ -99,6 +99,11 @@ const en: LocaleType = {
       SubTitle:
         "Will compress if uncompressed messages length exceeds the value",
     },
+    Host: {
+      Title: "Personal Host",
+      SubTitle: "Use your host to ignore access code limit",
+      Placeholder: "api.openai.com",
+    },
     Token: {
       Title: "API Key",
       SubTitle: "Use your key to ignore access code limit",
diff --git a/app/locales/es.ts b/app/locales/es.ts
index f3714ef3b..bc0eb6e7b 100644
--- a/app/locales/es.ts
+++ b/app/locales/es.ts
@@ -99,6 +99,11 @@ const es: LocaleType = {
       SubTitle:
         "Se comprimirán los mensajes si la longitud de los mensajes no comprimidos supera el valor",
     },
+    Host: {
+      Title: "Clave de Host",
+      SubTitle: "Utiliza tu clave para ignorar el límite de código de acceso",
+      Placeholder: "api.openai.com",
+    },
     Token: {
       Title: "Clave de API",
       SubTitle: "Utiliza tu clave para ignorar el límite de código de acceso",
diff --git a/app/locales/it.ts b/app/locales/it.ts
index c4736c1e0..5a3bbd751 100644
--- a/app/locales/it.ts
+++ b/app/locales/it.ts
@@ -99,6 +99,12 @@ const it: LocaleType = {
       SubTitle:
         "Comprimerà se la lunghezza dei messaggi non compressi supera il valore",
     },
+    Host: {
+      Title: "Chiave Host",
+      SubTitle:
+        "Utilizzare la chiave per ignorare il limite del codice di accesso",
+      Placeholder: "api.openai.com",
+    },
     Token: {
       Title: "Chiave API",
       SubTitle:
diff --git a/app/locales/tw.ts b/app/locales/tw.ts
index fff2f15dc..accb1f5f0 100644
--- a/app/locales/tw.ts
+++ b/app/locales/tw.ts
@@ -97,6 +97,11 @@ const tw: LocaleType = {
       Title: "歷史訊息長度壓縮閾值",
       SubTitle: "當未壓縮的歷史訊息超過該值時，將進行壓縮",
     },
+    Host: {
+      Title: "Personal Host",
+      SubTitle: "使用自己的 Host 可規避受控訪問限制",
+      Placeholder: "api.openai.com",
+    },
     Token: {
       Title: "API Key",
       SubTitle: "使用自己的 Key 可規避受控訪問限制",
diff --git a/app/requests.ts b/app/requests.ts
index da9b5c97f..170be2d72 100644
--- a/app/requests.ts
+++ b/app/requests.ts
@@ -38,6 +38,7 @@ function getHeaders() {
 
   if (accessStore.token && accessStore.token.length > 0) {
     headers["token"] = accessStore.token;
+    headers["userhost"] = accessStore.host;
   }
 
   return headers;
diff --git a/app/store/access.ts b/app/store/access.ts
index 9c61dfa01..ff636330a 100644
--- a/app/store/access.ts
+++ b/app/store/access.ts
@@ -5,7 +5,9 @@ import { queryMeta } from "../utils";
 export interface AccessControlStore {
   accessCode: string;
   token: string;
+  host: string;
 
+  updateHost: (_: string) => void;
   updateToken: (_: string) => void;
   updateCode: (_: string) => void;
   enabledAccessControl: () => boolean;
@@ -18,6 +20,7 @@ export const useAccessStore = create<AccessControlStore>()(
     (set, get) => ({
       token: "",
       accessCode: "",
+      host: "",
       enabledAccessControl() {
         return queryMeta("access") === "enabled";
       },
@@ -27,10 +30,13 @@ export const useAccessStore = create<AccessControlStore>()(
       updateToken(token: string) {
         set((state) => ({ token }));
       },
+      updateHost(host: string) {
+        set((state) => ({ host }));
+      },
     }),
     {
       name: ACCESS_KEY,
       version: 1,
-    }
-  )
+    },
+  ),
 );