import { eq, and } from "drizzle-orm";
import { database } from "~/database/context";
import * as schema from "~/database/schema";

export type Season = typeof schema.seasons.$inferSelect;
export type NewSeason = typeof schema.seasons.$inferInsert;
export type SeasonStatus = "pre_draft" | "draft" | "active" | "completed";

export async function createSeason(data: NewSeason): Promise<Season> {
  const db = database();
  const [season] = await db
    .insert(schema.seasons)
    .values(data)
    .returning();
  return season;
}

export async function findSeasonById(id: string): Promise<Season | undefined> {
  const db = database();
  return await db.query.seasons.findFirst({
    where: eq(schema.seasons.id, id),
  });
}

export async function findSeasonsByLeagueId(leagueId: string): Promise<Season[]> {
  const db = database();
  return await db.query.seasons.findMany({
    where: eq(schema.seasons.leagueId, leagueId),
    orderBy: (seasons, { desc }) => [desc(seasons.year)],
  });
}

export async function findCurrentSeason(
  leagueId: string
): Promise<Season | undefined> {
  const db = database();
  return await db.query.seasons.findFirst({
    where: eq(schema.seasons.leagueId, leagueId),
    orderBy: (seasons, { desc }) => [desc(seasons.year)],
  });
}

export async function findSeasonByLeagueAndYear(
  leagueId: string,
  year: number
): Promise<Season | undefined> {
  const db = database();
  return await db.query.seasons.findFirst({
    where: and(
      eq(schema.seasons.leagueId, leagueId),
      eq(schema.seasons.year, year)
    ),
  });
}

export async function updateSeason(
  id: string,
  data: Partial<NewSeason>
): Promise<Season> {
  const db = database();
  const [season] = await db
    .update(schema.seasons)
    .set({ ...data, updatedAt: new Date() })
    .where(eq(schema.seasons.id, id))
    .returning();
  return season;
}

export async function updateSeasonStatus(
  id: string,
  status: SeasonStatus
): Promise<Season> {
  return await updateSeason(id, { status });
}

export async function deleteSeason(id: string): Promise<void> {
  const db = database();
  await db.delete(schema.seasons).where(eq(schema.seasons.id, id));
}