Update hf.js

hf.js

@@ -3,18 +3,14 @@ const morgan = require('morgan');
 const { createProxyMiddleware } = require('http-proxy-middleware');
 const axios = require('axios');
 const fs = require('fs');
-const url = require('url'); // 用于解析代理 URL
+const url = require('url');
 const app = express();
 
 app.use(morgan('dev'));
 
-// 从环境变量获取代理池
-const proxyPool = process.env.PROXY ? process.env.PROXY.split(',').map(p => p.trim()) : [];
-if (proxyPool.length > 0) {
-  console.log('Proxy pool initialized with:', proxyPool);
-} else {
-  console.log('No proxies configured in PROXY environment variable');
-}
+// 全局代理池
+let proxyPool = process.env.PROXY ? process.env.PROXY.split(',').map(p => p.trim()) : [];
+console.log('Initial proxy pool:', proxyPool);
 
 // 从代理池中随机选择一个代理
 function getRandomProxy() {
@@ -22,7 +18,7 @@ function getRandomProxy() {
   const randomIndex = Math.floor(Math.random() * proxyPool.length);
   const proxyUrl = proxyPool[randomIndex];
   const parsedUrl = url.parse(proxyUrl);
-  
+
   return {
     host: parsedUrl.hostname,
     port: parsedUrl.port || 80,
@@ -40,10 +36,39 @@ function configureAxiosProxy() {
     axios.defaults.proxy = proxy;
     console.log(`Axios using proxy: ${proxy.host}:${proxy.port}`);
   } else {
-    delete axios.defaults.proxy; // 如果没有代理，移除默认代理
+    delete axios.defaults.proxy;
+    console.log('No proxy available for axios');
+  }
+}
+
+// 从外部 API 更新代理池
+async function updateProxyPool() {
+  const proxyApiUrl = process.env.PROXY_API_URL || 'http://example.com/api/proxies';
+  try {
+    const response = await axios.get(proxyApiUrl);
+    // 假设 API 返回格式为 { proxies: ["http://user:pass@host:port", ...] }
+    const newProxies = response.data.proxies || [];
+    if (newProxies.length > 0) {
+      proxyPool = newProxies;
+      console.log('Proxy pool updated:', proxyPool);
+      configureAxiosProxy(); // 更新 axios 代理
+    } else {
+      console.warn('No proxies received from API');
+    }
+  } catch (error) {
+    console.error('Failed to update proxy pool:', error.message);
   }
 }
 
+// 定期更新代理池
+const updateInterval = parseInt(process.env.PROXY_UPDATE_INTERVAL) || 300; // 默认 5 分钟
+if (updateInterval > 0) {
+  setInterval(updateProxyPool, updateInterval * 1000);
+  console.log(`Proxy pool will update every ${updateInterval} seconds`);
+  // 启动时立即更新一次
+  updateProxyPool();
+}
+
 // 模型列表 API
 app.get('/hf/v1/models', (req, res) => {
   const models = {
@@ -180,13 +205,15 @@ app.get('/hf/v1/models', (req, res) => {
   res.json(models);
 });
 
-// 代理转发，使用代理池
+// 代理转发，使用动态代理池
 app.use('/hf/v1/chat/completions', (req, res, next) => {
   const proxy = getRandomProxy();
   const middleware = createProxyMiddleware({
-    target: 'http://localhost:3010/v1/chat/completions',
+    target: 'http://localhost:3010/v1/chat/completions', // 可替换为实际目标服务
     changeOrigin: true,
-    proxy: proxy ? proxy : undefined, // 如果有代理则使用
+    proxy: proxy ? proxy : undefined,
+    timeout: 30000, // 超时 30 秒
+    proxyTimeout: 30000, // 代理超时 30 秒
     onProxyReq: (proxyReq, req, res) => {
       if (req.body) {
         const bodyData = JSON.stringify(req.body);
@@ -203,6 +230,8 @@ app.use('/hf/v1/chat/completions', (req, res, next) => {
   });
   if (proxy) {
     console.log(`Request proxied via ${proxy.host}:${proxy.port}`);
+  } else {
+    console.log('No proxy available, direct connection');
   }
   middleware(req, res, next);
 });
@@ -222,7 +251,6 @@ app.get('/', (req, res) => {
           --bg-color: #f8fafc;
           --card-bg: #ffffff;
         }
-        
         body { 
           padding: 20px;
           font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
@@ -312,7 +340,6 @@ app.get('/', (req, res) => {
           body {
             padding: 10px;
           }
-          
           .container {
             padding: 10px;
           }
@@ -374,10 +401,11 @@ app.get('/', (req, res) => {
   res.send(htmlContent);
 });
 
-// 启动服务前配置 axios 代理
+// 启动服务前配置初始 axios 代理
 configureAxiosProxy();
 
 const port = process.env.HF_PORT || 7860;
 app.listen(port, () => {
   console.log(`HF Proxy server is running at PORT: ${port}`);
 });
+