@42ailab/42plugin 0.1.0-beta.0 → 0.1.2

package/src/db.ts

@@ -0,0 +1,593 @@
+/**
+ * 本地存储
+ *
+ * 单文件封装 SQLite 存储和文件操作
+ */
+
+import { createClient, type Client } from '@libsql/client';
+import fs from 'fs/promises';
+import path from 'path';
+import crypto from 'crypto';
+import * as tar from 'tar';
+import cliProgress from 'cli-progress';
+import { config } from './config';
+import { formatBytes } from './utils';
+import type { LocalProject, LocalCache, LocalInstallation, PluginType, PluginDownloadInfo } from './types';
+
+// 文件大小超过 1MB 时显示下载进度条
+const PROGRESS_THRESHOLD = 1024 * 1024; // 1MB
+
+// ============================================================================
+// SQLite 客户端
+// ============================================================================
+
+let db: Client | null = null;
+
+async function getDb(): Promise<Client> {
+  if (!db) {
+    await fs.mkdir(path.dirname(config.dbPath), { recursive: true });
+    db = createClient({ url: `file:${config.dbPath}` });
+    await initSchema();
+  }
+  return db;
+}
+
+async function initSchema(): Promise<void> {
+  const client = db!;
+
+  await client.execute(`
+    CREATE TABLE IF NOT EXISTS projects (
+      id TEXT PRIMARY KEY,
+      path TEXT UNIQUE NOT NULL,
+      name TEXT NOT NULL,
+      registered_at TEXT NOT NULL,
+      last_used_at TEXT NOT NULL
+    )
+  `);
+
+  await client.execute(`
+    CREATE TABLE IF NOT EXISTS 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)
+    )
+  `);
+
+  await client.execute(`
+    CREATE TABLE IF NOT EXISTS installations (
+      id TEXT PRIMARY KEY,
+      project_id TEXT NOT NULL,
+      full_name TEXT NOT NULL,
+      type TEXT NOT NULL,
+      version TEXT NOT NULL,
+      cache_path TEXT NOT NULL,
+      link_path TEXT NOT NULL,
+      source TEXT NOT NULL,
+      source_kit TEXT,
+      installed_at TEXT NOT NULL,
+      FOREIGN KEY (project_id) REFERENCES projects(id),
+      UNIQUE (project_id, full_name)
+    )
+  `);
+}
+
+// ============================================================================
+// 项目管理
+// ============================================================================
+
+export async function getOrCreateProject(projectPath: string): Promise<LocalProject> {
+  const client = await getDb();
+  const absPath = path.resolve(projectPath);
+  const name = path.basename(absPath);
+
+  // 查找现有项目
+  const result = await client.execute({
+    sql: 'SELECT * FROM projects WHERE path = ?',
+    args: [absPath],
+  });
+
+  if (result.rows.length > 0) {
+    const row = result.rows[0];
+    // 更新最后使用时间
+    await client.execute({
+      sql: 'UPDATE projects SET last_used_at = ? WHERE id = ?',
+      args: [new Date().toISOString(), row.id],
+    });
+    return {
+      id: row.id as string,
+      path: row.path as string,
+      name: row.name as string,
+      registeredAt: row.registered_at as string,
+      lastUsedAt: new Date().toISOString(),
+    };
+  }
+
+  // 创建新项目
+  const id = crypto.randomUUID();
+  const now = new Date().toISOString();
+  await client.execute({
+    sql: 'INSERT INTO projects (id, path, name, registered_at, last_used_at) VALUES (?, ?, ?, ?, ?)',
+    args: [id, absPath, name, now, now],
+  });
+
+  return { id, path: absPath, name, registeredAt: now, lastUsedAt: now };
+}
+
+// ============================================================================
+// 缓存管理
+// ============================================================================
+
+export async function getCache(fullName: string, version: string): Promise<LocalCache | null> {
+  const client = await getDb();
+  const result = await client.execute({
+    sql: 'SELECT * FROM plugin_cache WHERE full_name = ? AND version = ?',
+    args: [fullName, version],
+  });
+
+  if (result.rows.length === 0) return null;
+
+  const row = result.rows[0];
+  return {
+    fullName: row.full_name as string,
+    type: row.type as PluginType,
+    version: row.version as string,
+    cachePath: row.cache_path as string,
+    checksum: row.checksum as string,
+    sizeBytes: row.size_bytes as number,
+    cachedAt: row.cached_at as string,
+  };
+}
+
+export async function setCache(cache: Omit<LocalCache, 'cachedAt'>): Promise<void> {
+  const client = await getDb();
+  const now = new Date().toISOString();
+
+  await client.execute({
+    sql: `INSERT OR REPLACE INTO plugin_cache
+          (full_name, version, type, cache_path, checksum, size_bytes, cached_at)
+          VALUES (?, ?, ?, ?, ?, ?, ?)`,
+    args: [
+      cache.fullName,
+      cache.version,
+      cache.type,
+      cache.cachePath,
+      cache.checksum,
+      cache.sizeBytes,
+      now,
+    ],
+  });
+}
+
+export async function removeCache(fullName: string, version: string): Promise<boolean> {
+  const client = await getDb();
+
+  // 先获取缓存路径
+  const cache = await getCache(fullName, version);
+  if (!cache) return false;
+
+  // 删除缓存文件/目录
+  try {
+    // cachePath 可能是文件或目录，需要获取版本目录
+    const versionDir = path.dirname(cache.cachePath);
+    await fs.rm(versionDir, { recursive: true, force: true });
+
+    // 尝试清理空的父目录
+    const pluginDir = path.dirname(versionDir);
+    const remaining = await fs.readdir(pluginDir);
+    if (remaining.length === 0) {
+      await fs.rm(pluginDir, { recursive: true, force: true });
+    }
+  } catch {
+    // 文件不存在，忽略
+  }
+
+  // 删除数据库记录
+  const result = await client.execute({
+    sql: 'DELETE FROM plugin_cache WHERE full_name = ? AND version = ?',
+    args: [fullName, version],
+  });
+
+  return result.rowsAffected > 0;
+}
+
+// ============================================================================
+// 安装记录
+// ============================================================================
+
+export async function addInstallation(data: {
+  projectId: string;
+  fullName: string;
+  type: PluginType;
+  version: string;
+  cachePath: string;
+  linkPath: string;
+  source: 'direct' | 'kit';
+  sourceKit?: string;
+}): Promise<LocalInstallation> {
+  const client = await getDb();
+  const id = crypto.randomUUID();
+  const now = new Date().toISOString();
+
+  await client.execute({
+    sql: `INSERT OR REPLACE INTO installations
+          (id, project_id, full_name, type, version, cache_path, link_path, source, source_kit, installed_at)
+          VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+    args: [
+      id,
+      data.projectId,
+      data.fullName,
+      data.type,
+      data.version,
+      data.cachePath,
+      data.linkPath,
+      data.source,
+      data.sourceKit || null,
+      now,
+    ],
+  });
+
+  return { id, ...data, installedAt: now };
+}
+
+export async function getInstallations(projectPath: string): Promise<LocalInstallation[]> {
+  const client = await getDb();
+  const absPath = path.resolve(projectPath);
+
+  const result = await client.execute({
+    sql: `SELECT i.* FROM installations i
+          JOIN projects p ON i.project_id = p.id
+          WHERE p.path = ?
+          ORDER BY i.installed_at DESC`,
+    args: [absPath],
+  });
+
+  return result.rows.map((row) => ({
+    id: row.id as string,
+    projectId: row.project_id as string,
+    fullName: row.full_name as string,
+    type: row.type as PluginType,
+    version: row.version as string,
+    cachePath: row.cache_path as string,
+    linkPath: row.link_path as string,
+    source: row.source as 'direct' | 'kit',
+    sourceKit: row.source_kit as string | undefined,
+    installedAt: row.installed_at as string,
+  }));
+}
+
+export async function removeInstallation(projectPath: string, fullName: string): Promise<boolean> {
+  const client = await getDb();
+  const absPath = path.resolve(projectPath);
+
+  const result = await client.execute({
+    sql: `DELETE FROM installations
+          WHERE full_name = ?
+          AND project_id IN (SELECT id FROM projects WHERE path = ?)`,
+    args: [fullName, absPath],
+  });
+
+  return result.rowsAffected > 0;
+}
+
+// ============================================================================
+// 密钥管理
+// ============================================================================
+
+const SECRETS_FILE = path.join(config.dataDir, 'secrets.json');
+
+interface Secrets {
+  sessionToken?: string;
+}
+
+async function readSecrets(): Promise<Secrets> {
+  try {
+    const content = await fs.readFile(SECRETS_FILE, 'utf-8');
+    return JSON.parse(content);
+  } catch {
+    return {};
+  }
+}
+
+async function writeSecrets(secrets: Secrets): Promise<void> {
+  await fs.mkdir(path.dirname(SECRETS_FILE), { recursive: true });
+  await fs.writeFile(SECRETS_FILE, JSON.stringify(secrets, null, 2), { mode: 0o600 });
+}
+
+export async function getSessionToken(): Promise<string | null> {
+  const secrets = await readSecrets();
+  return secrets.sessionToken || null;
+}
+
+export async function saveSessionToken(token: string): Promise<void> {
+  const secrets = await readSecrets();
+  secrets.sessionToken = token;
+  await writeSecrets(secrets);
+}
+
+export async function clearSessionToken(): Promise<void> {
+  const secrets = await readSecrets();
+  delete secrets.sessionToken;
+  await writeSecrets(secrets);
+}
+
+// ============================================================================
+// 文件操作
+// ============================================================================
+
+export async function downloadAndExtract(
+  url: string,
+  expectedChecksum: string,
+  fullName: string,
+  version: string
+): Promise<string> {
+  const parts = fullName.split('/');
+  const targetDir = path.join(config.cacheDir, ...parts, version);
+  await fs.mkdir(targetDir, { recursive: true });
+
+  const tempFile = path.join(targetDir, '.download.tmp');
+
+  try {
+    // 下载
+    const response = await fetch(url, {
+      headers: { 'User-Agent': '42plugin-cli/1.0.0' },
+    });
+
+    if (!response.ok) {
+      throw new Error(`下载失败: ${response.status}`);
+    }
+
+    const contentLength = parseInt(response.headers.get('content-length') || '0');
+    const showProgress = contentLength > PROGRESS_THRESHOLD;
+
+    let buffer: Buffer;
+
+    if (showProgress && response.body) {
+      // 流式下载，显示进度条
+      const reader = response.body.getReader();
+      const chunks: Uint8Array[] = [];
+      let downloadedBytes = 0;
+      const startTime = Date.now();
+
+      const progressBar = new cliProgress.SingleBar({
+        format: '  下载中... {bar} {percentage}% ({downloaded}/{total}) {speed}',
+        barCompleteChar: '█',
+        barIncompleteChar: '░',
+        hideCursor: true,
+      });
+
+      progressBar.start(contentLength, 0, {
+        downloaded: '0 B',
+        total: formatBytes(contentLength),
+        speed: '0 B/s',
+      });
+
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+
+        chunks.push(value);
+        downloadedBytes += value.length;
+
+        const elapsed = (Date.now() - startTime) / 1000;
+        const speed = elapsed > 0 ? downloadedBytes / elapsed : 0;
+
+        progressBar.update(downloadedBytes, {
+          downloaded: formatBytes(downloadedBytes),
+          speed: formatBytes(speed) + '/s',
+        });
+      }
+
+      progressBar.stop();
+      buffer = Buffer.concat(chunks.map((c) => Buffer.from(c)));
+    } else {
+      // 小文件直接下载
+      buffer = Buffer.from(await response.arrayBuffer());
+    }
+
+    await fs.writeFile(tempFile, buffer);
+
+    // 校验
+    const actualChecksum = crypto.createHash('sha256').update(buffer).digest('hex');
+    const expectedHash = expectedChecksum.replace(/^sha256:/, '');
+
+    if (actualChecksum !== expectedHash) {
+      throw new Error(`校验和不匹配: 期望 ${expectedHash}, 实际 ${actualChecksum}`);
+    }
+
+    // 解压
+    await tar.extract({ file: tempFile, cwd: targetDir, strip: 0 });
+
+    // 清理
+    await fs.unlink(tempFile).catch(() => {});
+
+    // 返回最终路径（如果只有一个子目录则进入）
+    return resolveFinalPath(targetDir);
+  } catch (error) {
+    await fs.rm(targetDir, { recursive: true, force: true }).catch(() => {});
+    throw error;
+  }
+}
+
+/**
+ * 修正旧缓存路径（如果是目录但应该是文件）
+ */
+async function correctCachePath(cachePath: string): Promise<string> {
+  try {
+    const stat = await fs.stat(cachePath);
+    if (!stat.isDirectory()) {
+      // 已经是文件，不需要修正
+      return cachePath;
+    }
+
+    // 是目录，检查是否应该返回其中的文件
+    const entries = await fs.readdir(cachePath, { withFileTypes: true });
+    const files = entries.filter((e) => e.isFile() && !e.name.startsWith('.'));
+    const dirs = entries.filter((e) => e.isDirectory() && !e.name.startsWith('.'));
+    const nonMetadataFiles = files.filter((f) => f.name !== 'metadata.json');
+
+    // 如果只有一个主文件（排除 metadata.json），返回该文件路径
+    if (nonMetadataFiles.length === 1 && dirs.length === 0) {
+      return path.join(cachePath, nonMetadataFiles[0].name);
+    }
+
+    return cachePath;
+  } catch {
+    return cachePath;
+  }
+}
+
+async function resolveFinalPath(dir: string): Promise<string> {
+  const entries = await fs.readdir(dir, { withFileTypes: true });
+  const dirs = entries.filter((e) => e.isDirectory() && !e.name.startsWith('.'));
+  const files = entries.filter((e) => e.isFile() && !e.name.startsWith('.'));
+
+  // 如果只有一个子目录（且没有其他文件，除了 metadata.json），进入该目录
+  const nonMetadataFiles = files.filter((f) => f.name !== 'metadata.json');
+  if (dirs.length === 1 && nonMetadataFiles.length === 0) {
+    const subDir = path.join(dir, dirs[0].name);
+    const subEntries = await fs.readdir(subDir);
+    if (subEntries.length > 0) {
+      return subDir;
+    }
+  }
+
+  // 如果只有一个主文件（排除 metadata.json），返回该文件路径
+  // 这是单文件插件（如 agent/command）的情况
+  if (nonMetadataFiles.length === 1 && dirs.length === 0) {
+    return path.join(dir, nonMetadataFiles[0].name);
+  }
+
+  return dir;
+}
+
+export async function getDirectorySize(pathOrDir: string): Promise<number> {
+  const stat = await fs.stat(pathOrDir);
+
+  // 如果是文件，直接返回文件大小
+  if (stat.isFile()) {
+    return stat.size;
+  }
+
+  // 如果是目录，递归计算总大小
+  let totalSize = 0;
+
+  const walk = async (dir: string): Promise<void> => {
+    const entries = await fs.readdir(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      const fullPath = path.join(dir, entry.name);
+      if (entry.isDirectory()) {
+        await walk(fullPath);
+      } else {
+        const fileStat = await fs.stat(fullPath);
+        totalSize += fileStat.size;
+      }
+    }
+  };
+
+  await walk(pathOrDir);
+  return totalSize;
+}
+
+export async function createLink(
+  sourcePath: string,
+  targetPath: string
+): Promise<void> {
+  // 移除目标路径末尾的斜杠
+  const normalizedTarget = targetPath.replace(/\/+$/, '');
+  const targetDir = path.dirname(normalizedTarget);
+  await fs.mkdir(targetDir, { recursive: true });
+
+  // 移除已存在的链接
+  try {
+    const stat = await fs.lstat(normalizedTarget);
+    if (stat.isSymbolicLink() || stat.isDirectory() || stat.isFile()) {
+      await fs.rm(normalizedTarget, { recursive: true, force: true });
+    }
+  } catch {
+    // 不存在，继续
+  }
+
+  // 检测源是文件还是目录，创建对应类型的符号链接
+  const sourceStat = await fs.stat(sourcePath);
+  const linkType = sourceStat.isDirectory() ? 'dir' : 'file';
+  await fs.symlink(sourcePath, normalizedTarget, linkType);
+}
+
+export async function removeLink(linkPath: string): Promise<void> {
+  try {
+    const stat = await fs.lstat(linkPath);
+    if (stat.isSymbolicLink()) {
+      await fs.unlink(linkPath);
+    } else if (stat.isDirectory()) {
+      await fs.rm(linkPath, { recursive: true, force: true });
+    }
+  } catch {
+    // 不存在，忽略
+  }
+}
+
+// ============================================================================
+// 缓存路径解析
+// ============================================================================
+
+/**
+ * 解析缓存路径（检查缓存或下载新文件）
+ */
+export async function resolveCachePath(
+  downloadInfo: PluginDownloadInfo,
+  force: boolean,
+  useCache: boolean
+): Promise<{ cachePath: string; fromCache: boolean }> {
+  // 检查缓存
+  if (!force && useCache) {
+    const cached = await getCache(downloadInfo.fullName, downloadInfo.version);
+    if (cached && cached.checksum === downloadInfo.checksum) {
+      // 验证缓存文件是否仍然存在
+      try {
+        await fs.access(cached.cachePath);
+        // 修正旧缓存的路径（可能是目录而应该是文件）
+        const correctedPath = await correctCachePath(cached.cachePath);
+        // 如果路径被修正，更新缓存记录
+        if (correctedPath !== cached.cachePath) {
+          await setCache({
+            fullName: cached.fullName,
+            type: cached.type,
+            version: cached.version,
+            cachePath: correctedPath,
+            checksum: cached.checksum,
+            sizeBytes: cached.sizeBytes,
+          });
+        }
+        return { cachePath: correctedPath, fromCache: true };
+      } catch {
+        // 缓存文件不存在，继续下载
+      }
+    }
+  }
+
+  // 下载
+  const cachePath = await downloadAndExtract(
+    downloadInfo.downloadUrl,
+    downloadInfo.checksum,
+    downloadInfo.fullName,
+    downloadInfo.version
+  );
+
+  // 更新缓存记录
+  const size = await getDirectorySize(cachePath);
+  await setCache({
+    fullName: downloadInfo.fullName,
+    type: downloadInfo.type,
+    version: downloadInfo.version,
+    cachePath,
+    checksum: downloadInfo.checksum,
+    sizeBytes: size,
+  });
+
+  return { cachePath, fromCache: false };
+}

package/src/errors.ts

@@ -0,0 +1,40 @@
+/**
+ * 发布相关错误类型
+ */
+
+export class ValidationError extends Error {
+  constructor(
+    message: string,
+    public code: string
+  ) {
+    super(message);
+    this.name = 'ValidationError';
+  }
+}
+
+export class VersionConflictError extends Error {
+  constructor(
+    message: string,
+    public existingVersion: string
+  ) {
+    super(message);
+    this.name = 'VersionConflictError';
+  }
+}
+
+export class UploadError extends Error {
+  constructor(
+    message: string,
+    public statusCode?: number
+  ) {
+    super(message);
+    this.name = 'UploadError';
+  }
+}
+
+export class AuthRequiredError extends Error {
+  constructor(message: string = '请先登录: 42plugin auth') {
+    super(message);
+    this.name = 'AuthRequiredError';
+  }
+}

package/src/index.ts

@@ -1,35 +1,8 @@
 #!/usr/bin/env bun
+/**
+ * 42plugin CLI
+ */
 
 import { program } from './cli';
-import { initDb } from './db/client';
-import { api } from './services/api';
-import { CliError } from './utils/errors';
-import chalk from 'chalk';
 
-async function main() {
-  try {
-    // Initialize database
-    await initDb();
-
-    // Initialize API client
-    await api.init();
-
-    // Parse and execute command
-    await program.parseAsync(process.argv);
-  } catch (error) {
-    if (error instanceof CliError) {
-      console.error(chalk.red(`错误: ${error.message}`));
-      process.exit(error.exitCode);
-    }
-
-    // Unknown error
-    if (process.env.DEBUG) {
-      console.error(error);
-    } else {
-      console.error(chalk.red(`发生错误: ${(error as Error).message}`));
-    }
-    process.exit(1);
-  }
-}
-
-main();
+program.parse();