feat: 支持代理ENDPOINT

nihaoshijie2718 · Sep 12, 2023 · c3d6380 · c3d6380
1 parent a5c24cc
commit c3d6380
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "bingo",
-  "version": "0.6.4",
+  "version": "0.7.0",
   "private": true,
   "scripts": {
     "dev": "cross-env DEBUG=bingo* next dev --hostname 0.0.0.0",

diff --git a/src/lib/utils.ts b/src/lib/utils.ts
@@ -179,6 +179,7 @@ export function createHeaders(cookies: Partial<{ [key: string]: string }>, type?
         ...cookies,
       })
       headers['x-forwarded-for'] = BING_IP || randomIP()
+      headers['x-ms-useragent'] = headers['x-ms-useragent'] || 'azsdk-js-api-client-factory/1.0.0-beta.1 core-rest-pipeline/1.10.3 OS/Win32'
       return headers
     }
   }

diff --git a/src/pages/api/create.ts b/src/pages/api/create.ts
@@ -11,7 +11,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     const headers = createHeaders(req.cookies)
     do {
       headers['x-forwarded-for'] = headers['x-forwarded-for'] || randomIP()
-      const endpoints = [...(process.env.ENDPOINT || '').split(',').filter(Boolean), 'www.bing.com']
+      const endpoints = [req.headers['x-endpoint'], ...(process.env.ENDPOINT || '').split(',').filter(Boolean), 'www.bing.com']
       const endpoint = endpoints[count % endpoints.length]
       const { conversationId } = req.query
       const query = conversationId ? new URLSearchParams({

diff --git a/src/pages/api/sydney.ts b/src/pages/api/sydney.ts
@@ -21,7 +21,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
   debug(id, conversationContext, headers)
   res.setHeader('Content-Type', 'text/stream; charset=UTF-8')
 
-  const ws = new WebSocket(`wss://${WS_ENDPOINT}/sydney/ChatHub`, {
+  const ws = new WebSocket(`wss://${req.headers['x-ws-endpoint'] || WS_ENDPOINT}/sydney/ChatHub`, {
     headers: {
       ...headers,
       'accept-language': 'zh-CN,zh;q=0.9',