import { SystemRole } from "@capakraken/shared";
import { describe, expect, it, vi } from "vitest";
import { staffingRouter } from "../router/staffing.js";
import { createCallerFactory } from "../trpc.js";

// Mock the pure-logic packages — we focus on the router/DB layer
vi.mock("@capakraken/staffing", () => ({
  rankResources: vi.fn().mockImplementation((input: { resources: { id: string }[] }) =>
    input.resources.map((r: { id: string }, i: number) => ({
      resourceId: r.id,
      score: 80 - i * 10,
      breakdown: {
        skillScore: 70,
        availabilityScore: 90,
        costScore: 80,
        utilizationScore: 75,
      },
    })),
  ),
}));

vi.mock("@capakraken/application", () => ({
  listAssignmentBookings: vi.fn().mockResolvedValue([]),
}));

const createCaller = createCallerFactory(staffingRouter);

// ── Caller factories ─────────────────────────────────────────────────────────

function createProtectedCaller(db: Record<string, unknown>) {
  return createCaller({
    session: {
      user: { email: "user@example.com", name: "User", image: null },
      expires: "2099-01-01T00:00:00.000Z",
    },
    db: db as never,
    dbUser: {
      id: "user_1",
      systemRole: SystemRole.USER,
      permissionOverrides: null,
    },
  });
}

// ── Sample data ──────────────────────────────────────────────────────────────

function sampleResource(overrides: Record<string, unknown> = {}) {
  return {
    id: "res_1",
    displayName: "Alice",
    eid: "alice",
    lcrCents: 7500,
    chargeabilityTarget: 80,
    availability: { monday: 8, tuesday: 8, wednesday: 8, thursday: 8, friday: 8 },
    skills: [
      { skill: "Compositing", proficiency: 4, isMainSkill: true },
      { skill: "Nuke", proficiency: 3, isMainSkill: false, category: "Software" },
    ],
    isActive: true,
    valueScore: 85,
    chapter: "VFX",
    countryId: "country_de",
    federalState: "BY",
    metroCityId: null,
    country: { code: "DE" },
    metroCity: null,
    ...overrides,
  };
}

// ─── getSuggestions ──────────────────────────────────────────────────────────

describe("staffing.getSuggestions", () => {
  it("returns ranked suggestions for a staffing demand", async () => {
    const resources = [
      sampleResource(),
      sampleResource({ id: "res_2", displayName: "Bob", eid: "bob", valueScore: 70 }),
    ];
    const db = {
      resource: {
        findMany: vi.fn().mockResolvedValue(resources),
      },
    };

    const caller = createProtectedCaller(db);
    const result = await caller.getSuggestions({
      requiredSkills: ["Compositing"],
      startDate: new Date("2026-04-01"),
      endDate: new Date("2026-04-30"),
      hoursPerDay: 8,
    });

    expect(result).toHaveLength(2);
    expect(result[0]).toHaveProperty("resourceId");
    expect(result[0]).toHaveProperty("score");
    expect(result[0]).toMatchObject({
      resourceName: "Alice",
      eid: "alice",
      location: {
        countryCode: "DE",
        federalState: "BY",
      },
      capacity: expect.objectContaining({
        requestedHoursPerDay: 8,
        baseAvailableHours: expect.any(Number),
        effectiveAvailableHours: expect.any(Number),
        remainingHoursPerDay: expect.any(Number),
        holidayHoursDeduction: expect.any(Number),
      }),
      conflicts: {
        count: expect.any(Number),
        conflictDays: expect.any(Array),
        details: expect.any(Array),
      },
      ranking: expect.objectContaining({
        rank: 1,
        components: expect.any(Array),
      }),
    });
  });

  it("filters resources by chapter when provided", async () => {
    const db = {
      resource: {
        findMany: vi.fn().mockResolvedValue([]),
      },
    };

    const caller = createProtectedCaller(db);
    await caller.getSuggestions({
      requiredSkills: ["Compositing"],
      startDate: new Date("2026-04-01"),
      endDate: new Date("2026-04-30"),
      hoursPerDay: 8,
      chapter: "VFX",
    });

    expect(db.resource.findMany).toHaveBeenCalledWith(
      expect.objectContaining({
        where: expect.objectContaining({
          isActive: true,
          chapter: "VFX",
        }),
      }),
    );
  });

  it("returns empty array when no resources match", async () => {
    const db = {
      resource: {
        findMany: vi.fn().mockResolvedValue([]),
      },
    };

    const caller = createProtectedCaller(db);
    const result = await caller.getSuggestions({
      requiredSkills: ["Compositing"],
      startDate: new Date("2026-04-01"),
      endDate: new Date("2026-04-30"),
      hoursPerDay: 8,
    });

    expect(result).toHaveLength(0);
  });

  it("passes budget constraint to ranking", async () => {
    const resources = [sampleResource()];
    const db = {
      resource: {
        findMany: vi.fn().mockResolvedValue(resources),
      },
    };

    const { rankResources } = await import("@capakraken/staffing");
    const caller = createProtectedCaller(db);
    await caller.getSuggestions({
      requiredSkills: ["Compositing"],
      startDate: new Date("2026-04-01"),
      endDate: new Date("2026-04-30"),
      hoursPerDay: 8,
      budgetLcrCentsPerHour: 10000,
    });

    expect(rankResources).toHaveBeenCalledWith(
      expect.objectContaining({
        budgetLcrCentsPerHour: 10000,
      }),
    );
  });

  it("uses value score as a transparent tiebreaker within two score points", async () => {
    const resources = [
      sampleResource({ id: "res_1", displayName: "Alice", eid: "alice", valueScore: 60 }),
      sampleResource({ id: "res_2", displayName: "Bob", eid: "bob", valueScore: 95 }),
    ];
    const db = {
      resource: {
        findMany: vi.fn().mockResolvedValue(resources),
      },
    };

    const { rankResources } = await import("@capakraken/staffing");
    vi.mocked(rankResources).mockImplementationOnce((input: { resources: Array<{ id: string }> }) => ([
      {
        resourceId: input.resources[0]!.id,
        score: 80,
        breakdown: {
          skillScore: 80,
          availabilityScore: 80,
          costScore: 80,
          utilizationScore: 80,
        },
      },
      {
        resourceId: input.resources[1]!.id,
        score: 79,
        breakdown: {
          skillScore: 79,
          availabilityScore: 79,
          costScore: 79,
          utilizationScore: 79,
        },
      },
    ]));

    const caller = createProtectedCaller(db);
    const result = await caller.getSuggestions({
      requiredSkills: ["Compositing"],
      startDate: new Date("2026-04-01"),
      endDate: new Date("2026-04-30"),
      hoursPerDay: 8,
    });

    expect(result[0]?.resourceId).toBe("res_2");
    expect(result[0]?.ranking).toMatchObject({
      rank: 1,
      baseRank: 2,
      tieBreakerApplied: true,
    });
    expect(result[0]?.ranking.tieBreakerReason).toContain("value score");
  });
});

