import { expect, test, vi, it, describe, beforeEach, afterEach } from "vitest";
import {
  nextCheckInDate,
  initAndSyncTables,
  Nag,
  CheckIn,
  findGuiltyNags,
  getCheckIn,
} from "./service";
import { Sequelize, literal, Op } from "sequelize";

describe("nextCheckInDate", () => {
  beforeEach(() => {
    vi.useFakeTimers(); // Tell vitest to use fake timers
  });
  afterEach(() => {
    vi.useRealTimers(); // Reset date after test runs
  });
  it("Returns 9AM if called before 9AM that day", () => {
    const now = new Date(Date.now());
    let at9AM = new Date(
      now.getFullYear(),
      now.getMonth(),
      now.getDate(),
      9,
      0,
    );
    vi.setSystemTime(
      new Date(now.getFullYear(), now.getMonth(), now.getDate(), 8, 0),
    );
    expect(nextCheckInDate()).toEqual(at9AM);
  });
  it("Returns 9AM tomorrow if called after 9AM", () => {
    const dayInMS = 24 * 60 * 60 * 1000;
    const now = new Date(Date.now());
    const tomorrow = new Date(Date.now() + dayInMS);
    let tomorrow9AM = new Date(
      tomorrow.getFullYear(),
      tomorrow.getMonth(),
      tomorrow.getDate(),
      9,
      0,
    );
    vi.setSystemTime(
      new Date(now.getFullYear(), now.getMonth(), now.getDate(), 9, 30),
    );
    expect(nextCheckInDate()).toEqual(tomorrow9AM);
  });
});

describe("Finding nags without check-ins", async () => {
  const sequelize = new Sequelize("sqlite://:memory:");
  const exampleNag = {
    userId: "1234",
    guildId: "1234",
    channelId: "1234",
    messageId: "1234",
    text: "Example nag 1",
    mentionHere: false,
  };

  await initAndSyncTables(sequelize);

  beforeEach(async () => {
    vi.useFakeTimers();
  });

  afterEach(async () => {
    await Nag.destroy({ where: {} });
    await CheckIn.destroy({ where: {} });
    vi.useRealTimers();
  });

  it("Finds nags without any check-ins", async () => {
    const now = new Date();
    vi.setSystemTime(
      new Date(now.getFullYear(), now.getMonth(), now.getDate(), 9),
    );
    await Nag.create(exampleNag);
    const results = await findGuiltyNags();
    expect(results.map((nag) => nag.userId)).toEqual(["1234"]);
  });

  it("Ignores nags with a recent check-in", async () => {
    const newNag = await Nag.create(exampleNag);
    newNag.save();
    const currentCheckInTime = getCheckIn(9, 0);
    const newCheckIn = await CheckIn.create({
      nagId: newNag.id,
      // 1 hour previously; i.e. we checked in before the required time
      lastCheckIn: new Date(currentCheckInTime.getTime() - 60 * 60 * 1000),
    });
    newCheckIn.save();
    const results = await findGuiltyNags();
    expect(results.map((nag) => nag.userId)).toEqual([]);
  });
});