Upload 27 files

src/utils/storage.js

@@ -15,9 +15,15 @@ const logger = createLogger('Storage');
  */
 class StorageManager {
   constructor() {
+    // 优先使用环境变量 DATA_DIR，如果未设置，则默认为项目根目录下的 'data' 文件夹
+    // 这样可以灵活配置数据存储路径，以适应不同的部署环境（如 Hugging Face Spaces）
+    const dataDir = process.env.DATA_DIR || path.join(process.cwd(), 'data');
+    
+    logger.info(`Data directory is set to: ${dataDir}`);
+
     // 数据文件路径
-    this.dataFilePath = path.join(dirname(dirname(__dirname)), 'data', 'cookies-data.json');
-    this.backupFilePath = path.join(dirname(dirname(__dirname)), 'data', 'cookies-data.backup.json');
+    this.dataFilePath = path.join(dataDir, 'cookies-data.json');
+    this.backupFilePath = path.join(dataDir, 'cookies-data.backup.json');
     
     // 确保数据目录存在
     this.ensureDataDirectory();
@@ -29,8 +35,14 @@ class StorageManager {
   ensureDataDirectory() {
     const dataDir = path.dirname(this.dataFilePath);
     if (!fs.existsSync(dataDir)) {
-      fs.mkdirSync(dataDir, { recursive: true });
-      logger.info(`创建数据目录: ${dataDir}`);
+      try {
+        fs.mkdirSync(dataDir, { recursive: true });
+        logger.info(`创建数据目录: ${dataDir}`);
+      } catch (error) {
+        logger.error(`创建数据目录失败: ${dataDir}. 请确保此路径具有写入权限。`);
+        logger.error('在 Hugging Face Spaces 等环境中，您可能需要设置 DATA_DIR 环境变量。');
+        throw error;
+      }
     }
   }