feat(organize): 整理、季包选文件、刮削、文件管理识别统一套用集数映射;单集沿用资源判定,待确认文件不按原号入库
Co-Authored-By: Claude Opus 5.5 <[email protected]> Claude-Session: https://claude.ai/code/session_01Cxwgnn2pbzQ7rXv3rq7fei
This commit is contained in:
1 parent
691671a1d2
commit
fe597187dd
12 files changed
+848
-21
No files matched your search
@@ -0,0 +1,100 @@
|
||||
import { writeFile } from "node:fs/promises";
|
||||
import { join } from "node:path";
|
||||
import { afterAll, beforeAll, describe, expect, it } from "vitest";
|
||||
import { createFmHarness, type FmHarness, pgAvailable, READ_KEY, seedEntity } from "./harness.ts";
|
||||
|
||||
// 文件管理识别 / 推荐名按作品映射集数(docs/plan/episode-resolution.md 批次 3,只读):识别到本地剧集作品后,
|
||||
// 展示的季集、建议单元与推荐文件名用映射后的规范季集;集数待确认的不给单元与推荐名,也不开待处理。
|
||||
|
||||
const skip = !(await pgAvailable());
|
||||
|
||||
describe.skipIf(skip)("文件管理识别与推荐名按作品映射集数", () => {
|
||||
let h: FmHarness;
|
||||
let show: string;
|
||||
let s2e13: string;
|
||||
|
||||
beforeAll(async () => {
|
||||
h = await createFmHarness("episode_mapping_identify");
|
||||
show = await seedEntity(h.db, { kind: "series", title: "Kusuriya no Hitorigoto", year: 2023 });
|
||||
await h.db
|
||||
.updateTable("media_entity")
|
||||
.set({ metadata: JSON.stringify({ airStatus: { value: "Ended", source: "tmdb" } }) })
|
||||
.where("id", "=", show)
|
||||
.execute();
|
||||
for (const [season, total] of [
|
||||
[1, 12],
|
||||
[2, 13],
|
||||
] as const) {
|
||||
await h.db
|
||||
.insertInto("content_unit")
|
||||
.values({
|
||||
media_entity_id: show,
|
||||
kind: "season",
|
||||
season_number: season,
|
||||
episode_total: total,
|
||||
status: "confirmed",
|
||||
source: "tmdb",
|
||||
})
|
||||
.execute();
|
||||
}
|
||||
s2e13 = (
|
||||
await h.db
|
||||
.insertInto("content_unit")
|
||||
.values({
|
||||
media_entity_id: show,
|
||||
kind: "episode",
|
||||
season_number: 2,
|
||||
episode_number: 13,
|
||||
status: "confirmed",
|
||||
source: "tmdb",
|
||||
})
|
||||
.returning("id")
|
||||
.executeTakeFirstOrThrow()
|
||||
).id;
|
||||
await writeFile(join(h.dl, "[ANi] Kusuriya no Hitorigoto - 25 [1080P].mkv"), "v25");
|
||||
await writeFile(join(h.dl, "[ANi] Kusuriya no Hitorigoto - 40 [1080P].mkv"), "v40");
|
||||
});
|
||||
afterAll(async () => h?.close());
|
||||
|
||||
it("识别:`- 25` 展示与建议单元为 S02E13;`- 40` 待确认不给单元、带提示,不开待处理", async () => {
|
||||
const r = await h.api(
|
||||
"POST",
|
||||
"/files/identify",
|
||||
{
|
||||
items: [
|
||||
{ root: h.dlRoot, path: "[ANi] Kusuriya no Hitorigoto - 25 [1080P].mkv" },
|
||||
{ root: h.dlRoot, path: "[ANi] Kusuriya no Hitorigoto - 40 [1080P].mkv" },
|
||||
],
|
||||
minSizeBytes: 0,
|
||||
},
|
||||
READ_KEY,
|
||||
);
|
||||
expect(r.status).toBe(200);
|
||||
const [ok, bad] = r.body.data.items;
|
||||
expect(ok.candidates[0]).toMatchObject({ mediaEntityId: show });
|
||||
expect(ok.parsed).toMatchObject({ season: 2, episodes: [13] });
|
||||
expect(ok.suggestion).toMatchObject({ mediaEntityId: show, contentUnit: "S02E13", contentUnitId: s2e13 });
|
||||
expect(bad.suggestion).toMatchObject({ mediaEntityId: show, contentUnit: null, contentUnitId: null });
|
||||
expect(bad.warnings.some((w: string) => w.includes("集数对不上"))).toBe(true);
|
||||
const decisions = await h.db.selectFrom("decision").select("id").where("kind", "=", "episode_mapping").execute();
|
||||
expect(decisions).toHaveLength(0);
|
||||
});
|
||||
|
||||
it("推荐名:`- 25` 按 S02E13 命名;`- 40` 不给推荐名(episode_unresolved)", async () => {
|
||||
const ok = await h.api(
|
||||
"POST",
|
||||
"/files/suggest-name",
|
||||
{ root: h.dlRoot, path: "[ANi] Kusuriya no Hitorigoto - 25 [1080P].mkv", mediaEntityId: show },
|
||||
READ_KEY,
|
||||
);
|
||||
expect(ok.status).toBe(200);
|
||||
expect(ok.body.data.name).toMatch(/S02E13/);
|
||||
const bad = await h.api(
|
||||
"POST",
|
||||
"/files/suggest-name",
|
||||
{ root: h.dlRoot, path: "[ANi] Kusuriya no Hitorigoto - 40 [1080P].mkv", mediaEntityId: show },
|
||||
READ_KEY,
|
||||
);
|
||||
expect(bad.body.data).toMatchObject({ name: null, reason: "episode_unresolved", mediaEntityId: show });
|
||||
});
|
||||
});
|
||||
Reference in new issue
Block a user