fix: fetch balance with proxy

Author: Kerwin

service/src/chatgpt/index.ts

@@ -180,10 +180,23 @@ async function fetchBalance() {
     'Authorization': `Bearer ${OPENAI_API_KEY}`,
     'Content-Type': 'application/json',
   }
+  let socksAgent
+  let httpsAgent
+  if (isNotEmptyString(config.socksProxy)) {
+    socksAgent = new SocksProxyAgent({
+      hostname: config.socksProxy.split(':')[0],
+      port: parseInt(config.socksProxy.split(':')[1]),
+      userId: isNotEmptyString(config.socksAuth) ? config.socksAuth.split(':')[0] : undefined,
+      password: isNotEmptyString(config.socksAuth) ? config.socksAuth.split(':')[1] : undefined,
+    })
+  }
+  else if (isNotEmptyString(config.httpsProxy)) {
+    httpsAgent = new HttpsProxyAgent(config.httpsProxy)
+  }
 
   try {
     // 获取API限额
-    let response = await fetch(urlSubscription, { headers })
+    let response = await fetch(urlSubscription, { agent: socksAgent === undefined ? httpsAgent : socksAgent, headers })
     if (!response.ok) {
       console.error('您的账户已被封禁，请登录OpenAI进行查看。')
       return
@@ -192,7 +205,7 @@ async function fetchBalance() {
     const totalAmount = subscriptionData.hard_limit_usd
 
     // 获取已使用量
-    response = await fetch(urlUsage, { headers })
+    response = await fetch(urlUsage, { agent: socksAgent === undefined ? httpsAgent : socksAgent, headers })
     const usageData = await response.json()
     const totalUsage = usageData.total_usage / 100
 
@@ -202,7 +215,8 @@ async function fetchBalance() {
 
     return Promise.resolve(cachedBanlance.toFixed(3))
   }
-  catch {
+  catch (error) {
+    global.console.error(error)
     return Promise.resolve('-')
   }
 }