import Image from "next/image";

import { Badge } from "@/components/ui/badge";
import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs";
import {
  StygianBossCard,
  StygianBossTab,
} from "@/components/stygian/stygian-boss-display";
import { elementIconFilter } from "@/lib/catalog/element";
import { characterRarityGradientClass, rarityGradientClass } from "@/lib/catalog/rarity";
import type { getActiveCatalog } from "@/lib/guides/queries";
import type { getStygianSchedule } from "@/lib/stygian/repository";
import { cn } from "@/lib/utils";

type Schedule = NonNullable<Awaited<ReturnType<typeof getStygianSchedule>>>;
type Catalog = NonNullable<Awaited<ReturnType<typeof getActiveCatalog>>>;
type Item = Schedule["recommendations"][number]["items"][number];

function recommendation(item: Item, catalog: Catalog) {
  if (item.kind === "element" && item.element) {
    const source = catalog.characters.find((character) => character.element === item.element && character.elementIconUrl);
    return { name: item.element, imageUrl: source?.elementIconUrl ?? null, badge: "Element", rarity: null };
  }
  if (item.kind === "character") {
    const character = catalog.characters.find((candidate) => candidate.key === item.catalogKey);
    const weapon = catalog.weapons.find((candidate) => candidate.key === item.weaponKey);
    const artifact = catalog.artifacts.find((candidate) => candidate.key === item.artifactKey);
    return {
      name: character?.name ?? item.catalogKey ?? "Unknown",
      imageUrl: character?.imageUrl ?? null,
      badge: item.constellationLevel === null ? "Character" : `C${item.constellationLevel}`,
      rarity: character?.rarity ?? null,
      weapon,
      artifact,
      refinement: item.refinement,
    };
  }
  const collection = item.kind === "weapon" ? catalog.weapons : catalog.artifacts;
  const value = collection.find((candidate) => candidate.key === item.catalogKey);
  return { name: value?.name ?? item.catalogKey ?? "Unknown", imageUrl: value?.imageUrl ?? null, badge: item.kind, rarity: value?.rarity ?? null };
}

function RecommendationIcon({ item, catalog, teamLayout = false }: { item: Item; catalog: Catalog; teamLayout?: boolean }) {
  const value = recommendation(item, catalog);
  if (item.kind === "character") {
    const hasEquipment = Boolean(value.weapon || value.artifact);
    return (
      <div className={cn("flex min-w-0 flex-col gap-1 rounded-lg border bg-card p-1", teamLayout && "w-full sm:p-2 xl:p-1", teamLayout && hasEquipment && "sm:w-[164px] xl:w-full", teamLayout && !hasEquipment && "sm:w-[116px] xl:w-full", teamLayout && !hasEquipment && item.constellationLevel === null && "aspect-square justify-center")}>
        <div className={teamLayout && hasEquipment ? "flex flex-col gap-1 sm:grid sm:grid-cols-[100px_40px] sm:gap-2 xl:grid-cols-1 xl:gap-1" : "flex flex-col gap-1"}>
          <div className={cn("relative aspect-square overflow-hidden rounded-lg", teamLayout && "sm:size-[100px] xl:size-auto", teamLayout && hasEquipment && "sm:row-span-2 xl:row-auto", characterRarityGradientClass(item.catalogKey, value.rarity))}>
            {value.imageUrl ? <Image src={value.imageUrl} alt={value.name} fill sizes={teamLayout ? "(min-width: 640px) 128px, 25vw" : "128px"} className="object-contain" /> : null}
          </div>
          {hasEquipment ? <div className={teamLayout ? "grid grid-cols-2 gap-1 sm:row-span-2 sm:grid-cols-1 sm:gap-2 sm:self-center xl:row-auto xl:grid-cols-2 xl:gap-1 xl:self-auto" : "grid grid-cols-2 gap-1"}>
            {value.weapon ? <div className={cn("relative aspect-square overflow-hidden rounded-md border", rarityGradientClass(value.weapon.rarity))}><Image src={value.weapon.imageUrl} alt={value.weapon.name} fill sizes="64px" className="object-contain p-1" /></div> : null}
            {value.artifact ? <div className={cn("relative aspect-square overflow-hidden rounded-md border", rarityGradientClass(value.artifact.rarity))}><Image src={value.artifact.imageUrl} alt={value.artifact.name} fill sizes="64px" className="object-contain p-1" /></div> : null}
          </div> : null}
        </div>
        {teamLayout ? item.constellationLevel !== null || value.weapon && value.refinement ? <div className="flex justify-center gap-1">
          {item.constellationLevel !== null ? <Badge variant="secondary">C{item.constellationLevel}</Badge> : null}
          {value.weapon && value.refinement ? <Badge variant="secondary">R{value.refinement}</Badge> : null}
        </div> : null : <span className="truncate text-center text-xs font-medium" title={value.name}>{value.name}</span>}
      </div>
    );
  }
  return (
    <div className="flex min-w-0 flex-col items-center gap-1 text-center">
      <div className={cn("relative aspect-square w-full overflow-hidden rounded-lg border", item.kind !== "element" && rarityGradientClass(value.rarity))}>
        {value.imageUrl ? <Image src={value.imageUrl} alt={value.name} fill sizes="80px" className="object-contain" style={item.kind === "element" && item.element ? { filter: elementIconFilter(item.element) } : undefined} /> : null}
      </div>
      <span className="max-w-[100px] truncate text-xs font-medium" title={value.name}>{value.name}</span>
    </div>
  );
}

