Update server.js

server.js

@@ -1,10 +1,11 @@
 const express = require('express');
 const path = require('path');
 const axios = require('axios');
+const { HubApi } = require('@huggingface/hub');
 const app = express();
 const port = process.env.PORT || 8080;
 
-// 从环境变量 HF_USER 获取用户和对应的 API Token 映射
+// 从环境变量获取 HuggingFace 用户名和对应的 API Token 映射
 const userTokenMapping = {};
 const usernames = [];
 const hfUserConfig = process.env.HF_USER || '';
@@ -22,6 +23,30 @@ if (hfUserConfig) {
   });
 }
 
+// 缓存管理
+class SpaceCache {
+  constructor() {
+    this.spaces = {};
+    this.lastUpdate = null;
+  }
+
+  updateAll(spacesData) {
+    this.spaces = spacesData.reduce((acc, space) => ({ ...acc, [space.repo_id]: space }), {});
+    this.lastUpdate = Date.now();
+  }
+
+  getAll() {
+    return Object.values(this.spaces);
+  }
+
+  isExpired(expireMinutes = 5) {
+    if (!this.lastUpdate) return true;
+    return (Date.now() - this.lastUpdate) > (expireMinutes * 60 * 1000);
+  }
+}
+
+const spaceCache = new SpaceCache();
+
 // 提供静态文件（前端文件）
 app.use(express.static(path.join(__dirname, 'public')));
 
@@ -30,36 +55,196 @@ app.get('/api/config', (req, res) => {
   res.json({ usernames: usernames.join(',') });
 });
 