// ─── analyzeUtilization ──────────────────────────────────────────────────────

describe("staffing.analyzeUtilization", () => {
  it("returns utilization analysis for an existing resource", async () => {
    const resource = {
      id: "res_1",
      displayName: "Alice",
      chargeabilityTarget: 80,
      availability: { monday: 8, tuesday: 8, wednesday: 8, thursday: 8, friday: 8 },
      countryId: "country_de",
      federalState: "BY",
      metroCityId: null,
      country: { code: "DE" },
      metroCity: null,
    };
    const db = {
      resource: {
        findUnique: vi.fn().mockResolvedValue(resource),
      },
    };

    const caller = createProtectedCaller(db);
    const result = await caller.analyzeUtilization({
      resourceId: "res_1",
      startDate: new Date("2026-04-01"),
      endDate: new Date("2026-04-30"),
    });

    expect(result).toHaveProperty("currentChargeability");
    expect(result.resourceId).toBe("res_1");
  });

  it("excludes Bavarian public holidays from chargeability analysis", async () => {
    const resource = {
      id: "res_1",
      displayName: "Alice",
      chargeabilityTarget: 80,
      availability: { monday: 8, tuesday: 8, wednesday: 8, thursday: 8, friday: 8 },
      countryId: "country_de",
      federalState: "BY",
      metroCityId: null,
      country: { code: "DE" },
      metroCity: null,
    };
    const db = {
      resource: {
        findUnique: vi.fn().mockResolvedValue(resource),
      },
    };

    const { listAssignmentBookings } = await import("@capakraken/application");
    vi.mocked(listAssignmentBookings).mockResolvedValue([
      {
        id: "a1",
        projectId: "project_1",
        resourceId: "res_1",
        startDate: new Date("2026-01-05T00:00:00.000Z"),
        endDate: new Date("2026-01-06T00:00:00.000Z"),
        hoursPerDay: 8,
        dailyCostCents: 0,
        status: "CONFIRMED",
        project: { id: "project_1", name: "Chargeable", shortCode: "CHG", status: "ACTIVE", orderType: "CHARGEABLE", clientId: null, dynamicFields: null },
        resource: { id: "res_1", displayName: "Alice", chapter: "VFX" },
      },
    ]);

    const caller = createProtectedCaller(db);
    const result = await caller.analyzeUtilization({
      resourceId: "res_1",
      startDate: new Date("2026-01-05T00:00:00.000Z"),
      endDate: new Date("2026-01-06T00:00:00.000Z"),
    });

    expect(result.currentChargeability).toBe(100);
    expect(result.overallocatedDays).toEqual([]);
    expect(result.underutilizedDays).toEqual([]);
  });

  it("throws NOT_FOUND when resource does not exist", async () => {
    const db = {
      resource: {
        findUnique: vi.fn().mockResolvedValue(null),
      },
    };

    const caller = createProtectedCaller(db);
    await expect(
      caller.analyzeUtilization({
        resourceId: "nonexistent",
        startDate: new Date("2026-04-01"),
        endDate: new Date("2026-04-30"),
      }),
    ).rejects.toThrow("Resource not found");
  });
});

