agens-studio 0.1.0

package/src/services/assetStore.js

@@ -0,0 +1,224 @@
+import fs from 'node:fs/promises';
+import path from 'node:path';
+import { randomUUID } from 'node:crypto';
+import { config } from '../../config.js';
+
+/**
+ * 资产管理服务
+ * - 生成物下载到 assets/{images|videos}
+ * - 元数据维护在 data/index.json
+ * - 支持列表 / 删除 / 保留标记 / 备注
+ */
+
+let _cache = null; // 内存缓存，避免每次请求都读文件
+
+async function ensureDirs() {
+  await fs.mkdir(config.dirs.images, { recursive: true });
+  await fs.mkdir(config.dirs.videos, { recursive: true });
+  await fs.mkdir(config.dirs.data, { recursive: true });
+  await fs.mkdir(config.dirs.uploads, { recursive: true });
+  try {
+    await fs.access(config.dirs.indexFile);
+  } catch {
+    await fs.writeFile(config.dirs.indexFile, '[]', 'utf8');
+  }
+}
+
+async function loadIndex() {
+  if (_cache) return _cache;
+  try {
+    const raw = await fs.readFile(config.dirs.indexFile, 'utf8');
+    _cache = JSON.parse(raw);
+    if (!Array.isArray(_cache)) _cache = [];
+  } catch {
+    _cache = [];
+  }
+  return _cache;
+}
+
+async function saveIndex() {
+  if (!_cache) return;
+  await fs.writeFile(config.dirs.indexFile, JSON.stringify(_cache, null, 2), 'utf8');
+}
+
+/** 提取文件扩展名（带点），默认根据 mime */
+function extFromMime(mime) {
+  const map = {
+    'image/png': '.png',
+    'image/jpeg': '.jpg',
+    'image/webp': '.webp',
+    'image/jpg': '.jpg',
+    'video/mp4': '.mp4',
+    'video/quicktime': '.mov',
+    'video/webm': '.webm',
+  };
+  return map[mime?.toLowerCase()] || '';
+}
+
+/** 从 URL 推断扩展名，否则用 mime */
+function extFromUrl(url, mime) {
+  try {
+    const u = new URL(url);
+    const m = u.pathname.match(/\.(png|jpe?g|webp|mp4|mov|webm)$/i);
+    if (m) return '.' + m[1].toLowerCase().replace('jpeg', 'jpg');
+  } catch { /* ignore */ }
+  return extFromMime(mime);
+}
+
+/**
+ * 下载远程文件到本地并登记为资产
+ * @param {object} opts
+ * @param {string} opts.url 远程文件 URL
+ * @param {'image'|'video'} opts.kind
+ * @param {string} [opts.mime] 可选，明确 mime
+ * @param {object} opts.meta 元数据（prompt/params/mode/taskId 等）
+ * @returns {Promise<object>} 新建资产记录
+ */
+export async function downloadAndRegister({ url, kind, mime, meta = {} }) {
+  await ensureDirs();
+  // 下载重试：视频 URL 可能临时不可访问（CDN 未同步），最多重试 3 次
+  let res;
+  let lastErr;
+  for (let attempt = 0; attempt < 3; attempt++) {
+    try {
+      res = await fetch(url);
+      if (res.ok) break;
+      lastErr = new Error(`HTTP ${res.status}`);
+    } catch (e) {
+      lastErr = e;
+    }
+    if (attempt < 2) await new Promise((r) => setTimeout(r, 2000 * (attempt + 1))); // 2s, 4s
+  }
+  if (!res || !res.ok) throw new Error(`下载失败（重试3次）: ${lastErr?.message || res?.status || '未知'}`);
+  const contentType = mime || res.headers.get('content-type') || '';
+  const buf = Buffer.from(await res.arrayBuffer());
+  const ext = extFromUrl(url, contentType);
+  const id = randomUUID();
+  const ts = Date.now();
+  const filename = `${id}${ext}`;
+  const subdir = kind === 'video' ? 'videos' : 'images';
+  const absPath = path.join(config.dirs[subdir], filename);
+  await fs.writeFile(absPath, buf);
+
+  const record = {
+    id,
+    type: kind, // 'image' | 'video'
+    filename,
+    // 浏览器可访问的相对路径
+    path: `/assets/${subdir}/${filename}`,
+    size: buf.length,
+    mime: contentType,
+    prompt: meta.prompt || '',
+    mode: meta.mode || '',
+    params: meta.params || {},
+    taskId: meta.taskId || null,
+    kept: true, // 默认保留，false 表示待清理
+    note: '',
+    createdAt: ts,
+  };
+
+  const list = await loadIndex();
+  list.unshift(record);
+  await saveIndex();
+  return record;
+}
+
+/**
+ * 登记一个本地已存在的文件（例如用户上传后用作参考帧）
+ */
+export async function registerLocal({ absPath, kind, meta = {} }) {
+  await ensureDirs();
+  const id = randomUUID();
+  const ts = Date.now();
+  const filename = path.basename(absPath);
+  const subdir = kind === 'video' ? 'videos' : 'images';
+  const record = {
+    id,
+    type: kind,
+    filename,
+    path: `/assets/${subdir}/${filename}`,
+    size: (await fs.stat(absPath)).size,
+    mime: '',
+    prompt: meta.prompt || '',
+    mode: meta.mode || '',
+    params: meta.params || {},
+    taskId: meta.taskId || null,
+    kept: true,
+    note: '',
+    createdAt: ts,
+  };
+  const list = await loadIndex();
+  list.unshift(record);
+  await saveIndex();
+  return record;
+}
+
+/**
+ * 列出资产
+ * @param {object} q
+ * @param {'image'|'video'|'all'} [q.type]
+ * @param {string} [q.search] 搜索 prompt / note
+ * @param {boolean} [q.keptOnly] 只看保留的
+ * @param {number} [q.page]
+ * @param {number} [q.pageSize]
+ */
+export async function listAssets({ type = 'all', search = '', keptOnly = false, page = 1, pageSize = 60 } = {}) {
+  const list = await loadIndex();
+  let filtered = list;
+  if (type && type !== 'all') filtered = filtered.filter((a) => a.type === type);
+  if (keptOnly) filtered = filtered.filter((a) => a.kept);
+  if (search) {
+    const s = search.toLowerCase();
+    filtered = filtered.filter(
+      (a) =>
+        (a.prompt || '').toLowerCase().includes(s) ||
+        (a.note || '').toLowerCase().includes(s)
+    );
+  }
+  // 已按 createdAt 倒序（新在前）
+  const total = filtered.length;
+  const start = (page - 1) * pageSize;
+  const items = filtered.slice(start, start + pageSize);
+  return { items, total, page, pageSize };
+}
+
+/** 根据 id 查单个资产 */
+export async function getAsset(id) {
+  const list = await loadIndex();
+  return list.find((a) => a.id === id) || null;
+}
+
+/**
+ * 更新资产：保留标记 / 备注
+ */
+export async function updateAsset(id, patch) {
+  const list = await loadIndex();
+  const item = list.find((a) => a.id === id);
+  if (!item) return null;
+  if (typeof patch.kept === 'boolean') item.kept = patch.kept;
+  if (typeof patch.note === 'string') item.note = patch.note;
+  await saveIndex();
+  return item;
+}
+
+/**
+ * 删除资产（同时删文件）
+ */
+export async function deleteAsset(id) {
+  const list = await loadIndex();
+  const idx = list.findIndex((a) => a.id === id);
+  if (idx === -1) return false;
+  const [item] = list.splice(idx, 1);
+  await saveIndex();
+  try {
+    const subdir = item.type === 'video' ? 'videos' : 'images';
+    await fs.unlink(path.join(config.dirs[subdir], item.filename));
+  } catch { /* 文件可能已不在 */ }
+  return true;
+}
+
+/** 启动时调用，确保目录就绪 */
+export async function init() {
+  await ensureDirs();
+  await loadIndex();
+}