feat(file-manager): v2 backend gaps — TMDB candidate entities, min size, on-existing, naming source (W9-H)

- G1 createEntityFromTmdbCandidate (sources-identity service): entity + accepted manual external identity + season/episode units, idempotent per TMDB id; POST /files/entity-from-candidate and organize items with `external` - G2 minSizeBytes for organize (excluded below_min_size) and identify (skipped + excludedBySize) - G3 onExisting passed to planImport; dryRun marks each file (same/skip/replace/conflict); replace = explicit grant - G4 naming template|keep; planImport namingMode/namingSource, frozen plan.naming {mode, templateSource, templateVersion}; targetSegments - G5 identify prior-record hints, GET /files/record-sources, organize refreshMetadata, POST /files/scrape-batch (dedupe by entity) - G8 GET /files/tree (parent, subdirs hasChildren, video/registered counts) - G9 POST /files/suggest-name returns 501 until the naming engine lands Co-Authored-By: Claude Opus 5.5 <[email protected]>
2026-09-28 15:08:17 +08:00
1 parent 88bbed409c
commit 13132e7ad8
@@ -0,0 +1,130 @@
+import { lstat, readdir } from "node:fs/promises";
+import { join } from "node:path";
+import type {
+  FileRootId,
+  FilesRecordSourcesResult,
+  FilesTreeQuery,
+  FilesTreeResult,
+  OrganizationStatus,
+} from "@umt/contracts";
+import { sql } from "@umt/db";
+import { FilesError, type FmDeps, isTempName, loadRootDefs, resolveIn, within } from "./common.ts";
+import { kindOfName } from "./list.ts";
+
+// G8 目录摘要(树折叠 / 上一级):只列子目录(是否还有下级)+ 本目录递归视频数与已登记数;
+// G5 整理记录 → 源 FileRef(整理记录"重新整理"按登记目录 hostPath 换算)。
+
+const likePrefix = (rel: string) => `${rel.replace(/[\\%_]/g, (m) => `\\${m}`)}/%`;
+const hiddenName = (n: string) => n.startsWith(".") || isTempName(n);
+
+export async function treeOf(d: FmDeps, q: FilesTreeQuery): Promise<FilesTreeResult> {
+  const r = await resolveIn(d, q.root, q.path);
+  if (!r.lst?.isDirectory() || r.isSymlink) throw new FilesError("bad_request", `不是目录: ${r.rel || "/"}`);
+  const names = await readdir(r.abs, { withFileTypes: true });
+  const subdirs = names
+    .filter((n) => n.isDirectory() && !hiddenName(n.name))
+    .map((n) => n.name)
+    .sort((a, b) => a.localeCompare(b));
+  const max = d.options.listMaxEntries;
+  const truncated = subdirs.length > max;
+  const dirs = await Promise.all(
+    subdirs.slice(0, max).map(async (name) => {
+      const path = r.rel ? `${r.rel}/${name}` : name;
+      const kids = await readdir(join(r.abs, name), { withFileTypes: true }).catch(() => []);
+      return { name, path, hasChildren: kids.some((k) => k.isDirectory() && !hiddenName(k.name)) };
+    }),
+  );
+
+  // 递归数视频(上限 dirSummaryMax 项,同列表目录摘要口径)
+  let seen = 0;
+  let videoCount = 0;
+  let partial = false;
+  const stack = [r.abs];
+  while (stack.length > 0 && !partial) {
+    const dir = stack.pop()!;
+    for (const n of await readdir(dir, { withFileTypes: true }).catch(() => [])) {
+      if (++seen > d.options.dirSummaryMax) {
+        partial = true;
+        break;
+      }
+      if (n.name.startsWith(".")) continue;
+      if (n.isDirectory()) stack.push(join(dir, n.name));
+      else if (n.isFile() && kindOfName(n.name) === "video") videoCount++;
+    }
+  }
+  let registeredCount = 0;
+  if (r.def.libraryRootId) {
+    const row = await sql<{ n: number }>`
+      select count(*)::int as n from asset_location
+       where library_root_id = ${r.def.libraryRootId} and status = 'available'
+         and (${r.rel} = '' or relative_path like ${likePrefix(r.rel)})`.execute(d.ctx.db);
+    registeredCount = row.rows[0]?.n ?? 0;
+  }
+  const slash = r.rel.lastIndexOf("/");
+  return {
+    root: q.root,
+    path: r.rel,
+    name: r.rel ? r.rel.slice(slash + 1) : r.def.name,
+    parent: r.rel === "" ? null : slash < 0 ? "" : r.rel.slice(0, slash),
+    dirs,
+    truncated,
+    summary: { videoCount, registeredCount, partial },
+  };
+}
+
+export async function recordSources(d: FmDeps, recordId: string): Promise<FilesRecordSourcesResult> {
+  const db = d.ctx.db;
+  const rec = await db
+    .selectFrom("organization_record")
+    .select(["id", "media_entity_id", "action", "status", "source_path", "source_path_view"])
+    .where("id", "=", recordId)
+    .executeTakeFirst();
+  if (!rec) throw new FilesError("not_found", "整理记录不存在");
+  const mains = await db
+    .selectFrom("organization_file_result")
+    .select(["source_path"])
+    .where("organization_record_id", "=", rec.id)
+    .where("role", "=", "main")
+    .orderBy("ordinal")
+    .limit(500)
+    .execute();
+  const defs = (await loadRootDefs(db)).sort((a, b) => b.hostPath.length - a.hostPath.length);
+  const toRef = (abs: string) => {
+    const def = defs.find((x) => within(abs, x.hostPath));
+    return def
+      ? { root: def.id as FileRootId, path: abs === def.hostPath ? "" : abs.slice(def.hostPath.length + 1) }
+      : null;
+  };
+  // 多个主视频都在记录的源目录下、且源目录不是登记目录本身 → 给目录(向导按目录展开);否则逐个主视频
+  let sources = mains.map((m) => m.source_path);
+  const dir = rec.source_path;
+  if (
+    dir &&
+    mains.length > 1 &&
+    mains.every((m) => within(m.source_path, dir)) &&
+    !defs.some((x) => x.hostPath === dir.replace(/\/+$/, ""))
+  )
+    sources = [dir];
+  if (sources.length === 0 && dir) sources = [dir];
+  const items: FilesRecordSourcesResult["items"] = [];
+  const unresolved: string[] = [];
+  for (const abs of [...new Set(sources)]) {
+    const ref = rec.source_path_view === "local" ? toRef(abs) : null;
+    const exists = ref
+      ? await lstat(abs).then(
+          () => true,
+          () => false,
+        )
+      : false;
+    if (ref && exists) items.push(ref);
+    else unresolved.push(abs);
+  }
+  return {
+    recordId: rec.id,
+    mediaEntityId: rec.media_entity_id,
+    action: rec.action,
+    status: rec.status as OrganizationStatus,
+    items,
+    unresolved,
+  };
+}