Update server.js

server.js

@@ -3,26 +3,24 @@ const http = require('http');
 const os = require('os');
 const fs = require('fs');
 const { exec } = require('child_process');
-const path = require('path');
 
 // --- Hugging Face Spaces 核心配置 ---
-// 必须监听所有接口，以响应外部请求
 const HOST = '0.0.0.0';
-// 必须使用 Hugging Face 指定的端口
 const PORT = process.env.PORT || 7860;
 
-// --- 辅助函数：获取磁盘使用情况 ---
+// --- 辅助函数：获取磁盘使用情况（需要处理 /data 可能不存在）---
 function getDiskUsage() {
     try {
-        const stats = fs.statfsSync('/'); // 获取根目录的磁盘信息
+        // 获取当前工作目录所在的文件系统（通常是容器根目录）
+        const stats = fs.statfsSync('/');
         const total = stats.bsize * stats.blocks;
         const free = stats.bsize * stats.bfree;
         const used = total - free;
         const usagePercent = ((used / total) * 100).toFixed(2);
         return {
-            total: (total / (1024 ** 3)).toFixed(2),   // GB
-            used: (used / (1024 ** 3)).toFixed(2),     // GB
-            free: (free / (1024 ** 3)).toFixed(2),     // GB
+            total: (total / (1024 ** 3)).toFixed(2),
+            used: (used / (1024 ** 3)).toFixed(2),
+            free: (free / (1024 ** 3)).toFixed(2),
             usagePercent: `${usagePercent}%`
         };
     } catch (error) {
@@ -30,21 +28,17 @@ function getDiskUsage() {
     }
 }
 
-// --- 辅助函数：获取出口公网IP（cURL方式）---
+// --- 辅助函数：获取出口公网IP---
 function getPublicIP() {
     return new Promise(resolve => {
-        // Hugging Face Spaces 容器内通常可以访问外部网络
         exec('curl -s --max-time 5 ifconfig.me', (err, stdout) => {
-            if (err) {
-                resolve('无法获取公网IP (可能为NAT环境)');
-            } else {
-                resolve(stdout.trim());
-            }
+            if (err) resolve('无法获取公网IP (可能为NAT环境)');
+            else resolve(stdout.trim());
         });
     });
 }
 
-// --- 辅助函数：获取所有环境变量（过滤掉敏感信息）---
+// --- 辅助函数：安全获取环境变量---
 function getSafeEnvVars() {
     const safeVars = {};
     const sensitiveKeys = ['PASSWORD', 'SECRET', 'TOKEN', 'KEY'];
@@ -58,29 +52,47 @@ function getSafeEnvVars() {
     return safeVars;
 }
 
+// --- 辅助函数：检查 /data 目录状态（兼容不存在的情况）---
+function checkDataDir() {
+    try {
+        if (fs.existsSync('/data')) {
+            // 目录存在，检查是否可写
+            fs.accessSync('/data', fs.constants.W_OK);
+            return { exists: true, writable: true };
+        } else {
+            return { exists: false, writable: false, reason: '目录不存在（免费层未挂载持久存储）' };
+        }
+    } catch (err) {
+        return { exists: true, writable: false, reason: err.message };
+    }
+}
+
 // --- 创建 HTTP 服务器 ---
 const server = http.createServer(async (req, res) => {
-    // 1. 获取动态信息
+    // 1. 动态信息
     const publicIP = await getPublicIP();
     const diskInfo = getDiskUsage();
+    const dataDirStatus = checkDataDir();
     
-    // 2. 系统与硬件信息 (通过 os 模块)
+    // 2. 系统与硬件信息
     const cpuInfo = os.cpus()[0];
-    const totalMem = (os.totalmem() / (1024 ** 3)).toFixed(2);     // GB
-    const freeMem = (os.freemem() / (1024 ** 3)).toFixed(2);       // GB
-    const usedMem = (totalMem - freeMem).toFixed(2);               // GB
+    const totalMem = (os.totalmem() / (1024 ** 3)).toFixed(2);
+    const freeMem = (os.freemem() / (1024 ** 3)).toFixed(2);
+    const usedMem = (totalMem - freeMem).toFixed(2);
     const systemUptime = Math.floor(os.uptime() / 60) + ' 分钟';
     
-    // 3. 关键环境检查
-    const isNodePathSet = !!process.env.NODE_PATH;
-    const isDataDirWritable = fs.accessSync('/data', fs.constants.W_OK) === undefined; // 检查 /data 是否可写
-    const isHfHomeSet = !!process.env.HF_HOME;
+    // 3. 用户ID和权限
+    let userId = null, groupId = null;
+    try {
+        userId = process.getuid();
+        groupId = process.getgid();
+    } catch(e) { /* Windows 可能不支持 */ }
     
-    // 4. 构建用于展示的数据模型
+    // 4. 构建环境报告
     const envReport = {
-        // ========== 1. 访问信息 ==========
+        // 访问信息
         access: {
-            spaceUrl: `https://${process.env.SPACE_ID?.replace('/', '-')}.hf.space` || '未知 (请通过公网域名访问)',
+            spaceUrl: `https://${process.env.SPACE_ID?.replace('/', '-') || 'unknown'}.hf.space`,
             publicIP: publicIP,
             internalIP: (() => {
                 const interfaces = os.networkInterfaces();
@@ -92,10 +104,10 @@ const server = http.createServer(async (req, res) => {
                 return '127.0.0.1';
             })(),
             hostname: os.hostname(),
-            spaceId: process.env.SPACE_ID || '未知',
-            spaceRepoName: process.env.SPACE_REPO_NAME || '未知',
+            spaceId: process.env.SPACE_ID || '未设置',
+            spaceRepoName: process.env.SPACE_REPO_NAME || '未设置',
         },
-        // ========== 2. 硬件资源 (免费层标准) ==========
+        // 硬件资源
         hardware: {
             cpuModel: cpuInfo?.model || '未知',
             cpuCores: os.cpus().length,
@@ -106,38 +118,37 @@ const server = http.createServer(async (req, res) => {
             uptime: systemUptime,
             expectedFreeTier: '2 vCPU / 16GB RAM / 50GB 临时磁盘'
         },
-        // ========== 3. Node.js 环境 ==========
+        // Node.js 环境
         nodeEnv: {
             version: process.version,
             platform: process.platform,
             arch: process.arch,
             execPath: process.execPath,
             cwd: process.cwd(),
-            nodePath: isNodePathSet ? process.env.NODE_PATH : '未设置'
         },
-        // ========== 4. 容器与权限 ==========
+        // 容器与权限
         container: {
-            userId: process.getuid?.(),
-            groupId: process.getgid?.(),
-            expectedUserId: '1000 (Hugging Face 默认用户ID)',
-            isDataDirWritable: isDataDirWritable,
-            hfHome: process.env.HF_HOME || '未设置 (可设置为 /data/.huggingface 加速重启)',
-            // 检测是否使用了 Persistent Storage
-            persistentStorageMounted: fs.existsSync('/data'),
+            userId: userId,
+            groupId: groupId,
+            expectedUserId: '通常为 1000 (node 用户)',
+            isDataDirPresent: dataDirStatus.exists,
+            isDataDirWritable: dataDirStatus.writable,
+            dataDirDetail: dataDirStatus.reason || (dataDirStatus.writable ? '可写' : '不可写'),
+            hfHome: process.env.HF_HOME || '未设置',
+            persistentStorageAvailable: dataDirStatus.exists && dataDirStatus.writable,
         },
-        // ========== 5. 关键环境变量 (安全过滤) ==========
+        // 关键环境变量（安全过滤）
         envVars: getSafeEnvVars(),
-        // ========== 6. 部署建议 ==========
+        // 部署建议
         suggestions: [
             '✅ 必须监听 0.0.0.0，并在 README.md 中设置 app_port: 7860',
-            '✅ 强烈建议为敏感信息（如 OPENCODE_PASSWORD）使用 HF Secrets，勿硬编码',
-            '✅ 如需数据持久化，请考虑升级 Persistent Storage 或将数据备份到 HF Dataset',
-            '✅ 容器默认以用户 ID 1000 运行，Dockerfile 中需创建此用户避免权限问题',
-            '✅ 可设置 HF_HOME=/data/.huggingface 以加速 Hugging Face 库的重复启动',
+            '✅ 强烈建议为敏感信息使用 HF Secrets，勿硬编码',
+            '✅ 免费层磁盘非持久化，重启会丢失数据',
+            '✅ 如需持久化，考虑升级 Persistent Storage 或将数据备份到 HF Dataset',
+            '✅ 容器默认用户 UID 可能为 1000，Dockerfile 中建议使用现有 node 用户',
         ]
     };
     
-    // 5. 输出美观的 JSON 报告
     res.writeHead(200, { 'Content-Type': 'application/json; charset=utf-8' });
     res.end(JSON.stringify(envReport, null, 2));
 });