@42ailab/42plugin 0.1.17 → 0.1.18

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@42ailab/42plugin",
-  "version": "0.1.17",
+  "version": "0.1.18",
   "description": "活水插件",
   "type": "module",
   "main": "src/index.ts",

package/src/config.ts

@@ -5,11 +5,12 @@
 import path from 'path';
 import os from 'os';
 
-// 数据目录
+// 数据目录（支持环境变量覆盖，用于测试）
 const dataDir =
-  process.platform === 'win32'
+  process.env.PLUGIN_DATA_DIR ||
+  (process.platform === 'win32'
     ? path.join(process.env.APPDATA || os.homedir(), '42plugin')
-    : path.join(os.homedir(), '.42plugin');
+    : path.join(os.homedir(), '.42plugin'));
 
 export const config = {
   // 目录

package/src/db.ts

@@ -24,9 +24,53 @@ const PROGRESS_THRESHOLD = 1024 * 1024; // 1MB
 
 let db: Database | null = null;
 
+/**
+ * 检查目录权限，返回友好的错误信息
+ */
+async function checkDirectoryPermissions(dir: string): Promise<void> {
+  try {
+    // 检查目录是否存在
+    const stat = await fs.stat(dir);
+
+    // 检查是否可写
+    await fs.access(dir, fs.constants.W_OK);
+
+    // 检查所有者（仅在非 Windows 系统）
+    if (process.platform !== 'win32') {
+      const expectedUid = process.getuid?.();
+      if (expectedUid !== undefined && stat.uid !== expectedUid && stat.uid === 0) {
+        throw new Error('OWNED_BY_ROOT');
+      }
+    }
+  } catch (error) {
+    if ((error as NodeJS.ErrnoException).code === 'ENOENT') {
+      // 目录不存在，稍后会创建
+      return;
+    }
+
+    if ((error as Error).message === 'OWNED_BY_ROOT' ||
+        (error as NodeJS.ErrnoException).code === 'EACCES') {
+      const homeDir = os.homedir();
+      throw new Error(
+        `数据目录权限错误: ${dir}\n\n` +
+        `这通常是因为之前使用了 sudo 运行 CLI 导致的。\n` +
+        `请执行以下命令修复权限:\n\n` +
+        `  sudo chown -R $(whoami) ${path.join(homeDir, '.42plugin')}\n`
+      );
+    }
+
+    throw error;
+  }
+}
+
 async function getDb(): Promise<Database> {
   if (!db) {
-    await fs.mkdir(path.dirname(config.dbPath), { recursive: true });
+    const dir = path.dirname(config.dbPath);
+
+    // 检查权限
+    await checkDirectoryPermissions(dir);
+
+    await fs.mkdir(dir, { recursive: true });
     db = new Database(config.dbPath);
     initSchema();
   }
@@ -34,6 +78,9 @@ async function getDb(): Promise<Database> {
 }
 
 function initSchema(): void {
+  // 检查并执行数据库迁移
+  migrateIfNeeded();
+
   db!.run(`
     CREATE TABLE IF NOT EXISTS projects (
       id TEXT PRIMARY KEY,
@@ -75,6 +122,82 @@ function initSchema(): void {
   `);
 }
 
+/**
+ * 数据库迁移：处理旧版本 schema 升级
+ */
+function migrateIfNeeded(): void {
+  try {
+    // 检查 plugin_cache 表是否存在
+    const tableExists = db!.prepare(`
+      SELECT name FROM sqlite_master
+      WHERE type='table' AND name='plugin_cache'
+    `).get();
+
+    if (!tableExists) {
+      // 表不存在，无需迁移
+      return;
+    }
+
+    // 获取表结构
+    const columns = db!.prepare('PRAGMA table_info(plugin_cache)').all() as { name: string }[];
+    const columnNames = columns.map(c => c.name);
+
+    // 检查是否是旧版本 schema（有 downloaded_at 但没有 cached_at）
+    const hasDownloadedAt = columnNames.includes('downloaded_at');
+    const hasCachedAt = columnNames.includes('cached_at');
+
+    if (hasDownloadedAt && !hasCachedAt) {
+      console.log('检测到旧版本数据库，正在迁移...');
+
+      // 开始事务
+      db!.run('BEGIN TRANSACTION');
+
+      try {
+        // 1. 重命名旧表
+        db!.run('ALTER TABLE plugin_cache RENAME TO plugin_cache_old');
+
+        // 2. 创建新表
+        db!.run(`
+          CREATE TABLE plugin_cache (
+            full_name TEXT NOT NULL,
+            version TEXT NOT NULL,
+            type TEXT NOT NULL,
+            cache_path TEXT NOT NULL,
+            checksum TEXT NOT NULL,
+            size_bytes INTEGER NOT NULL,
+            cached_at TEXT NOT NULL,
+            PRIMARY KEY (full_name, version)
+          )
+        `);
+
+        // 3. 迁移数据（downloaded_at → cached_at）
+        db!.run(`
+          INSERT INTO plugin_cache (full_name, version, type, cache_path, checksum, size_bytes, cached_at)
+          SELECT full_name, version, type, cache_path, checksum, COALESCE(size_bytes, 0), downloaded_at
+          FROM plugin_cache_old
+        `);
+
+        // 4. 删除旧表
+        db!.run('DROP TABLE plugin_cache_old');
+
+        // 提交事务
+        db!.run('COMMIT');
+
+        console.log('数据库迁移完成');
+      } catch (error) {
+        // 回滚事务
+        db!.run('ROLLBACK');
+        throw error;
+      }
+    }
+  } catch (error) {
+    // 迁移失败不应阻止程序运行，只记录警告
+    if (config.debug) {
+      console.warn('数据库迁移检查失败:', error);
+    }
+  }
+}
+
 // ============================================================================
 // 项目管理
 // ============================================================================
@@ -295,15 +418,27 @@ interface Secrets {
 
 async function readSecrets(): Promise<Secrets> {
   try {
+    // 先检查目录权限
+    await checkDirectoryPermissions(path.dirname(SECRETS_FILE));
+
     const content = await fs.readFile(SECRETS_FILE, 'utf-8');
     return JSON.parse(content);
-  } catch {
+  } catch (error) {
+    // 权限错误会直接抛出友好提示
+    if ((error as Error).message.includes('数据目录权限错误')) {
+      throw error;
+    }
     return {};
   }
 }
 
 async function writeSecrets(secrets: Secrets): Promise<void> {
-  await fs.mkdir(path.dirname(SECRETS_FILE), { recursive: true });
+  const dir = path.dirname(SECRETS_FILE);
+
+  // 检查目录权限
+  await checkDirectoryPermissions(dir);
+
+  await fs.mkdir(dir, { recursive: true });
   await fs.writeFile(SECRETS_FILE, JSON.stringify(secrets, null, 2), { mode: 0o600 });
 }