import { Box, Text, Popover, Progress, Title, Image } from "@mantine/core";
import { usePlayerBadges, useAllBadges } from "../queries";
import { useAuth } from "@/contexts/auth-context";
import { Badge, BadgeProgress } from "../types";
import { useMemo, useState } from "react";
import { MedalIcon, LockKeyIcon } from "@phosphor-icons/react";

interface BadgeShowcaseProps {
  playerId: string;
}

interface BadgeDisplay {
  badge: Badge;
  progress?: BadgeProgress;
  earned: boolean;
  progressText: string;
}

interface BadgeIconProps {
  badge: Badge;
  earned: boolean;
}

const BadgeIcon = ({ badge, earned }: BadgeIconProps) => {
  const [imageError, setImageError] = useState(false);
  const imagePath = `/static/img/${badge.key}.png`;

  if (imageError) {
    // Fallback to icon if image fails to load
    return earned ? (
      <MedalIcon
        size={48}
        weight="fill"
        color="var(--mantine-primary-color-6)"
      />
    ) : (
      <LockKeyIcon
        size={44}
        weight="regular"
        color="var(--mantine-color-dimmed)"
      />
    );
  }

  return (
    <Image
      src={imagePath}
      alt={badge.name}
      width={48}
      height={48}
      onError={() => setImageError(true)}
      style={{
        objectFit: 'contain',
        opacity: earned ? 1 : 0.4,
      }}
    />
  );
};

