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 });
  });
});