// ─── findCapacity ────────────────────────────────────────────────────────────

describe("staffing.findCapacity", () => {
  it("returns capacity windows for an existing resource", async () => {
    const resource = {
      id: "res_1",
      displayName: "Alice",
      availability: { monday: 8, tuesday: 8, wednesday: 8, thursday: 8, friday: 8 },
      countryId: "country_de",
      federalState: "BY",
      metroCityId: null,
      country: { code: "DE" },
      metroCity: null,
    };
    const db = {
      resource: {
        findUnique: vi.fn().mockResolvedValue(resource),
      },
    };

    const caller = createProtectedCaller(db);
    const result = await caller.findCapacity({
      resourceId: "res_1",
      startDate: new Date("2026-04-01"),
      endDate: new Date("2026-04-30"),
    });

    expect(result.length).toBeGreaterThan(0);
    expect(result[0]).toHaveProperty("availableHoursPerDay");
    expect(result.every((window) => window.availableHoursPerDay > 0)).toBe(true);
    expect(result.reduce((sum, window) => sum + window.availableDays, 0)).toBeGreaterThan(0);
  });

  it("splits capacity windows around Bavarian public holidays", async () => {
    const resource = {
      id: "res_1",
      displayName: "Alice",
      availability: { monday: 8, tuesday: 8, wednesday: 8, thursday: 8, friday: 8 },
      countryId: "country_de",
      federalState: "BY",
      metroCityId: null,
      country: { code: "DE" },
      metroCity: null,
    };
    const db = {
      resource: {
        findUnique: vi.fn().mockResolvedValue(resource),
      },
    };

    const { listAssignmentBookings } = await import("@capakraken/application");
    vi.mocked(listAssignmentBookings).mockResolvedValue([]);

    const caller = createProtectedCaller(db);
    const result = await caller.findCapacity({
      resourceId: "res_1",
      startDate: new Date("2026-01-05T00:00:00.000Z"),
      endDate: new Date("2026-01-07T00:00:00.000Z"),
      minAvailableHoursPerDay: 4,
    });

    expect(result).toHaveLength(2);
    expect(result[0]).toEqual(
      expect.objectContaining({
        startDate: new Date("2026-01-05T00:00:00.000Z"),
        endDate: new Date("2026-01-05T00:00:00.000Z"),
      }),
    );
    expect(result[1]).toEqual(
      expect.objectContaining({
        startDate: new Date("2026-01-07T00:00:00.000Z"),
        endDate: new Date("2026-01-07T00:00:00.000Z"),
      }),
    );
  });

  it("throws NOT_FOUND when resource does not exist", async () => {
    const db = {
      resource: {
        findUnique: vi.fn().mockResolvedValue(null),
      },
    };

    const caller = createProtectedCaller(db);
    await expect(
      caller.findCapacity({
        resourceId: "nonexistent",
        startDate: new Date("2026-04-01"),
        endDate: new Date("2026-04-30"),
      }),
    ).rejects.toThrow("Resource not found");
  });

  it("honors minAvailableHoursPerDay when computing holiday-aware windows", async () => {
    const resource = {
      id: "res_1",
      displayName: "Alice",
      availability: { monday: 8, tuesday: 8, wednesday: 8, thursday: 8, friday: 8 },
      countryId: "country_de",
      federalState: "BY",
      metroCityId: null,
      country: { code: "DE" },
      metroCity: null,
    };
    const db = {
      resource: {
        findUnique: vi.fn().mockResolvedValue(resource),
      },
    };

    const { listAssignmentBookings } = await import("@capakraken/application");
    vi.mocked(listAssignmentBookings).mockResolvedValue([
      {
        id: "a1",
        projectId: "project_1",
        resourceId: "res_1",
        startDate: new Date("2026-04-01T00:00:00.000Z"),
        endDate: new Date("2026-04-30T00:00:00.000Z"),
        hoursPerDay: 3,
        dailyCostCents: 0,
        status: "CONFIRMED",
        project: { id: "project_1", name: "Project", shortCode: "PRJ", status: "ACTIVE", orderType: "CHARGEABLE", clientId: null, dynamicFields: null },
        resource: { id: "res_1", displayName: "Alice", chapter: "VFX" },
      },
    ]);

    const caller = createProtectedCaller(db);
    const result = await caller.findCapacity({
      resourceId: "res_1",
      startDate: new Date("2026-04-01"),
      endDate: new Date("2026-04-30"),
      minAvailableHoursPerDay: 6,
    });

    expect(result.every((window) => window.availableHoursPerDay >= 6)).toBe(true);
  });
});