const BadgeShowcase = ({ playerId }: BadgeShowcaseProps) => {
  const { user } = useAuth();
  const { data: badgeProgress } = usePlayerBadges(playerId);
  const { data: allBadges } = useAllBadges();

  const isCurrentUser = user?.id === playerId;

  const badgesToDisplay = useMemo(() => {
    const displays: BadgeDisplay[] = [];

    if (isCurrentUser) {
      for (const badge of allBadges) {
        const progress = badgeProgress.find(bp => bp.badge.id === badge.id);
        const earned = progress?.earned || false;

        if (badge.type === 'manual' && !earned) {
          continue;
        }

        const isVeteranBadge = /^veteran_\d+_badge$/.test(badge.key);
        if (isVeteranBadge && !earned) {
          continue;
        }

        if (badge.key === 'new_player_badge') {
          continue;
        }

        let progressText = "";
        if (progress) {
          const target = getTargetProgress(badge);
          progressText = `${progress.progress} / ${target}`;
        } else {
          const target = getTargetProgress(badge);
          progressText = `0 / ${target}`;
        }

        displays.push({
          badge,
          progress,
          earned,
          progressText,
        });
      }

      displays.sort((a, b) => {
        if (a.earned && !b.earned) return -1;
        if (!a.earned && b.earned) return 1;
        return a.badge.order - b.badge.order;
      });
    } else {
      const earnedProgress = badgeProgress.filter(bp => bp.earned);
      for (const progress of earnedProgress) {
        const badge: Badge = {
          ...progress.badge,
          criteria: {},
          created: progress.created,
          updated: progress.updated,
        };

        const target = getTargetProgress(badge);
        displays.push({
          badge,
          progress,
          earned: true,
          progressText: `${progress.progress} / ${target}`,
        });
      }

      displays.sort((a, b) => a.badge.order - b.badge.order);
    }

    return displays;
  }, [allBadges, badgeProgress, isCurrentUser]);

  if (badgesToDisplay.length === 0) {
    return null;
  }

  return (
    <Box mb="lg">
      <Box
        px="md"
        style={{
          maxHeight: '220px',
          overflowY: 'auto',
          overflowX: 'hidden',
          width: '100%',
        }}
      >
        <Box
          style={{
            display: 'grid',
            gridTemplateColumns: 'repeat(auto-fill, minmax(85px, 1fr))',
            gap: 'var(--mantine-spacing-md)',
            paddingBottom: 'var(--mantine-spacing-sm)',
            width: '100%',
          }}
        >
          {badgesToDisplay.map((display) => {
            const isStackableBadge = ['winner_badge', 'silver_medal_badge', 'bronze_medal_badge'].includes(display.badge.key);
            const stackCount = display.earned && isStackableBadge
              ? (display.progress?.progress || 0)
              : 1;
            const showStack = stackCount > 1;

            return (
              <Popover key={display.badge.id} width={220} position="top" withArrow shadow="md">
                <Popover.Target>
                  <Box
                    style={{
                      cursor: "pointer",
                      transition: 'all 0.2s ease',
                      position: 'relative',
                    }}
                  >
                    {showStack && (
                      <>
                        {[...Array(Math.min(stackCount - 1, 2))].map((_, i) => (
                          <Box
                            key={i}
                            style={{
                              width: '85px',
                              height: '85px',
                              borderRadius: '12px',
                              background: 'transparent',
                              border: '2px solid var(--mantine-primary-color-5)',
                              position: 'absolute',
                              top: `${(i + 1) * 3}px`,
                              left: `${(i + 1) * 3}px`,
                              opacity: 0.4 - (i * 0.15),
                              zIndex: -(i + 1),
                            }}
                          />
                        ))}
                      </>
                    )}

                    <Box
                      style={{
                        width: '100px',
                        height: '100px',
                        borderRadius: '12px',
                        background: 'light-dark(var(--mantine-color-white), var(--mantine-color-dark-6))',
                        border: display.earned
                          ? '2px solid var(--mantine-primary-color-6)'
                          : '2px dashed var(--mantine-primary-color-4)',
                        display: 'flex',
                        alignItems: 'center',
                        justifyContent: 'center',
                        flexDirection: 'column',
                        gap: '4px',
                        position: 'relative',
                        boxShadow: display.earned
                          ? '0 0 0 1px color-mix(in srgb, var(--mantine-primary-color-6) 20%, transparent)'
                          : 'none',
                        opacity: display.earned ? 1 : 0.4,
                        zIndex: 1,
                      }}
                    >
                      <BadgeIcon badge={display.badge} earned={display.earned} />

                      {showStack && (
                        <Box
                          style={{
                            position: 'absolute',
                            top: '4px',
                            right: '4px',
                            color: 'var(--mantine-primary-color-6)',
                            width: '20px',
                            height: '20px',
                            display: 'flex',
                            alignItems: 'center',
                            justifyContent: 'center',
                            fontSize: '10px',
                            fontWeight: 700,
                          }}
                        >
                          x{stackCount}
                        </Box>
                      )}

                      <Text
                        size="xs"
                        fw={display.earned ? 600 : 500}
                        ta="center"
                        c={display.earned ? undefined : 'dimmed'}
                        style={{ lineHeight: 1.1 }}
                      >
                        {display.badge.name}
                      </Text>
                    </Box>
                  </Box>
                </Popover.Target>
              <Popover.Dropdown>
                <Box>
                  <Title order={5}>
                    {display.badge.name}
                  </Title>
                  <Text size="xs" c="dimmed" mb={isCurrentUser ? "sm" : undefined}>
                    {display.badge.description}
                  </Text>
                  {isCurrentUser && (
                    <Box>
                      <Box mb="xs" style={{ display: 'flex', justifyContent: 'space-between', alignItems: 'center' }}>
                        <Text size="xs" fw={500} c="dimmed">
                          Progress
                        </Text>
                        <Text size="xs" fw={600} c="dimmed">
                          {display.progressText}
                        </Text>
                      </Box>
                      <Progress
                        value={(display.progress?.progress || 0) / getTargetProgress(display.badge) * 100}
                        size="sm"
                        radius="sm"
                        color={display.earned ? "green" : undefined}
                      />
                    </Box>
                  )}
                </Box>
              </Popover.Dropdown>
            </Popover>
            );
          })}
        </Box>
      </Box>
    </Box>
  );
};

function getTargetProgress(badge: Badge): number {
  const criteria = badge.criteria;
  return (
    criteria.matches_played ||
    criteria.tournament_wins ||
    criteria.tournaments_attended ||
    criteria.overtime_matches ||
    criteria.overtime_wins ||
    criteria.consecutive_wins ||
    1
  );
}

export default BadgeShowcase;