Update server.js

server.js

@@ -22,6 +22,10 @@ if (hfUserConfig) {
   });
 }
 
+// 从环境变量获取登录凭据
+const ADMIN_USERNAME = process.env.USER_NAME || 'admin';
+const ADMIN_PASSWORD = process.env.USER_PASSWORD || 'password';
+
 // 缓存管理
 class SpaceCache {
   constructor() {
@@ -160,7 +164,7 @@ app.post('/api/proxy/rebuild/:repoId(*)', async (req, res) => {
     const payload = { factoryReboot: true };
     console.log(`发送重建请求，payload: ${JSON.stringify(payload)}`);
     const response = await axios.post(`https://huggingface.co/api/spaces/${repoId}/restart`, payload, { headers });
-    console.log(`重建 Space ${repoId} 成功，状态码: ${response.status}, 响应数据:`, response.data);
+    console.log(`重建 Space ${repoId} 成功，状态码: ${response.status}`);
     res.json({ success: true, message: `Space ${repoId} 重建成功` });
   } catch (error) {
     console.error(`重建 space 失败 (${req.params.repoId}):`, error.message);
@@ -260,9 +264,9 @@ app.post('/api/v1/action/:token/:spaceId(*)/rebuild', async (req, res) => {
 
     const headers = { 'Authorization': `Bearer ${token}`, 'Content-Type': 'application/json' };
     const payload = { factoryReboot: true };
-    console.log(`外部 API 发送重建请求，spaceId: ${spaceId}, payload: ${JSON.stringify(payload)}`);
+    console.log(`外部 API 发送重建请求，spaceId: ${spaceId}`);
     const response = await axios.post(`https://huggingface.co/api/spaces/${spaceId}/restart`, payload, { headers });
-    console.log(`外部 API 重建 Space ${spaceId} 成功，状态码: ${response.status}, 响应数据:`, response.data);
+    console.log(`外部 API 重建 Space ${spaceId} 成功，状态码: ${response.status}`);
     res.json({ success: true, message: `Space ${spaceId} 重建成功` });
   } catch (error) {
     console.error(`重建 space 失败 (外部 API):`, error.message);
@@ -295,29 +299,22 @@ app.get('/api/proxy/live-metrics/:username/:instanceId', async (req, res) => {
 
     const token = userTokenMapping[username];
     let headers = {
-      'Accept': 'text/event-stream', // 明确指定接受 SSE 格式
+      'Accept': 'text/event-stream',
       'Cache-Control': 'no-cache',
       'Connection': 'keep-alive'
     };
     if (token) {
       headers['Authorization'] = `Bearer ${token}`;
-      console.log(`使用 Token 访问监控数据 (${username}/${instanceId})`);
-    } else {
-      console.warn(`用户 ${username} 没有配置 API Token，将尝试无认证访问公开数据`);
     }
 
-    console.log(`尝试获取监控数据: ${url}`);
-
     const response = await axios({
       method: 'get',
       url,
       headers,
       responseType: 'stream',
-      timeout: 10000 // 设置超时，防止请求挂起
+      timeout: 10000
     });
 
-    console.log(`成功连接监控数据流 (${username}/${instanceId})，状态码: ${response.status}`);
-
     res.set({
       'Content-Type': 'text/event-stream',
       'Cache-Control': 'no-cache',
@@ -325,28 +322,11 @@ app.get('/api/proxy/live-metrics/:username/:instanceId', async (req, res) => {
     });
     response.data.pipe(res);
 
-    response.data.on('data', (chunk) => {
-      console.log(`接收到监控数据片段 (${username}/${instanceId}): ${chunk.length} 字节`);
-    });
-
-    response.data.on('error', (err) => {
-      console.error(`直播监控数据流错误 (${username}/${instanceId}):`, err.message);
-      res.status(500).end();
-    });
-
     req.on('close', () => {
-      console.log(`客户端关闭监控数据连接 (${username}/${instanceId})`);
       response.data.destroy();
     });
   } catch (error) {
     console.error(`代理获取直播监控数据失败 (${req.params.username}/${req.params.instanceId}):`, error.message);
-    if (error.response) {
-      console.error(`状态码: ${error.response.status}, 响应头:`, error.response.headers);
-    } else if (error.request) {
-      console.error(`请求未收到响应:`, error.message);
-    } else {
-      console.error(`请求配置错误:`, error.message);
-    }
     res.status(error.response?.status || 500).json({ error: '获取监控数据失败', details: error.message });
   }
 });
@@ -359,4 +339,5 @@ app.get('*', (req, res) => {
 app.listen(port, () => {
   console.log(`Server running on port ${port}`);
   console.log(`User configurations:`, usernames.map(user => `${user}: ${userTokenMapping[user] ? 'Token Configured' : 'No Token'}`).join(', ') || 'None');
+  console.log(`Admin login enabled: Username=${ADMIN_USERNAME}, Password=${ADMIN_PASSWORD ? 'Configured' : 'Not Configured'}`);
 });