function BossCard({ schedule, level, position, catalog }: { schedule: Schedule; level: 105 | 110; position: number; catalog: Catalog }) {
  const boss = schedule.bosses.find((candidate) => candidate.level === level && candidate.position === position);
  if (!boss) return null;
  const groups = schedule.recommendations.filter((group) => group.level === level && group.bossPosition === position);
  const teams = groups.filter((group) => group.kind === "team");
  const units = groups.filter((group) => group.kind === "unit").flatMap((group) => group.items);
  return (
    <StygianBossCard boss={boss} level={level}>
      <section className="flex flex-col gap-3" aria-label={`Recommended teams for ${boss.name}`}>
        <h3 className="font-heading text-lg font-semibold">Recommended team</h3>
        {teams.length ? <div className="grid grid-cols-1 items-start gap-3 xl:grid-cols-2">
          {teams.map((team) => (
            <div key={team.id} className="grid w-full min-w-0 grid-cols-4 items-start gap-1 rounded-lg border bg-muted/20 p-2 sm:flex sm:w-fit sm:flex-wrap sm:gap-3 xl:grid xl:grid-cols-4 xl:gap-1">
              {team.items.slice(0, 4).map((item) => <RecommendationIcon key={`${item.groupId}-${item.position}`} item={item} catalog={catalog} teamLayout />)}
              {Array.from({ length: Math.max(0, 4 - team.items.length) }, (_, index) => <div key={`empty-${index}`} className="aspect-square w-full rounded-lg border border-dashed bg-muted/30 sm:w-[116px] xl:w-full" aria-hidden="true" />)}
            </div>
          ))}
        </div> : <p className="text-sm text-muted-foreground">No team recommendations yet.</p>}
      </section>
      <section className="flex flex-col gap-3" aria-label={`Recommended units for ${boss.name}`}>
        <h3 className="font-heading text-lg font-semibold">Recommended unit</h3>
        {units.length ? <div className="grid grid-cols-[repeat(auto-fill,minmax(5rem,6.25rem))] gap-2">{units.map((item) => <RecommendationIcon key={`${item.groupId}-${item.position}`} item={item} catalog={catalog} />)}</div> : <p className="text-sm text-muted-foreground">No unit recommendations yet.</p>}
      </section>
    </StygianBossCard>
  );
}

export function StygianGuide({ schedule, catalog }: { schedule: Schedule; catalog: Catalog }) {
  return (
    <Tabs defaultValue="110">
      <TabsList aria-label="Stygian difficulty">
        <TabsTrigger value="105">Level 105</TabsTrigger>
        <TabsTrigger value="110">Level 110</TabsTrigger>
      </TabsList>
      {([105, 110] as const).map((level) => (
        <TabsContent key={level} value={String(level)}>
          <Tabs defaultValue="0">
            <TabsList variant="line" className="grid w-full grid-cols-3 overflow-hidden group-data-horizontal/tabs:h-auto group-data-horizontal/tabs:min-h-96" aria-label={`Level ${level} bosses`}>
              {[0, 1, 2].map((position) => {
                const boss = schedule.bosses.find((candidate) => candidate.level === level && candidate.position === position);
                const label = boss?.name ?? `Boss ${position + 1}`;
                const positionLabel = ["บน", "กลาง", "ล่าง"][position];
                return <TabsTrigger key={position} value={String(position)} aria-label={`${positionLabel}: ${label}`} title={label} className="h-full min-w-0 flex-col justify-start overflow-hidden rounded-none px-2 py-3 data-active:bg-accent/50 data-active:ring-2 data-active:ring-primary data-active:ring-inset">
                  <StygianBossTab boss={boss} position={position} priority={level === 110} />
                </TabsTrigger>;
              })}
            </TabsList>
            {[0, 1, 2].map((position) => <TabsContent key={position} value={String(position)} className="pt-4"><BossCard schedule={schedule} level={level} position={position} catalog={catalog} /></TabsContent>)}
          </Tabs>
        </TabsContent>
      ))}
    </Tabs>
  );
}