-// 代理 HuggingFace API：获取用户的 spaces 列表（包括私有）
+// 获取所有 spaces 列表（包括私有）
 app.get('/api/proxy/spaces', async (req, res) => {
   try {
-    const { author } = req.query;
-    if (!author) {
-      return res.status(400).json({ error: '缺少 author 参数' });
+    if (!spaceCache.isExpired()) {
+      console.log('从缓存获取 Spaces 数据');
+      return res.json(spaceCache.getAll());
     }
 
-    const token = userTokenMapping[author];
-    let response;
-    if (!token) {
-      console.warn(`用户 ${author} 没有配置 API Token，将尝试无认证访问公开数据`);
-      response = await axios.get(`https://huggingface.co/api/spaces?author=${author}`);
-    } else {
-      console.log(`用户 ${author} 使用 Token 访问私有数据`);
-      response = await axios.get(`https://huggingface.co/api/spaces?author=${author}`, {
-        headers: {
-          'Authorization': `Bearer ${token}`
+    const allSpaces = [];
+    for (const username of usernames) {
+      const token = userTokenMapping[username];
+      if (!token) {
+        console.warn(`用户 ${username} 没有配置 API Token，将尝试无认证访问公开数据`);
+      }
+
+      try {
+        const hubApi = new HubApi({ accessToken: token || undefined });
+        const spaces = await hubApi.listSpaces({ author: username });
+        console.log(`获取到 ${spaces.length} 个 Spaces for ${username}`);
+
+        for (const space of spaces) {
+          try {
+            const spaceInfo = await hubApi.getSpaceInfo({ repoId: space.id });
+            const spaceRuntime = spaceInfo.runtime || {};
+
+            allSpaces.push({
+              repo_id: spaceInfo.id,
+              name: spaceInfo.cardData?.title || spaceInfo.id.split('/')[1],
+              owner: spaceInfo.author,
+              username: username,
+              token: token || '',
+              url: `https://${spaceInfo.author}-${spaceInfo.id.split('/')[1]}.hf.space`,
+              status: spaceRuntime.stage || 'unknown',
+              last_modified: spaceInfo.lastModified || 'unknown',
+              created_at: spaceInfo.createdAt || 'unknown',
+              sdk: spaceInfo.sdk || 'unknown',
+              tags: spaceInfo.tags || [],
+              private: spaceInfo.private || false,
+              app_port: spaceInfo.cardData?.app_port || 'unknown'
+            });
+          } catch (error) {
+            console.error(`处理 Space ${space.id} 失败:`, error.message);
+          }
         }
-      });
+      } catch (error) {
+        console.error(`获取 Spaces 列表失败 for ${username}:`, error.message);
+      }
     }
-    console.log(`获取 spaces 列表成功 (${author}): 共 ${response.data.length} 个实例`);
-    res.json(response.data);
+
+    allSpaces.sort((a, b) => a.name.localeCompare(b.name));
+    spaceCache.updateAll(allSpaces);
+    console.log(`总共获取到 ${allSpaces.length} 个 Spaces`);
+    res.json(allSpaces);
+  } catch (error) {
+    console.error(`代理获取 spaces 列表失败:`, error.message);
+    res.status(500).json({ error: '获取 spaces 列表失败', details: error.message });
+  }
+});
+
+// 代理重启 Space
+app.post('/api/proxy/restart/:repoId', async (req, res) => {
+  try {
+    const { repoId } = req.params;
+    const spaces = spaceCache.getAll();
+    const space = spaces.find(s => s.repo_id === repoId);
+    if (!space || !space.token) {
+      return res.status(404).json({ error: 'Space 未找到或无 Token 配置' });
+    }
+
+    const hubApi = new HubApi({ accessToken: space.token });
+    await hubApi.restartSpace({ repoId });
+    res.json({ success: true, message: `Space ${repoId} 重启成功` });
+  } catch (error) {
+    console.error(`重启 space 失败 (${req.params.repoId}):`, error.message);
+    res.status(error.response?.status || 500).json({ error: '重启 space 失败', details: error.message });
+  }
+});
+
+// 代理重建 Space
+app.post('/api/proxy/rebuild/:repoId', async (req, res) => {
+  try {
+    const { repoId } = req.params;
+    const spaces = spaceCache.getAll();
+    const space = spaces.find(s => s.repo_id === repoId);
+    if (!space || !space.token) {
+      return res.status(404).json({ error: 'Space 未找到或无 Token 配置' });
+    }
+
+    const hubApi = new HubApi({ accessToken: space.token });
+    await hubApi.restartSpace({ repoId, factoryReboot: true });
+    res.json({ success: true, message: `Space ${repoId} 重建成功` });
+  } catch (error) {
+    console.error(`重建 space 失败 (${req.params.repoId}):`, error.message);
+    res.status(error.response?.status || 500).json({ error: '重建 space 失败', details: error.message });
+  }
+});
+
+// 外部 API 服务（类似于 Flask 的 /api/v1）
+app.get('/api/v1/info/:token', async (req, res) => {
+  try {
+    const { token } = req.params;
+    const authHeader = req.headers.authorization;
+    if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.split(' ')[1] !== process.env.API_KEY) {
+      return res.status(401).json({ error: '无效的 API 密钥' });
+    }
+
+    const hubApi = new HubApi({ accessToken: token });
+    const userInfo = await hubApi.whoami();
+    const username = userInfo.name;
+    const spaces = await hubApi.listSpaces({ author: username });
+    const spaceList = [];
+
+    for (const space of spaces) {
+      try {
+        const spaceInfo = await hubApi.getSpaceInfo({ repoId: space.id });
+        spaceList.push(spaceInfo.id);
+      } catch (error) {
+        console.error(`获取 Space 信息失败 (${space.id}):`, error.message);
+      }
+    }
+
+    res.json({ spaces: spaceList, total: spaceList.length });
+  } catch (error) {
+    console.error(`获取 spaces 列表失败 (外部 API):`, error.message);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+app.get('/api/v1/info/:token/:spaceId(*) ', async (req, res) => {
+  try {
+    const { token, spaceId } = req.params;
+    const authHeader = req.headers.authorization;
+    if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.split(' ')[1] !== process.env.API_KEY) {
+      return res.status(401).json({ error: '无效的 API 密钥' });
+    }
+
+    const hubApi = new HubApi({ accessToken: token });
+    const spaceInfo = await hubApi.getSpaceInfo({ repoId: spaceId });
+    const spaceRuntime = spaceInfo.runtime || {};
+
+    res.json({
+      id: spaceInfo.id,
+      status: spaceRuntime.stage || 'unknown',
+      last_modified: spaceInfo.lastModified || null,
+      created_at: spaceInfo.createdAt || null,
+      sdk: spaceInfo.sdk || 'unknown',
+      tags: spaceInfo.tags || [],
+      private: spaceInfo.private || false
+    });
+  } catch (error) {
+    console.error(`获取 space 信息失败 (外部 API):`, error.message);
+    res.status(error.response?.status || 404).json({ error: error.message });
+  }
+});
+
+app.post('/api/v1/action/:token/:spaceId(*)/restart', async (req, res) => {
+  try {
+    const { token, spaceId } = req.params;
+    const authHeader = req.headers.authorization;
+    if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.split(' ')[1] !== process.env.API_KEY) {
+      return res.status(401).json({ error: '无效的 API 密钥' });
+    }
+
+    const hubApi = new HubApi({ accessToken: token });
+    await hubApi.restartSpace({ repoId: spaceId });
+    res.json({ success: true, message: `Space ${spaceId} 重启成功` });
+  } catch (error) {
+    console.error(`重启 space 失败 (外部 API):`, error.message);
+    res.status(error.response?.status || 500).json({ success: false, error: error.message });
+  }
+});
+
+app.post('/api/v1/action/:token/:spaceId(*)/rebuild', async (req, res) => {
+  try {
+    const { token, spaceId } = req.params;
+    const authHeader = req.headers.authorization;
+    if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.split(' ')[1] !== process.env.API_KEY) {
+      return res.status(401).json({ error: '无效的 API 密钥' });
+    }
+
+    const hubApi = new HubApi({ accessToken: token });
+    await hubApi.restartSpace({ repoId: spaceId, factoryReboot: true });
+    res.json({ success: true, message: `Space ${spaceId} 重建成功` });
   } catch (error) {
-    console.error(`代理获取 spaces 列表失败 (${req.query.author}):`, error.message, error.response?.status || '无状态码');
-    res.status(error.response?.status || 500).json({ error: '获取 spaces 列表失败', details: error.message });
+    console.error(`重建 space 失败 (外部 API):`, error.message);
+    res.status(error.response?.status || 500).json({ success: false, error: error.message });
   }
 });
 
-// 代理 HuggingFace API：获取实时监控数据（SSE 需要特殊处理）
+// 代理 HuggingFace API：获取实时监控数据（SSE）
 app.get('/api/proxy/live-metrics/:username/:instanceId', async (req, res) => {
   try {
     const { username, instanceId } = req.params;
@@ -69,12 +254,10 @@ app.get('/api/proxy/live-metrics/:username/:instanceId', async (req, res) => {
     let headers = {};
     if (token) {
       headers = { 'Authorization': `Bearer ${token}` };
-      console.log(`用戶 ${username}/${instanceId} 使用 Token 連接監控數據`);
     } else {
       console.warn(`用户 ${username} 没有配置 API Token，将尝试无认证访问公开数据`);
     }
 
-    // 发起请求并设置流式响应
     const response = await axios({
       method: 'get',
       url,
@@ -82,7 +265,6 @@ app.get('/api/proxy/live-metrics/:username/:instanceId', async (req, res) => {
       responseType: 'stream'
     });
 
-    // 将响应流转发到客户端
     res.set({
       'Content-Type': 'text/event-stream',
       'Cache-Control': 'no-cache',
@@ -90,7 +272,6 @@ app.get('/api/proxy/live-metrics/:username/:instanceId', async (req, res) => {
     });
     response.data.pipe(res);
 
-    // 监听错误
     response.data.on('error', (err) => {
       console.error(`直播监控数据流错误 (${username}/${instanceId}):`, err);
       res.status(500).end();
@@ -100,7 +281,7 @@ app.get('/api/proxy/live-metrics/:username/:instanceId', async (req, res) => {
       response.data.destroy();
     });
   } catch (error) {
-    console.error(`代理获取直播监控数据失败 (${req.params.username}/${req.params.instanceId}):`, error.message, error.response?.status || '无状态码');
+    console.error(`代理获取直播监控数据失败 (${req.params.username}/${req.params.instanceId}):`, error.message);
     res.status(error.response?.status || 500).json({ error: '获取监控数据失败', details: error.message });
   }
 });